1 files changed, 78 insertions, 0 deletions
diff --git a/cmd/praefect/subcmd_sql_migrate_test.go b/cmd/praefect/subcmd_sql_migrate_test.go
new file mode 100644
index 000000000..d7381b82f
--- /dev/null
+++ b/cmd/praefect/subcmd_sql_migrate_test.go
@@ -0,0 +1,78 @@
+package main
+
+import (
+	"bytes"
+	"flag"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"gitlab.com/gitlab-org/gitaly/v14/internal/praefect/config"
+	"gitlab.com/gitlab-org/gitaly/v14/internal/praefect/datastore/migrations"
+	"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper/testdb"
+)
+
+func TestSubCmdSqlMigrate(t *testing.T) {
+	db := testdb.New(t)
+	dbCfg := testdb.GetConfig(t, db.Name)
+	cfg := config.Config{DB: dbCfg}
+
+	migrationCt := len(migrations.All())
+
+	for _, tc := range []struct {
+		desc           string
+		up             int
+		verbose        bool
+		expectedOutput []string
+	}{
+		{
+			desc:           "All migrations up",
+			up:             migrationCt,
+			expectedOutput: []string{"praefect sql-migrate: all migrations are up"},
+		},
+		{
+			desc: "All migrations down",
+			up:   0,
+			expectedOutput: []string{
+				fmt.Sprintf("praefect sql-migrate: migrations to apply: %d", migrationCt),
+				"20200109161404_hello_world: migrating",
+				"20200109161404_hello_world: applied (",
+				fmt.Sprintf("praefect sql-migrate: OK (applied %d migrations)", migrationCt),
+			},
+		},
+		{
+			desc: "Some migrations down",
+			up:   10,
+			expectedOutput: []string{
+				fmt.Sprintf("praefect sql-migrate: migrations to apply: %d", migrationCt-10),
+				"20201126165633_repository_assignments_table: migrating",
+				"20201126165633_repository_assignments_table: applied (",
+				fmt.Sprintf("praefect sql-migrate: OK (applied %d migrations)", migrationCt-10),
+			},
+		},
+		{
+			desc:    "Verbose output",
+			up:      0,
+			verbose: true,
+			expectedOutput: []string{
+				fmt.Sprintf("praefect sql-migrate: migrations to apply: %d", migrationCt),
+				"20200109161404_hello_world: migrating",
+				"[CREATE TABLE hello_world (id integer)]",
+				"20200109161404_hello_world: applied (",
+				fmt.Sprintf("praefect sql-migrate: OK (applied %d migrations)", migrationCt),
+			},
+		},
+	} {
+		t.Run(tc.desc, func(t *testing.T) {
+			testdb.SetMigrations(t, db, cfg, tc.up)
+
+			var stdout bytes.Buffer
+			migrateCmd := sqlMigrateSubcommand{w: &stdout, ignoreUnknown: true, verbose: tc.verbose}
+			assert.NoError(t, migrateCmd.Exec(flag.NewFlagSet("", flag.PanicOnError), cfg))
+
+			for _, out := range tc.expectedOutput {
+				assert.Contains(t, stdout.String(), out)
+			}
+		})
+	}
+}