spec/utils/db_producer_spec.rb in deimos-ruby-1.10.1 vs spec/utils/db_producer_spec.rb in deimos-ruby-1.10.2

@@ -260,14 +260,16 @@
         with('my-topic', 'abc').once
       producer.process_topic('my-topic')
     end
 
     it 'should register an error if it gets an error' do
+      allow(producer).to receive(:shutdown_producer)
       expect(producer).to receive(:retrieve_messages).and_raise('OH NOES')
       expect(Deimos::KafkaTopicInfo).to receive(:register_error).
         with('my-topic', 'abc')
       expect(producer).not_to receive(:produce_messages)
       producer.process_topic('my-topic')
+      expect(producer).to have_received(:shutdown_producer)
     end
 
     it 'should move on if it gets a partial batch' do
       expect(producer).to receive(:retrieve_messages).ordered.
         and_return([Deimos::KafkaMessage.new(