graphql-engine/cli/integration_test/v2/migrate.go

191 lines
4.5 KiB
Go
Raw Normal View History

package v2
import (
"os"
"path/filepath"
"testing"
"github.com/ghodss/yaml"
"github.com/hasura/graphql-engine/cli"
"github.com/hasura/graphql-engine/cli/commands"
"github.com/hasura/graphql-engine/cli/migrate"
"github.com/hasura/graphql-engine/cli/util"
"github.com/stretchr/testify/assert"
)
type migrateInterface interface {
Run() error
}
func TestMigrateCmd(t *testing.T, ec *cli.ExecutionContext) {
// copy migrations to ec.Execution.Directory/migrations
os.RemoveAll(ec.MigrationDir)
currDir, _ := os.Getwd()
err := util.CopyDir(filepath.Join(currDir, "v2/migrations"), ec.MigrationDir)
if err != nil {
t.Fatalf("unable to copy migrations directory %v", err)
}
tt := []struct {
name string
opts migrateInterface
err error
status migrate.Status
}{
{"apply-up-all-migrations", &commands.MigrateApplyOptions{
EC: ec,
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
},
}},
{"apply-down-1-migration", &commands.MigrateApplyOptions{
EC: ec,
DownMigration: "1",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
},
}},
{"apply-down-all-migration", &commands.MigrateApplyOptions{
EC: ec,
DownMigration: "all",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
},
}},
{"apply-goto-2-migration", &commands.MigrateApplyOptions{
EC: ec,
GotoVersion: "2",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
},
}},
{"apply-goto-nil-migration", &commands.MigrateApplyOptions{
EC: ec,
GotoVersion: "-1",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
},
}},
{"apply-up-1-migration", &commands.MigrateApplyOptions{
EC: ec,
UpMigration: "1",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
},
}},
{"apply-version-2-up-migration", &commands.MigrateApplyOptions{
EC: ec,
VersionMigration: "2",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
},
}},
{"apply-version-2-down-migration", &commands.MigrateApplyOptions{
EC: ec,
VersionMigration: "2",
MigrationType: "down",
}, nil, migrate.Status{
Index: []uint64{1, 2},
Migrations: map[uint64]*migrate.MigrationStatus{
1: &migrate.MigrationStatus{
IsApplied: true,
IsPresent: true,
},
2: &migrate.MigrationStatus{
IsApplied: false,
IsPresent: true,
},
},
}},
}
for _, tc := range tt {
t.Run(tc.name, func(t *testing.T) {
err := tc.opts.Run()
if err != tc.err {
t.Fatalf("%s: expected %v, got %v", tc.name, tc.err, err)
}
expectedStatusByt, err := yaml.Marshal(tc.status)
if err != nil {
t.Fatal(err)
}
statusOpts := &commands.MigrateStatusOptions{
EC: ec,
}
actualStatus, err := statusOpts.Run()
if err != nil {
t.Fatalf("%s: unable to fetch migrate status, got %v", tc.name, err)
}
actualStatusByt, err := yaml.Marshal(actualStatus)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, string(expectedStatusByt), string(actualStatusByt))
})
}
}