From 4e2f0381ccd5aa231167a168e9ba98fa00d34730 Mon Sep 17 00:00:00 2001
From: Mark VanderVoord <mvandervoord@gmail.com>
Date: Mon, 21 Nov 2016 15:19:16 -0500
Subject: [PATCH] Add ability to inject correct mock prefix

---
 auto/generate_module.rb | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/auto/generate_module.rb b/auto/generate_module.rb
index 43f5b0d..e68aa72 100644
--- a/auto/generate_module.rb
+++ b/auto/generate_module.rb
@@ -96,6 +96,7 @@ class UnityModuleGenerator
       :update_svn      => false,
       :boilerplates    => {},
       :test_prefix     => 'Test',
+      :mock_prefix     => 'Mock',
     }
   end
 
@@ -136,7 +137,7 @@ class UnityModuleGenerator
           :includes => case(triad[:inc])
                          when :src then @options[:includes][:src] | pattern_traits[:inc].map{|f| f % [module_name]}
                          when :inc then @options[:includes][:inc]
-                         when :tst then @options[:includes][:tst] | pattern_traits[:inc].map{|f| "Mock#{f}"% [module_name]}
+                         when :tst then @options[:includes][:tst] | pattern_traits[:inc].map{|f| "#{@options[:mock_prefix]}#{f}"% [module_name]}
                        end
         }
       end