JT suggestion - move code to tests

Author: rustagir

docs/includes/schema-builder/galaxies_migration.php

@@ -2,15 +2,22 @@
 
 declare(strict_types=1);
 
-use Illuminate\Database\Migrations\Migration;
+namespace App\Http\Controllers;
+
 use Illuminate\Support\Facades\Schema;
+use MongoDB\Collection;
 use MongoDB\Laravel\Schema\Blueprint;
+use MongoDB\Laravel\Tests\TestCase;
 
-return new class extends Migration
-{
-    protected $connection = 'mongodb';
+use function assert;
 
-    public function up(): void
+class AtlasIdxSchemaBuilderTest extends TestCase
+{
+    /**
+     * @runInSeparateProcess
+     * @preserveGlobalState disabled
+     */
+    public function testAtlasSearchIdx(): void
     {
         // begin-create-search-indexes
         Schema::create('galaxies', function (Blueprint $collection) {
@@ -33,6 +40,22 @@ public function up(): void
         });
         // end-create-search-indexes
 
+        $index = $this->getSearchIndex('galaxies', 'dynamic_index');
+        self::assertNotNull($index);
+
+        self::assertSame('dynamic_index', $index['name']);
+        self::assertSame('search', $index['type']);
+        self::assertTrue($index['latestDefinition']['mappings']['dynamic']);
+
+        $index = $this->getSearchIndex('galaxies', 'auto_index');
+        self::assertNotNull($index);
+
+        self::assertSame('auto_index', $index['name']);
+        self::assertSame('search', $index['type']);
+    }
+
+    public function testVectorSearchIdx(): void
+    {
         // begin-create-vs-index
         Schema::create('galaxies', function (Blueprint $collection) {
             $collection->vectorSearchIndex([
@@ -47,14 +70,50 @@ public function up(): void
             ], 'vs_index');
         });
         // end-create-vs-index
+
+        $index = $this->getSearchIndex('galaxies', 'vs_index');
+        self::assertNotNull($index);
+
+        self::assertSame('vs_index', $index['name']);
+        self::assertSame('vectorSearch', $index['type']);
+        self::assertSame('vector', $index['latestDefinition']['fields'][0]['type']);
     }
 
-    public function down(): void
+    public function testDropIndexes(): void
     {
         // begin-drop-search-index
         Schema::table('galaxies', function (Blueprint $collection) {
             $collection->dropSearchIndex('auto_index');
         });
         // end-drop-search-index
+
+        Schema::table('galaxies', function (Blueprint $collection) {
+            $collection->dropSearchIndex('dynamic_index');
+        });
+
+        Schema::table('galaxies', function (Blueprint $collection) {
+            $collection->dropSearchIndex('vs_index');
+        });
+
+        $index = $this->getSearchIndex('galaxies', 'auto_index');
+        self::assertNull($index);
+
+        $index = $this->getSearchIndex('galaxies', 'dynamic_index');
+        self::assertNull($index);
+
+        $index = $this->getSearchIndex('galaxies', 'vs_index');
+        self::assertNull($index);
+    }
+
+    protected function getSearchIndex(string $collection, string $name): ?array
+    {
+        $collection = $this->getConnection('mongodb')->getCollection($collection);
+        assert($collection instanceof Collection);
+
+        foreach ($collection->listSearchIndexes(['name' => $name, 'typeMap' => ['root' => 'array', 'array' => 'array', 'document' => 'array']]) as $index) {
+            return $index;
+        }
+
+        return null;
     }
-};
+}