lib/bogus/minitest.rb in bogus-0.1.3 vs lib/bogus/minitest.rb in bogus-0.1.4

@@ -35,12 +35,15 @@
     Bogus.ensure_all_expectations_satisfied!
     super
   end
 end
 
-class Minitest::Test
-  include Bogus::Minitest
-end if defined? Minitest::Test
-
-class MiniTest::Unit::TestCase
-  include Bogus::Minitest
-end if defined? MiniTest::Unit::TestCase
+# minitest 5 vs 4.7
+if defined? Minitest::Test
+  class Minitest::Test
+    include Bogus::Minitest
+  end
+else
+  class MiniTest::Unit::TestCase
+    include Bogus::Minitest
+  end
+end