Fix failing KMS test by fixing test config

Signed-off-by: Jeffrey Aaron Jeyasingh <jeffreyaaron06@gmail.com>

Author: jeffreyAaron

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

@@ -108,25 +108,6 @@ void setUp() {
 
         topicName = "buffer-" + RandomStringUtils.randomAlphabetic(5);
 
-        final Map<String, Object> topicConfigMap = Map.of(
-                "name", topicName,
-                "group_id", "buffergroup-" + RandomStringUtils.randomAlphabetic(6),
-                "create_topic", true
-        );
-
-        topicConfig = objectMapper.convertValue(topicConfigMap, BufferTopicConfig.class);
-
-        bootstrapServersCommaDelimited = System.getProperty("tests.kafka.bootstrap_servers");
-
-        LOG.info("Using Kafka bootstrap servers: {}", bootstrapServersCommaDelimited);
-
-        final Map<String, Object> bufferConfigMap = Map.of(
-                "topics", List.of(topicConfigMap),
-                "bootstrap_servers", List.of(bootstrapServersCommaDelimited),
-                "encryption", Map.of("type", "none")
-        );
-        kafkaBufferConfig = objectMapper.convertValue(bufferConfigMap, KafkaBufferConfig.class);
-
         kmsKey = System.getProperty("tests.kafka.kms_key");
         kmsClient = KmsClient.create();
 
@@ -164,14 +145,25 @@ void setUp() throws NoSuchAlgorithmException, InvalidKeyException, NoSuchPadding
             encryptCipher = Cipher.getInstance("AES");
             encryptCipher.init(Cipher.ENCRYPT_MODE, secretKey);
 
+            final Map<String, Object> topicConfigMap = Map.of(
+                    "name", topicName,
+                    "group_id", "buffergroup-" + RandomStringUtils.randomAlphabetic(6),
+                    "create_topic", true,
+                    "encryption_key", aesKey,
+                    "kms", Map.of("key_id", kmsKey)
+            );
+
+            topicConfig = objectMapper.convertValue(topicConfigMap, BufferTopicConfig.class);
+
+            bootstrapServersCommaDelimited = System.getProperty("tests.kafka.bootstrap_servers");
+
+            LOG.info("Using Kafka bootstrap servers: {}", bootstrapServersCommaDelimited);
 
-            final Map<String, Object> topicConfigMap = objectMapper.convertValue(topicConfig, Map.class);
-            topicConfigMap.put("encryption_key", aesKey);
-            topicConfigMap.put("kms", Map.of(
-                    "key_id", kmsKey
-            ));
-            final Map<String, Object> bufferConfigMap = objectMapper.convertValue(kafkaBufferConfig, Map.class);
-            bufferConfigMap.put("topics", List.of(topicConfigMap));
+            final Map<String, Object> bufferConfigMap = Map.of(
+                    "topics", List.of(topicConfigMap),
+                    "bootstrap_servers", List.of(bootstrapServersCommaDelimited),
+                    "encryption", Map.of("type", "none")
+            );
             kafkaBufferConfig = objectMapper.convertValue(bufferConfigMap, KafkaBufferConfig.class);
         }