refactor: Update projectCsvUtils to use consistent capitalization for…

… headers
bacpop · Sep 6, 2024 · 985258d · 985258d
1 parent 8d382ce
commit 985258d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/app/client-v2/src/__tests__/utils/projectCsvUtils.spec.ts b/app/client-v2/src/__tests__/utils/projectCsvUtils.spec.ts
@@ -72,7 +72,7 @@ describe("projectCsvUtils", () => {
       const result = generateCsvContent(data);
 
       expect(result).toBe(
-        "filename,Penicillin,cluster\nsample1,Penicillin-0.9,cluster1\nsample2,Penicillin-0.8,cluster2"
+        'filename,Penicillin,cluster\n"sample1","Penicillin-0.9","cluster1"\n"sample2","Penicillin-0.8","cluster2"'
       );
     });
 

diff --git a/app/client-v2/src/utils/projectCsvUtils.ts b/app/client-v2/src/utils/projectCsvUtils.ts
@@ -3,9 +3,9 @@ import { convertProbabilityToWord } from "./amrDisplayUtils";
 
 export const downloadCsv = (samples: ProjectSample[], filename: string) => {
   const csvData = samples.map((sample) => ({
-    filename: sample.filename,
+    Filename: sample.filename,
     ...(sample.amr && convertAmrForCsv(sample.amr)),
-    cluster: sample.cluster || "run for cluster assignment"
+    Cluster: sample.cluster || ""
   }));
 
   const csvContent = generateCsvContent(csvData);
@@ -24,7 +24,7 @@ export const generateCsvContent = (data: Record<string, string>[]) => {
   if (data.length === 0) return "";
 
   const headers = Object.keys(data[0]);
-  const rows = data.map((row) => headers.map((header) => row[header] || "").join(","));
+  const rows = data.map((row) => headers.map((header) => `"${row[header]}"`).join(","));
   return [headers.join(","), ...rows].join("\n");
 };