89337ea01f
* add regex support for api tests * revert dumb thing * add api tests * add unit test for core async workflow * add xorm store unit tests * fix typo * remove unnecessary assignment
232 lines
7.6 KiB
Go
232 lines
7.6 KiB
Go
package cloudmigrationimpl
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/api/routing"
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/infra/db"
|
|
"github.com/grafana/grafana/pkg/infra/tracing"
|
|
"github.com/grafana/grafana/pkg/services/cloudmigration"
|
|
"github.com/grafana/grafana/pkg/services/contexthandler/ctxkey"
|
|
contextmodel "github.com/grafana/grafana/pkg/services/contexthandler/model"
|
|
"github.com/grafana/grafana/pkg/services/dashboards"
|
|
"github.com/grafana/grafana/pkg/services/datasources"
|
|
datafakes "github.com/grafana/grafana/pkg/services/datasources/fakes"
|
|
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
|
"github.com/grafana/grafana/pkg/services/folder"
|
|
"github.com/grafana/grafana/pkg/services/folder/foldertest"
|
|
secretsfakes "github.com/grafana/grafana/pkg/services/secrets/fakes"
|
|
"github.com/grafana/grafana/pkg/services/user"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/require"
|
|
"go.opentelemetry.io/otel/sdk/trace/tracetest"
|
|
)
|
|
|
|
func Test_NoopServiceDoesNothing(t *testing.T) {
|
|
s := &NoopServiceImpl{}
|
|
_, e := s.CreateToken(context.Background())
|
|
assert.ErrorIs(t, e, cloudmigration.ErrFeatureDisabledError)
|
|
}
|
|
|
|
func Test_CreateGetAndDeleteToken(t *testing.T) {
|
|
s := setUpServiceTest(t, false)
|
|
|
|
createResp, err := s.CreateToken(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, createResp.Token)
|
|
|
|
token, err := s.GetToken(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, token.Name)
|
|
|
|
err = s.DeleteToken(context.Background(), token.ID)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = s.GetToken(context.Background())
|
|
assert.ErrorIs(t, cloudmigration.ErrTokenNotFound, err)
|
|
|
|
cm := cloudmigration.CloudMigrationSession{}
|
|
err = s.ValidateToken(context.Background(), cm)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func Test_CreateGetRunMigrationsAndRuns(t *testing.T) {
|
|
s := setUpServiceTest(t, true)
|
|
|
|
createTokenResp, err := s.CreateToken(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, createTokenResp.Token)
|
|
|
|
cmd := cloudmigration.CloudMigrationSessionRequest{
|
|
AuthToken: createTokenResp.Token,
|
|
}
|
|
|
|
createResp, err := s.CreateSession(context.Background(), cmd)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, createResp.UID)
|
|
require.NotEmpty(t, createResp.Slug)
|
|
|
|
getMigResp, err := s.GetSession(context.Background(), createResp.UID)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, getMigResp)
|
|
require.Equal(t, createResp.UID, getMigResp.UID)
|
|
require.Equal(t, createResp.Slug, getMigResp.Slug)
|
|
|
|
listResp, err := s.GetSessionList(context.Background())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, listResp)
|
|
require.Equal(t, 1, len(listResp.Sessions))
|
|
require.Equal(t, createResp.UID, listResp.Sessions[0].UID)
|
|
require.Equal(t, createResp.Slug, listResp.Sessions[0].Slug)
|
|
|
|
runResp, err := s.RunMigration(ctxWithSignedInUser(), createResp.UID)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, runResp)
|
|
resultItemsByType := make(map[string]int)
|
|
for _, item := range runResp.Items {
|
|
resultItemsByType[string(item.Type)] = resultItemsByType[string(item.Type)] + 1
|
|
}
|
|
require.Equal(t, 1, resultItemsByType["DASHBOARD"])
|
|
require.Equal(t, 2, resultItemsByType["DATASOURCE"])
|
|
require.Equal(t, 2, len(resultItemsByType))
|
|
|
|
runStatusResp, err := s.GetMigrationStatus(context.Background(), runResp.RunUID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, runResp.RunUID, runStatusResp.UID)
|
|
|
|
listRunResp, err := s.GetMigrationRunList(context.Background(), createResp.UID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(listRunResp.Runs))
|
|
require.Equal(t, runResp.RunUID, listRunResp.Runs[0].RunUID)
|
|
|
|
delMigResp, err := s.DeleteSession(context.Background(), createResp.UID)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, createResp.UID, delMigResp.UID)
|
|
}
|
|
|
|
func Test_ExecuteAsyncWorkflow(t *testing.T) {
|
|
s := setUpServiceTest(t, false)
|
|
|
|
createTokenResp, err := s.CreateToken(context.Background())
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, createTokenResp.Token)
|
|
|
|
cmd := cloudmigration.CloudMigrationSessionRequest{
|
|
AuthToken: createTokenResp.Token,
|
|
}
|
|
|
|
createResp, err := s.CreateSession(context.Background(), cmd)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, createResp.UID)
|
|
require.NotEmpty(t, createResp.Slug)
|
|
|
|
getSessionResp, err := s.GetSession(context.Background(), createResp.UID)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, getSessionResp)
|
|
require.Equal(t, createResp.UID, getSessionResp.UID)
|
|
require.Equal(t, createResp.Slug, getSessionResp.Slug)
|
|
|
|
listResp, err := s.GetSessionList(context.Background())
|
|
require.NoError(t, err)
|
|
require.NotNil(t, listResp)
|
|
require.Equal(t, 1, len(listResp.Sessions))
|
|
require.Equal(t, createResp.UID, listResp.Sessions[0].UID)
|
|
require.Equal(t, createResp.Slug, listResp.Sessions[0].Slug)
|
|
|
|
sessionUid := createResp.UID
|
|
snapshotResp, err := s.CreateSnapshot(ctxWithSignedInUser(), sessionUid)
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, snapshotResp.UID)
|
|
require.Equal(t, sessionUid, snapshotResp.SessionUID)
|
|
snapshotUid := snapshotResp.UID
|
|
|
|
snapshot, err := s.GetSnapshot(ctxWithSignedInUser(), sessionUid, snapshotUid)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, snapshotResp.UID, snapshot.UID)
|
|
assert.Equal(t, snapshotResp.EncryptionKey, snapshot.EncryptionKey)
|
|
assert.Empty(t, snapshot.Result) // will change once we create a new table for migration items
|
|
|
|
snapshots, err := s.GetSnapshotList(ctxWithSignedInUser(), cloudmigration.ListSnapshotsQuery{SessionUID: sessionUid, Limit: 100})
|
|
require.NoError(t, err)
|
|
assert.Len(t, snapshots, 1)
|
|
assert.Equal(t, snapshotResp.UID, snapshots[0].UID)
|
|
assert.Equal(t, snapshotResp.EncryptionKey, snapshots[0].EncryptionKey)
|
|
assert.Empty(t, snapshots[0].Result) // should remain this way even after we create a new table
|
|
|
|
err = s.UploadSnapshot(ctxWithSignedInUser(), sessionUid, snapshotUid)
|
|
require.NoError(t, err)
|
|
|
|
assert.Panics(t, func() {
|
|
err = s.CancelSnapshot(ctxWithSignedInUser(), sessionUid, snapshotUid)
|
|
})
|
|
}
|
|
|
|
func ctxWithSignedInUser() context.Context {
|
|
c := &contextmodel.ReqContext{
|
|
SignedInUser: &user.SignedInUser{OrgID: 1},
|
|
}
|
|
k := ctxkey.Key{}
|
|
ctx := context.WithValue(context.Background(), k, c)
|
|
return ctx
|
|
}
|
|
|
|
func setUpServiceTest(t *testing.T, withDashboardMock bool) cloudmigration.Service {
|
|
sqlStore := db.InitTestDB(t)
|
|
secretsService := secretsfakes.NewFakeSecretsService()
|
|
rr := routing.NewRouteRegister()
|
|
spanRecorder := tracetest.NewSpanRecorder()
|
|
tracer := tracing.InitializeTracerForTest(tracing.WithSpanProcessor(spanRecorder))
|
|
mockFolder := &foldertest.FakeService{
|
|
ExpectedFolder: &folder.Folder{UID: "folderUID", Title: "Folder"},
|
|
}
|
|
|
|
cfg := setting.NewCfg()
|
|
section, err := cfg.Raw.NewSection("cloud_migration")
|
|
require.NoError(t, err)
|
|
_, err = section.NewKey("domain", "localhost:1234")
|
|
require.NoError(t, err)
|
|
// dont know if this is the best, but dont want to refactor at the moment
|
|
cfg.CloudMigration.IsDeveloperMode = true
|
|
|
|
dashboardService := dashboards.NewFakeDashboardService(t)
|
|
if withDashboardMock {
|
|
dashboardService.On("GetAllDashboards", mock.Anything).Return(
|
|
[]*dashboards.Dashboard{
|
|
{
|
|
UID: "1",
|
|
Data: simplejson.New(),
|
|
},
|
|
},
|
|
nil,
|
|
)
|
|
}
|
|
|
|
dsService := &datafakes.FakeDataSourceService{
|
|
DataSources: []*datasources.DataSource{
|
|
{Name: "mmm", Type: "mysql"},
|
|
{Name: "ZZZ", Type: "infinity"},
|
|
},
|
|
}
|
|
|
|
s, err := ProvideService(
|
|
cfg,
|
|
featuremgmt.WithFeatures(featuremgmt.FlagOnPremToCloudMigrations),
|
|
sqlStore,
|
|
dsService,
|
|
secretsService,
|
|
rr,
|
|
prometheus.DefaultRegisterer,
|
|
tracer,
|
|
dashboardService,
|
|
mockFolder,
|
|
)
|
|
require.NoError(t, err)
|
|
|
|
return s
|
|
}
|