diff --git a/lib/cmock_generator.rb b/lib/cmock_generator.rb index b603098..a676924 100644 --- a/lib/cmock_generator.rb +++ b/lib/cmock_generator.rb @@ -183,8 +183,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]})" : '' file << @plugins.run(:mock_implementation, function) - file << " return cmock_call_instance->ReturnVal;\n" unless (function[:return][:void?]) + file << " return #{return_type_cast}cmock_call_instance->ReturnVal;\n" unless (function[:return][:void?]) file << "}\n\n" end