Fix tests

mateusjunges · Jan 9, 2024 · 08a02ae · 08a02ae
1 parent ffece55
commit 08a02ae
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/Console/Commands/ConsumerCommand.php b/src/Console/Commands/ConsumerCommand.php
@@ -57,7 +57,7 @@ public function handle()
             return;
         }
 
-        $parsedOptions = array_map(fn ($value) => $value === '?' ? null : $value, $this->options());
+        $parsedOptions = array_map($this->parseOptions(...), $this->options());
 
         $options = new Options($parsedOptions, $this->config);
 
@@ -85,4 +85,17 @@ public function handle()
 
         $consumer->consume();
     }
+
+    private function parseOptions(int|string|null $option): int|string|null
+    {
+        if ($option === '?') {
+            return null;
+        }
+
+        if (is_numeric($option)) {
+            return (int) $option;
+        }
+
+        return $option;
+    }
 }
diff --git a/tests/Consumers/ConsumerTest.php b/tests/Consumers/ConsumerTest.php
@@ -317,7 +317,7 @@ public function testCanStopConsumeIfMaxTimeReached()
         $this->mockProducer();
 
         $fakeHandler = new CallableConsumer(
-            function (KafkaConsumerMessage $message) {
+            function (ConsumerMessage $message) {
                 sleep(2);
             },
             []