test(db): add on()/onDatabase tests and TestDatabaseTag enum

Author: laylatichy

tests/Integration/Database/Builder/IsDatabaseModelTest.php

@@ -49,6 +49,7 @@
 use Tests\Tempest\Fixtures\Modules\Books\Models\Book;
 use Tests\Tempest\Fixtures\Modules\Books\Models\Isbn;
 use Tests\Tempest\Integration\FrameworkIntegrationTestCase;
+use Tempest\Database\Builder\QueryBuilders\QueryBuilder;
 
 use function Tempest\Database\query;
 use function Tempest\Mapper\map;
@@ -694,6 +695,100 @@ public function test_nullable_relation_save(): void
         $this->assertNotNull($a->b);
         $this->assertSame('b', $a->b->name);
     }
+
+    public function test_on_returns_query_builder_with_database_tag(): void
+    {
+        $builder = Foo::on('analytics');
+
+        $this->assertInstanceOf(QueryBuilder::class, $builder);
+        $this->assertSame('analytics', $builder->onDatabase);
+    }
+
+    public function test_on_propagates_tag_to_select_builder(): void
+    {
+        $selectBuilder = Foo::on('analytics')->select();
+
+        $this->assertSame('analytics', $selectBuilder->onDatabase);
+    }
+
+    public function test_on_propagates_tag_to_insert_builder(): void
+    {
+        $insertBuilder = Foo::on('analytics')->insert();
+
+        $this->assertSame('analytics', $insertBuilder->onDatabase);
+    }
+
+    public function test_on_propagates_tag_to_count_builder(): void
+    {
+        $countBuilder = Foo::on('analytics')->count();
+
+        $this->assertSame('analytics', $countBuilder->onDatabase);
+    }
+
+    public function test_on_propagates_tag_to_delete_builder(): void
+    {
+        $deleteBuilder = Foo::on('analytics')->delete();
+
+        $this->assertSame('analytics', $deleteBuilder->onDatabase);
+    }
+
+    public function test_on_with_null_sets_null_tag(): void
+    {
+        $builder = Foo::on(null);
+
+        $this->assertNull($builder->onDatabase);
+    }
+
+    public function test_on_with_enum_tag(): void
+    {
+        $builder = Foo::on(TestDatabaseTag::Analytics);
+
+        $this->assertSame(TestDatabaseTag::Analytics, $builder->onDatabase);
+    }
+
+    public function test_on_database_returns_clone(): void
+    {
+        $this->database->migrate(
+            CreateMigrationsTable::class,
+            FooDatabaseMigration::class,
+        );
+
+        $foo = Foo::create(bar: 'baz');
+        $clone = $foo->onDatabase('analytics');
+
+        $this->assertNotSame($foo, $clone);
+    }
+
+    public function test_on_database_does_not_mutate_original(): void
+    {
+        $this->database->migrate(
+            CreateMigrationsTable::class,
+            FooDatabaseMigration::class,
+        );
+
+        $foo = Foo::create(bar: 'baz');
+        $foo->onDatabase('analytics');
+
+        // Original still works against default database
+        $foo->update(bar: 'updated');
+        $refreshed = Foo::get($foo->id);
+
+        $this->assertSame('updated', $refreshed->bar);
+    }
+
+    public function test_on_database_preserves_model_data(): void
+    {
+        $this->database->migrate(
+            CreateMigrationsTable::class,
+            FooDatabaseMigration::class,
+        );
+
+        $foo = Foo::create(bar: 'baz');
+        $clone = $foo->onDatabase('analytics');
+
+        $this->assertSame('baz', $clone->bar);
+        $this->assertEquals($foo->id, $clone->id);
+    }
 }
 
 final class Foo
@@ -1117,3 +1212,9 @@ final class ModelWithHookedVirtualProperty
         get => strtoupper($this->name);
     }
 }
+
+enum TestDatabaseTag
+{
+    case Analytics;
+    case Reporting;
+}