package daemon_test

import (
	"os"
	"testing"

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

func TestMain(m *testing.M) {
	os.Exit(testkit.RunTestsWithPackageTiDBRuntimeRoots(m, "internal/api/daemon"))
}

func openDaemonIntegrationTestTiDB(t *testing.T) coredb.DB {
	t.Helper()
	dsn := testkit.StartInitializedTiDBWithPackageRuntime(t, testkit.TiDBTestRuntimeUnistore)
	return coredb.WrapSQLDB(testkit.OpenTiDB(t, dsn))
}