diff --git a/lib/cmock_config.rb b/lib/cmock_config.rb index 2f29be6..2ad85ba 100644 --- a/lib/cmock_config.rb +++ b/lib/cmock_config.rb @@ -31,6 +31,7 @@ class CMockConfig :includes_h_post_orig_header => nil, :includes_c_pre_header => nil, :includes_c_post_header => nil, + :orig_header_include_fmt => '"%s"', } def initialize(options=nil) diff --git a/lib/cmock_generator.rb b/lib/cmock_generator.rb index 949f4e7..ca2d3bf 100644 --- a/lib/cmock_generator.rb +++ b/lib/cmock_generator.rb @@ -69,7 +69,7 @@ class CMockGenerator file << "#ifndef _#{define_name}_H\n" file << "#define _#{define_name}_H\n\n" @includes_h_pre_orig_header.each {|inc| file << "#include #{inc}\n"} - file << "#include \"#{orig_filename}\"\n" + file << "#include " + @config.orig_header_include_fmt % "#{orig_filename}" + "\n" @includes_h_post_orig_header.each {|inc| file << "#include #{inc}\n"} plugin_includes = @plugins.run(:include_files) file << plugin_includes if (!plugin_includes.empty?) diff --git a/test/unit/cmock_generator_main_test.rb b/test/unit/cmock_generator_main_test.rb index 35c8ffd..ce4fba5 100644 --- a/test/unit/cmock_generator_main_test.rb +++ b/test/unit/cmock_generator_main_test.rb @@ -103,6 +103,7 @@ class CMockGeneratorTest < Test::Unit::TestCase "\n", ] + @config.expect.orig_header_include_fmt.returns('"%s"') @plugins.expect.run(:include_files).returns("#include \"PluginRequiredHeader.h\"\n") @cmock_generator.create_mock_header_header(output, "MockPoutPoutFish.h") @@ -147,6 +148,7 @@ class CMockGeneratorTest < Test::Unit::TestCase "\n", ] + @config.expect.orig_header_include_fmt.returns('"%s"') @plugins.expect.run(:include_files).returns("#include \"PluginRequiredHeader.h\"\n") @cmock_generator2.create_mock_header_header(output, "MockPout-Pout Fish.h") @@ -177,6 +179,7 @@ class CMockGeneratorTest < Test::Unit::TestCase "\n", ] + @config.expect.orig_header_include_fmt.returns('"%s"') @plugins.expect.run(:include_files).returns('') @cmock_generator.create_mock_header_header(output, "MockPoutPoutFish.h") @@ -208,6 +211,7 @@ class CMockGeneratorTest < Test::Unit::TestCase "\n", ] + @config.expect.orig_header_include_fmt.returns('"%s"') @plugins.expect.run(:include_files).returns("#include \"PluginRequiredHeader.h\"\n") @cmock_generator.create_mock_header_header(output, "MockPoutPoutFish.h")