From af59c531f7fc3c88e52b50a409d920e5f0e0bc48 Mon Sep 17 00:00:00 2001
From: merazmus <merazmus@users.noreply.github.com>
Date: Fri, 22 Mar 2019 09:11:58 +0100
Subject: [PATCH] Update unit tests

---
 test/unit/cmock_generator_main_test.rb | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/unit/cmock_generator_main_test.rb b/test/unit/cmock_generator_main_test.rb
index fbfe41c..2993082 100644
--- a/test/unit/cmock_generator_main_test.rb
+++ b/test/unit/cmock_generator_main_test.rb
@@ -92,6 +92,7 @@ describe CMockGenerator, "Verify CMockGenerator Module" do
       "/* AUTOGENERATED FILE. DO NOT EDIT. */\n",
       "#ifndef _#{define_name}\n",
       "#define _#{define_name}\n\n",
+      "#include \"unity.h\"\n",
       "#include \"ConfigRequiredHeader1.h\"\n",
       "#include \"ConfigRequiredHeader2.h\"\n",
       "#include \"#{orig_filename}\"\n",
@@ -147,6 +148,7 @@ describe CMockGenerator, "Verify CMockGenerator Module" do
       "/* AUTOGENERATED FILE. DO NOT EDIT. */\n",
       "#ifndef _#{define_name}\n",
       "#define _#{define_name}\n\n",
+      "#include \"unity.h\"\n",
       "#include \"ConfigRequiredHeader1.h\"\n",
       "#include \"ConfigRequiredHeader2.h\"\n",
       "#include \"#{orig_filename}\"\n",
@@ -185,6 +187,7 @@ describe CMockGenerator, "Verify CMockGenerator Module" do
       "/* AUTOGENERATED FILE. DO NOT EDIT. */\n",
       "#ifndef _#{define_name}\n",
       "#define _#{define_name}\n\n",
+      "#include \"unity.h\"\n",
       "#include \"ConfigRequiredHeader1.h\"\n",
       "#include \"ConfigRequiredHeader2.h\"\n",
       "#include \"#{orig_filename}\"\n",
@@ -222,6 +225,7 @@ describe CMockGenerator, "Verify CMockGenerator Module" do
       "/* AUTOGENERATED FILE. DO NOT EDIT. */\n",
       "#ifndef _#{define_name}\n",
       "#define _#{define_name}\n\n",
+      "#include \"unity.h\"\n",
       "#include \"ConfigRequiredHeader1.h\"\n",
       "#include \"ConfigRequiredHeader2.h\"\n",
       "#include \"#{orig_filename}\"\n",
@@ -308,7 +312,6 @@ describe CMockGenerator, "Verify CMockGenerator Module" do
                  "#include <string.h>\n",
                  "#include <stdlib.h>\n",
                  "#include <setjmp.h>\n",
-                 "#include \"unity.h\"\n",
                  "#include \"cmock.h\"\n",
                  "#include \"MockPoutPoutFish.h\"\n",
                  "\n",