examples/rr/probe_creator_example.rb in rr-0.1.11 vs examples/rr/probe_creator_example.rb in rr-0.1.12

@@ -19,10 +19,21 @@
   it_should_behave_like "RR::ProbeCreator"
 
   before do
     @creator = ProbeCreator.new(@space, @subject)
   end
+
+  it "clears out all methods from creator" do
+    creator_subclass = Class.new(ProbeCreator) do
+      def i_should_be_a_scenario
+      end
+    end
+    creator_subclass.instance_methods.should include('i_should_be_a_scenario')
+
+    creator = creator_subclass.new(@space, @subject)
+    creator.i_should_be_a_scenario.should be_instance_of(Scenario)
+  end
 end
 
 describe ProbeCreator, ".new with block" do
   it_should_behave_like "RR::ProbeCreator"
 
@@ -40,9 +51,21 @@
   it "creates doubles" do
     @subject.foobar(1, 2).should == :original_foobar
     @subject.foobar(1).should == :original_foobar
     @subject.foobar(:something).should == :original_foobar
     proc {@subject.foobar(:nasty)}.should raise_error
+  end
+
+  it "clears out all methods from creator" do
+    creator_subclass = Class.new(ProbeCreator) do
+      def i_should_be_a_scenario
+      end
+    end
+    creator_subclass.instance_methods.should include('i_should_be_a_scenario')
+
+    creator_subclass.new(@space, @subject) do |m|
+      m.i_should_be_a_scenario.should be_instance_of(Scenario)
+    end
   end
 end
 
 describe ProbeCreator, ".new where method takes a block" do
   it_should_behave_like "RR::ProbeCreator"
\ No newline at end of file