feat: Convert internal interfaces to use protos

googleapis · Feb 20, 2023 · b8148ec · b8148ec
1 parent c14b41b
commit b8148ec
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/...e/src/test/java/com/google/cloud/pubsublite/internal/wire/ConnectedCommitterImplTest.java b/...e/src/test/java/com/google/cloud/pubsublite/internal/wire/ConnectedCommitterImplTest.java
@@ -224,10 +224,18 @@ public void commitResponseProxied() {
 
   @Test
   public void receiveTimeout_closesConnection() throws Exception {
-    CountDownLatch connectionClosed = new CountDownLatch(1);
+    CountDownLatch requestClosed = new CountDownLatch(1);
     doAnswer(
             args -> {
-              connectionClosed.countDown();
+              requestClosed.countDown();
+              return null;
+            })
+        .when(mockRequestStream)
+        .closeSendWithError(any());
+    CountDownLatch outputClosed = new CountDownLatch(1);
+    doAnswer(
+            args -> {
+              outputClosed.countDown();
               return null;
             })
         .when(mockOutputStream)
@@ -245,7 +253,8 @@ public void receiveTimeout_closesConnection() throws Exception {
     verify(mockRequestStream).send(initialRequest());
 
     // No subsequent stream responses should close the stream.
-    assertThat(connectionClosed.await(30, SECONDS)).isTrue();
+    assertThat(requestClosed.await(30, SECONDS)).isTrue();
+    assertThat(outputClosed.await(30, SECONDS)).isTrue();
 
     verify(mockRequestStream).closeSendWithError(argThat(new ApiExceptionMatcher(Code.ABORTED)));
     verify(mockOutputStream).onError(argThat(new ApiExceptionMatcher(Code.ABORTED)));