lib/autowatchr.rb in autowatchr-0.1.3 vs lib/autowatchr.rb in autowatchr-0.1.4

@@ -1,11 +1,12 @@
 require 'erb'
 
 class Autowatchr
   class Config
-    attr_writer :command, :ruby, :include, :lib_dir, :test_dir, :lib_re,
-      :test_re, :failed_results_re, :completed_re, :failing_only, :run_suite
+    attr_writer :command, :ruby, :include, :require, :lib_dir, :test_dir,
+      :lib_re, :test_re, :failed_results_re, :completed_re, :failing_only,
+      :run_suite
 
     def initialize(options = {})
       @failing_only = @run_suite = true
 
       options.each_pair do |key, value|
@@ -15,18 +16,32 @@
         end
       end
     end
 
     def command
-      @command ||= "<%= ruby %> -I<%= include %> <%= predicate %>"
+      @command ||= "<%= ruby %> -I<%= include %> <% list_of_requires.each { |lib| %>-r<%= lib %> <% } %><%= predicate %>"
     end
 
     def ruby
       @ruby ||= "ruby"
     end
 
     def include
       @include ||= ".:#{self.lib_dir}:#{self.test_dir}"
+    end
+
+    def require
+      @require
+    end
+
+    def list_of_requires
+      if @require.nil? || @require.empty?
+        []
+      elsif @require.is_a?(Array)
+        @require
+      else
+        [@require]
+      end
     end
 
     def lib_dir
       @lib_dir ||= "lib"
     end