diff --git a/auto/generate_test_runner.rb b/auto/generate_test_runner.rb index 84696e3..134be65 100644 --- a/auto/generate_test_runner.rb +++ b/auto/generate_test_runner.rb @@ -23,6 +23,7 @@ class UnityTestRunnerGenerator def self.default_options { :includes => [], + :defines => [], :plugins => [], :framework => :unity, :test_prefix => "test|spec|should", @@ -167,6 +168,9 @@ class UnityTestRunnerGenerator output.puts('#include ') output.puts('#include ') output.puts('#include "CException.h"') if @options[:plugins].include?(:cexception) + if (@options[:defines] && !@options[:defines].empty?) + @options[:defines].each {|d| output.puts("#define #{d}")} + end if (@options[:header_file] && !@options[:header_file].empty?) output.puts("#include \"#{File.basename(@options[:header_file])}\"") else