Move assertion setup into formula_assertions.rb

1 files changed, 3 insertions, 26 deletions

diff --git a/Library/Homebrew/cmd/test.rb b/Library/Homebrew/cmd/test.rb
index 139cebfed..218668dab 100644
--- a/Library/Homebrew/cmd/test.rb
+++ b/Library/Homebrew/cmd/test.rb
@@ -1,33 +1,11 @@
 require "extend/ENV"
 require "timeout"
 require "debrew"
+require "formula_assertions"
 
 module Homebrew
   TEST_TIMEOUT_SECONDS = 5*60
 
-  if defined?(Gem)
-    begin
-      gem "minitest", "< 5.0.0"
-    rescue Gem::LoadError
-      require "test/unit/assertions"
-    else
-      require "minitest/unit"
-      require "test/unit/assertions"
-    end
-  else
-    require "test/unit/assertions"
-  end
-
-  if defined?(MiniTest::Assertion)
-    FailedAssertion = MiniTest::Assertion
-  elsif defined?(Minitest::Assertion)
-    FailedAssertion = Minitest::Assertion
-  else
-    FailedAssertion = Test::Unit::AssertionFailedError
-  end
-
-  require "formula_assertions"
-
   def test
     raise FormulaUnspecifiedError if ARGV.named.empty?
 
@@ -49,8 +27,7 @@ module Homebrew
 
       puts "Testing #{f.name}"
 
-      f.extend(Test::Unit::Assertions)
-      f.extend(Homebrew::Assertions)
+      f.extend(Assertions)
       f.extend(Debrew::Formula) if ARGV.debug?
 
       env = ENV.to_hash
@@ -60,7 +37,7 @@ module Homebrew
         Timeout::timeout TEST_TIMEOUT_SECONDS do
           raise if f.run_test == false
         end
-      rescue FailedAssertion => e
+      rescue Assertions::FailedAssertion => e
         ofail "#{f.name}: failed"
         puts e.message
       rescue Exception => e