included gears module file in project. Fixes #1

Author: jruaux

src/main/java/com/redislabs/testcontainers/RedisEnterpriseContainer.java

@@ -116,6 +116,11 @@ public RedisEnterpriseContainer withDatabaseName(String name) {
         return this;
     }
 
+    public RedisEnterpriseContainer withOSSCluster() {
+        this.ossCluster = true;
+        return this;
+    }
+
     @SneakyThrows
     @Override
     protected void containerIsStarted(InspectContainerResponse containerInfo) {
@@ -161,11 +166,6 @@ private void createCluster(String username, String password, String externalAddr
         }
     }
 
-    public RedisEnterpriseContainer withOSSCluster() {
-        this.ossCluster = true;
-        return this;
-    }
-
     @Override
     public String getRedisURI() {
         return RedisServer.redisURI(getHost(), ENDPOINT_PORT);

src/main/java/com/redislabs/testcontainers/support/enterprise/DatabaseProvisioner.java

@@ -18,7 +18,6 @@
 
 import java.io.ByteArrayOutputStream;
 import java.io.InputStream;
-import java.net.URL;
 import java.time.Duration;
 import java.util.HashMap;
 import java.util.Map;
@@ -27,7 +26,6 @@
 public class DatabaseProvisioner {
 
     public static final String GEARS_MODULE_FILE = "redisgears.linux-bionic-x64.1.0.6.zip";
-    public static final String URL_GEARS_MODULE = "https://redismodules.s3.amazonaws.com/redisgears/" + GEARS_MODULE_FILE;
 
     private final RestAPI restAPI;
     private final Options options;
@@ -94,9 +92,7 @@ private Map<String, String> availableModules() throws Exception {
     }
 
     private void installGears() throws Exception {
-        URL url = new URL(URL_GEARS_MODULE);
-        log.info("Downloading module file {}", url);
-        try (InputStream zipInputStream = url.openStream()) {
+        try (InputStream zipInputStream = getClass().getClassLoader().getResourceAsStream(GEARS_MODULE_FILE)) {
             ByteArrayOutputStream baos = new ByteArrayOutputStream();
             IOUtils.copy(zipInputStream, baos);
             log.info("Installing module {}", GEARS_MODULE_FILE);

src/main/resources/redisgears.linux-bionic-x64.1.0.6.zip

@@ -1,10 +1,10 @@
 package com.redislabs.testcontainers;
 
-import com.redislabs.mesclun.api.sync.RedisGearsCommands;
+import com.redislabs.mesclun.RedisModulesClient;
+import com.redislabs.mesclun.api.StatefulRedisModulesConnection;
 import com.redislabs.mesclun.cluster.RedisModulesClusterClient;
 import com.redislabs.mesclun.cluster.api.StatefulRedisModulesClusterConnection;
 import com.redislabs.mesclun.gears.RedisGearsUtils;
-import com.redislabs.mesclun.gears.output.ExecutionResults;
 import com.redislabs.mesclun.search.Field;
 import com.redislabs.mesclun.search.SearchResults;
 import com.redislabs.testcontainers.support.enterprise.rest.Database;
@@ -85,32 +85,25 @@ void searchCluster() {
             container.stop();
         }
     }
-//
-//    @Test
-//    void gearsCluster() {
-//        RedisEnterpriseContainer container = new RedisEnterpriseContainer();
-//        container.withShardCount(3);
-//        container.withOSSCluster();
-//        container.withModules(Database.Module.GEARS);
-//        try {
-//            container.start();
-//            RedisModulesClusterClient client = RedisModulesClusterClient.create(container.getRedisURI());
-//            try (StatefulRedisModulesClusterConnection<String, String> connection = client.connect()) {
-//                connection.sync().set("foo", "bar");
-//                ExecutionResults results = pyExecute(connection.sync(), "sleep.py");
-//                Assertions.assertEquals("1", results.getResults().get(0));
-//            }
-//        } finally {
-//            container.stop();
-//        }
-//    }
-//
-//    private ExecutionResults pyExecute(RedisGearsCommands<String, String> sync, String resourceName) {
-//        return sync.pyExecute(load(resourceName));
-//    }
-//
-//    private String load(String resourceName) {
-//        return RedisGearsUtils.toString(getClass().getClassLoader().getResourceAsStream(resourceName));
-//    }
+
+    @Test
+    void gears() {
+        RedisEnterpriseContainer container = new RedisEnterpriseContainer();
+        container.withModules(Database.Module.GEARS);
+        try {
+            container.start();
+            RedisModulesClient client = RedisModulesClient.create(container.getRedisURI());
+            try (StatefulRedisModulesConnection<String, String> connection = client.connect()) {
+                connection.sync().set("foo", "bar");
+                connection.sync().pyExecute(load("sleep.py"));
+            }
+        } finally {
+            container.stop();
+        }
+    }
+
+    private String load(String resourceName) {
+        return RedisGearsUtils.toString(getClass().getClassLoader().getResourceAsStream(resourceName));
+    }
 
 }