Merge pull request #25 from MacPaw/feat(PLATECO-1668)/move-messenger-transport-factory-decorator-to-package

Feat(plateco-1668): add default schema stamp

Author: serhiidonii

config/services.yaml

@@ -23,5 +23,7 @@ services:
       - { name: kernel.event_subscriber }
 
   Macpaw\SchemaContextBundle\Messenger\Middleware\BaggageSchemaMiddleware:
+    arguments:
+      $defaultSchema: '%schema_context.default_schema%'
     tags:
       - { name: messenger.middleware }

src/Messenger/Middleware/BaggageSchemaMiddleware.php

@@ -16,7 +16,8 @@ class BaggageSchemaMiddleware implements MiddlewareInterface
 {
     public function __construct(
         private BaggageSchemaResolver $baggageSchemaResolver,
-        private BaggageCodec $baggageCodec
+        private BaggageCodec $baggageCodec,
+        private string $defaultSchema,
     ) {
     }
 
@@ -43,6 +44,8 @@ public function handle(Envelope $envelope, StackInterface $stack): Envelope
 
         if ($schema !== null && $schema !== '') {
             $envelope = $envelope->with(new BaggageSchemaStamp($schema, $baggage));
+        } else {
+            $envelope = $envelope->with(new BaggageSchemaStamp($this->defaultSchema, $baggage));
         }
 
         return $stack->next()->handle($envelope, $stack);

tests/Messenger/Middleware/BaggageSchemaMiddlewareTest.php

@@ -12,6 +12,7 @@
 use Symfony\Component\Messenger\Envelope;
 use Symfony\Component\Messenger\Middleware\MiddlewareInterface;
 use Symfony\Component\Messenger\Middleware\StackInterface;
+use Symfony\Component\Messenger\Stamp\ReceivedStamp;
 
 class BaggageSchemaMiddlewareTest extends TestCase
 {
@@ -25,9 +26,10 @@ public function testSchemaIsSetFromStamp(): void
 
         $resolver = new BaggageSchemaResolver();
         $baggageCodec = new BaggageCodec();
-        $middleware = new BaggageSchemaMiddleware($resolver, $baggageCodec);
+        $middleware = new BaggageSchemaMiddleware($resolver, $baggageCodec, 'public');
         $stamp = new BaggageSchemaStamp($schema, $rawBaggage);
         $envelope = (new Envelope(new \stdClass()))->with($stamp);
+        $envelope = $envelope->with(new ReceivedStamp('async'));
         $stack = $this->createMock(StackInterface::class);
         $nextMiddleware = new class implements MiddlewareInterface {
             public function handle(Envelope $envelope, StackInterface $stack): Envelope
@@ -68,7 +70,7 @@ public function testSchemaStampIsInjectedIfMissing(): void
             ->setSchema($schema)
             ->setBaggage($baggage);
         $baggageCodec = new BaggageCodec();
-        $middleware = new BaggageSchemaMiddleware($resolver, $baggageCodec);
+        $middleware = new BaggageSchemaMiddleware($resolver, $baggageCodec, 'public');
         $originalEnvelope = new Envelope(new \stdClass());
         $stack = $this->createMock(StackInterface::class);
 
@@ -91,4 +93,32 @@ public function handle(Envelope $envelope, StackInterface $stack): Envelope
         $this->assertSame($schema, $stamp->schema);
         $this->assertSame($rawBaggage, $stamp->baggage);
     }
+
+    public function testSchemaStampIsDefaultSchema(): void
+    {
+        $resolver = new BaggageSchemaResolver();
+        $baggageCodec = new BaggageCodec();
+        $middleware = new BaggageSchemaMiddleware($resolver, $baggageCodec, 'public');
+        $originalEnvelope = new Envelope(new \stdClass());
+        $stack = $this->createMock(StackInterface::class);
+
+        $stack->expects($this->once())
+            ->method('next')
+            ->willReturnCallback(function () {
+                return new class implements MiddlewareInterface {
+                    public function handle(Envelope $envelope, StackInterface $stack): Envelope
+                    {
+                        return $envelope;
+                    }
+                };
+            });
+
+        $resultEnvelope = $middleware->handle($originalEnvelope, $stack);
+
+        $stamp = $resultEnvelope->last(BaggageSchemaStamp::class);
+
+        $this->assertInstanceOf(BaggageSchemaStamp::class, $stamp);
+        $this->assertSame('public', $stamp->schema);
+        $this->assertSame('', $stamp->baggage);
+    }
 }