From 279e2bdf4f7f10aaa6c9e4a4fc05b00c74ed0525 Mon Sep 17 00:00:00 2001
From: Philip Durbin <philip_durbin@harvard.edu>
Date: Mon, 6 Feb 2017 13:20:30 -0500
Subject: [PATCH] improve code coverage a bit #3584

---
 .../edu/harvard/iq/dataverse/util/FileUtilTest.java  | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/test/java/edu/harvard/iq/dataverse/util/FileUtilTest.java b/src/test/java/edu/harvard/iq/dataverse/util/FileUtilTest.java
index 50051cc4999..d980c47d5a8 100644
--- 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"));
+    }
 }