spec/unit/realtime/channel_spec.rb in ably-0.8.5 vs spec/unit/realtime/channel_spec.rb in ably-0.8.6

@@ -66,37 +66,38 @@
     end
   end
 
   describe '#publish name argument' do
     let(:encoded_value) { random_str.encode(encoding) }
+    let(:message) { instance_double('Ably::Models::Message', client_id: nil) }
 
     before do
-      allow(subject).to receive(:create_message).and_return('message_stubbed')
+      allow(subject).to receive(:create_message).and_return(message)
       allow(subject).to receive(:attach).and_return(:true)
     end
 
     context 'as UTF_8 string' do
       let(:encoding) { Encoding::UTF_8 }
 
       it 'is permitted' do
-        expect(subject.publish(encoded_value, 'data')).to eql('message_stubbed')
+        expect(subject.publish(encoded_value, 'data')).to eql(message)
       end
     end
 
     context 'as SHIFT_JIS string' do
       let(:encoding) { Encoding::SHIFT_JIS }
 
       it 'is permitted' do
-        expect(subject.publish(encoded_value, 'data')).to eql('message_stubbed')
+        expect(subject.publish(encoded_value, 'data')).to eql(message)
       end
     end
 
     context 'as ASCII_8BIT string' do
       let(:encoding) { Encoding::ASCII_8BIT }
 
       it 'is permitted' do
-        expect(subject.publish(encoded_value, 'data')).to eql('message_stubbed')
+        expect(subject.publish(encoded_value, 'data')).to eql(message)
       end
     end
 
     context 'as Integer' do
       let(:encoded_value) { 1 }
@@ -108,10 +109,10 @@
 
     context 'as Nil' do
       let(:encoded_value) { nil }
 
       it 'is permitted' do
-        expect(subject.publish(encoded_value, 'data')).to eql('message_stubbed')
+        expect(subject.publish(encoded_value, 'data')).to eql(message)
       end
     end
   end
 
   describe 'callbacks' do