Retryable reads prose tests

Author: jmikola

tests/SpecTests/RetryableReads/Prose2_RetryOnMongosTest.php

@@ -0,0 +1,167 @@
+<?php
+
+namespace MongoDB\Tests\SpecTests\RetryableReads;
+
+use MongoDB\Driver\Exception\CommandException;
+use MongoDB\Driver\Monitoring\CommandFailedEvent;
+use MongoDB\Driver\Monitoring\CommandStartedEvent;
+use MongoDB\Driver\Monitoring\CommandSubscriber;
+use MongoDB\Driver\Monitoring\CommandSucceededEvent;
+use MongoDB\Driver\Server;
+use MongoDB\Tests\SpecTests\FunctionalTestCase;
+
+use function assert;
+use function count;
+
+/**
+ * Prose test 2: Retry on different or same mongos
+ *
+ * @see https://github.com/mongodb/specifications/blob/master/source/retryable-writes/tests/README.md
+ */
+class Prose2_RetryOnMongosTest extends FunctionalTestCase
+{
+    public const HOST_UNREACHABLE = 6;
+
+    public function testRetryOnDifferentMongos(): void
+    {
+        if (! $this->isMongos()) {
+            $this->markTestSkipped('Test requires connections to mongos');
+        }
+
+        $this->skipIfServerVersion('<', '4.1.7', 'Test requires mongos support for configureFailPoint');
+
+        /* By default, the Manager under test is created with a single-mongos
+         * URI. Explicitly create a Client with multiple mongoses and invoke
+         * server selection to initialize SDAM. */
+        $client = static::createTestClient(static::getUri(true), ['retryReads' => true]);
+        $client->getManager()->selectServer();
+
+        /* Step 1: Select servers for each mongos in the cluster.
+         *
+         * TODO: Support topologies with 3+ servers by selecting only two and
+         * recreating a client URI.
+         */
+        $servers = $client->getManager()->getServers();
+        assert(count($servers) === 2);
+
+        // Step 2: Configure the following fail point on each mongos
+        foreach ($servers as $server) {
+            $this->configureFailPoint(
+                [
+                    'configureFailPoint' => 'failCommand',
+                    'mode' => ['times' => 1],
+                    'data' => [
+                        'failCommands' => ['find'],
+                        'errorCode' => self::HOST_UNREACHABLE,
+                    ],
+                ],
+                $server
+            );
+        }
+
+        /* Step 3: Use the previously created client with retryReads=true,
+         * which is connected to a cluster with two mongoses */
+
+        // Step 4: Enable failed command event monitoring for client
+        $subscriber = new class implements CommandSubscriber {
+            public $commandFailedServers = [];
+
+            public function commandStarted(CommandStartedEvent $event): void
+            {
+            }
+
+            public function commandSucceeded(CommandSucceededEvent $event): void
+            {
+            }
+
+            public function commandFailed(CommandFailedEvent $event): void
+            {
+                $this->commandFailedServers[] = $event->getServer();
+            }
+        };
+
+        $client->addSubscriber($subscriber);
+
+        // Step 5: Execute a find command. Assert that the command failed.
+        try {
+            $client->selectCollection($this->getDatabaseName(), $this->getCollectionName())->find(['x' => 1]);
+            $this->fail('BulkWriteException was not thrown');
+        } catch (CommandException $e) {
+            $this->assertSame(self::HOST_UNREACHABLE, $e->getCode());
+        }
+
+        $client->removeSubscriber($subscriber);
+
+        /* Step 6: Assert that two failed command events occurred. Assert that
+         * the failed command events occurred on different mongoses. */
+        $this->assertCount(2, $subscriber->commandFailedServers);
+        $this->assertNotEquals($subscriber->commandFailedServers[0], $subscriber->commandFailedServers[1]);
+
+        // Step 7: The fail points will be disabled during tearDown()
+    }
+
+    public function testRetryOnSameMongos(): void
+    {
+        if (! $this->isMongos()) {
+            $this->markTestSkipped('Test requires connections to mongos');
+        }
+
+        $this->skipIfServerVersion('<', '4.1.7', 'Test requires mongos support for configureFailPoint');
+
+        // Step 1: Create a client that connects to a single mongos
+        $client = static::createTestClient(null, ['directConnection' => false, 'retryReads' => true]);
+        $server = $client->getManager()->selectServer();
+
+        // Step 2: Configure the following fail point on the mongos
+        $this->configureFailPoint(
+            [
+                'configureFailPoint' => 'failCommand',
+                'mode' => ['times' => 1],
+                'data' => [
+                    'failCommands' => ['find'],
+                    'errorCode' => self::HOST_UNREACHABLE,
+                ],
+            ],
+            $server,
+        );
+
+        // Step 3 is omitted because we can re-use the same client
+
+        // Step 4: Enable succeeded and failed command event monitoring
+        $subscriber = new class implements CommandSubscriber {
+            public Server $commandSucceededServer;
+            public Server $commandFailedServer;
+
+            public function commandStarted(CommandStartedEvent $event): void
+            {
+            }
+
+            public function commandSucceeded(CommandSucceededEvent $event): void
+            {
+                $this->commandSucceededServer = $event->getServer();
+            }
+
+            public function commandFailed(CommandFailedEvent $event): void
+            {
+                $this->commandFailedServer = $event->getServer();
+            }
+        };
+
+        $client->addSubscriber($subscriber);
+
+        // Step 5: Execute a find command. Assert that the command succeeded.
+        $cursor = $client->selectCollection($this->getDatabaseName(), $this->getCollectionName())->find(['x' => 1]);
+        $this->assertSame([], $cursor->toArray());
+
+        $client->removeSubscriber($subscriber);
+
+        /* Step 6: Assert that exactly one failed command event and one
+         * succeeded command event occurred. Assert that both events occurred on
+         * the same mongos. */
+        $this->assertNotNull($subscriber->commandSucceededServer);
+        $this->assertNotNull($subscriber->commandFailedServer);
+        $this->assertEquals($subscriber->commandSucceededServer, $subscriber->commandFailedServer);
+
+        // Step 7: The fail point will be disabled during tearDown()
+    }
+}