Unit test fixed

kyma-project · Mar 22, 2024 · 6dabb41 · 6dabb41
1 parent 3f542e8
commit 6dabb41
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/components/compass-runtime-agent/internal/compassconnection/package_test.go b/components/compass-runtime-agent/internal/compassconnection/package_test.go
@@ -325,10 +325,12 @@ func TestCompassConnectionController(t *testing.T) {
 		// given
 		clearMockCalls(&configurationClientMock.Mock)
 		configurationClientMock.On("FetchConfiguration", requestIDCtxMatcher).Return(nil, nil, errors.New("error"))
+		configurationClientMock.On("SetRuntimeStatusCondition", mock.Anything, graphql.RuntimeStatusConditionConnected).Return(nil)
 
 		// when
 		err = waitFor(checkInterval, testTimeout, func() bool {
-			return mockFunctionCalled(&configurationClientMock.Mock, "FetchConfiguration", requestIDCtxMatcher)
+			return mockFunctionCalled(&configurationClientMock.Mock, "FetchConfiguration", requestIDCtxMatcher) &&
+				mockFunctionCalled(&configurationClientMock.Mock, "SetRuntimeStatusCondition", mock.Anything, graphql.RuntimeStatusConditionConnected)
 		})
 
 		// then