Skip to content

Commit b656122

Browse files
committed
Fixed Encrypted tests
Signed-off-by: Jeffrey Aaron Jeyasingh <jeffreyaaron06@gmail.com>
1 parent 135a574 commit b656122

1 file changed

Lines changed: 27 additions & 2 deletions

File tree

  • data-prepper-plugins/kafka-plugins/src/integrationTest/java/org/opensearch/dataprepper/plugins/kafka/buffer

data-prepper-plugins/kafka-plugins/src/integrationTest/java/org/opensearch/dataprepper/plugins/kafka/buffer/KafkaBufferIT.java

Lines changed: 27 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,31 @@ class Encrypted {
448448

449449
@BeforeEach
450450
void setUp() throws NoSuchAlgorithmException, InvalidKeyException, NoSuchPaddingException {
451+
random = new Random();
452+
acknowledgementSetManager = mock(AcknowledgementSetManager.class);
453+
acknowledgementSet = mock(AcknowledgementSet.class);
454+
lenient().doAnswer((a) -> null).when(acknowledgementSet).complete();
455+
lenient().when(acknowledgementSetManager.create(any(), any(Duration.class))).thenReturn(acknowledgementSet);
456+
objectMapper = new ObjectMapper().registerModule(new JavaTimeModule());
457+
458+
when(pluginSetting.getPipelineName()).thenReturn(UUID.randomUUID().toString());
459+
460+
topicName = "buffer-" + RandomStringUtils.randomAlphabetic(5);
461+
462+
Map<String, Object> topicConfigMap = new java.util.HashMap<>(Map.of(
463+
"name", topicName,
464+
"group_id", "buffergroup-" + RandomStringUtils.randomAlphabetic(6),
465+
"create_topic", true
466+
));
467+
468+
topicConfig = objectMapper.convertValue(topicConfigMap, BufferTopicConfig.class);
469+
470+
final Map<String, Object> bufferConfigMap = new java.util.HashMap<>(Map.of(
471+
"topics", List.of(topicConfigMap),
472+
"bootstrap_servers", List.of(bootstrapServersCommaDelimited),
473+
"encryption", Map.of("type", "none")
474+
));
475+
451476
final KeyGenerator aesKeyGenerator = KeyGenerator.getInstance("AES");
452477
aesKeyGenerator.init(256);
453478
final SecretKey secretKey = aesKeyGenerator.generateKey();
@@ -459,11 +484,11 @@ void setUp() throws NoSuchAlgorithmException, InvalidKeyException, NoSuchPadding
459484
final byte[] base64Bytes = Base64.getEncoder().encode(secretKey.getEncoded());
460485
aesKey = new String(base64Bytes);
461486

462-
final Map<String, Object> topicConfigMap = objectMapper.convertValue(topicConfig, Map.class);
463487
topicConfigMap.put("encryption_key", aesKey);
464-
final Map<String, Object> bufferConfigMap = objectMapper.convertValue(kafkaBufferConfig, Map.class);
465488
bufferConfigMap.put("topics", List.of(topicConfigMap));
466489
kafkaBufferConfig = objectMapper.convertValue(bufferConfigMap, KafkaBufferConfig.class);
490+
491+
byteDecoder = null;
467492
}
468493

469494
@Test

0 commit comments

Comments
 (0)