Add connection tests

Author: w1am

.github/workflows/tests.yml

@@ -20,7 +20,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        test: [Streams, PersistentSubscriptions, Telemetry, MultiStreamAppend]
+        test: [Streams, PersistentSubscriptions, Telemetry, MultiStreamAppend, Connection]
 
     runs-on: ubuntu-latest
     steps:

build.gradle

@@ -132,6 +132,7 @@ tasks.register("singleNodeTests", Test) {
         include("**/PersistentSubscriptionsTests.class")
         include("**/TelemetryTests.class")
         include("**/ConnectionShutdownTests.class")
+        include("**/ConnectionTests.class")
     }
 }
 
@@ -155,6 +156,7 @@ tasks.register("clusterTests", Test) {
         include("**/StreamsTests.class")
         include("**/PersistentSubscriptionsTests.class")
         include("**/ConnectionShutdownTests.class")
+        include("**/ConnectionTests.class")
     }
 }
 

src/test/java/io/kurrent/dbclient/connection/ConnectionTests.java

@@ -0,0 +1,58 @@
+package io.kurrent.dbclient.connection;
+
+import io.grpc.Status;
+import io.grpc.StatusRuntimeException;
+import io.kurrent.dbclient.*;
+import io.kurrent.dbclient.databases.DockerContainerDatabase;
+import org.junit.jupiter.api.*;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+
+public class ConnectionTests {
+    static private DockerContainerDatabase database;
+    static private Logger logger;
+
+    @BeforeEach
+    public void setup() {
+        database = (DockerContainerDatabase) DatabaseFactory.spawn();
+        logger = LoggerFactory.getLogger(PersistentSubscriptionsTests.class);
+    }
+
+    @AfterEach
+    public void cleanup() {
+        unpauseDatabase();
+        database.dispose();
+    }
+
+    @Test
+    @Timeout(value = 30, unit = TimeUnit.SECONDS)
+    public void testCallTerminationWhenServerUnreachable() throws Throwable {
+        KurrentDBClient client = database.defaultClient();
+
+        ReadResult initialResult = client.readAll(ReadAllOptions.get()).get(5, TimeUnit.SECONDS);
+
+        Assertions.assertFalse(initialResult.getEvents().isEmpty());
+
+        pauseDatabase();
+
+        ExecutionException e = Assertions.assertThrows(ExecutionException.class, () ->
+                client.readAll().get(30, TimeUnit.SECONDS)
+        );
+
+        StatusRuntimeException status = (StatusRuntimeException) e.getCause();
+        Assertions.assertEquals(Status.Code.UNAVAILABLE, status.getStatus().getCode());
+    }
+
+    static void pauseDatabase() {
+        logger.debug("Pausing database container: {}", database.getContainerId());
+        database.getDockerClient().pauseContainerCmd(database.getContainerId()).exec();
+    }
+
+    static void unpauseDatabase() {
+        logger.debug("Unpausing database container: {}", database.getContainerId());
+        database.getDockerClient().unpauseContainerCmd(database.getContainerId()).exec();
+    }
+}