spec/support/shared/destination_examples.rb in message-driver-0.4.0 vs spec/support/shared/destination_examples.rb in message-driver-0.5.0

@@ -1,42 +1,82 @@
-shared_examples 'a destination' do
-  its(:adapter) { should be adapter }
+RSpec.shared_examples 'a destination' do
+  describe '#adapter' do
+    it { expect(subject.adapter).to be adapter }
+  end
 
   describe '#pop_message' do
     let(:body) { 'The message body' }
-    let(:headers) { { 'foo' => 'bar', 'bar' => 'baz'} }
-    let(:properties) { {persistent: true, client_ack: true} }
+    let(:headers) { { 'foo' => 'bar', 'bar' => 'baz' } }
+    let(:properties) { { persistent: true, client_ack: true } }
 
     before do
       destination.publish(body, headers, properties)
     end
 
     context 'the result' do
       subject(:message) { destination.pop_message }
 
-      it { should be_a MessageDriver::Message::Base }
-      its(:body) { should eq(body) }
-      its(:headers) { should include(headers) }
-      its(:properties) { should_not be_nil }
+      it { is_expected.to be_a MessageDriver::Message::Base }
+
+      describe '#body' do
+        it { expect(subject.body).to eq(body) }
+      end
+
+      describe '#headers' do
+        it { expect(subject.headers).to include(headers) }
+      end
+
+      describe '#properties' do
+        it { expect(subject.properties).not_to be_nil }
+      end
     end
   end
 end
 
-shared_examples "doesn't support #message_count" do
+RSpec.shared_examples "doesn't support #message_count" do
   describe '#message_count' do
     it 'raises an error' do
-      expect {
+      expect do
         destination.message_count
-      }.to raise_error "#message_count is not supported by #{destination.class}"
+      end.to raise_error "#message_count is not supported by #{destination.class}"
     end
   end
 end
 
-shared_examples 'supports #message_count' do
+RSpec.shared_examples 'supports #message_count' do
   it "reports it's message_count" do
-    expect {
+    expect do
       destination.publish('msg1')
       destination.publish('msg2')
       pause_if_needed
-    }.to change{destination.message_count}.by(2)
+    end.to change { destination.message_count }.by(2)
+  end
+end
+
+RSpec.shared_examples "doesn't support #consumer_count" do
+  describe '#consumer_count' do
+    it 'raises an error' do
+      expect do
+        destination.consumer_count
+      end.to raise_error "#consumer_count is not supported by #{destination.class}"
+    end
+  end
+end
+
+RSpec.shared_examples 'supports #consumer_count' do
+  describe '#consumer_count' do
+    it "reports it's consumer count" do
+      consumer1 = ->(_) {}
+      consumer2 = ->(_) {}
+      sub1 = nil
+      sub2 = nil
+      expect do
+        sub1 = destination.subscribe(&consumer1)
+        sub2 = destination.subscribe(&consumer2)
+      end.to change { destination.consumer_count }.by(2)
+      expect do
+        sub1.unsubscribe
+        sub2.unsubscribe
+      end.to change { destination.consumer_count }.by(-2)
+    end
   end
 end