fix(batch): avoid Assertions.catch*

Author: bevzzz

src/test/java/io/weaviate/client6/v1/api/collections/batch/BatchContextTest.java

@@ -21,6 +21,7 @@
 import org.junit.After;
 import org.junit.AfterClass;
 import org.junit.Before;
+import org.junit.Ignore;
 import org.junit.Test;
 
 import io.grpc.stub.StreamObserver;
@@ -109,7 +110,7 @@ public void startContext() throws InterruptedException {
 
   @After
   public void reset() throws Exception {
-    if (context != null && !contextClosed) {
+    if (!contextClosed) {
       closeContext();
     }
     context = null;
@@ -136,9 +137,13 @@ private void closeContext() throws Exception {
         throw new RuntimeException(e);
       }
     }, BACKGROUND).thenCompose(__ -> out.eof(true));
-    context.close();
-    eof.get();
-    contextClosed = true;
+
+    try {
+      context.close();
+      eof.get();
+    } finally {
+      contextClosed = true;
+    }
   }
 
   @Test
@@ -418,15 +423,15 @@ public void test_maxReconnectRetries() throws Exception {
     }
 
     out.hangup();
-    Assertions.assertThat(in.done).completesExceptionallyWithin(5, TimeUnit.SECONDS);
-
-    Assertions.assertThatThrownBy(() -> context.close())
+    
+    try {
+    this.closeContext();
+    }
+    catch (Throwable t) {
+      Assertions.assertThat(t)
         .isInstanceOf(IOException.class)
         .hasMessageContaining("Server unavailable");
-
-    // Cleanup: unset the context to prevent test teardown code
-    // from tripping on it while trying to close the context.
-    context = null;
+    }
   }
 
   @Test(expected = IllegalStateException.class)