test/unit/trapper_test.rb in hardmock-1.2.0 vs test/unit/trapper_test.rb in hardmock-1.2.3

@@ -1,8 +1,10 @@
 require File.expand_path(File.dirname(__FILE__) + "/../test_helper")
-require 'hardmock'
+require 'hardmock/method_cleanout'
+require 'hardmock/trapper'
 
 class TrapperTest < Test::Unit::TestCase
+  include Hardmock
 
   def setup
     @mock = Object.new
     @mock_control = MyControl.new
     @builder = ExpBuilder.new