Chore: Add unit test for cloudmigration package (#88868)

2024-06-07 13:46:19 -03:00
parent 3a8b59f5e8
commit 330da7916d
6 changed files with 952 additions and 53 deletions
@@ -0,0 +1,139 @@
+package fake
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"time"
+
+	"github.com/grafana/grafana/pkg/services/cloudmigration"
+	"github.com/grafana/grafana/pkg/services/gcom"
+)
+
+var fixedDate = time.Date(2024, 6, 5, 17, 30, 40, 0, time.UTC)
+
+// FakeServiceImpl fake implementation of cloudmigration.Service for testing purposes
+type FakeServiceImpl struct {
+	ReturnError bool
+}
+
+var _ cloudmigration.Service = (*FakeServiceImpl)(nil)
+
+func (m FakeServiceImpl) GetToken(_ context.Context) (gcom.TokenView, error) {
+	if m.ReturnError {
+		return gcom.TokenView{}, fmt.Errorf("mock error")
+	}
+	return gcom.TokenView{ID: "mock_id", DisplayName: "mock_name"}, nil
+}
+
+func (m FakeServiceImpl) CreateToken(_ context.Context) (cloudmigration.CreateAccessTokenResponse, error) {
+	if m.ReturnError {
+		return cloudmigration.CreateAccessTokenResponse{}, fmt.Errorf("mock error")
+	}
+	return cloudmigration.CreateAccessTokenResponse{Token: "mock_token"}, nil
+}
+
+func (m FakeServiceImpl) ValidateToken(ctx context.Context, migration cloudmigration.CloudMigration) error {
+	panic("implement me")
+}
+
+func (m FakeServiceImpl) DeleteToken(_ context.Context, _ string) error {
+	if m.ReturnError {
+		return fmt.Errorf("mock error")
+	}
+	return nil
+}
+
+func (m FakeServiceImpl) CreateMigration(_ context.Context, _ cloudmigration.CloudMigrationRequest) (*cloudmigration.CloudMigrationResponse, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	return &cloudmigration.CloudMigrationResponse{
+		UID:     "fake_uid",
+		Stack:   "fake_stack",
+		Created: fixedDate,
+		Updated: fixedDate,
+	}, nil
+}
+
+func (m FakeServiceImpl) GetMigration(_ context.Context, _ string) (*cloudmigration.CloudMigration, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	return &cloudmigration.CloudMigration{UID: "fake"}, nil
+}
+
+func (m FakeServiceImpl) DeleteMigration(_ context.Context, _ string) (*cloudmigration.CloudMigration, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	return &cloudmigration.CloudMigration{UID: "fake"}, nil
+}
+
+func (m FakeServiceImpl) UpdateMigration(ctx context.Context, uid string, request cloudmigration.CloudMigrationRequest) (*cloudmigration.CloudMigrationResponse, error) {
+	panic("implement me")
+}
+
+func (m FakeServiceImpl) GetMigrationList(_ context.Context) (*cloudmigration.CloudMigrationListResponse, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	return &cloudmigration.CloudMigrationListResponse{
+		Migrations: []cloudmigration.CloudMigrationResponse{
+			{UID: "mock_uid_1", Stack: "mock_stack_1", Created: fixedDate, Updated: fixedDate},
+			{UID: "mock_uid_2", Stack: "mock_stack_2", Created: fixedDate, Updated: fixedDate},
+		},
+	}, nil
+}
+
+func (m FakeServiceImpl) RunMigration(_ context.Context, _ string) (*cloudmigration.MigrateDataResponseDTO, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	r := fakeMigrateDataResponseDTO()
+	return &r, nil
+}
+
+func fakeMigrateDataResponseDTO() cloudmigration.MigrateDataResponseDTO {
+	return cloudmigration.MigrateDataResponseDTO{
+		RunUID: "fake_uid",
+		Items: []cloudmigration.MigrateDataResponseItemDTO{
+			{Type: "type", RefID: "make_refid", Status: "ok", Error: "none"},
+		},
+	}
+}
+
+func (m FakeServiceImpl) CreateMigrationRun(ctx context.Context, run cloudmigration.CloudMigrationRun) (string, error) {
+	panic("implement me")
+}
+
+func (m FakeServiceImpl) GetMigrationStatus(_ context.Context, _ string) (*cloudmigration.CloudMigrationRun, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	result, err := json.Marshal(fakeMigrateDataResponseDTO())
+	if err != nil {
+		return nil, err
+	}
+	return &cloudmigration.CloudMigrationRun{
+		ID:                0,
+		UID:               "fake_uid",
+		CloudMigrationUID: "fake_mig_uid",
+		Result:            result,
+		Created:           fixedDate,
+		Updated:           fixedDate,
+		Finished:          fixedDate,
+	}, nil
+}
+
+func (m FakeServiceImpl) GetMigrationRunList(_ context.Context, _ string) (*cloudmigration.CloudMigrationRunList, error) {
+	if m.ReturnError {
+		return nil, fmt.Errorf("mock error")
+	}
+	return &cloudmigration.CloudMigrationRunList{
+		Runs: []cloudmigration.MigrateDataResponseListDTO{
+			{RunUID: "fake_run_uid_1"},
+			{RunUID: "fake_run_uid_2"},
+		},
+	}, nil
+}