spec/punchblock/component_spec.rb in punchblock-0.4.3 vs spec/punchblock/component_spec.rb in punchblock-0.5.0

@@ -65,30 +65,24 @@
       end # #add_event
 
       describe "#response=" do
         before do
           subject.request!
-          subject.connection = mock(:record_command_id_for_iq_id => true)
+          subject.client = Client.new
         end
 
         let(:component_id) { 'abc123' }
 
         let :ref do
           Ref.new.tap do |ref|
             ref.id = component_id
           end
         end
 
-        let :iq do
-          Blather::Stanza::Iq.new(:result, 'blah').tap do |iq|
-            iq.from = "[e-mail]"
-            iq << ref
-          end
-        end
-
         it "should set the component ID from the ref" do
-          subject.response = iq
+          subject.response = ref
           subject.component_id.should == component_id
+          subject.client.find_component_by_id(component_id).should be subject
         end
       end
     end # ComponentNode
   end # Component
 end # Punchblock