Prevent uninitialized constant error

you can load Test::Unit without Test::Unit::TestCase
justinweiss · Aug 17, 2015 · a8f79fe · a8f79fe
1 parent e0fca9a
commit a8f79fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/resque_unit.rb b/lib/resque_unit.rb
@@ -13,7 +13,7 @@ module ResqueUnit
 require 'resque_unit/assertions'
 require 'resque_unit/plugin'
 
-if defined?(Test::Unit)
+if defined?(Test::Unit::TestCase)
   Test::Unit::TestCase.send(:include, ResqueUnit::Assertions)
 end
 

diff --git a/lib/resque_unit_scheduler.rb b/lib/resque_unit_scheduler.rb
@@ -1,7 +1,7 @@
 require 'resque_unit/scheduler'
 require 'resque_unit/scheduler_assertions'
 
-if defined?(Test::Unit)
+if defined?(Test::Unit::TestCase)
   Test::Unit::TestCase.send(:include, ResqueUnit::SchedulerAssertions)
 end