[Chat] Fix MongoDB identifier context in MessageNormalizer denormalize

Author: camilleislasse

src/chat/src/Bridge/MongoDb/MessageStore.php

@@ -79,7 +79,9 @@ public function load(): MessageBag
         ]);
 
         return new MessageBag(...array_map(
-            fn (array $message): MessageInterface => $this->serializer->denormalize($message, MessageInterface::class),
+            fn (array $message): MessageInterface => $this->serializer->denormalize($message, MessageInterface::class, context: [
+                'identifier' => '_id',
+            ]),
             $cursor->toArray(),
         ));
     }

src/chat/src/MessageNormalizer.php

@@ -74,8 +74,9 @@ public function denormalize(mixed $data, string $type, ?string $format = null, a
             default => throw new LogicException(\sprintf('Unknown message type "%s".', $type)),
         };
 
+        $identifier = $context['identifier'] ?? 'id';
         /** @var AbstractUid&TimeBasedUidInterface&Uuid $existingUuid */
-        $existingUuid = Uuid::fromString($data['id']);
+        $existingUuid = Uuid::fromString($data[$identifier]);
 
         $messageWithExistingUuid = $message->withId($existingUuid);
 

src/chat/tests/Bridge/MongoDb/MessageStoreTest.php

@@ -89,7 +89,7 @@ public function testMessageStoreCanLoad()
 
         $cursor = $this->createMock(CursorInterface::class);
         $cursor->expects($this->once())->method('toArray')->willReturn([
-            $serializer->normalize(Message::ofUser('Hello world')),
+            $serializer->normalize(Message::ofUser('Hello world'), context: ['identifier' => '_id']),
         ]);
 
         $collection = $this->createMock(Collection::class);

src/chat/tests/MessageNormalizerTest.php

@@ -76,4 +76,19 @@ public function testItCanDenormalize()
         $this->assertSame(Role::User, $message->getRole());
         $this->assertArrayHasKey('addedAt', $message->getMetadata()->all());
     }
+
+    public function testItCanDenormalizeWithCustomIdentifier()
+    {
+        $normalizer = new MessageNormalizer();
+        $message = Message::ofUser('Hello World');
+
+        // Normalize with _id (like MongoDB)
+        $payload = $normalizer->normalize($message, context: ['identifier' => '_id']);
+        $this->assertArrayHasKey('_id', $payload);
+        $this->assertArrayNotHasKey('id', $payload);
+
+        $denormalized = $normalizer->denormalize($payload, MessageInterface::class, context: ['identifier' => '_id']);
+
+        $this->assertSame($message->getId()->toRfc4122(), $denormalized->getId()->toRfc4122());
+    }
 }