From 5ac390281e926c3ea41b8b67a393d31b783d8231 Mon Sep 17 00:00:00 2001
From: Jon Chambers <jon@signal.org>
Date: Fri, 21 Aug 2020 12:51:46 -0400
Subject: [PATCH] Add an abstract base class for Redis singleton tests.

---
 .../redis/AbstractRedisSingletonTest.java     | 71 +++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 service/src/test/java/org/whispersystems/textsecuregcm/redis/AbstractRedisSingletonTest.java

diff --git a/service/src/test/java/org/whispersystems/textsecuregcm/redis/AbstractRedisSingletonTest.java b/service/src/test/java/org/whispersystems/textsecuregcm/redis/AbstractRedisSingletonTest.java
new file mode 100644
index 000000000..5ddd05d8b
--- /dev/null
+++ b/service/src/test/java/org/whispersystems/textsecuregcm/redis/AbstractRedisSingletonTest.java
@@ -0,0 +1,71 @@
+package org.whispersystems.textsecuregcm.redis;
+
+import io.lettuce.core.RedisClient;
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.whispersystems.textsecuregcm.configuration.CircuitBreakerConfiguration;
+import org.whispersystems.textsecuregcm.providers.RedisClientFactory;
+import redis.embedded.RedisServer;
+
+import java.time.Duration;
+import java.util.List;
+
+import static org.junit.Assume.assumeFalse;
+
+public class AbstractRedisSingletonTest {
+
+    private static RedisServer redisServer;
+
+    private FaultTolerantRedisClient redisClient;
+    private ReplicatedJedisPool replicatedJedisPool;
+
+    @BeforeClass
+    public static void setUpBeforeClass() throws Exception {
+        assumeFalse(System.getProperty("os.name").equalsIgnoreCase("windows"));
+
+        redisServer = RedisServer.builder()
+                .setting("appendonly no")
+                .setting("dir " + System.getProperty("java.io.tmpdir"))
+                .port(AbstractRedisClusterTest.getNextRedisClusterPort())
+                .build();
+
+        redisServer.start();
+    }
+
+    @Before
+    public void setUp() throws Exception {
+        final String redisUrl = String.format("redis://127.0.0.1:%d", redisServer.ports().get(0));
+
+        redisClient = new FaultTolerantRedisClient("test-client",
+                RedisClient.create(redisUrl),
+                Duration.ofSeconds(2),
+                new CircuitBreakerConfiguration());
+
+        replicatedJedisPool = new RedisClientFactory("test-pool",
+                redisUrl,
+                List.of(redisUrl),
+                new CircuitBreakerConfiguration()).getRedisClientPool();
+
+        redisClient.useClient(connection -> connection.sync().flushall());
+    }
+
+    protected FaultTolerantRedisClient getRedisClient() {
+        return redisClient;
+    }
+
+    protected ReplicatedJedisPool getJedisPool() {
+        return replicatedJedisPool;
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        redisClient.shutdown();
+    }
+
+    @AfterClass
+    public static void tearDownAfterClass() {
+        redisServer.stop();
+    }
+}