package automation

import (
	"testing"

	"github.com/stretchr/testify/require"
	automationstore "github.com/tidbcloud/AgentTeam/internal/store/automation"
	"gorm.io/gorm"
)

func migrateAutomationServiceTestSchema(t *testing.T, db *gorm.DB, extraModels ...any) {
	t.Helper()
	models := []any{
		&automationstore.AutomationRun{},
		&automationstore.AutomationRepoStatusOutbox{},
		&automationstore.AutomationContentRepository{},
		&automationstore.AutomationRunSlot{},
		&automationstore.AutomationJobRun{},
		&automationstore.AutomationEvent{},
		&automationstore.AutomationJob{},
		&automationstore.AutomationWait{},
	}
	models = append(models, extraModels...)
	require.NoError(t, db.AutoMigrate(models...))
}