28
28
import org .junit .jupiter .api .BeforeEach ;
29
29
import org .junit .jupiter .params .ParameterizedTest ;
30
30
import org .springframework .util .Assert ;
31
+ import org .springframework .util .unit .DataSize ;
31
32
import org .testcontainers .junit .jupiter .Container ;
32
33
import org .testcontainers .shaded .com .google .common .collect .ImmutableList ;
33
34
import org .testcontainers .shaded .com .google .common .collect .ImmutableMap ;
36
37
import com .fasterxml .jackson .core .JsonProcessingException ;
37
38
import com .fasterxml .jackson .databind .ObjectMapper ;
38
39
import com .github .jcustenborder .kafka .connect .utils .SinkRecordHelper ;
40
+ import com .redis .enterprise .Database ;
41
+ import com .redis .enterprise .RedisModule ;
39
42
import com .redis .kafka .connect .sink .RedisEnterpriseSinkConfig ;
40
43
import com .redis .kafka .connect .sink .RedisEnterpriseSinkTask ;
41
44
import com .redis .lettucemod .timeseries .Sample ;
42
45
import com .redis .lettucemod .timeseries .TimeRange ;
46
+ import com .redis .testcontainers .RedisEnterpriseContainer ;
43
47
import com .redis .testcontainers .RedisModulesContainer ;
44
48
import com .redis .testcontainers .RedisServer ;
45
- import com .redis .testcontainers .junit .jupiter . AbstractTestcontainersRedisTestBase ;
46
- import com .redis .testcontainers .junit .jupiter . RedisTestContext ;
47
- import com .redis .testcontainers .junit .jupiter . RedisTestContextsSource ;
49
+ import com .redis .testcontainers .junit .AbstractTestcontainersRedisTestBase ;
50
+ import com .redis .testcontainers .junit .RedisTestContext ;
51
+ import com .redis .testcontainers .junit .RedisTestContextsSource ;
48
52
49
53
import io .lettuce .core .KeyValue ;
50
54
import io .lettuce .core .Range ;
@@ -57,9 +61,15 @@ class RedisEnterpriseSinkTaskIT extends AbstractTestcontainersRedisTestBase {
57
61
private static final RedisModulesContainer REDIS = new RedisModulesContainer (
58
62
RedisModulesContainer .DEFAULT_IMAGE_NAME .withTag (RedisModulesContainer .DEFAULT_TAG ));
59
63
64
+ @ Container
65
+ private static final RedisEnterpriseContainer REDIS_ENTERPRISE = new RedisEnterpriseContainer (
66
+ RedisEnterpriseContainer .DEFAULT_IMAGE_NAME .withTag (RedisEnterpriseContainer .DEFAULT_TAG ))
67
+ .withDatabase (Database .name ("RedisEnterpriseKafkaTests" ).memory (DataSize .ofMegabytes (100 )).ossCluster (true )
68
+ .modules (RedisModule .SEARCH , RedisModule .JSON , RedisModule .TIMESERIES ).build ());
69
+
60
70
@ Override
61
- protected Collection <RedisServer > servers () {
62
- return Arrays .asList (REDIS );
71
+ protected Collection <RedisServer > redisServers () {
72
+ return Arrays .asList (REDIS , REDIS_ENTERPRISE );
63
73
}
64
74
65
75
protected Map <String , String > map (String ... args ) {
0 commit comments