Function preparePostgresTest

api/datastore/postgres/postgres_test.go:17–67

(logf, fatalf func(string, ...interface{}))

const tmpPostgres = "postgres://postgres@%v:15432/funcs?sslmode=disable"

func preparePostgresTest(logf, fatalf func(string, ...interface{})) (func(), func()) {
	fmt.Println("initializing postgres for test")
	tryRun(logf, "remove old postgres container", exec.Command("docker", "rm", "-f", "iron-postgres-test"))
	mustRun(fatalf, "start postgres container", exec.Command("docker", "run", "--name", "iron-postgres-test", "-p", "15432:5432", "-d", "postgres"))

	wait := 1 * time.Second
	for {
		db, err := sql.Open("postgres", fmt.Sprintf("postgres://postgres@%v:15432?sslmode=disable",
			datastoretest.GetContainerHostIP()))
		if err != nil {
			fmt.Println("failed to connect to postgres:", err)
			fmt.Println("retrying in:", wait)
			time.Sleep(wait)
			wait = 2 * wait
			continue
		}

		_, err = db.Exec(`CREATE DATABASE funcs;`)
		if err != nil {
			fmt.Println("failed to create database:", err)
			fmt.Println("retrying in:", wait)
			time.Sleep(wait)
			wait = 2 * wait
			continue
		}
		_, err = db.Exec(`GRANT ALL PRIVILEGES ON DATABASE funcs TO postgres;`)
		if err == nil {
			break
		}
		fmt.Println("failed to grant privileges:", err)
		fmt.Println("retrying in:", wait)
		time.Sleep(wait)
		wait = 2 * wait
	}
	fmt.Println("postgres for test ready")
	return func() {
			db, err := sql.Open("postgres", fmt.Sprintf(tmpPostgres, datastoretest.GetContainerHostIP()))
			if err != nil {
				fatalf("failed to connect for truncation: %s\n", err)
			}
			for _, table := range []string{"routes", "apps", "extras"} {
				_, err = db.Exec(`TRUNCATE TABLE ` + table)
				if err != nil {
					fatalf("failed to truncate table %q: %s\n", table, err)
				}
			}
		},
		func() {
			tryRun(logf, "stop postgres container", exec.Command("docker", "rm", "-f", "iron-postgres-test"))
		}
}

func TestDatastore(t *testing.T) {
	_, close := preparePostgresTest(t.Logf, t.Fatalf)