Explicit wantServices in TestRegisterGRPCServices

canonical · Sep 6, 2024 · 729c4c6 · 729c4c6
1 parent 6f4c70d
commit 729c4c6
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/windows-agent/internal/proservices/proservices_test.go b/windows-agent/internal/proservices/proservices_test.go
@@ -128,20 +128,16 @@ func TestRegisterGRPCServices(t *testing.T) {
 		wantServices []string
 		wantErr      bool
 	}{
-		"Success with WSL net adapter":    {},
+		"Success with WSL net adapter":    {wantServices: defaultServices},
 		"Success without WSL net adapter": {withoutWSLNet: true, wantServices: []string{"agentapi.UI"}},
 
-		"Error with insecure requests": {insecureClient: true, wantErr: true},
+		"Error with insecure requests": {insecureClient: true, wantServices: defaultServices, wantErr: true},
 	}
 
 	for name, tc := range testCases {
 		t.Run(name, func(t *testing.T) {
 			t.Parallel()
 
-			if tc.wantServices == nil {
-				tc.wantServices = defaultServices
-			}
-
 			ctx, cancel := context.WithCancel(context.Background())
 			t.Cleanup(cancel)