Add macroable trait to Kafka class (#221)

In order to make use of macros the Illuminate Macroable trait has been added to Junges\Kafka\Kafka. Co-authored-by: Alexander Lentner <[email protected]>
mateusjunges · Sep 23, 2023 · 445aa85 · 445aa85
1 parent 5097762
commit 445aa85
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/Kafka.php b/src/Kafka.php
@@ -2,6 +2,7 @@
 
 namespace Junges\Kafka;
 
+use Illuminate\Support\Traits\Macroable;
 use Junges\Kafka\Consumers\ConsumerBuilder;
 use Junges\Kafka\Contracts\CanConsumeMessagesFromKafka;
 use Junges\Kafka\Contracts\CanProduceMessages;
@@ -10,6 +11,8 @@
 
 class Kafka implements CanPublishMessagesToKafka, CanConsumeMessagesFromKafka
 {
+    use Macroable;
+
     /**
      * Creates a new ProducerBuilder instance, setting brokers and topic.
      *

diff --git a/tests/KafkaTest.php b/tests/KafkaTest.php
@@ -3,6 +3,7 @@
 namespace Junges\Kafka\Tests;
 
 use Illuminate\Support\Str;
+use Junges\Kafka\Config\Sasl;
 use Junges\Kafka\Consumers\ConsumerBuilder;
 use Junges\Kafka\Contracts\KafkaProducerMessage;
 use Junges\Kafka\Exceptions\CouldNotPublishMessage;
@@ -293,4 +294,18 @@ public function testSendMessageBatch()
 
         Kafka::publishOn('test')->withBodyKey('foo', 'bar')->sendBatch($messageBatch);
     }
+
+    public function testMacro()
+    {
+        $sasl = new Sasl(username: 'username', password: 'password', mechanisms: 'mechanisms');
+
+        Kafka::macro('defaultProducer', function (string $topic) use ($sasl) {
+            return $this->publishOn($topic)->withSasl($sasl);
+        });
+
+        $producer = Kafka::defaultProducer('test-topic');
+
+        $this->assertInstanceOf(ProducerBuilder::class, $producer);
+        $this->assertEquals($sasl, $this->getPropertyWithReflection('saslConfig', $producer));
+    }
 }