CloudMigrations: Add unit tests for snapshot management (#89521)

* 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

pkg/services/cloudmigration/cloudmigrationimpl/cloudmigration_test.go

@@ -109,6 +109,63 @@ func Test_CreateGetRunMigrationsAndRuns(t *testing.T) {
 	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},