spec/punchblock/client_spec.rb in punchblock-2.0.1 vs spec/punchblock/client_spec.rb in punchblock-2.0.2

@@ -12,11 +12,12 @@
     its(:component_registry)  { should be_a Client::ComponentRegistry }
 
     let(:call_id)         { 'abc123' }
     let(:mock_event)      { double('Event').as_null_object }
     let(:component_id)    { 'abc123' }
-    let(:mock_component)  { double 'Component', :component_id => component_id }
+    let(:component_uri)   { 'callid@server/abc123' }
+    let(:mock_component)  { double 'Component', source_uri: component_uri }
     let(:mock_command)    { double 'Command' }
 
     describe '#run' do
       it 'should start up the connection' do
         connection.should_receive(:run).once
@@ -75,10 +76,10 @@
       end
     end
 
     it 'should be able to register and retrieve components' do
       subject.register_component mock_component
-      subject.find_component_by_id(component_id).should be mock_component
+      subject.find_component_by_uri(component_uri).should be mock_component
     end
 
     describe '#execute_command' do
       let(:component) { Component::Output.new }
       let(:event)     { Event::Complete.new }