mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-16 18:42:30 +03:00
c7ac1ede3f
> ### Description > This PR adds two new packages which implement the CLI requirements specified in RFC:https://github.com/hasura/lux/blob/cloud/docs/rfcs/20210614_github_integration.md 1. `pkg/metadata`  2. `pkg/migrate`  ### Changelog - [x] `CHANGELOG.md` is updated with user-facing content relevant to this PR. If no changelog is required, then add the `no-changelog-required` label. ### Affected components - [x] CLI https://github.com/hasura/graphql-engine-mono/pull/1598 GitOrigin-RevId: 0e2bce498386c5aae68dbca0fe383a6afff9d1a9
542 lines
13 KiB
Go
542 lines
13 KiB
Go
package migrate
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/hasura/graphql-engine/cli/v2/internal/testutil"
|
|
)
|
|
|
|
func TestProjectMigrate_ApplyConfig_v3(t *testing.T) {
|
|
port, teardown := testutil.StartHasuraWithMetadataDatabase(t, testutil.HasuraDockerImage)
|
|
hasuraEndpoint := fmt.Sprintf("%s:%s", testutil.BaseURL, port)
|
|
connectionStringSource1, teardownPG1 := testutil.StartPGContainer(t)
|
|
connectionStringSource2, teardownPG2 := testutil.StartPGContainer(t)
|
|
testutil.AddPGSourceToHasura(t, hasuraEndpoint, connectionStringSource1, "s1")
|
|
testutil.AddPGSourceToHasura(t, hasuraEndpoint, connectionStringSource2, "s2")
|
|
defer func() {
|
|
teardownPG2()
|
|
teardownPG1()
|
|
teardown()
|
|
}()
|
|
hgeEndpoint := fmt.Sprintf("http://localhost:%s", port)
|
|
type fields struct {
|
|
projectDirectory string
|
|
endpointString string
|
|
}
|
|
type args struct {
|
|
opts []ProjectMigrationApplierOption
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
"can apply migrations in config v3 project",
|
|
fields{
|
|
projectDirectory: "testdata/projectv3",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyOnAllDatabases()},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"can apply a version in config v3 project",
|
|
fields{
|
|
projectDirectory: "testdata/projectv3",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyOnDatabaseName("s1"), ApplyVersion("1623841477474", MigrationDirectionDown)},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"can apply a version in config v3 project",
|
|
fields{
|
|
projectDirectory: "testdata/projectv3",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyOnDatabaseName("s1"), ApplyVersion("1623841477474", MigrationDirectionUp)},
|
|
},
|
|
false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
p, err := NewProjectMigrate(tt.fields.projectDirectory, WithAdminSecret(testutil.TestAdminSecret), WithEndpoint(tt.fields.endpointString))
|
|
require.NoError(t, err)
|
|
err = p.Apply(tt.args.opts...)
|
|
if tt.wantErr {
|
|
require.Error(t, err)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestProjectMigrate_Apply_Configv2(t *testing.T) {
|
|
port, teardown := testutil.StartHasura(t, testutil.HasuraDockerImage)
|
|
defer teardown()
|
|
hgeEndpoint := fmt.Sprintf("http://localhost:%s", port)
|
|
type fields struct {
|
|
projectDirectory string
|
|
adminSecret string
|
|
endpointString string
|
|
}
|
|
type args struct {
|
|
opts []ProjectMigrationApplierOption
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
"can apply migrations in config v2 project",
|
|
fields{
|
|
projectDirectory: "testdata/projectv2",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyOnAllDatabases()},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"can apply down migration on a version in config v2 project",
|
|
fields{
|
|
projectDirectory: "testdata/projectv2",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyVersion("1623842054907", MigrationDirectionDown)},
|
|
},
|
|
false,
|
|
},
|
|
{
|
|
"throws error when trying to do a down migration which is not applied",
|
|
fields{
|
|
projectDirectory: "testdata/projectv2",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyVersion("1623842054907", MigrationDirectionDown)},
|
|
},
|
|
true,
|
|
},
|
|
{
|
|
"can apply up migrations of a version on a config v2 project",
|
|
fields{
|
|
projectDirectory: "testdata/projectv2",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
[]ProjectMigrationApplierOption{ApplyVersion("1623842054907", MigrationDirectionUp)},
|
|
},
|
|
false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
p, err := NewProjectMigrate(tt.fields.projectDirectory, WithAdminSecret(testutil.TestAdminSecret), WithEndpoint(tt.fields.endpointString))
|
|
require.NoError(t, err)
|
|
err = p.Apply(tt.args.opts...)
|
|
if tt.wantErr {
|
|
require.Error(t, err)
|
|
} else {
|
|
require.NoError(t, err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestProjectMigrate_Status_ConfigV2(t *testing.T) {
|
|
port, teardown := testutil.StartHasura(t, testutil.HasuraDockerImage)
|
|
defer teardown()
|
|
hgeEndpoint := fmt.Sprintf("http://localhost:%s", port)
|
|
type fields struct {
|
|
projectDirectory string
|
|
adminSecret string
|
|
endpointString string
|
|
}
|
|
type args struct {
|
|
opts []ProjectMigrationStatusOption
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
want string
|
|
wantErr bool
|
|
before func(t *testing.T, p *ProjectMigrate)
|
|
}{
|
|
{
|
|
"can get status of migrations",
|
|
fields{
|
|
projectDirectory: "testdata/projectv2",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
opts: []ProjectMigrationStatusOption{},
|
|
},
|
|
`
|
|
[
|
|
{
|
|
"databaseName": "default",
|
|
"status": {
|
|
"migrations": [
|
|
1623842054907,
|
|
1623842062104,
|
|
1623842069725,
|
|
1623842076537,
|
|
1623842087940
|
|
],
|
|
"status": {
|
|
"1623842054907": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623842062104": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623842069725": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623842076537": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623842087940": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
]`,
|
|
false,
|
|
func(t *testing.T, p *ProjectMigrate) {},
|
|
},
|
|
{
|
|
"can get status of migrations",
|
|
fields{
|
|
projectDirectory: "testdata/projectv2",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
opts: []ProjectMigrationStatusOption{},
|
|
},
|
|
`
|
|
[
|
|
{
|
|
"databaseName": "default",
|
|
"status": {
|
|
"migrations": [
|
|
1623842054907,
|
|
1623842062104,
|
|
1623842069725,
|
|
1623842076537,
|
|
1623842087940
|
|
],
|
|
"status": {
|
|
"1623842054907": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623842062104": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623842069725": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623842076537": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623842087940": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
]`,
|
|
false,
|
|
func(t *testing.T, p *ProjectMigrate) {
|
|
assert.NoError(t, p.Apply(ApplyOnAllDatabases()))
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
p, err := NewProjectMigrate(tt.fields.projectDirectory, WithAdminSecret(testutil.TestAdminSecret), WithEndpoint(tt.fields.endpointString))
|
|
require.NoError(t, err)
|
|
applier, err := NewProjectMigrate(tt.fields.projectDirectory, WithAdminSecret(testutil.TestAdminSecret), WithEndpoint(tt.fields.endpointString))
|
|
require.NoError(t, err)
|
|
tt.before(t, applier)
|
|
got, err := p.status(tt.args.opts...)
|
|
if tt.wantErr {
|
|
require.Error(t, err)
|
|
}
|
|
require.NoError(t, err)
|
|
gotJSON, err := json.Marshal(got)
|
|
require.NoError(t, err)
|
|
require.JSONEq(t, tt.want, string(gotJSON))
|
|
|
|
statusJson, err := p.StatusJSON(tt.args.opts...)
|
|
require.NoError(t, err)
|
|
statusJsonb, err := ioutil.ReadAll(statusJson)
|
|
require.NoError(t, err)
|
|
require.JSONEq(t, tt.want, string(statusJsonb))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestProjectMigrate_Status_ConfigV3(t *testing.T) {
|
|
port, teardown := testutil.StartHasuraWithMetadataDatabase(t, testutil.HasuraDockerImage)
|
|
hasuraEndpoint := fmt.Sprintf("%s:%s", testutil.BaseURL, port)
|
|
connectionStringSource1, teardownPG1 := testutil.StartPGContainer(t)
|
|
connectionStringSource2, teardownPG2 := testutil.StartPGContainer(t)
|
|
testutil.AddPGSourceToHasura(t, hasuraEndpoint, connectionStringSource1, "s1")
|
|
testutil.AddPGSourceToHasura(t, hasuraEndpoint, connectionStringSource2, "s2")
|
|
defer func() {
|
|
teardownPG2()
|
|
teardownPG1()
|
|
teardown()
|
|
}()
|
|
hgeEndpoint := fmt.Sprintf("http://localhost:%s", port)
|
|
type fields struct {
|
|
projectDirectory string
|
|
adminSecret string
|
|
endpointString string
|
|
}
|
|
type args struct {
|
|
opts []ProjectMigrationStatusOption
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
want string
|
|
wantErr bool
|
|
before func(t *testing.T, p *ProjectMigrate)
|
|
}{
|
|
{
|
|
"can get status of migrations",
|
|
fields{
|
|
projectDirectory: "testdata/projectv3",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
opts: []ProjectMigrationStatusOption{},
|
|
},
|
|
`[
|
|
{
|
|
"databaseName": "s1",
|
|
"status": {
|
|
"migrations": [
|
|
1623841477474,
|
|
1623841485323,
|
|
1623841492743,
|
|
1623841500466,
|
|
1623841510619
|
|
],
|
|
"status": {
|
|
"1623841477474": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841485323": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841492743": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841500466": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841510619": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
"databaseName": "s2",
|
|
"status": {
|
|
"migrations": [
|
|
1623841477474,
|
|
1623841485323,
|
|
1623841492743,
|
|
1623841500466,
|
|
1623841510619
|
|
],
|
|
"status": {
|
|
"1623841477474": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841485323": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841492743": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841500466": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
},
|
|
"1623841510619": {
|
|
"database_status": false,
|
|
"source_status": true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
]`,
|
|
false,
|
|
func(t *testing.T, p *ProjectMigrate) {},
|
|
},
|
|
{
|
|
"can get status of migrations",
|
|
fields{
|
|
projectDirectory: "testdata/projectv3",
|
|
adminSecret: "",
|
|
endpointString: hgeEndpoint,
|
|
},
|
|
args{
|
|
opts: []ProjectMigrationStatusOption{},
|
|
},
|
|
`
|
|
[
|
|
{
|
|
"databaseName": "s1",
|
|
"status": {
|
|
"migrations": [
|
|
1623841477474,
|
|
1623841485323,
|
|
1623841492743,
|
|
1623841500466,
|
|
1623841510619
|
|
],
|
|
"status": {
|
|
"1623841477474": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841485323": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841492743": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841500466": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841510619": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
}
|
|
}
|
|
}
|
|
},
|
|
{
|
|
"databaseName": "s2",
|
|
"status": {
|
|
"migrations": [
|
|
1623841477474,
|
|
1623841485323,
|
|
1623841492743,
|
|
1623841500466,
|
|
1623841510619
|
|
],
|
|
"status": {
|
|
"1623841477474": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841485323": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841492743": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841500466": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
},
|
|
"1623841510619": {
|
|
"database_status": true,
|
|
"source_status": true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
]`,
|
|
false,
|
|
func(t *testing.T, p *ProjectMigrate) {
|
|
assert.NoError(t, p.Apply(ApplyOnAllDatabases()))
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
p, err := NewProjectMigrate(tt.fields.projectDirectory, WithAdminSecret(testutil.TestAdminSecret), WithEndpoint(tt.fields.endpointString))
|
|
require.NoError(t, err)
|
|
applier, err := NewProjectMigrate(tt.fields.projectDirectory, WithAdminSecret(testutil.TestAdminSecret), WithEndpoint(tt.fields.endpointString))
|
|
require.NoError(t, err)
|
|
tt.before(t, applier)
|
|
got, err := p.status(tt.args.opts...)
|
|
if tt.wantErr {
|
|
require.Error(t, err)
|
|
}
|
|
require.NoError(t, err)
|
|
gotJSON, err := json.Marshal(got)
|
|
require.NoError(t, err)
|
|
require.JSONEq(t, tt.want, string(gotJSON))
|
|
|
|
statusJson, err := p.StatusJSON(tt.args.opts...)
|
|
require.NoError(t, err)
|
|
statusJsonb, err := ioutil.ReadAll(statusJson)
|
|
require.NoError(t, err)
|
|
require.JSONEq(t, tt.want, string(statusJsonb))
|
|
})
|
|
}
|
|
}
|