Make COMMAND options customizable

Author: hoppergee

lib/matrixeval/ruby/command_line.rb

@@ -38,7 +38,7 @@ def rest_arguments
 
       def seperator_index
         argv.index do |argument|
-          COMMANDS.include?(argument)
+          Config.commands.include?(argument)
         end
       end
 

lib/matrixeval/ruby/command_line/parse_context_arguments.rb

@@ -50,7 +50,7 @@ def parse!
             end
 
             opts.separator ""
-            opts.separator "Commands: #{COMMANDS.join("/")} (Customizable)"
+            opts.separator "Commands: #{Config.commands.join("/")} (Customizable)"
 
             opts.separator ""
             opts.separator "MatrixEval Options:"

lib/matrixeval/ruby/config.rb

@@ -49,6 +49,11 @@ def parallel_workers
           YAML["parallel_workers"] || "number_of_processors"
         end
 
+        def commands
+          cmds = YAML["commands"] || []
+          COMMANDS + cmds
+        end
+
       end
     end
   end

test/matrixeval/ruby/command_line_test.rb

@@ -4,6 +4,10 @@
 
 class Matrixeval::Ruby::CommandLineTest < MatrixevalTest
 
+  def setup
+    Matrixeval::Ruby::Config::YAML.stubs(:yaml).returns({"matrix" => {}})
+  end
+
   def test_init?
     command = Matrixeval::Ruby::CommandLine.new(["init"])
     assert command.init?
@@ -13,8 +17,6 @@ def test_init?
   end
 
   def test_all?
-    Matrixeval::Ruby::Config::YAML.stubs(:yaml).returns({"matrix" => {}})
-
     command = Matrixeval::Ruby::CommandLine.new(["--all", "rake", "test"])
     assert command.all?
 

test/matrixeval/ruby/config_test.rb

@@ -82,4 +82,12 @@ def test_exclusions
     assert_equal exclusions, Matrixeval::Ruby::Config.exclusions
   end
 
+  def test_commands
+    Matrixeval::Ruby::Config::YAML.stubs(:yaml).returns({})
+    assert_equal ['rake', 'rspec', 'bundle', 'bash'], Matrixeval::Ruby::Config.commands
+
+    Matrixeval::Ruby::Config::YAML.stubs(:yaml).returns({'commands' => ['ls']})
+    assert_equal ['rake', 'rspec', 'bundle', 'bash', 'ls'], Matrixeval::Ruby::Config.commands
+  end
+
 end