feat: ExportFormat test

InseeFr · Nov 27, 2023 · 429bee7 · 429bee7
1 parent 8c7509c
commit 429bee7
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 1 deletion.
diff --git a/.../src/main/java/fr/insee/arc/ws/services/importServlet/ImportStep3GetTableDataService.java b/.../src/main/java/fr/insee/arc/ws/services/importServlet/ImportStep3GetTableDataService.java
@@ -37,7 +37,7 @@ public void execute(SendResponse resp) throws ArcException {
 
 		// binary transfer
 		ServiceDao.execQueryExportDataToResponse(resp.getWr(),
-				ViewEnum.normalizeTableName(arcClientIdentifier.getClient()), this.arcClientIdentifier.getFormat().equals(ExportFormat.CSV_GZIP.getFormat()));
+				ViewEnum.normalizeTableName(arcClientIdentifier.getClient()), ExportFormat.isCsv(this.arcClientIdentifier.getFormat()));
 
 		if (this.clientDao.isWebServiceNotPending()) {
 			this.clientDao.dropTable(arcClientIdentifier.getClient());

diff --git a/arc-ws/src/main/java/fr/insee/arc/ws/services/importServlet/bo/ExportFormat.java b/arc-ws/src/main/java/fr/insee/arc/ws/services/importServlet/bo/ExportFormat.java
@@ -14,5 +14,9 @@ private ExportFormat(String format)
 	public String getFormat() {
 		return format;
 	}
+
+	public static boolean isCsv(String clientFormat) {
+		return clientFormat.equals(CSV_GZIP.toString());
+	}
 
 }
diff --git a/arc-ws/src/test/java/fr/insee/arc/ws/services/importServlet/bo/ExportFormatTest.java b/arc-ws/src/test/java/fr/insee/arc/ws/services/importServlet/bo/ExportFormatTest.java
@@ -0,0 +1,15 @@
+package fr.insee.arc.ws.services.importServlet.bo;
+
+import static org.junit.Assert.*;
+
+import org.junit.Test;
+
+public class ExportFormatTest {
+
+	@Test
+	public void isCsv() {
+		assertFalse(ExportFormat.isCsv(ExportFormat.BINARY.toString()));
+		assertTrue(ExportFormat.isCsv(ExportFormat.CSV_GZIP.toString()));
+	}
+
+}