symfony
diff --git a/‎src/Symfony/Component/Messenger/Tests/Transport/Serialization/SerializerTest.php
Copy file name to clipboardExpand all lines: src/Symfony/Component/Messenger/Tests/Transport/Serialization/SerializerTest.php
+52-8Lines changed: 52 additions & 8 deletions b/‎src/Symfony/Component/Messenger/Tests/Transport/Serialization/SerializerTest.php
Copy file name to clipboardExpand all lines: src/Symfony/Component/Messenger/Tests/Transport/Serialization/SerializerTest.php
+52-8Lines changed: 52 additions & 8 deletions
diff --git a/‎src/Symfony/Component/Messenger/Transport/Serialization/Serializer.php
Copy file name to clipboardExpand all lines: src/Symfony/Component/Messenger/Transport/Serialization/Serializer.php
+18-2Lines changed: 18 additions & 2 deletions b/‎src/Symfony/Component/Messenger/Transport/Serialization/Serializer.php
Copy file name to clipboardExpand all lines: src/Symfony/Component/Messenger/Transport/Serialization/Serializer.php
+18-2Lines changed: 18 additions & 2 deletions
@@ -19,6 +19,7 @@
 use Symfony\Component\Messenger\Transport\Serialization\Serializer;
 use Symfony\Component\Serializer as SerializerComponent;
 use Symfony\Component\Serializer\Normalizer\ObjectNormalizer;
+use Symfony\Component\Serializer\SerializerInterface as SerializerComponentInterface;
 
 class SerializerTest extends TestCase
 {
@@ -74,24 +75,67 @@ public function testUsesTheCustomFormatAndContext()
 
     public function testEncodedWithSymfonySerializerForStamps()
     {
-        $serializer = new Serializer();
+        $serializer = new Serializer(
+            $symfonySerializer = $this->createMock(SerializerComponentInterface::class)
+        );
 
-        $envelope = (new Envelope(new DummyMessage('Hello')))
+        $envelope = (new Envelope($message = new DummyMessage('test')))
             ->with($serializerStamp = new SerializerStamp([ObjectNormalizer::GROUPS => ['foo']]))
-            ->with($validationStamp = new ValidationStamp(['foo', 'bar']))
+            ->with($validationStamp = new ValidationStamp(['foo', 'bar']));
+
+
+        $symfonySerializer
+            ->expects($this->at(2))
+            ->method('serialize')->with(
+                $message,
+                'json',
+                [
+                    ObjectNormalizer::GROUPS => ['foo']
+                ]
+            )
         ;
 
         $encoded = $serializer->encode($envelope);
 
         $this->assertArrayHasKey('body', $encoded);
         $this->assertArrayHasKey('headers', $encoded);
         $this->assertArrayHasKey('type', $encoded['headers']);
-        $this->assertArrayHasKey('X-Message-Stamp-'.SerializerStamp::class, $encoded['headers']);
-        $this->assertArrayHasKey('X-Message-Stamp-'.ValidationStamp::class, $encoded['headers']);
+        $this->assertArrayHasKey('X-Message-Stamp-' . SerializerStamp::class, $encoded['headers']);
+        $this->assertArrayHasKey('X-Message-Stamp-' . ValidationStamp::class, $encoded['headers']);
+    }
 
-        $decoded = $serializer->decode($encoded);
+    public function testDecodeWithSymfonySerializerStamp()
+    {
+        $serializer = new Serializer(
+            $symfonySerializer = $this->createMock(SerializerComponentInterface::class)
+        );
+
+        $symfonySerializer
+            ->expects($this->at(0))
+            ->method('deserialize')
+            ->with('[{"context":{"groups":["foo"]}}]', SerializerStamp::class.'[]', 'json', [])
+            ->willReturn([ new SerializerStamp(['groups' => ['foo']]) ])
+        ;
+
+        $symfonySerializer
+            ->expects($this->at(1))
+            ->method('deserialize')->with(
+                '{}',
+                DummyMessage::class,
+                'json',
+                [
+                    ObjectNormalizer::GROUPS => ['foo']
+                ]
+            )
+            ->willReturn(new DummyMessage('test'))
+        ;
 
-        $this->assertEquals($serializerStamp, $decoded->last(SerializerStamp::class));
-        $this->assertEquals($validationStamp, $decoded->last(ValidationStamp::class));
+        $serializer->decode([
+            'body' => '{}',
+            'headers' => [
+                'type' => DummyMessage::class,
+                'X-Message-Stamp-'.SerializerStamp::class => '[{"context":{"groups":["foo"]}}]',
+            ]
+        ]);
     }
 }
@@ -15,6 +15,7 @@
 use Symfony\Component\Messenger\Exception\InvalidArgumentException;
 use Symfony\Component\Messenger\Exception\LogicException;
 use Symfony\Component\Messenger\Stamp\SerializerStamp;
+use Symfony\Component\Messenger\Stamp\StampInterface;
 use Symfony\Component\Serializer\Encoder\JsonEncoder;
 use Symfony\Component\Serializer\Encoder\XmlEncoder;
 use Symfony\Component\Serializer\Normalizer\ArrayDenormalizer;
@@ -69,10 +70,11 @@ public function decode(array $encodedEnvelope): Envelope
         }
 
         $stamps = $this->decodeStamps($encodedEnvelope);
+        $serializerStamp = $this->findFirstSerializerStamp($stamps);
 
         $context = $this->context;
-        if (isset($stamps[SerializerStamp::class])) {
-            $context = end($stamps[SerializerStamp::class])->getContext() + $context;
+        if ($serializerStamp !== null) {
+            $context = $serializerStamp->getContext() + $context;
         }
 
         $message = $this->serializer->deserialize($encodedEnvelope['body'], $encodedEnvelope['headers']['type'], $this->format, $context);
@@ -129,4 +131,18 @@ private function encodeStamps(Envelope $envelope): array
 
         return $headers;
     }
+
+    /**
+     * @param StampInterface[] $stamps
+     */
+    private function findFirstSerializerStamp(array $stamps): ?SerializerStamp
+    {
+        foreach ($stamps as $stamp) {
+            if ($stamp instanceof SerializerStamp) {
+                return $stamp;
+            }
+        }
+
+        return null;
+    }
 }