Stop casting unless necessary for custom types or const when returning values.

lib/cmock_generator.rb

@@ -23,7 +23,7 @@ class CMockGenerator
     @includes_h_post_orig_header = (@config.includes_h_post_orig_header || []).map{|h| h =~ /</ ? h : "\"#{h}\""}
     @includes_c_pre_header       = (@config.includes_c_pre_header || []).map{|h| h =~ /</ ? h : "\"#{h}\""}
     @includes_c_post_header      = (@config.includes_c_post_header || []).map{|h| h =~ /</ ? h : "\"#{h}\""}
-    
+
     here = File.dirname __FILE__
     unity_path_in_ceedling = "#{here}/../../unity" # path to Unity from within Ceedling
     unity_path_in_cmock = "#{here}/../vendor/unity" # path to Unity from within CMock
@@ -34,7 +34,7 @@ class CMockGenerator
     else
       raise "Failed to find an instance of Unity to pull in type_sanitizer module!"
     end
-    
+
   end
 
   def create_mock(module_name, parsed_stuff)
@@ -211,9 +211,9 @@ class CMockGenerator
       file << "    UNITY_TEST_FAIL(cmock_line, \"Function '#{function[:name]}' called later than expected.\");\n"
       # file << "  UNITY_TEST_ASSERT((cmock_call_instance->CallOrder == ++GlobalVerifyOrder), cmock_line, \"Out of order function calls. Function '#{function[:name]}'\");\n"
     end
-    return_type_cast = function[:return][:const?] ? "(const #{function[:return][:type]})" : ''
+    return_type = function[:return][:const?] ? "(const #{function[:return][:type]})" : ((function[:return][:type] =~ /cmock/) ? "(#{function[:return][:type]})" : '')
     file << @plugins.run(:mock_implementation, function)
-    file << "  return #{return_type_cast}cmock_call_instance->ReturnVal;\n" unless (function[:return][:void?])
+    file << "  return #{return_type}cmock_call_instance->ReturnVal;\n" unless (function[:return][:void?])
     file << "}\n\n"
   end
 

lib/cmock_generator_plugin_callback.rb

@@ -2,19 +2,19 @@
 #   CMock Project - Automatic Mock Generation for C
 #   Copyright (c) 2007 Mike Karlesky, Mark VanderVoord, Greg Williams
 #   [Released under MIT License. Please refer to license.txt for details]
-# ========================================== 
+# ==========================================
 
 class CMockGeneratorPluginCallback
 
   attr_accessor :include_count
   attr_reader :priority
   attr_reader :config, :utils
-  
+
   def initialize(config, utils)
     @config = config
     @utils = utils
     @priority = 6
-    
+
     @include_count = @config.callback_include_count
     if (@config.callback_after_arg_check)
       alias :mock_implementation          :mock_implementation_for_callbacks
@@ -54,14 +54,14 @@ class CMockGeneratorPluginCallback
       when 7 then "    return Mock.#{func_name}_CallbackFunctionPointer(#{function[:args].map{|m| m[:name]}.join(', ')}, Mock.#{func_name}_CallbackCalls++);\n  }\n"
     end
   end
-  
+
   def nothing(function)
     return ""
   end
 
   def mock_interfaces(function)
     func_name = function[:name]
-    "void #{func_name}_StubWithCallback(CMOCK_#{func_name}_CALLBACK Callback)\n{\n" + 
+    "void #{func_name}_StubWithCallback(CMOCK_#{func_name}_CALLBACK Callback)\n{\n" +
     "  Mock.#{func_name}_CallbackFunctionPointer = Callback;\n}\n\n"
   end
 
@@ -69,7 +69,7 @@ class CMockGeneratorPluginCallback
     "  Mock.#{function[:name]}_CallbackFunctionPointer = NULL;\n" +
     "  Mock.#{function[:name]}_CallbackCalls = 0;\n"
   end
-  
+
   def mock_verify(function)
     func_name = function[:name]
     "  if (Mock.#{func_name}_CallbackFunctionPointer != NULL)\n    Mock.#{func_name}_CallInstance = CMOCK_GUTS_NONE;\n"

lib/cmock_generator_plugin_expect_any_args.rb

@@ -45,8 +45,8 @@ class CMockGeneratorPluginExpectAnyArgs
     else
       retval = function[:return].merge( { :name => "cmock_call_instance->ReturnVal"} )
       lines << "  " + @utils.code_assign_argument_quickly("Mock.#{function[:name]}_FinalReturn", retval) unless (retval[:void?])
-      return_type_cast = function[:return][:const?] ? "(const #{function[:return][:type]})" : ''
-      lines << "    return #{return_type_cast}cmock_call_instance->ReturnVal;\n  }\n"
+      return_type = function[:return][:const?] ? "(const #{function[:return][:type]})" : ((function[:return][:type] =~ /cmock/) ? "(#{function[:return][:type]})" : '')
+      lines << "    return #{return_type}cmock_call_instance->ReturnVal;\n  }\n"
     end
     lines
   end

lib/cmock_generator_plugin_ignore.rb

@@ -39,10 +39,10 @@ class CMockGeneratorPluginIgnore
       lines << "    return;\n  }\n"
     else
       retval = function[:return].merge( { :name => "cmock_call_instance->ReturnVal"} )
-      return_type = function[:return][:const?] ? "const #{function[:return][:type]}" : function[:return][:type]
-      lines << "    if (cmock_call_instance == NULL)\n      return (#{return_type})Mock.#{function[:name]}_FinalReturn;\n"
+      return_type = function[:return][:const?] ? "(const #{function[:return][:type]})" : ((function[:return][:type] =~ /cmock/) ? "(#{function[:return][:type]})" : '')
+      lines << "    if (cmock_call_instance == NULL)\n      return #{return_type}Mock.#{function[:name]}_FinalReturn;\n"
       lines << "  " + @utils.code_assign_argument_quickly("Mock.#{function[:name]}_FinalReturn", retval) unless (retval[:void?])
-      lines << "    return (#{return_type})cmock_call_instance->ReturnVal;\n  }\n"
+      lines << "    return #{return_type}cmock_call_instance->ReturnVal;\n  }\n"
     end
     lines
   end