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

@@ -1,15 +1,16 @@
 require File.expand_path(File.dirname(__FILE__) + "/../test_helper")
-require 'hardmock'
+require 'hardmock/expectation_builder'
 
 class ExpectationBuilderTest < Test::Unit::TestCase
+  include Hardmock
   
   def test_build_expectation
     builder = ExpectationBuilder.new
     
     ex = builder.build_expectation( :stuff => 'inside' )
     assert_not_nil ex, "Didn't build an expectation"
-    assert_kind_of SimpleExpectation, ex, "Wrong type!"
+    assert_kind_of Expectation, ex, "Wrong type!"
 
     # Shhhh... fragile, yes, whatever.  The functional tests do the 
     # real testing of this anyway
     assert_equal({:stuff => 'inside'}, ex.instance_variable_get('@options'), "Hash not sent to SimpleExpectation constructor")
   end