[gRPC/ObjC] Fix flaky test InteropTests::testHijackingInterceptor (gr…

…pc#27116)
dragan-radovic · Aug 26, 2021 · cef46fe · cef46fe
1 parent 764232f
commit cef46fe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/objective-c/tests/InteropTests/InteropTests.m b/src/objective-c/tests/InteropTests/InteropTests.m
@@ -1603,6 +1603,8 @@ - (void)testHijackingInterceptor {
   XCTAssertNotNil([[self class] host]);
   __weak XCTestExpectation *expectUserCallComplete =
       [self expectationWithDescription:@"User call completed."];
+  __weak XCTestExpectation *expectResponseCallbackComplete =
+      [self expectationWithDescription:@"Hook interceptor response callback completed"];
 
   NSArray *responses = @[ @1, @2, @3, @4 ];
   __block int index = 0;
@@ -1659,6 +1661,7 @@ - (void)testHijackingInterceptor {
         XCTAssertNil(trailingMetadata);
         XCTAssertNotNil(error);
         XCTAssertEqual(error.code, GRPC_STATUS_CANCELLED);
+        [expectResponseCallbackComplete fulfill];
       }
       didWriteDataHook:nil];