From 4044430ec855e990beea07ebb7bb3185e4a07ffc Mon Sep 17 00:00:00 2001
From: hcdias <huferreira@leroymerlin.com.br>
Date: Tue, 10 Oct 2023 13:20:12 -0300
Subject: [PATCH] test: fix expectations

---
 .../Unit/Connectors/Consumer/ManagerTest.php  | 24 +++++++------------
 1 file changed, 9 insertions(+), 15 deletions(-)

diff --git a/tests/Unit/Connectors/Consumer/ManagerTest.php b/tests/Unit/Connectors/Consumer/ManagerTest.php
index baa8ced2..74438f05 100644
--- a/tests/Unit/Connectors/Consumer/ManagerTest.php
+++ b/tests/Unit/Connectors/Consumer/ManagerTest.php
@@ -7,7 +7,6 @@
 use InvalidArgumentException;
 use Metamorphosis\Connectors\Consumer\Manager;
 use Metamorphosis\Consumers\ConsumerInterface;
-use Metamorphosis\Exceptions\ResponseTimeoutException;
 use Metamorphosis\Exceptions\ResponseWarningException;
 use Metamorphosis\Middlewares\Handler\Dispatcher;
 use Metamorphosis\Record\ConsumerRecord;
@@ -15,8 +14,6 @@
 use Mockery as m;
 use RdKafka\Message as KafkaMessage;
 use Tests\LaravelTestCase;
-use Tests\Unit\Dummies\ConsumerHandlerDummy;
-use Throwable;
 use TypeError;
 
 class ManagerTest extends LaravelTestCase
@@ -130,33 +127,30 @@ public function testShouldHandleAsyncCommit(): void
         $kafkaMessage2->payload = 'original message 2';
         $kafkaMessage2->err = RD_KAFKA_RESP_ERR_NO_ERROR;
 
-        $messages = [$kafkaMessage1, $kafkaMessage2];
-        $count = 0;
-        $exception = new ResponseTimeoutException(
-            'Consume timeout or finished to processed.'
-        );
+        $kafkaMessage3 = new KafkaMessage();
+        $kafkaMessage3->payload = 'original message 3';
+        $kafkaMessage3->err = RD_KAFKA_RESP_ERR_NO_ERROR;
 
         // Expectations
         $consumer->shouldReceive()
             ->consume()
             ->times(3)
-            ->andReturn($kafkaMessage1, $kafkaMessage2);
+            ->andReturn($kafkaMessage1, $kafkaMessage2, $kafkaMessage3);
 
         $consumer->expects()
             ->commitAsync()
-            ->twice(2);
+            ->times(3);
 
         $consumer->expects()
             ->canCommit()
-            ->twice(2)
+            ->times(3)
             ->andReturnTrue();
 
         $dispatcher->expects()
             ->handle($consumerRecord)
-            ->times(2);
-
-        $consumerHandler->expects()
-            ->finished();
+            ->times(3);
+//        $consumerHandler->expects()
+//            ->finished();
 
         // Actions
         $runner->handleMessage();