fix tests

Author: xprojects-de

src/chat/src/Bridge/Doctrine/DoctrineDbalMessageStore.php

@@ -54,6 +54,11 @@ public function setup(array $options = []): void
         }
 
         $schemaManager = $this->dbalConnection->createSchemaManager();
+        $currentSchema = $schemaManager->introspectSchema();
+
+        if ($currentSchema->hasTable($this->tableName)) {
+            return;
+        }
 
         if (class_exists(ComparatorConfig::class)) {
             $comparator = $schemaManager->createComparator(new ComparatorConfig(false, false));
@@ -62,7 +67,6 @@ public function setup(array $options = []): void
             $comparator = $schemaManager->createComparator();
         }
 
-        $currentSchema = $schemaManager->introspectSchema();
         $migrations = $this->dbalConnection->getDatabasePlatform()->getAlterSchemaSQL($comparator->compareSchemas($currentSchema, $this->defineTableSchema($currentSchema)));
 
         foreach ($migrations as $sql) {
@@ -126,10 +130,6 @@ private function defineTableSchema(Schema $currentSchema): Schema
     {
         $schema = clone $currentSchema;
 
-        if ($schema->hasTable($this->tableName)) {
-            $schema->dropTable($this->tableName);
-        }
-
         $table = $schema->createTable($this->tableName);
         $table->addOption('_symfony_ai_chat_table_name', $this->tableName);
         $idColumn = $table->addColumn('id', Types::BIGINT)

src/chat/tests/Bridge/Doctrine/DoctrineDbalMessageStoreTest.php

@@ -17,7 +17,9 @@
 use Doctrine\DBAL\Result;
 use Doctrine\DBAL\Schema\AbstractSchemaManager;
 use Doctrine\DBAL\Schema\Column;
+use Doctrine\DBAL\Schema\Comparator;
 use Doctrine\DBAL\Schema\Schema;
+use Doctrine\DBAL\Schema\SchemaDiff;
 use Doctrine\DBAL\Schema\Table;
 use PHPUnit\Framework\TestCase;
 use Symfony\AI\Chat\Bridge\Doctrine\DoctrineDbalMessageStore;
@@ -76,15 +78,23 @@ public function testMessageStoreTableCanBeSetup()
         $schema = $this->createMock(Schema::class);
         $schema->expects($this->once())->method('hasTable')->willReturn(false);
         $schema->expects($this->once())->method('createTable')->with('foo')->willReturn($table);
-        $schema->expects($this->once())->method('toSql')->with($platform)->willReturn([]);
 
         $sqliteSchemaManager = $this->createMock(AbstractSchemaManager::class);
         $sqliteSchemaManager->expects($this->once())->method('introspectSchema')->willReturn($schema);
 
+        $comparator = $this->createMock(Comparator::class);
+        $sqliteSchemaManager->expects($this->once())->method('createComparator')->willReturn($comparator);
+
         $connection = $this->createMock(Connection::class);
         $connection->expects($this->once())->method('createSchemaManager')->willReturn($sqliteSchemaManager);
         $connection->expects($this->exactly(2))->method('getDatabasePlatform')->willReturn($platform);
 
+        $schemaDiff = $this->createMock(SchemaDiff::class);
+
+        $comparator->expects($this->once())->method('compareSchemas')->willReturn($schemaDiff);
+        $platform->expects($this->once())->method('getAlterSchemaSQL')->willReturn(['SQL STATEMENT']);
+        $connection->expects($this->once())->method('executeQuery')->with('SQL STATEMENT');
+
         $messageStore = new DoctrineDbalMessageStore('foo', $connection);
         $messageStore->setup();
     }