diff --git a/lib/cmock.rb b/lib/cmock.rb index 72f86418..a8a0105c 100644 --- a/lib/cmock.rb +++ b/lib/cmock.rb @@ -45,7 +45,7 @@ def generate_mock(src, folder) name = File.basename(src, '.*') ext = File.extname(src) puts "Creating mock for #{name}..." unless @silent - @cm_generator.create_mock(name, @cm_parser.parse(name, File.read(src)), ext, folder) + @cm_generator.create_mock(name, @cm_parser.parse(name, File.read(src)), ext, folder,src) end def generate_skeleton(src) diff --git a/lib/cmock_generator.rb b/lib/cmock_generator.rb index a98cfad1..aec391cd 100644 --- a/lib/cmock_generator.rb +++ b/lib/cmock_generator.rb @@ -44,7 +44,7 @@ def initialize(config, file_writer, utils, plugins) end end - def create_mock(module_name, parsed_stuff, module_ext = nil, folder = nil) + def create_mock(module_name, parsed_stuff, module_ext = nil, folder = nil, src = nil) # determine the name for our new mock mock_name = @prefix + module_name + @suffix @@ -68,7 +68,8 @@ def create_mock(module_name, parsed_stuff, module_ext = nil, folder = nil) :clean_name => TypeSanitizer.sanitize_c_identifier(mock_name), :folder => mock_folder, :parsed_stuff => parsed_stuff, - :skeleton => false + :skeleton => false, + :source => src } create_mock_subdir(mock_project) @@ -129,6 +130,7 @@ def create_mock_header_file(mock_project) end @file_writer.create_file(mock_project[:mock_name] + mock_project[:module_ext], mock_project[:folder]) do |file, filename| + file << "/* Source File: #{mock_project[:source]} */\n" # if mock_project[:source] create_mock_header_header(file, filename, mock_project) create_mock_header_service_call_declarations(file, mock_project) create_typedefs(file, mock_project) @@ -142,6 +144,7 @@ def create_mock_header_file(mock_project) def create_mock_source_file(mock_project) @file_writer.create_file(mock_project[:mock_name] + '.c', mock_project[:folder]) do |file, filename| + file << "/* Source File: #{mock_project[:source]} */\n" # if mock_project[:source] create_source_header_section(file, filename, mock_project) create_instance_structure(file, mock_project) create_extern_declarations(file)