Allow assertPublishedOnTimes to be used with batch messages

src/Producers/MessageBatch.php

@@ -32,15 +32,15 @@ public function __construct()
      * @param Message $message
      * @return void
      */
-    public function push(Message $message)
+    public function push(Message $message): void
     {
         $this->messages->push($message);
     }
 
     /**
      * Returns all messages from batch before producing them to kafka
      *
-     * @return SplDoublyLinkedList
+     * @return SplDoublyLinkedList<\Junges\Kafka\Message\Message>
      */
     public function getMessages(): SplDoublyLinkedList
     {

src/Support/Testing/Fakes/ProducerFake.php

@@ -47,7 +47,9 @@ public function produceBatch(MessageBatch $messageBatch): int
         if ($this->producerCallback !== null) {
             $callback = $this->producerCallback;
 
+            /** @var Message $message */
             foreach ($messageBatch->getMessages() as $message) {
+                $message->setTopicName($this->topic);
                 $callback($message);
                 $produced++;
             }

tests/KafkaFakeTest.php

@@ -152,6 +152,30 @@ public function testAssertPublishedOnTimes(): void
         }
     }
 
+    public function testAssertPublishedOnTimesForBatchMessages(): void
+    {
+        $producer = $this->fake->publishOn('batch-topic')
+            ->withConfigOption('key', 'value');
+
+        $message = new Message(
+            headers: ['header-key' => 'header-value'],
+            body: ['body-key' => 'body-value'],
+            key: 2
+        );
+
+        $messageBatch = new MessageBatch();
+        $messageBatch->push($message);
+        $messageBatch->push($message);
+
+        $producer->sendBatch($messageBatch);
+
+        $this->fake->assertPublishedTimes(2);
+        $this->fake->assertpublished();
+        $this->fake->assertPublishedOnTimes('batch-topic', 2);
+        $this->fake->assertPublishedOn('batch-topic');
+
+    }
+
     public function testICanPerformAssertionsUsingAssertPublishedOn(): void
     {
         $producer = $this->fake->publishOn('topic')