diff --git a/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/BatchTests.java b/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/BatchTests.java index f4904fb7..17a09bbb 100644 --- a/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/BatchTests.java +++ b/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/BatchTests.java @@ -81,7 +81,6 @@ import com.redis.spring.batch.writer.Sugadd; import com.redis.spring.batch.writer.TsAdd; import com.redis.spring.batch.writer.TsAddAll; -import com.redis.spring.batch.writer.Xadd; import io.lettuce.core.Consumer; import io.lettuce.core.GeoArgs; @@ -808,29 +807,6 @@ void writeWait(TestInfo info) throws Exception { assertEquals("Insufficient replication level (0/1)", exceptions.get(0).getCause().getCause().getMessage()); } - @Test - void writeStream(TestInfo info) throws Exception { - String stream = "stream:0"; - List> messages = new ArrayList<>(); - for (int index = 0; index < 100; index++) { - Map body = new HashMap<>(); - body.put("field1", "value1"); - body.put("field2", "value2"); - messages.add(body); - } - ListItemReader> reader = new ListItemReader<>(messages); - Xadd> xadd = new Xadd<>(keyFunction(stream), Function.identity()); - RedisItemWriter> writer = writer(xadd); - run(info, reader, writer); - Assertions.assertEquals(messages.size(), redisCommands.xlen(stream)); - List> xrange = redisCommands.xrange(stream, - io.lettuce.core.Range.create("-", "+")); - for (int index = 0; index < xrange.size(); index++) { - StreamMessage message = xrange.get(index); - Assertions.assertEquals(messages.get(index), message.getBody()); - } - } - private void replicateLive(TestInfo info, RedisItemReader reader, RedisItemWriter writer, RedisItemReader liveReader, RedisItemWriter liveWriter) throws Exception { diff --git a/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/StackToStackTests.java b/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/StackToStackTests.java index c0aaa717..1fcb9600 100644 --- a/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/StackToStackTests.java +++ b/subprojects/spring-batch-redis-test/src/test/java/com/redis/spring/batch/test/StackToStackTests.java @@ -513,6 +513,29 @@ void writeSet(TestInfo info) throws Exception { assertEquals(values.size(), redisCommands.scard(key)); } + @Test + void writeStream(TestInfo info) throws Exception { + String stream = "stream:0"; + List> messages = new ArrayList<>(); + for (int index = 0; index < 100; index++) { + Map body = new HashMap<>(); + body.put("field1", "value1"); + body.put("field2", "value2"); + messages.add(body); + } + ListItemReader> reader = new ListItemReader<>(messages); + Xadd> xadd = new Xadd<>(keyFunction(stream), Function.identity()); + RedisItemWriter> writer = writer(xadd); + run(info, reader, writer); + Assertions.assertEquals(messages.size(), redisCommands.xlen(stream)); + List> xrange = redisCommands.xrange(stream, + io.lettuce.core.Range.create("-", "+")); + for (int index = 0; index < xrange.size(); index++) { + StreamMessage message = xrange.get(index); + Assertions.assertEquals(messages.get(index), message.getBody()); + } + } + private MapOptions hashOptions(Range fieldCount) { MapOptions options = new MapOptions(); options.setFieldCount(fieldCount); @@ -550,4 +573,4 @@ void writeStructMerge(TestInfo info) throws Exception { assertEquals(10, actual.size()); } -} +} \ No newline at end of file