spec/unit/realtime/connection_spec.rb in ably-0.6.2 vs spec/unit/realtime/connection_spec.rb in ably-0.7.0

@@ -1,18 +1,18 @@
 require 'spec_helper'
-require 'support/protocol_msgbus_helper'
-require 'support/event_machine_helper'
+require 'shared/protocol_msgbus_behaviour'
 
 describe Ably::Realtime::Connection do
   let(:client) { instance_double('Ably::Realtime::Client', logger: double('logger').as_null_object) }
 
   subject do
     Ably::Realtime::Connection.new(client)
   end
 
   before do
-    expect(EventMachine::Timer).to receive(:new) # Connection Manager #initializer
     expect(EventMachine).to receive(:next_tick)  # non_blocking_loop_while for delivery of messages async
+    subject.__incoming_protocol_msgbus__.off
+    subject.__outgoing_protocol_msgbus__.off
   end
 
   describe 'callbacks' do
     specify 'are supported for valid STATE events' do
       state = nil