Fix tests after rebase

Signed-off-by: Hai Yan <[email protected]>
opensearch-project · Nov 12, 2024 · 9aea7db · 9aea7db
1 parent cf1a566
commit 9aea7db
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/...test/java/org/opensearch/dataprepper/plugins/source/rds/export/DataFileSchedulerTest.java b/...test/java/org/opensearch/dataprepper/plugins/source/rds/export/DataFileSchedulerTest.java
@@ -213,7 +213,11 @@ private void mockDbTableMetadata() {
         final String hostName = UUID.randomUUID().toString();
         final int port = new Random().nextInt();
 
-        final DbMetadata dbMetadata = new DbMetadata(dbIdentifier, hostName, port);
+        final DbMetadata dbMetadata = DbMetadata.builder()
+                .dbIdentifier(dbIdentifier)
+                .endpoint(hostName)
+                .port(port)
+                .build();
         final Map<String, Object> map = new HashMap<>();
         map.put("dbMetadata", dbMetadata.toMap());
         map.put("tableColumnDataTypeMap", tableColumnDataTypeMap);

diff --git a/...a/org/opensearch/dataprepper/plugins/source/rds/stream/StreamWorkerTaskRefresherTest.java b/...a/org/opensearch/dataprepper/plugins/source/rds/stream/StreamWorkerTaskRefresherTest.java
@@ -240,7 +240,11 @@ private Map<String, Object> getDbTableMetaDataMap() {
         final int port = new Random().nextInt();
         final String tableName = UUID.randomUUID().toString();
 
-        final DbMetadata dbMetadata = new DbMetadata(dbIdentifier, hostName, port);
+        final DbMetadata dbMetadata = DbMetadata.builder()
+                .dbIdentifier(dbIdentifier)
+                .endpoint(hostName)
+                .port(port)
+                .build();
         final Map<String, Map<String, String>> tableColumnDataTypeMap = new HashMap<>();
         final Map<String, String> columnDataTypeMap = new HashMap<>();
         columnDataTypeMap.put("int_column", "INTEGER");