From cb1178c82917945653e38005f7a39cc0747e6baf Mon Sep 17 00:00:00 2001
From: Nikita <abracham.mitchell@gmail.com>
Date: Wed, 14 Jun 2017 17:12:31 +0300
Subject: [PATCH] Tests added for tasks batch feature

---
 .../executor/RedissonExecutorServiceTest.java | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/redisson/src/test/java/org/redisson/executor/RedissonExecutorServiceTest.java b/redisson/src/test/java/org/redisson/executor/RedissonExecutorServiceTest.java
index 3f6cfddc9..2d2340fe0 100644
--- a/redisson/src/test/java/org/redisson/executor/RedissonExecutorServiceTest.java
+++ b/redisson/src/test/java/org/redisson/executor/RedissonExecutorServiceTest.java
@@ -22,11 +22,15 @@ import org.junit.Test;
 import org.redisson.BaseTest;
 import org.redisson.RedissonNode;
 import org.redisson.RedissonRuntimeEnvironment;
+import org.redisson.api.RExecutorBatchFuture;
 import org.redisson.api.RExecutorFuture;
 import org.redisson.api.RExecutorService;
 import org.redisson.config.Config;
 import org.redisson.config.RedissonNodeConfig;
 
+import com.jayway.awaitility.Awaitility;
+import com.jayway.awaitility.Duration;
+
 public class RedissonExecutorServiceTest extends BaseTest {
 
     private static RedissonNode node;
@@ -83,6 +87,42 @@ public class RedissonExecutorServiceTest extends BaseTest {
         }
         assertThat(canceled).isTrue();
     }
+
+    @Test
+    public void testBatchSubmitRunnable() throws InterruptedException, ExecutionException, TimeoutException {
+        RExecutorService e = redisson.getExecutorService("test");
+        RExecutorBatchFuture future = e.submit(new IncrementRunnableTask("myCounter"), new IncrementRunnableTask("myCounter"), 
+                    new IncrementRunnableTask("myCounter"), new IncrementRunnableTask("myCounter"));
+        
+        future.get(5, TimeUnit.SECONDS);
+        future.getTaskFutures().stream().forEach(x -> x.syncUninterruptibly());
+    }
+    
+    @Test
+    public void testBatchSubmitCallable() throws InterruptedException, ExecutionException, TimeoutException {
+        RExecutorService e = redisson.getExecutorService("test");
+        RExecutorBatchFuture future = e.submit(new IncrementCallableTask("myCounter"), new IncrementCallableTask("myCounter"), 
+                    new IncrementCallableTask("myCounter"), new IncrementCallableTask("myCounter"));
+        
+        future.get(5, TimeUnit.SECONDS);
+        future.getTaskFutures().stream().forEach(x -> assertThat(x.getNow()).isEqualTo("1234"));
+    }
+
+    
+    @Test(expected = NullPointerException.class)
+    public void testBatchExecuteNPE() {
+        RExecutorService e = redisson.getExecutorService("test");
+        e.execute();
+    }
+    
+    @Test
+    public void testBatchExecute() {
+        RExecutorService e = redisson.getExecutorService("test");
+        e.execute(new IncrementRunnableTask("myCounter"), new IncrementRunnableTask("myCounter"), 
+                    new IncrementRunnableTask("myCounter"), new IncrementRunnableTask("myCounter"));
+        
+        Awaitility.await().atMost(Duration.FIVE_SECONDS).until(() -> redisson.getAtomicLong("myCounter").get() == 4);
+    }
     
     @Test
     public void testCancelAndInterrupt() throws InterruptedException, ExecutionException {