Use a distinct collection names for each test using atlas search indexes

Author: GromNaN

tests/AtlasSearchTest.php

@@ -12,6 +12,7 @@
 use MongoDB\Laravel\Schema\Builder;
 use MongoDB\Laravel\Tests\Models\Book;
 use PHPUnit\Framework\Attributes\Group;
+use ReflectionClass;
 
 use function array_map;
 use function assert;
@@ -29,6 +30,10 @@ class AtlasSearchTest extends TestCase
 
     public function setUp(): void
     {
+        // Use a unique prefix per test to avoid collisions when the search index is
+        // deleted asynchronously while we try to create it in setup of the next test
+        $_SERVER['DB_PREFIX'] = (new ReflectionClass($this))->getShortName() . '_' . $this->name() . '_';
+
         parent::setUp();
 
         Book::insert($this->addVector([
@@ -54,7 +59,7 @@ public function setUp(): void
             ['title' => 'Pattern Recognition and Machine Learning'],
         ]));
 
-        $collection = $this->getConnection('mongodb')->getCollection('books');
+        $collection = $this->getConnection('mongodb')->getCollection($this->getBookCollectionName());
         assert($collection instanceof MongoDBCollection);
 
         try {
@@ -92,25 +97,24 @@ public function setUp(): void
         // Wait for the index to be ready
         do {
             $ready = true;
-            usleep(10_000);
+            usleep(100);
             foreach ($collection->listSearchIndexes() as $index) {
-                if ($index['status'] !== 'READY') {
-                    $ready = false;
-                }
+                $ready = $ready && $index['queryable'];
             }
         } while (! $ready);
     }
 
     public function tearDown(): void
     {
-        $this->getConnection('mongodb')->getCollection('books')->drop();
+        $this->getConnection('mongodb')->getCollection($this->getBookCollectionName())->drop();
+        unset($_SERVER['DB_PREFIX']);
 
         parent::tearDown();
     }
 
     public function testGetIndexes()
     {
-        $indexes = Schema::getIndexes('books');
+        $indexes = Schema::getIndexes($this->getBookCollectionName());
 
         self::assertIsArray($indexes);
         self::assertCount(4, $indexes);
@@ -171,7 +175,7 @@ public function testEloquentBuilderSearch()
 
     public function testDatabaseBuilderSearch()
     {
-        $results = $this->getConnection('mongodb')->table('books')
+        $results = $this->getConnection('mongodb')->table($this->getBookCollectionName())
             ->search(Search::text('title', 'systems'), sort: ['title' => 1]);
 
         self::assertInstanceOf(LaravelCollection::class, $results);
@@ -199,7 +203,7 @@ public function testEloquentBuilderAutocomplete()
 
     public function testDatabaseBuilderAutocomplete()
     {
-        $results = $this->getConnection('mongodb')->table('books')
+        $results = $this->getConnection('mongodb')->table($this->getBookCollectionName())
             ->autocomplete('title', 'system');
 
         self::assertInstanceOf(LaravelCollection::class, $results);
@@ -213,7 +217,7 @@ public function testDatabaseBuilderAutocomplete()
 
     public function testDatabaseBuilderVectorSearch()
     {
-        $results = $this->getConnection('mongodb')->table('books')
+        $results = $this->getConnection('mongodb')->table($this->getBookCollectionName())
             ->vectorSearch(
                 index: 'vector',
                 path: 'vector4',
@@ -253,6 +257,15 @@ public function testEloquentBuilderVectorSearch()
         );
     }
 
+    private function getBookCollectionName(): string
+    {
+        $name = (new Book())->getTable();
+
+        self::assertNotEquals('books', $name, 'DB_PREFIX is not set correctly');
+
+        return $name;
+    }
+
     /** Generate random vectors using fixed seed to make tests deterministic */
     private function addVector(array $items): array
     {

tests/Models/Book.php

@@ -23,6 +23,15 @@ class Book extends Model
     protected $table = 'books';
     protected static $unguarded = true;
 
+    public function __construct(array $attributes = [])
+    {
+        /* @TODO remove when connection prefix is supported
+         * @see https://jira.mongodb.org/browse/PHPORM-433 */
+        $this->table = ($_SERVER['DB_PREFIX'] ?? '') . $this->table;
+
+        parent::__construct($attributes);
+    }
+
     public function author(): BelongsTo
     {
         return $this->belongsTo(User::class, 'author_id');