Prose test 22.2

Author: jmikola

tests/SpecTests/ClientSideEncryption/Prose22_RangeExplicitEncryptionTest.php

@@ -2,12 +2,15 @@
 
 namespace MongoDB\Tests\SpecTests\ClientSideEncryption;
 
+use ArrayIterator;
 use Generator;
+use Iterator;
 use MongoDB\BSON\Binary;
 use MongoDB\BSON\Decimal128;
 use MongoDB\BSON\Document;
 use MongoDB\BSON\UTCDateTime;
 use MongoDB\Driver\ClientEncryption;
+use MultipleIterator;
 
 use function base64_decode;
 use function file_get_contents;
@@ -207,6 +210,59 @@ public function testCase1_CanDecryptAPayload(string $type, array $rangeOpts): vo
         $this->assertEquals($originalValue, $decryptedValue);
     }
 
+    /**
+     * @see https://github.com/mongodb/specifications/blob/master/source/client-side-encryption/tests/README.rst#case-2-can-find-encrypted-range-and-return-the-maximum
+     * @dataProvider provideTypeAndRangeOpts
+     */
+    public function testCase2_CanFindEncryptedRangeAndReturnTheMaximum(string $type, array $rangeOpts): void
+    {
+        $this->setUpWithTypeAndRangeOpts($type, $rangeOpts);
+
+        $encryptOpts = [
+            'keyId' => $this->key1Id,
+            'algorithm' => ClientEncryption::ALGORITHM_RANGE_PREVIEW,
+            'queryType' => ClientEncryption::QUERY_TYPE_RANGE_PREVIEW,
+            'contentionFactor' => 0,
+            'rangeOpts' => $rangeOpts,
+        ];
+
+        $cast = self::getCastCallableForType($type);
+        $fieldName = 'encrypted' . $type;
+
+        $expr = [
+            '$and' => [
+                [$fieldName => ['$gte' => $cast(6)]],
+                [$fieldName => ['$lte' => $cast(200)]],
+            ],
+        ];
+
+        $encryptedExpr = $this->clientEncryption->encryptExpression($expr, $encryptOpts);
+        $cursor = $this->collection->find($encryptedExpr, ['sort' => ['_id' => 1]]);
+
+        $expectedDocuments = [
+            ['_id' => 1, $fieldName => $cast(6)],
+            ['_id' => 2, $fieldName => $cast(30)],
+            ['_id' => 3, $fieldName => $cast(200)],
+        ];
+
+        $this->assertMultipleDocumentsMatch($expectedDocuments, $cursor);
+    }
+
+    private function assertMultipleDocumentsMatch(array $expectedDocuments, Iterator $actualDocuments): void
+    {
+        $mi = new MultipleIterator(MultipleIterator::MIT_NEED_ANY);
+        $mi->attachIterator(new ArrayIterator($expectedDocuments));
+        $mi->attachIterator($actualDocuments);
+
+        foreach ($mi as $documents) {
+            [$expectedDocument, $actualDocument] = $documents;
+            $this->assertNotNull($expectedDocument);
+            $this->assertNotNull($actualDocument);
+
+            $this->assertDocumentsMatch($expectedDocument, $actualDocument);
+        }
+    }
+
     private static function getCastCallableForType(string $type): callable
     {
         switch ($type) {