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

kylecarbs · kylecarbs · commit be5d5171783c · 2022-06-17T14:54:30.000Z
diff --git a/.github/workflows/coder.yaml b/.github/workflows/coder.yaml
@@ -293,8 +293,21 @@ jobs:
               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 \
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() {
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)
+}
