Add cleanUp

azhwani · azhwani · commit d7d834f625c3 · 2020-09-25T18:40:26.000+01:00
diff --git a/core-java-modules/core-java-io-3/src/test/java/com/baeldung/lastmodifiedfile/LastModifiedFileAppUnitTest.java b/core-java-modules/core-java-io-3/src/test/java/com/baeldung/lastmodifiedfile/LastModifiedFileAppUnitTest.java
@@ -9,6 +9,7 @@
 import java.nio.file.Paths;
 
 import org.apache.commons.io.FileUtils;
+import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.junit.jupiter.api.Test;
 
@@ -57,8 +58,7 @@ public void givenDirectory_whenUsingNioApi_thenFindLastModfile() throws IOExcept
         Path lastModPath = LastModifiedFileApp.findUsingNIOApi(SOURCEDIRECTORY);
 
         assertThat(lastModPath).isNotNull();
-        assertThat(lastModPath.toFile()
-          .getName()).isEqualTo("file02.txt");
+        assertThat(lastModPath.toFile().getName()).isEqualTo("file02.txt");
     }
 
     @Test
@@ -69,4 +69,10 @@ public void givenDirectory_whenUsingApacheCommons_thenFindLastModfile() throws I
         assertThat(lastModFile.getName()).isEqualTo("file02.txt");
     }
 
+    @AfterAll
+    public static void cleanUp() throws IOException {
+        File srcDir = new File(SOURCEDIRECTORY);
+        FileUtils.deleteDirectory(srcDir);
+    }
+
 }
