Merge pull request #133 from netMedi/master

include minitest plugin for Minitest::Test class
mattheworiordan · Jul 26, 2015 · 468cb2e · 468cb2e
2 parents 346a045 + aeb87df
commit 468cb2e
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/lib/capybara-screenshot/minitest.rb b/lib/capybara-screenshot/minitest.rb
@@ -22,6 +22,15 @@ def after_teardown
   end
 end
 
-class MiniTest::Unit::TestCase
-  include Capybara::Screenshot::MiniTestPlugin
+begin
+  Minitest.const_get('Test')
+  class Minitest::Test
+    include Capybara::Screenshot::MiniTestPlugin
+  end
+rescue NameError => e
+  class MiniTest::Unit::TestCase
+    include Capybara::Screenshot::MiniTestPlugin
+  end
 end
+
+