From 34eaf6ba3d1bf223f4b7bfcb71870f92a7dfaea6 Mon Sep 17 00:00:00 2001
From: Maciej Mensfeld <maciej@mensfeld.pl>
Date: Mon, 16 Sep 2024 10:31:01 +0200
Subject: [PATCH] stabilize spec for slow ci

---
 spec/lib/waterdrop/producer/transactions_spec.rb | 6 ++----
 spec/support/factories/producer.rb               | 4 +++-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/spec/lib/waterdrop/producer/transactions_spec.rb b/spec/lib/waterdrop/producer/transactions_spec.rb
index 2392739..1db3833 100644
--- a/spec/lib/waterdrop/producer/transactions_spec.rb
+++ b/spec/lib/waterdrop/producer/transactions_spec.rb
@@ -535,17 +535,15 @@
     end
 
     context 'when we abort the nested transaction' do
+      subject(:producer) { build(:transactional_producer, queue_buffering_max_ms: 5_000) }
+
       it 'expect to abort all levels' do
         handlers = []
 
         producer.transaction do
           handlers << producer.produce_async(topic: 'example_topic', payload: 'data')
 
-          sleep(0.1)
-
           producer.transaction do
-            sleep(0.1)
-
             handlers << producer.produce_async(topic: 'example_topic', payload: 'data')
             raise(WaterDrop::AbortTransaction)
           end
diff --git a/spec/support/factories/producer.rb b/spec/support/factories/producer.rb
index 2fef6f4..82f88d5 100644
--- a/spec/support/factories/producer.rb
+++ b/spec/support/factories/producer.rb
@@ -43,6 +43,7 @@
       transaction_timeout_ms { 30_000 }
       request_required_acks { 'all' }
       idempotent { true }
+      queue_buffering_max_ms { 5 }
     end
 
     kafka do
@@ -52,7 +53,8 @@
         'transactional.id': transactional_id,
         'transaction.timeout.ms': transaction_timeout_ms,
         'message.timeout.ms': transaction_timeout_ms,
-        'enable.idempotence': idempotent
+        'enable.idempotence': idempotent,
+        'queue.buffering.max.ms': queue_buffering_max_ms
       }
     end
   end