MINI Sh3ll

<?php

namespace MongoDB\Tests\Operation;

use MongoDB\Driver\Exception\CommandException;
use MongoDB\Operation\DropCollection;
use MongoDB\Operation\FindOne;
use MongoDB\Operation\InsertOne;
use MongoDB\Operation\RenameCollection;
use MongoDB\Tests\CommandObserver;

use function version_compare;

class RenameCollectionFunctionalTest extends FunctionalTestCase
{
    /** @var integer */
    private static $errorCodeNamespaceNotFound = 26;

    /** @var integer */
    private static $errorCodeNamespaceExists = 48;

    /** @var string */
    private $toCollectionName;

    public function setUp(): void
    {
        parent::setUp();

        $this->toCollectionName = $this->getCollectionName() . '.renamed';
        $operation = new DropCollection($this->getDatabaseName(), $this->toCollectionName);
        $operation->execute($this->getPrimaryServer());
    }

    public function tearDown(): void
    {
        if ($this->hasFailed()) {
            return;
        }

        $operation = new DropCollection($this->getDatabaseName(), $this->toCollectionName);
        $operation->execute($this->getPrimaryServer());

        parent::tearDown();
    }

    public function testDefaultWriteConcernIsOmitted(): void
    {
        (new CommandObserver())->observe(
            function (): void {
                $server = $this->getPrimaryServer();

                $insertOne = new InsertOne($this->getDatabaseName(), $this->getCollectionName(), ['x' => 1]);
                $writeResult = $insertOne->execute($server);
                $this->assertEquals(1, $writeResult->getInsertedCount());

                $operation = new RenameCollection(
                    $this->getDatabaseName(),
                    $this->getCollectionName(),
                    $this->getDatabaseName(),
                    $this->toCollectionName,
                    ['writeConcern' => $this->createDefaultWriteConcern()]
                );

                $operation->execute($server);
            },
            function (array $event): void {
                $this->assertObjectNotHasAttribute('writeConcern', $event['started']->getCommand());
            }
        );
    }

    public function testRenameCollectionToNonexistentTarget(): void
    {
        $server = $this->getPrimaryServer();

        $insertOne = new InsertOne($this->getDatabaseName(), $this->getCollectionName(), ['_id' => 1]);
        $writeResult = $insertOne->execute($server);
        $this->assertEquals(1, $writeResult->getInsertedCount());

        $operation = new RenameCollection(
            $this->getDatabaseName(),
            $this->getCollectionName(),
            $this->getDatabaseName(),
            $this->toCollectionName
        );
        $commandResult = $operation->execute($server);

        $this->assertCommandSucceeded($commandResult);
        $this->assertCollectionDoesNotExist($this->getCollectionName());
        $this->assertCollectionExists($this->toCollectionName);

        $operation = new FindOne($this->getDatabaseName(), $this->toCollectionName, []);
        $this->assertSameDocument(['_id' => 1], $operation->execute($server));
    }

    public function testRenameCollectionExistingTarget(): void
    {
        $server = $this->getPrimaryServer();

        $insertOne = new InsertOne($this->getDatabaseName(), $this->getCollectionName(), ['_id' => 1]);
        $writeResult = $insertOne->execute($server);
        $this->assertEquals(1, $writeResult->getInsertedCount());

        $insertOne = new InsertOne($this->getDatabaseName(), $this->toCollectionName, ['_id' => 1]);
        $writeResult = $insertOne->execute($server);
        $this->assertEquals(1, $writeResult->getInsertedCount());

        $this->expectException(CommandException::class);

        /* TODO: mongos returns an inconsistent error code (see: SERVER-60632)
         *
         * Note: pre-3.2 server versions omit an error code. libmongoc will
         * substitute MONGOC_ERROR_QUERY_FAILURE in _mongoc_cmd_check_ok. */
        if (! $this->isShardedCluster() && version_compare($this->getServerVersion(), '3.2.0', '>=')) {
            $this->expectExceptionCode(self::$errorCodeNamespaceExists);
        }

        $operation = new RenameCollection(
            $this->getDatabaseName(),
            $this->getCollectionName(),
            $this->getDatabaseName(),
            $this->toCollectionName
        );
        $operation->execute($server);
    }

    public function testRenameNonexistentCollection(): void
    {
        $this->expectException(CommandException::class);

        /* Note: pre-3.2 server versions omit an error code. libmongoc will
         * substitute MONGOC_ERROR_QUERY_FAILURE in _mongoc_cmd_check_ok */
        if (version_compare($this->getServerVersion(), '3.2.0', '>=')) {
            $this->expectExceptionCode(self::$errorCodeNamespaceNotFound);
        }

        $operation = new RenameCollection(
            $this->getDatabaseName(),
            $this->getCollectionName(),
            $this->getDatabaseName(),
            $this->toCollectionName
        );
        $operation->execute($this->getPrimaryServer());
    }

    public function testSessionOption(): void
    {
        if (version_compare($this->getServerVersion(), '3.6.0', '<')) {
            $this->markTestSkipped('Sessions are not supported');
        }

        (new CommandObserver())->observe(
            function (): void {
                $server = $this->getPrimaryServer();

                $insertOne = new InsertOne($this->getDatabaseName(), $this->getCollectionName(), ['x' => 1]);
                $writeResult = $insertOne->execute($server);
                $this->assertEquals(1, $writeResult->getInsertedCount());

                $operation = new RenameCollection(
                    $this->getDatabaseName(),
                    $this->getCollectionName(),
                    $this->getDatabaseName(),
                    $this->toCollectionName,
                    ['session' => $this->createSession()]
                );

                $operation->execute($server);
            },
            function (array $event): void {
                $this->assertObjectHasAttribute('lsid', $event['started']->getCommand());
            }
        );
    }
}