package taskboard

import (
	"context"
	"os"
	"testing"

	"github.com/tidbcloud/AgentTeam/internal/coredb"
	"github.com/tidbcloud/AgentTeam/internal/coredb/testkit"
)

var taskboardTiDB testkit.TiDBCoreDBFixture

func TestMain(m *testing.M) {
	os.Exit(testkit.RunTestsWithTiDBCoreDBFixture(m, "taskboard", &taskboardTiDB))
}

func startTaskboardTestPool(t *testing.T, ctx context.Context) coredb.DB {
	t.Helper()
	return startTaskboardRollbackDB(t, ctx)
}

func startTaskboardTiDBTestDB(t *testing.T) coredb.DB {
	t.Helper()
	return startTaskboardRollbackDB(t, context.Background())
}

func startTaskboardRollbackDB(t *testing.T, ctx context.Context) coredb.DB {
	t.Helper()
	return taskboardTiDB.RollbackDB(t, ctx)
}