test: rw performance tests

aws · Oct 25, 2024 · 38897d2 · 38897d2
1 parent c332625
commit 38897d2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/integration/container/tests/read_write_splitting_performance.test.ts b/tests/integration/container/tests/read_write_splitting_performance.test.ts
@@ -78,7 +78,7 @@ describe("rwperformance", () => {
 
       // Internal connection pool results.
       ConnectionProviderManager.setConnectionProvider(new InternalPooledConnectionProvider());
-      const rwPluginWithPoolConfig = initReadWritePluginConfig(env.databaseInfo.clusterEndpoint, env.databaseInfo.clusterEndpointPort);
+      const rwPluginWithPoolConfig = initReadWritePluginConfig(env.databaseInfo.writerInstanceEndpoint, env.databaseInfo.clusterEndpointPort);
       const rwPluginWithPoolData = await measurePerformance(rwPluginWithPoolConfig);
 
       readerData = calculateReaderOverhead("Switch to reader", rwPluginWithPoolData, noPluginData);

diff --git a/tests/integration/container/tests/utils/perf_util.ts b/tests/integration/container/tests/utils/perf_util.ts
@@ -65,7 +65,9 @@ export class PerfTestUtility {
     }
 
     const worksheet = XLSX.utils.json_to_sheet(rows);
-    XLSX.utils.book_append_sheet(workbook, worksheet, worksheetName);
+    if (!workbook.Sheets[worksheetName]) {
+      XLSX.utils.book_append_sheet(workbook, worksheet, worksheetName);
+    }
     XLSX.utils.sheet_add_aoa(worksheet, data[0].writeHeader(), { origin: "A1" });
     XLSX.writeFile(workbook, filePath);
   }