diff --git a/auto/generate_test_runner.rb b/auto/generate_test_runner.rb index 8debc52..a8c2c14 100644 --- a/auto/generate_test_runner.rb +++ b/auto/generate_test_runner.rb @@ -56,7 +56,7 @@ class UnityTestRunnerGenerator def generate(input_file, output_file, tests, includes, used_mocks) File.open(output_file, 'w') do |output| - create_header(output, used_mocks) + create_header(output, includes, used_mocks) create_externs(output, tests, used_mocks) create_mock_management(output, used_mocks) create_suite_setup_and_teardown(output) @@ -127,13 +127,13 @@ class UnityTestRunnerGenerator return mock_headers end - def create_header(output, mocks) + def create_header(output, includes, mocks) output.puts('/* AUTOGENERATED FILE. DO NOT EDIT. */') create_runtest(output, mocks) output.puts("\n//=======Automagically Detected Files To Include=====") output.puts("#include \"#{@options[:framework].to_s}.h\"") output.puts('#include "cmock.h"') unless (mocks.empty?) - @options[:includes].flatten.uniq.compact.each do |inc| + (@options[:includes].flatten.uniq.compact + includes.flatten.uniq.compact).each do |inc| output.puts("#include #{inc.include?('<') ? inc : "\"#{inc.gsub('.h','')}.h\""}") end output.puts('#include ')