azkaban-aplcache

Details

diff --git a/az-core/src/test/java/azkaban/utils/UtilsTest.java b/az-core/src/test/java/azkaban/utils/UtilsTest.java
index 7889b7c..d4e084a 100644
--- a/az-core/src/test/java/azkaban/utils/UtilsTest.java
+++ b/az-core/src/test/java/azkaban/utils/UtilsTest.java
@@ -40,15 +40,21 @@ public class UtilsTest {
   @Test
   public void testUnzipInsecureFile() throws IOException {
     final File zipFile = new File("myTest.zip");
-    try (final ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipFile))) {
-      final ZipEntry entry = new ZipEntry("../../../../../evil.txt");
-      out.putNextEntry(entry);
-    }
+    try {
+      try (final ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipFile))) {
+        final ZipEntry entry = new ZipEntry("../../../../../evil.txt");
+        out.putNextEntry(entry);
+      }
 
-    final ZipFile source = new ZipFile(zipFile);
-    final File dest = Utils.createTempDir();
-    assertThatThrownBy(() -> Utils.unzip(source, dest)).isInstanceOf(IOException.class)
-        .hasMessageContaining("Extracting zip entry would have resulted in a file outside the "
-            + "specified destination directory.");
+      final ZipFile source = new ZipFile(zipFile);
+      final File dest = Utils.createTempDir();
+      assertThatThrownBy(() -> Utils.unzip(source, dest)).isInstanceOf(IOException.class)
+          .hasMessageContaining("Extracting zip entry would have resulted in a file outside the "
+              + "specified destination directory.");
+    } finally {
+      if (zipFile.exists()) {
+        zipFile.delete();
+      }
+    }
   }
 }