Function createCleanDatabases

persistence/sql/persister_test.go:96–172

(t testing.TB)

}

func createCleanDatabases(t testing.TB) map[string]string {
	conns := map[string]string{
		"sqlite": dbal.NewSQLiteTestDatabase(t),
	}
	connsMtx := sync.Mutex{}

	if !testing.Short() {
		funcs := map[string]func(t testing.TB) string{
			"postgres": func(t testing.TB) string {
				return dockertest.RunTestPostgreSQLWithVersion(t, "16")
			},
			"mysql": func(t testing.TB) string {
				return dockertest.RunTestMySQLWithVersion(t, "8.4")
			},
			"cockroach": newLocalTestCRDBServer,
		}

		var wg sync.WaitGroup
		wg.Add(len(funcs))

		for k, f := range funcs {
			go func(s string, f func(t testing.TB) string) {
				defer wg.Done()
				db := f(t)
				connsMtx.Lock()
				conns[s] = db
				connsMtx.Unlock()
			}(k, f)
		}

		wg.Wait()
	}

	ps := make(map[string]string, len(conns))
	psMtx := sync.Mutex{}

	var wg sync.WaitGroup
	wg.Add(len(conns))
	for name, dsn := range conns {
		go func(name, dsn string) {
			defer wg.Done()

			if name != "sqlite" {
				require.EventuallyWithT(t, func(t *assert.CollectT) {
					c, err := pop.NewConnection(&pop.ConnectionDetails{URL: dsn})
					require.NoError(t, err)
					require.NoError(t, c.Open())
					dbName := "testdb" + strings.ReplaceAll(x.NewUUID().String(), "-", "")
					require.NoError(t, c.RawQuery("CREATE DATABASE "+dbName).Exec())
					dsn = regexp.MustCompile(`/[a-z0-9]+\?`).ReplaceAllString(dsn, "/"+dbName+"?")
				}, 20*time.Second, 100*time.Millisecond)
			}

			t.Logf("Connecting to %s: %s", name, dsn)

			_, reg := pkg.NewRegistryDefaultWithDSN(t, dsn)
			p := reg.Persister().(*sql.Persister)