Add functionality to allow custom templates in control repo, this all…

…ows a custom spec_helper.rb for different flows
voxpupuli · Mar 2, 2018 · ea28302 · ea28302
1 parent f3a09f8
commit ea28302
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/onceover/controlrepo.rb b/lib/onceover/controlrepo.rb
@@ -537,7 +537,12 @@ def self.create_dirs_and_log(dir)
     def self.evaluate_template(template_name, bind)
       logger.debug "Evaluating template #{template_name}"
       template_dir = File.expand_path('../../templates', File.dirname(__FILE__))
-      template = File.read(File.expand_path("./#{template_name}", template_dir))
+      if File.file?(File.expand_path("./spec/templates/#{template_name}", @root))
+        puts "Using Custom #{template_name}"
+        template = File.read(File.expand_path("./spec/templates/#{template_name}", @root))
+      else
+        template = File.read(File.expand_path("./#{template_name}", template_dir))
+      end
       ERB.new(template, nil, '-').result(bind)
     end