improve code coverage a bit #3584

IQSS · Feb 6, 2017 · 279e2bd · 279e2bd
1 parent c65a0df
commit 279e2bd
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/test/java/edu/harvard/iq/dataverse/util/FileUtilTest.java b/src/test/java/edu/harvard/iq/dataverse/util/FileUtilTest.java
@@ -158,4 +158,16 @@ public void testGetPublicDownloadUrl() {
         assertEquals("https://demo.dataverse.org/api/access/datafile/42", FileUtil.getPublicDownloadUrl("https://demo.dataverse.org", 42l));
     }
 
+    @Test
+    public void testGenerateOriginalExtension() {
+        assertEquals("", FileUtil.generateOriginalExtension("foo"));
+        // uh-oh, NullPointerException
+//        assertEquals("", FileUtil.generateOriginalExtension(null));
+        assertEquals(".sav", FileUtil.generateOriginalExtension("application/x-spss-sav"));
+        assertEquals(".por", FileUtil.generateOriginalExtension("application/x-spss-por"));
+        assertEquals(".dta", FileUtil.generateOriginalExtension("application/x-stata"));
+        assertEquals(".RData", FileUtil.generateOriginalExtension("application/x-rlang-transport"));
+        assertEquals(".csv", FileUtil.generateOriginalExtension("text/csv"));
+        assertEquals(".xlsx", FileUtil.generateOriginalExtension("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"));
+    }
 }