Factorize search indexes waiting helper

Author: GromNaN

docs/includes/fundamentals/as-avs/AtlasSearchTest.php

@@ -11,6 +11,7 @@
 use MongoDB\Collection;
 use MongoDB\Driver\Exception\ServerException;
 use MongoDB\Laravel\Schema\Builder;
+use MongoDB\Laravel\Tests\AtlasSearchIndexManagement;
 use MongoDB\Laravel\Tests\TestCase;
 use PHPUnit\Framework\Attributes\Group;
 
@@ -19,11 +20,12 @@
 use function rand;
 use function range;
 use function srand;
-use function usleep;
 
 #[Group('atlas-search')]
 class AtlasSearchTest extends TestCase
 {
+    use AtlasSearchIndexManagement;
+
     private array $vectors;
 
     protected function setUp(): void
@@ -51,10 +53,7 @@ protected function setUp(): void
             ['title' => 'D', 'plot' => 'Stranded on a distant planet, astronauts must repair their ship before supplies run out.'],
         ]));
 
-        // Waits for the search index created in the previous test to be deleted
-        while ($moviesCollection->listSearchIndexes()->count()) {
-            usleep(1000);
-        }
+        $this->waitForSearchIndexesDropped($moviesCollection);
 
         try {
             $moviesCollection->createSearchIndex([
@@ -87,14 +86,7 @@ protected function setUp(): void
             throw $e;
         }
 
-        // Waits for the index to be ready
-        do {
-            $ready = true;
-            usleep(10_000);
-            foreach ($moviesCollection->listSearchIndexes() as $index) {
-                $ready = $ready && $index['queryable'];
-            }
-        } while (! $ready);
+        $this->waitForSearchIndexesReady($moviesCollection);
     }
 
     /**

tests/AtlasSearchIndexManagement.php

@@ -0,0 +1,50 @@
+<?php
+
+namespace MongoDB\Laravel\Tests;
+
+use MongoDB\Collection;
+use RuntimeException;
+
+use function hrtime;
+use function usleep;
+
+/**
+ * Helpers for managing Atlas Search indexes in tests with awaiting mechanism.
+ */
+trait AtlasSearchIndexManagement
+{
+    /**
+     * Waits for the search index created in the previous test to be deleted
+     */
+    public function waitForSearchIndexesDropped(Collection $collection)
+    {
+        $timeout = hrtime()[0] + 30;
+        // Waits for the search index created in the previous test to be deleted
+        while ($collection->listSearchIndexes()->count()) {
+            if (hrtime()[0] > $timeout) {
+                throw new RuntimeException('Timed out waiting for search indexes to be dropped');
+            }
+
+            usleep(1000);
+        }
+    }
+
+    /**
+     * Waits for all search indexes to be ready
+     */
+    public function waitForSearchIndexesReady(Collection $collection)
+    {
+        $timeout = hrtime()[0] + 30;
+        do {
+            if (hrtime()[0] > $timeout) {
+                throw new RuntimeException('Timed out waiting for search indexes to be ready');
+            }
+
+            usleep(1000);
+            $ready = true;
+            foreach ($collection->listSearchIndexes() as $index) {
+                $ready = $ready && $index['queryable'];
+            }
+        } while (! $ready);
+    }
+}

tests/AtlasSearchTest.php

@@ -19,12 +19,13 @@
 use function rand;
 use function range;
 use function srand;
-use function usleep;
 use function usort;
 
 #[Group('atlas-search')]
 class AtlasSearchTest extends TestCase
 {
+    use AtlasSearchIndexManagement;
+
     private array $vectors;
 
     public function setUp(): void
@@ -59,10 +60,7 @@ public function setUp(): void
         ]));
 
         try {
-            // Waits for the search index created in the previous test to be deleted
-            while ($collection->listSearchIndexes()->count()) {
-                usleep(1000);
-            }
+            $this->waitForSearchIndexesDropped($collection);
 
             $collection->createSearchIndex([
                 'mappings' => [
@@ -95,14 +93,7 @@ public function setUp(): void
             throw $e;
         }
 
-        // Wait for the index to be ready
-        do {
-            $ready = true;
-            usleep(1000);
-            foreach ($collection->listSearchIndexes() as $index) {
-                $ready = $ready && $index['queryable'];
-            }
-        } while (! $ready);
+        $this->waitForSearchIndexesReady($collection);
     }
 
     public function tearDown(): void

tests/Scout/ScoutIntegrationTest.php

@@ -7,6 +7,7 @@
 use Illuminate\Support\LazyCollection;
 use Laravel\Scout\ScoutServiceProvider;
 use LogicException;
+use MongoDB\Laravel\Tests\AtlasSearchIndexManagement;
 use MongoDB\Laravel\Tests\Scout\Models\ScoutUser;
 use MongoDB\Laravel\Tests\Scout\Models\SearchableInSameNamespace;
 use MongoDB\Laravel\Tests\TestCase;
@@ -17,6 +18,7 @@
 use function array_merge;
 use function count;
 use function env;
+use function hrtime;
 use function iterator_to_array;
 use function Orchestra\Testbench\artisan;
 use function range;
@@ -26,6 +28,8 @@
 #[Group('atlas-search')]
 class ScoutIntegrationTest extends TestCase
 {
+    use AtlasSearchIndexManagement;
+
     #[Override]
     protected function getPackageProviders($app): array
     {
@@ -96,25 +100,29 @@ public function setUp(): void
     /** This test create the search index for tests performing search */
     public function testItCanCreateTheCollection()
     {
+        $this->skipIfSearchIndexManagementIsNotSupported();
+
         $collection = DB::connection('mongodb')->getCollection('prefix_scout_users');
         $collection->drop();
+        $this->waitForSearchIndexesDropped($collection);
 
         // Recreate the indexes using the artisan commands
         // Ensure they return a success exit code (0)
         self::assertSame(0, artisan($this, 'scout:delete-index', ['name' => ScoutUser::class]));
-        self::assertSame(0, artisan($this, 'scout:index', ['name' => ScoutUser::class]));
         self::assertSame(0, artisan($this, 'scout:import', ['model' => ScoutUser::class]));
+        self::assertSame(0, artisan($this, 'scout:index', ['name' => ScoutUser::class]));
 
         self::assertSame(44, $collection->countDocuments());
 
         $searchIndexes = $collection->listSearchIndexes(['name' => 'scout', 'typeMap' => ['root' => 'array', 'document' => 'array', 'array' => 'array']]);
         self::assertCount(1, $searchIndexes);
         self::assertSame(['mappings' => ['dynamic' => true, 'fields' => ['bool_field' => ['type' => 'boolean']]]], iterator_to_array($searchIndexes)[0]['latestDefinition']);
 
+        $this->waitForSearchIndexesReady($collection);
+
         // Wait for all documents to be indexed asynchronously
-        $i = 1000;
+        $timeout = hrtime()[0] + 30;
         while (true) {
-            usleep(10_000);
             $indexedDocuments = $collection->aggregate([
                 ['$search' => ['index' => 'scout', 'exists' => ['path' => 'name']]],
             ])->toArray();
@@ -123,9 +131,11 @@ public function testItCanCreateTheCollection()
                 break;
             }
 
-            if ($i-- === 0) {
-                self::fail('Documents not indexed');
+            if (hrtime()[0] > $timeout) {
+                self::fail('Timed out waiting for documents to be indexed');
             }
+
+            usleep(1000);
         }
 
         self::assertCount(44, $indexedDocuments);