From ae5b4c57990583222a70b5260e66f8a48a89691b Mon Sep 17 00:00:00 2001
From: Mark VanderVoord <mvandervoord@gmail.com>
Date: Thu, 10 Nov 2016 10:10:13 -0500
Subject: [PATCH] Added ability to inject defines, much like what was built
 into Ceedling before Ceedling switched to just using this generator.

---
 auto/generate_test_runner.rb | 4 ++++
 1 file changed, 4 insertions(+)

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 <setjmp.h>')
     output.puts('#include <stdio.h>')
     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