Add tests for $search and $searchMeta stages (#61)

Author: alcaeus

generator/config/stage/search.yaml

@@ -12,3 +12,30 @@ arguments:
         name: search
         type:
             - object
+
+tests:
+    -
+        name: 'Example'
+        link: 'https://www.mongodb.com/docs/atlas/atlas-search/query-syntax/#aggregation-variable'
+        pipeline:
+            -
+                $search:
+                    near:
+                        path: 'released'
+                        origin: !bson_utcdatetime '2011-09-01T00:00:00.000+00:00'
+                        pivot: 7776000000
+            -
+                $project:
+                    _id: 0
+                    title: 1
+                    released: 1
+            -
+                $limit: 5
+            -
+                $facet:
+                    docs: []
+                    meta:
+                        -
+                            $replaceWith: '$$SEARCH_META'
+                        -
+                            $limit: 1

generator/config/stage/searchMeta.yaml

@@ -12,3 +12,17 @@ arguments:
         name: meta
         type:
             - object
+
+tests:
+    -
+        name: 'Example'
+        link: 'https://www.mongodb.com/docs/atlas/atlas-search/query-syntax/#example'
+        pipeline:
+            -
+                $searchMeta:
+                    range:
+                        path: 'year'
+                        gte: 1998
+                        lt: 1999
+                    count:
+                        type: 'total'

tests/Builder/Stage/Pipelines.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MongoDB\Tests\Builder\Stage;
+
+use MongoDB\Builder\Pipeline;
+use MongoDB\Builder\Stage;
+use MongoDB\Tests\Builder\PipelineTestCase;
+
+use function MongoDB\object;
+
+/**
+ * Test $searchMeta stage
+ */
+class SearchMetaStageTest extends PipelineTestCase
+{
+    public function testExample(): void
+    {
+        $pipeline = new Pipeline(
+            Stage::searchMeta(object(
+                range: object(
+                    path: 'year',
+                    gte: 1998,
+                    lt: 1999,
+                ),
+                count: object(type: 'total'),
+            )),
+        );
+
+        $this->assertSamePipeline(Pipelines::SearchMetaExample, $pipeline);
+    }
+}

tests/Builder/Stage/SearchMetaStageTest.php

@@ -0,0 +1,44 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MongoDB\Tests\Builder\Stage;
+
+use DateTime;
+use MongoDB\BSON\UTCDateTime;
+use MongoDB\Builder\Expression;
+use MongoDB\Builder\Pipeline;
+use MongoDB\Builder\Stage;
+use MongoDB\Tests\Builder\PipelineTestCase;
+
+use function MongoDB\object;
+
+/**
+ * Test $search stage
+ */
+class SearchStageTest extends PipelineTestCase
+{
+    public function testExample(): void
+    {
+        $pipeline = new Pipeline(
+            Stage::search(object(
+                near: object(
+                    path: 'released',
+                    origin: new UTCDateTime(new DateTime('2011-09-01T00:00:00.000+00:00')),
+                    pivot: 7776000000,
+                ),
+            )),
+            Stage::project(_id: 0, title: 1, released: 1),
+            Stage::limit(5),
+            Stage::facet(
+                docs: [],
+                meta: new Pipeline(
+                    Stage::replaceWith(Expression::variable('SEARCH_META')),
+                    Stage::limit(1),
+                ),
+            ),
+        );
+
+        $this->assertSamePipeline(Pipelines::SearchExample, $pipeline);
+    }
+}