CRUD prose tests 3 and 4

Author: jmikola

tests/SpecTests/Crud/Prose3_BulkWriteSplitsOnMaxWriteBatchSizeTest.php

@@ -0,0 +1,68 @@
+<?php
+
+namespace MongoDB\Tests\SpecTests\Crud;
+
+use MongoDB\BulkWriteCommandBuilder;
+use MongoDB\Driver\Monitoring\CommandFailedEvent;
+use MongoDB\Driver\Monitoring\CommandStartedEvent;
+use MongoDB\Driver\Monitoring\CommandSubscriber;
+use MongoDB\Driver\Monitoring\CommandSucceededEvent;
+use MongoDB\Tests\SpecTests\FunctionalTestCase;
+
+/**
+ * Prose test 3: MongoClient.bulkWrite batch splits a writeModels input with greater than maxWriteBatchSize operations
+ *
+ * @see https://github.com/mongodb/specifications/blob/master/source/crud/tests/README.md
+ */
+class Prose3_BulkWriteSplitsOnMaxWriteBatchSizeTest extends FunctionalTestCase
+{
+    public function testSplitOnMaxWriteBatchSize(): void
+    {
+        if ($this->isServerless()) {
+            $this->markTestSkipped('bulkWrite command is not supported');
+        }
+
+        $this->skipIfServerVersion('<', '8.0', 'bulkWrite command is not supported');
+
+        $client = self::createTestClient();
+
+        $maxWriteBatchSize = $this->getPrimaryServer()->getInfo()['maxWriteBatchSize'] ?? null;
+        self::assertIsInt($maxWriteBatchSize);
+
+        $bulkWrite = BulkWriteCommandBuilder::createWithCollection($client->selectCollection($this->getDatabaseName(), $this->getCollectionName()));
+
+        for ($i = 0; $i < $maxWriteBatchSize + 1; ++$i) {
+            $bulkWrite->insertOne(['a' => 'b']);
+        }
+
+        $subscriber = new class implements CommandSubscriber {
+            public array $commandStartedEvents = [];
+
+            public function commandStarted(CommandStartedEvent $event): void
+            {
+                $this->commandStartedEvents[] = $event;
+            }
+
+            public function commandSucceeded(CommandSucceededEvent $event): void
+            {
+            }
+
+            public function commandFailed(CommandFailedEvent $event): void
+            {
+            }
+        };
+
+        $client->addSubscriber($subscriber);
+
+        $result = $client->bulkWrite($bulkWrite);
+
+        self::assertSame($maxWriteBatchSize + 1, $result->getInsertedCount());
+        self::assertCount(2, $subscriber->commandStartedEvents);
+        [$firstEvent, $secondEvent] = $subscriber->commandStartedEvents;
+        self::assertIsArray($firstCommandOps = $firstEvent->getCommand()->ops ?? null);
+        self::assertCount($maxWriteBatchSize, $firstCommandOps);
+        self::assertIsArray($secondCommandOps = $secondEvent->getCommand()->ops ?? null);
+        self::assertCount(1, $secondCommandOps);
+        self::assertEquals($firstEvent->getOperationId(), $secondEvent->getOperationId());
+    }
+}

tests/SpecTests/Crud/Prose4_BulkWriteSplitsOnMaxMessageSizeBytesTest.php

@@ -0,0 +1,74 @@
+<?php
+
+namespace MongoDB\Tests\SpecTests\Crud;
+
+use MongoDB\BulkWriteCommandBuilder;
+use MongoDB\Driver\Monitoring\CommandFailedEvent;
+use MongoDB\Driver\Monitoring\CommandStartedEvent;
+use MongoDB\Driver\Monitoring\CommandSubscriber;
+use MongoDB\Driver\Monitoring\CommandSucceededEvent;
+use MongoDB\Tests\SpecTests\FunctionalTestCase;
+
+use function str_repeat;
+
+/**
+ * Prose test 4: MongoClient.bulkWrite batch splits when an ops payload exceeds maxMessageSizeBytes
+ *
+ * @see https://github.com/mongodb/specifications/blob/master/source/crud/tests/README.md
+ */
+class Prose4_BulkWriteSplitsOnMaxMessageSizeBytesTest extends FunctionalTestCase
+{
+    public function testSplitOnMaxWriteBatchSize(): void
+    {
+        if ($this->isServerless()) {
+            $this->markTestSkipped('bulkWrite command is not supported');
+        }
+
+        $this->skipIfServerVersion('<', '8.0', 'bulkWrite command is not supported');
+
+        $client = self::createTestClient();
+
+        $hello = $this->getPrimaryServer()->getInfo();
+        self::assertIsInt($maxBsonObjectSize = $hello['maxBsonObjectSize'] ?? null);
+        self::assertIsInt($maxMessageSizeBytes = $hello['maxMessageSizeBytes'] ?? null);
+
+        $numModels = (int) ($maxMessageSizeBytes / $maxBsonObjectSize + 1);
+        $document = ['a' => str_repeat('b', $maxBsonObjectSize - 500)];
+
+        $bulkWrite = BulkWriteCommandBuilder::createWithCollection($client->selectCollection($this->getDatabaseName(), $this->getCollectionName()));
+
+        for ($i = 0; $i < $numModels; ++$i) {
+            $bulkWrite->insertOne($document);
+        }
+
+        $subscriber = new class implements CommandSubscriber {
+            public array $commandStartedEvents = [];
+
+            public function commandStarted(CommandStartedEvent $event): void
+            {
+                $this->commandStartedEvents[] = $event;
+            }
+
+            public function commandSucceeded(CommandSucceededEvent $event): void
+            {
+            }
+
+            public function commandFailed(CommandFailedEvent $event): void
+            {
+            }
+        };
+
+        $client->addSubscriber($subscriber);
+
+        $result = $client->bulkWrite($bulkWrite);
+
+        self::assertSame($numModels, $result->getInsertedCount());
+        self::assertCount(2, $subscriber->commandStartedEvents);
+        [$firstEvent, $secondEvent] = $subscriber->commandStartedEvents;
+        self::assertIsArray($firstCommandOps = $firstEvent->getCommand()->ops ?? null);
+        self::assertCount($numModels - 1, $firstCommandOps);
+        self::assertIsArray($secondCommandOps = $secondEvent->getCommand()->ops ?? null);
+        self::assertCount(1, $secondCommandOps);
+        self::assertEquals($firstEvent->getOperationId(), $secondEvent->getOperationId());
+    }
+}