CloudMigrations: Query Grafana Migration Status for status while the snapshot is in the cloud (#90314)

* implement querying gms for snapshot status

* add some documentation

* provide snapshot resources after snapshot is created

* add rate limiting to backend

* fix compilation error

* fix typo

* add unit tests

* finish merge

* lint

* swagger gen

* more testing

* remove duplicate test

* address a couple PR comments

* update switch statement to a map

* add timeouts to gms client through the http client

* remove extra whitespace

* put method back where it was so the PR is less confusing

* fix tests

* add todo

* fix final unit test

pkg/services/cloudmigration/gmsclient/inmemory_client.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"fmt"
 	"math/rand"
-	"time"
 
 	cryptoRand "crypto/rand"
 
@@ -68,33 +67,28 @@ func (c *memoryClientImpl) StartSnapshot(context.Context, cloudmigration.CloudMi
 	return c.snapshot, nil
 }
 
-func (c *memoryClientImpl) GetSnapshotStatus(ctx context.Context, session cloudmigration.CloudMigrationSession, snapshot cloudmigration.CloudMigrationSnapshot) (*cloudmigration.CloudMigrationSnapshot, error) {
-	results := []cloudmigration.CloudMigrationResource{
-		{
-			Type:   cloudmigration.DashboardDataType,
-			RefID:  "dash1",
-			Status: cloudmigration.ItemStatusOK,
-		},
-		{
-			Type:   cloudmigration.DatasourceDataType,
-			RefID:  "ds1",
-			Status: cloudmigration.ItemStatusError,
-			Error:  "fake error",
-		},
-		{
-			Type:   cloudmigration.FolderDataType,
-			RefID:  "folder1",
-			Status: cloudmigration.ItemStatusOK,
+func (c *memoryClientImpl) GetSnapshotStatus(ctx context.Context, session cloudmigration.CloudMigrationSession, snapshot cloudmigration.CloudMigrationSnapshot) (*cloudmigration.GetSnapshotStatusResponse, error) {
+	gmsResp := &cloudmigration.GetSnapshotStatusResponse{
+		State: cloudmigration.SnapshotStateFinished,
+		Results: []cloudmigration.CloudMigrationResource{
+			{
+				Type:   cloudmigration.DashboardDataType,
+				RefID:  "dash1",
+				Status: cloudmigration.ItemStatusOK,
+			},
+			{
+				Type:   cloudmigration.DatasourceDataType,
+				RefID:  "ds1",
+				Status: cloudmigration.ItemStatusError,
+				Error:  "fake error",
+			},
+			{
+				Type:   cloudmigration.FolderDataType,
+				RefID:  "folder1",
+				Status: cloudmigration.ItemStatusOK,
+			},
 		},
 	}
 
-	// just fake an entire response
-	gmsSnapshot := cloudmigration.CloudMigrationSnapshot{
-		Status:         cloudmigration.SnapshotStatusFinished,
-		GMSSnapshotUID: "gmssnapshotuid",
-		Resources:      results,
-		Finished:       time.Now(),
-	}
-
-	return &gmsSnapshot, nil
+	return gmsResp, nil
 }