package aiagent

import (
	"testing"

	"code.tczkiot.com/wlw/ai-agent/internal/models"

	"github.com/glebarez/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/schema"
)

func TestSyncSchemaCreatesEveryRegisteredTableWithModulePrefix(t *testing.T) {
	hostDB, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{
		NamingStrategy: schema.NamingStrategy{TablePrefix: "host_", SingularTable: true},
	})
	if err != nil {
		t.Fatalf("open database: %v", err)
	}
	if err := SyncSchema(hostDB, "iot_ai_"); err != nil {
		t.Fatalf("SyncSchema() error = %v", err)
	}

	var tableNames []string
	if err := hostDB.Raw("SELECT name FROM sqlite_master WHERE type = 'table' AND name LIKE 'iot_ai_%'").Scan(&tableNames).Error; err != nil {
		t.Fatalf("list AI Agent tables: %v", err)
	}
	if len(tableNames) != len(models.Models) {
		t.Fatalf("AI Agent table count = %d, registered model count = %d", len(tableNames), len(models.Models))
	}

	for _, forbidden := range []string{
		"iot_ai_migration",
		"iot_ai_user",
		"iot_ai_admin",
		"iot_ai_role",
		"iot_ai_permission",
		"iot_ai_token",
	} {
		var count int64
		if err := hostDB.Raw("SELECT COUNT(1) FROM sqlite_master WHERE type = 'table' AND name = ?", forbidden).Scan(&count).Error; err != nil {
			t.Fatalf("check forbidden table %s: %v", forbidden, err)
		}
		if count != 0 {
			t.Fatalf("forbidden table %s must not be created", forbidden)
		}
	}
}