RedisClusterSqlTestSuite introduced

2 years ago · 67c2392ccc
2 changed files with 123 additions and 0 deletions
--- a/dao/src/test/java/org/thingsboard/server/dao/AbstractRedisClusterContainer.java
+++ b/dao/src/test/java/org/thingsboard/server/dao/AbstractRedisClusterContainer.java
@ -0,0 +1,92 @@
+/**
+ * Copyright © 2016-2024 The Thingsboard Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.thingsboard.server.dao;
+
+import lombok.extern.slf4j.Slf4j;
+import org.junit.ClassRule;
+import org.junit.rules.ExternalResource;
+import org.testcontainers.containers.GenericContainer;
+import org.testcontainers.containers.Network;
+import org.testcontainers.containers.output.OutputFrame;
+import redis.clients.jedis.Jedis;
+
+import java.util.List;
+import java.util.concurrent.TimeUnit;
+
+@Slf4j
+public class AbstractRedisClusterContainer {
+
+    static final String nodes = "127.0.0.1:6371,127.0.0.1:6372,127.0.0.1:6373,127.0.0.1:6374,127.0.0.1:6375,127.0.0.1:6376";
+
+    @ClassRule(order = 0)
+    public static Network network = Network.newNetwork();
+    @ClassRule(order = 1)
+    public static GenericContainer redis1 = new GenericContainer("bitnami/redis-cluster:latest").withEnv("REDIS_PORT_NUMBER","6371").withNetworkMode("host").withLogConsumer(x->log.warn("{}", ((OutputFrame)x).getUtf8StringWithoutLineEnding())).withEnv("ALLOW_EMPTY_PASSWORD","yes").withEnv("REDIS_NODES",nodes);
+    @ClassRule(order = 2)
+    public static GenericContainer redis2 = new GenericContainer("bitnami/redis-cluster:latest").withEnv("REDIS_PORT_NUMBER","6372").withNetworkMode("host").withLogConsumer(x->log.warn("{}", ((OutputFrame)x).getUtf8StringWithoutLineEnding())).withEnv("ALLOW_EMPTY_PASSWORD","yes").withEnv("REDIS_NODES",nodes);
+    @ClassRule(order = 3)
+    public static GenericContainer redis3 = new GenericContainer("bitnami/redis-cluster:latest").withEnv("REDIS_PORT_NUMBER","6373").withNetworkMode("host").withLogConsumer(x->log.warn("{}", ((OutputFrame)x).getUtf8StringWithoutLineEnding())).withEnv("ALLOW_EMPTY_PASSWORD","yes").withEnv("REDIS_NODES",nodes);
+    @ClassRule(order = 4)
+    public static GenericContainer redis4 = new GenericContainer("bitnami/redis-cluster:latest").withEnv("REDIS_PORT_NUMBER","6374").withNetworkMode("host").withLogConsumer(x->log.warn("{}", ((OutputFrame)x).getUtf8StringWithoutLineEnding())).withEnv("ALLOW_EMPTY_PASSWORD","yes").withEnv("REDIS_NODES",nodes);
+    @ClassRule(order = 5)
+    public static GenericContainer redis5 = new GenericContainer("bitnami/redis-cluster:latest").withEnv("REDIS_PORT_NUMBER","6375").withNetworkMode("host").withLogConsumer(x->log.warn("{}", ((OutputFrame)x).getUtf8StringWithoutLineEnding())).withEnv("ALLOW_EMPTY_PASSWORD","yes").withEnv("REDIS_NODES",nodes);
+    @ClassRule(order = 6)
+    public static GenericContainer redis6 = new GenericContainer("bitnami/redis-cluster:latest").withEnv("REDIS_PORT_NUMBER","6376").withNetworkMode("host").withLogConsumer(x->log.warn("{}", ((OutputFrame)x).getUtf8StringWithoutLineEnding())).withEnv("ALLOW_EMPTY_PASSWORD","yes").withEnv("REDIS_NODES",nodes);
+
+
+    @ClassRule(order = 100)
+    public static ExternalResource resource = new ExternalResource() {
+        @Override
+        protected void before() throws Throwable {
+            redis1.start();
+            redis2.start();
+            redis3.start();
+            redis4.start();
+            redis5.start();
+            redis6.start();
+
+            String clusterCreateCommand = "echo yes | redis-cli --cluster create " +
+                     "127.0.0.1:6371 127.0.0.1:6372 127.0.0.1:6373 127.0.0.1:6374 127.0.0.1:6375 127.0.0.1:6376 " +
+                    "--cluster-replicas 1";
+
+            log.warn("Command to init Redis Cluster: {}", clusterCreateCommand);
+            log.warn("Connect to nodes : {}", nodes);
+            Thread.sleep(TimeUnit.SECONDS.toMillis(5));
+            var result = redis6.execInContainer("/bin/sh", "-c", clusterCreateCommand);
+            log.warn("Init cluster result: {}", result);
+
+            System.setProperty("cache.type", "redis");
+            System.setProperty("redis.connection.type", "cluster");
+            System.setProperty("redis.cluster.nodes", nodes);
+            System.setProperty("redis.password", "");
+
+            System.setProperty("redis.cluster.useDefaultPoolConfig", "false");
+        }
+
+        @Override
+        protected void after() {
+            redis1.stop();
+            redis2.stop();
+            redis3.stop();
+            redis4.stop();
+            redis5.stop();
+            redis6.stop();
+            List.of("cache.type", "redis.connection.type", "redis.standalone.host", "redis.standalone.port")
+                    .forEach(System.getProperties()::remove);
+        }
+    };
+
+}
--- a/dao/src/test/java/org/thingsboard/server/dao/RedisClusterSqlTestSuite.java
+++ b/dao/src/test/java/org/thingsboard/server/dao/RedisClusterSqlTestSuite.java
@ -0,0 +1,31 @@
+/**
+ * Copyright © 2016-2024 The Thingsboard Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.thingsboard.server.dao;
+
+import org.junit.extensions.cpsuite.ClasspathSuite;
+import org.junit.extensions.cpsuite.ClasspathSuite.ClassnameFilters;
+import org.junit.runner.RunWith;
+
+@RunWith(ClasspathSuite.class)
+@ClassnameFilters(
+        //All the same tests using redis instead of caffeine.
+        {
+                "org.thingsboard.server.dao.service.*ServiceSqlTest",
+        }
+)
+public class RedisClusterSqlTestSuite extends AbstractRedisClusterContainer {
+
+}