test: Use a template to prevent migrations from running for every test

kylecarbs · kylecarbs · commit d57cbd1aba60 · 2022-06-21T21:29:15.000Z
diff --git a/.github/workflows/coder.yaml b/.github/workflows/coder.yaml
@@ -284,17 +284,31 @@ jobs:
             -e POSTGRES_DB=postgres \
             -e PGDATA=/tmp \
             -p 5432:5432 \
-            -d postgres:11 \
+            -d postgres:13 \
             -c shared_buffers=1GB \
-            -c max_connections=1000
+            -c max_connections=1000 \
+            -c fsync=false
           while ! pg_isready -h 127.0.0.1
           do
               echo "$(date) - waiting for database to start"
               sleep 0.5
           done
 
+      - name: Create Migration Template
+        id: migrated
+        run: |
+          DB_FROM=$(go run scripts/migrate-ci/main.go)
+          echo "::set-output name=db-from::$DB_FROM"
+          echo $DB_FROM
+
       - name: Test with PostgreSQL Database
-        run: "make test-postgres"
+        run: |
+          gotestsum --junitfile="gotests.xml" --packages="./..." -- \
+            -covermode=atomic -coverprofile="gotests.coverage" -timeout=30m \
+            -coverpkg=./...,github.com/coder/coder/codersdk \
+            -count=2 -race -failfast
+        env:
+          DB_FROM: ${{ steps.migrated.outputs.db-from }}
 
       - name: Upload DataDog Trace
         if: always() && github.actor != 'dependabot[bot]' && !github.event.pull_request.head.repo.fork
diff --git a/Makefile b/Makefile
@@ -112,14 +112,6 @@ site/src/api/typesGenerated.ts: scripts/apitypings/main.go $(shell find codersdk
 test: test-clean
 	gotestsum -- -v -short ./...
 
-.PHONY: test-postgres
-test-postgres: test-clean
-	DB=ci gotestsum --junitfile="gotests.xml" --packages="./..." -- \
-          -covermode=atomic -coverprofile="gotests.coverage" -timeout=30m \
-          -coverpkg=./...,github.com/coder/coder/codersdk \
-          -count=1 -race -failfast
-
-
 .PHONY: test-postgres-docker
 test-postgres-docker:
 	docker run \
@@ -131,7 +123,7 @@ test-postgres-docker:
 		--name test-postgres-docker \
 		--restart unless-stopped \
 		--detach \
-		postgres:11 \
+		postgres:13 \
 		-c shared_buffers=1GB \
 		-c max_connections=1000
 
diff --git a/coderd/coderdtest/coderdtest.go b/coderd/coderdtest/coderdtest.go
@@ -115,8 +115,6 @@ func NewWithAPI(t *testing.T, options *Options) (*codersdk.Client, *coderd.API)
 		t.Cleanup(func() {
 			_ = sqlDB.Close()
 		})
-		err = database.MigrateUp(sqlDB)
-		require.NoError(t, err)
 		db = database.New(sqlDB)
 
 		pubsub, err = database.NewPubsub(context.Background(), sqlDB, connectionURL)
diff --git a/coderd/database/postgres/postgres.go b/coderd/database/postgres/postgres.go
@@ -13,6 +13,7 @@ import (
 	"github.com/ory/dockertest/v3/docker"
 	"golang.org/x/xerrors"
 
+	"github.com/coder/coder/coderd/database"
 	"github.com/coder/coder/cryptorand"
 )
 
@@ -39,9 +40,21 @@ func Open() (string, func(), error) {
 		}
 
 		dbName = "ci" + dbName
-		_, err = db.Exec("CREATE DATABASE " + dbName)
-		if err != nil {
-			return "", nil, xerrors.Errorf("create db: %w", err)
+		if os.Getenv("DB_FROM") == "" {
+			_, err = db.Exec("CREATE DATABASE " + dbName)
+			if err != nil {
+				return "", nil, xerrors.Errorf("create db: %w", err)
+			}
+
+			err = database.MigrateUp(db)
+			if err != nil {
+				return "", nil, xerrors.Errorf("migrate db: %w", err)
+			}
+		} else {
+			_, err = db.Exec("CREATE DATABASE " + dbName + " WITH TEMPLATE " + os.Getenv("DB_FROM"))
+			if err != nil {
+				return "", nil, xerrors.Errorf("create db with template: %w", err)
+			}
 		}
 
 		deleteDB := func() {
@@ -74,7 +87,7 @@ func Open() (string, func(), error) {
 
 	resource, err := pool.RunWithOptions(&dockertest.RunOptions{
 		Repository: "postgres",
-		Tag:        "11",
+		Tag:        "13",
 		Env: []string{
 			"POSTGRES_PASSWORD=postgres",
 			"POSTGRES_USER=postgres",
diff --git a/scripts/migrate-ci/main.go b/scripts/migrate-ci/main.go
@@ -0,0 +1,36 @@
+package main
+
+import (
+	"database/sql"
+	"fmt"
+
+	"github.com/coder/coder/coderd/database"
+	"github.com/coder/coder/cryptorand"
+)
+
+func main() {
+	dbURL := "postgres://postgres:postgres@127.0.0.1:5432/postgres?sslmode=disable"
+	db, err := sql.Open("postgres", dbURL)
+	if err != nil {
+		panic(err)
+	}
+	defer db.Close()
+
+	dbName, err := cryptorand.StringCharset(cryptorand.Lower, 10)
+	if err != nil {
+		panic(err)
+	}
+
+	dbName = "ci" + dbName
+	_, err = db.Exec("CREATE DATABASE " + dbName)
+	if err != nil {
+		panic(err)
+	}
+
+	err = database.MigrateUp(db)
+	if err != nil {
+		panic(err)
+	}
+
+	_, _ = fmt.Println(dbName)
+}
