Added test for utils

Author: Manuelvaas

internal/pkg/services/logs/utils/utils_test.go

@@ -12,8 +12,9 @@ import (
 )
 
 var (
-	testProjectId  = uuid.NewString()
-	testInstanceId = uuid.NewString()
+	testProjectId     = uuid.NewString()
+	testInstanceId    = uuid.NewString()
+	testAccessTokenId = uuid.NewString()
 )
 
 const (
@@ -22,9 +23,10 @@ const (
 )
 
 type logsClientMocked struct {
-	getInstanceFails   bool
-	getInstanceResp    *logs.LogsInstance
-	getAccessTokenResp *logs.AccessToken
+	getInstanceFails    bool
+	getInstanceResp     *logs.LogsInstance
+	getAccessTokenFails bool
+	getAccessTokenResp  *logs.AccessToken
 }
 
 func (m *logsClientMocked) GetLogsInstanceExecute(_ context.Context, _, _, _ string) (*logs.LogsInstance, error) {
@@ -35,8 +37,8 @@ func (m *logsClientMocked) GetLogsInstanceExecute(_ context.Context, _, _, _ str
 }
 
 func (m *logsClientMocked) GetAccessTokenExecute(_ context.Context, _, _, _, _ string) (*logs.AccessToken, error) {
-	if m.getInstanceFails {
-		return nil, fmt.Errorf("could not get instance")
+	if m.getAccessTokenFails {
+		return nil, fmt.Errorf("could not get access token")
 	}
 	return m.getAccessTokenResp, nil
 }
@@ -102,3 +104,65 @@ func TestGetInstanceName(t *testing.T) {
 		})
 	}
 }
+
+func TestGetAccessTokenName(t *testing.T) {
+	tests := []struct {
+		description         string
+		getAccessTokenFails bool
+		getAccessTokenResp  *logs.AccessToken
+		isValid             bool
+		expectedOutput      string
+	}{
+		{
+			description: "base",
+			getAccessTokenResp: &logs.AccessToken{
+				DisplayName: utils.Ptr(testInstanceName),
+			},
+			isValid:        true,
+			expectedOutput: testInstanceName,
+		},
+		{
+			description:         "get instance fails",
+			getAccessTokenFails: true,
+			isValid:             false,
+		},
+		{
+			description:         "response is nil",
+			getAccessTokenFails: false,
+			getAccessTokenResp:  nil,
+			isValid:             false,
+		},
+		{
+			description:         "name in response is nil",
+			getAccessTokenFails: false,
+			getAccessTokenResp: &logs.AccessToken{
+				DisplayName: nil,
+			},
+			isValid: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.description, func(t *testing.T) {
+			client := &logsClientMocked{
+				getAccessTokenFails: tt.getAccessTokenFails,
+				getAccessTokenResp:  tt.getAccessTokenResp,
+			}
+
+			output, err := GetAccessTokenName(context.Background(), client, testProjectId, testRegion, testInstanceId, testAccessTokenId)
+
+			if tt.isValid && err != nil {
+				t.Errorf("failed on valid input")
+			}
+			if !tt.isValid && err == nil {
+				t.Errorf("did not fail on invalid input")
+			}
+			if !tt.isValid {
+				return
+			}
+			if output != tt.expectedOutput {
+				t.Errorf("expected output to be %s, got %s", tt.expectedOutput, output)
+			}
+		})
+	}
+}