Cleanup extra databases used by transaction examples tests

Author: GromNaN

tests/DocumentationExamplesTest.php

@@ -12,8 +12,10 @@
 use MongoDB\Driver\Exception\Exception;
 use MongoDB\Driver\ReadPreference;
 use MongoDB\Driver\WriteConcern;
+use MongoDB\Operation\DropDatabase;
 use MongoDB\Tests\SpecTests\ClientSideEncryptionSpecTest;
 
+use function array_unique;
 use function base64_decode;
 use function in_array;
 use function microtime;
@@ -28,9 +30,14 @@
  * @see https://jira.mongodb.org/browse/DRIVERS-356
  * @see https://jira.mongodb.org/browse/DRIVERS-488
  * @see https://jira.mongodb.org/browse/DRIVERS-547
+ *
+ * Note: Example should use the "inventory" collection in the default database unless they need multiple databases.
  */
 class DocumentationExamplesTest extends FunctionalTestCase
 {
+    /** @var string[] Name of the databases created during the test that will be dropped in tearDown  */
+    private $dbNames = [];
+
     public function setUp(): void
     {
         parent::setUp();
@@ -46,6 +53,13 @@ public function tearDown(): void
 
         $this->dropCollection();
 
+        foreach (array_unique($this->dbNames) as $dbName) {
+            $options = ['writeConcern' => new WriteConcern(WriteConcern::MAJORITY)];
+
+            $operation = new DropDatabase($dbName, $options);
+            $operation->execute($this->getPrimaryServer());
+        }
+
         parent::tearDown();
     }
 
@@ -1235,6 +1249,7 @@ public function testIndex_example_2(): void
     // Start Transactions Intro Example 1
     private function updateEmployeeInfo1(\MongoDB\Client $client, \MongoDB\Driver\Session $session): void
     {
+        $this->dbNames = ['hr', 'reporting'];
         $session->startTransaction([
             'readConcern' => new \MongoDB\Driver\ReadConcern('snapshot'),
             'writeConcern' => new \MongoDB\Driver\WriteConcern(\MongoDB\Driver\WriteConcern::MAJORITY),
@@ -1291,6 +1306,8 @@ public function testTransactions_intro_example_1(): void
 
         $client = static::createTestClient();
 
+        $this->dbNames = ['hr', 'reporting'];
+
         /* The WC is required: https://mongodb.com/docs/manual/core/transactions/#transactions-and-locks */
         $client->hr->dropCollection('employees', ['writeConcern' => new WriteConcern('majority')]);
         $client->reporting->dropCollection('events', ['writeConcern' => new WriteConcern('majority')]);
@@ -1451,6 +1468,8 @@ private function updateEmployeeInfo3(\MongoDB\Client $client, \MongoDB\Driver\Se
 
     private function doUpdateEmployeeInfo(\MongoDB\Client $client): void
     {
+        $this->dbNames = ['hr', 'reporting'];
+
         // Start a session.
         $session = $client->startSession();
 
@@ -1471,6 +1490,8 @@ public function testTransactions_retry_example_3(): void
 
         $client = static::createTestClient();
 
+        $this->dbNames = ['hr', 'reporting'];
+
         /* The WC is required: https://mongodb.com/docs/manual/core/transactions/#transactions-and-locks */
         $client->hr->dropCollection('employees', ['writeConcern' => new WriteConcern('majority')]);
         $client->reporting->dropCollection('events', ['writeConcern' => new WriteConcern('majority')]);
@@ -1788,6 +1809,7 @@ public function testWithTransactionExample(): void
         $this->skipIfTransactionsAreNotSupported();
 
         $uriString = static::getUri(true);
+        $this->dbNames = ['mydb1', 'mydb2'];
 
         // phpcs:disable SlevomatCodingStandard.Namespaces.ReferenceUsedNamesOnly
         // Start Transactions withTxn API Example 1