2828import org .junit .jupiter .api .BeforeEach ;
2929import org .junit .jupiter .params .ParameterizedTest ;
3030import org .springframework .util .Assert ;
31+ import org .springframework .util .unit .DataSize ;
3132import org .testcontainers .junit .jupiter .Container ;
3233import org .testcontainers .shaded .com .google .common .collect .ImmutableList ;
3334import org .testcontainers .shaded .com .google .common .collect .ImmutableMap ;
3637import com .fasterxml .jackson .core .JsonProcessingException ;
3738import com .fasterxml .jackson .databind .ObjectMapper ;
3839import com .github .jcustenborder .kafka .connect .utils .SinkRecordHelper ;
40+ import com .redis .enterprise .Database ;
41+ import com .redis .enterprise .RedisModule ;
3942import com .redis .kafka .connect .sink .RedisEnterpriseSinkConfig ;
4043import com .redis .kafka .connect .sink .RedisEnterpriseSinkTask ;
4144import com .redis .lettucemod .timeseries .Sample ;
4245import com .redis .lettucemod .timeseries .TimeRange ;
46+ import com .redis .testcontainers .RedisEnterpriseContainer ;
4347import com .redis .testcontainers .RedisModulesContainer ;
4448import 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 ;
4852
4953import io .lettuce .core .KeyValue ;
5054import io .lettuce .core .Range ;
@@ -57,9 +61,15 @@ class RedisEnterpriseSinkTaskIT extends AbstractTestcontainersRedisTestBase {
5761 private static final RedisModulesContainer REDIS = new RedisModulesContainer (
5862 RedisModulesContainer .DEFAULT_IMAGE_NAME .withTag (RedisModulesContainer .DEFAULT_TAG ));
5963
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+
6070 @ Override
61- protected Collection <RedisServer > servers () {
62- return Arrays .asList (REDIS );
71+ protected Collection <RedisServer > redisServers () {
72+ return Arrays .asList (REDIS , REDIS_ENTERPRISE );
6373 }
6474
6575 protected Map <String , String > map (String ... args ) {
0 commit comments