Fix mongo createDocuments losing internalIds

Author: PineappleIOnic

src/Database/Adapter/Mongo.php

@@ -757,7 +757,18 @@ public function createDocuments(string $collection, array $documents): array
         $name = $this->getNamespace() . '_' . $this->filter($collection);
 
         $records = [];
+        $hasInternalId = null;
+        $documents = array_map(fn ($doc) => clone $doc, $documents);
+
         foreach ($documents as $document) {
+            $internalId = $document->getInternalId();
+
+            if ($hasInternalId === null) {
+                $hasInternalId = !empty($internalId);
+            } elseif ($hasInternalId == empty($internalId)) {
+                throw new DatabaseException('All documents must have an internalId if one is set');
+            }
+
             $document->removeAttribute('$internalId');
 
             if ($this->sharedTables) {
@@ -767,6 +778,10 @@ public function createDocuments(string $collection, array $documents): array
             $record = $this->replaceChars('$', '_', (array)$document);
             $record = $this->timeToMongo($record);
 
+            if (!empty($internalId)) {
+                $record['_id'] = $internalId;
+            }
+
             $records[] = $this->removeNullKeys($record);
         }
 

tests/e2e/Adapter/Base.php

@@ -2291,16 +2291,14 @@ public function testCreateDocumentsWithDifferentAttributes(): void
         try {
             static::getDatabase()->createDocuments($collection, $documents);
             $this->fail('Failed to throw exception');
-        } catch (Throwable $e) {
-            $this->assertTrue($e instanceof DatabaseException);
+        } catch (DatabaseException $e) {
         }
 
         $documents = array_reverse($documents);
         try {
             static::getDatabase()->createDocuments($collection, $documents);
             $this->fail('Failed to throw exception');
-        } catch (Throwable $e) {
-            $this->assertTrue($e instanceof DatabaseException);
+        } catch (DatabaseException $e) {
         }
 
         static::getDatabase()->deleteCollection($collection);