MINI Sh3ll

<?php

namespace MongoDB\Tests\UnifiedSpecTests;

use ArrayIterator;
use MongoDB\Client;
use MongoDB\Driver\ReadConcern;
use MongoDB\Driver\ReadPreference;
use MongoDB\Driver\WriteConcern;
use MongoDB\Tests\UnifiedSpecTests\Constraint\Matches;
use MultipleIterator;
use stdClass;

use function PHPUnit\Framework\assertContainsOnly;
use function PHPUnit\Framework\assertIsArray;
use function PHPUnit\Framework\assertIsString;
use function PHPUnit\Framework\assertNotNull;
use function PHPUnit\Framework\assertThat;
use function sprintf;

class CollectionData
{
    /** @var string */
    private $collectionName;

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

    /** @var array */
    private $documents;

    public function __construct(stdClass $o)
    {
        assertIsString($o->collectionName);
        $this->collectionName = $o->collectionName;

        assertIsString($o->databaseName);
        $this->databaseName = $o->databaseName;

        assertIsArray($o->documents);
        assertContainsOnly('object', $o->documents);
        $this->documents = $o->documents;
    }

    public function prepareInitialData(Client $client): void
    {
        $database = $client->selectDatabase(
            $this->databaseName,
            ['writeConcern' => new WriteConcern(WriteConcern::MAJORITY)]
        );

        $database->dropCollection($this->collectionName);

        if (empty($this->documents)) {
            $database->createCollection($this->collectionName);

            return;
        }

        $database->selectCollection($this->collectionName)->insertMany($this->documents);
    }

    public function assertOutcome(Client $client): void
    {
        $collection = $client->selectCollection(
            $this->databaseName,
            $this->collectionName,
            [
                'readConcern' => new ReadConcern(ReadConcern::LOCAL),
                'readPreference' => new ReadPreference(ReadPreference::PRIMARY),
            ]
        );

        $cursor = $collection->find([], ['sort' => ['_id' => 1]]);

        $mi = new MultipleIterator(MultipleIterator::MIT_NEED_ANY);
        $mi->attachIterator(new ArrayIterator($this->documents));
        $mi->attachIterator($cursor);

        foreach ($mi as $i => $documents) {
            [$expectedDocument, $actualDocument] = $documents;
            assertNotNull($expectedDocument);
            assertNotNull($actualDocument);

            /* Prohibit extra root keys and disable operators to enforce exact
             * matching of documents. Key order variation is still allowed. */
            $constraint = new Matches($expectedDocument, null, false, false);
            assertThat($actualDocument, $constraint, sprintf('documents[%d] match', $i));
        }
    }
}