cleanup

spec/lib/waterdrop/instrumentation/vendors/datadog/metrics_listener_spec.rb

@@ -96,7 +96,7 @@ def initialize
     let(:counts) { dummy_client.buffer[:count] }
     let(:histograms) { dummy_client.buffer[:histogram] }
     let(:guages) { dummy_client.buffer[:gauge] }
-    let(:broker_tag) { { tags: %w[broker:localhost:9092] } }
+    let(:broker_tag) { { tags: %w[broker:127.0.0.1:9092] } }
 
     # We add all expectations in one example not to sleep each time
     it 'expect to have proper metrics in place' do

spec/lib/waterdrop/producer/async_spec.rb

@@ -27,7 +27,11 @@
     end
 
     context 'when producing with good middleware' do
+      let(:message) { build(:valid_message, payload: nil) }
+
       before do
+        producer.produce_sync(topic: message[:topic], payload: nil)
+
         mid = lambda do |msg|
           msg[:partition_key] = '1'
           msg
@@ -36,12 +40,12 @@
         producer.middleware.append mid
       end
 
-      let(:message) { build(:valid_message, payload: nil) }
-
       it { expect(delivery).to be_a(Rdkafka::Producer::DeliveryHandle) }
     end
 
     context 'when producing with corrupted middleware' do
+      let(:message) { build(:valid_message, payload: nil) }
+
       before do
         mid = lambda do |msg|
           msg[:partition_key] = -1
@@ -51,8 +55,6 @@
         producer.middleware.append mid
       end
 
-      let(:message) { build(:valid_message, payload: nil) }
-
       it { expect { delivery }.to raise_error(WaterDrop::Errors::MessageInvalidError) }
     end
 

spec/lib/waterdrop/producer/sync_spec.rb

@@ -58,6 +58,8 @@
 
     let(:message) { build(:valid_message, partition_key: rand.to_s, topic: 'example_topic') }
 
+    before { producer.produce_sync(topic: 'example_topic', payload: '1') }
+
     it { expect(delivery).to be_a(Rdkafka::Producer::DeliveryReport) }
   end