From 9691acfb82bbafd88758cbef5621049df71a096b Mon Sep 17 00:00:00 2001 From: mkarlesky Date: Wed, 27 Apr 2011 04:24:05 +0000 Subject: [PATCH] fixed forgotten includes git-svn-id: http://unity.svn.sourceforge.net/svnroot/unity/trunk@131 e7d17a6e-8845-0410-bbbc-c8efb4fdad7e --- auto/generate_test_runner.rb | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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 ')