From 43de2c262037bd1b1a1f6e51891bc13c8bbb7a2d Mon Sep 17 00:00:00 2001
From: 0xb00bface <0xboobface@gmail.com>
Date: Fri, 25 Sep 2020 19:43:00 +0200
Subject: [PATCH] Add test for DeleteOriginal pp

---
 .../postprocessing/DeleteOriginalTest.java    | 28 ++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/common/src/test/java/ctbrec/recorder/postprocessing/DeleteOriginalTest.java b/common/src/test/java/ctbrec/recorder/postprocessing/DeleteOriginalTest.java
index 0a3e0b4f..3b6047ab 100644
--- a/common/src/test/java/ctbrec/recorder/postprocessing/DeleteOriginalTest.java
+++ b/common/src/test/java/ctbrec/recorder/postprocessing/DeleteOriginalTest.java
@@ -1,6 +1,9 @@
 package ctbrec.recorder.postprocessing;
 
+import static org.junit.Assert.*;
+
 import java.io.IOException;
+import java.nio.file.Files;
 
 import org.junit.Test;
 
@@ -14,12 +17,35 @@ public class DeleteOriginalTest extends AbstractPpTest {
         Recording rec = new Recording();
         rec.setModel(mockModel());
         rec.setAbsoluteFile(original);
-        rec.setAbsoluteFile(postProcessed);
+        rec.setPostProcessedFile(postProcessed);
         rec.setStartDate(now);
         rec.setSingleFile(true);
 
         Config config = mockConfig();
         DeleteOriginal pp = new DeleteOriginal();
         pp.postprocess(rec, null, config);
+
+        assertEquals(postProcessed, rec.getAbsoluteFile());
+        assertTrue(rec.getAbsoluteFile().exists());
+        assertFalse(original.exists());
+    }
+
+    @Test
+    public void testPostProcessWithDirectory() throws IOException, InterruptedException {
+        Recording rec = new Recording();
+        rec.setModel(mockModel());
+        rec.setAbsoluteFile(originalDir);
+        rec.setPostProcessedFile(postProcessedDir);
+        rec.setStartDate(now);
+        rec.setSingleFile(true);
+
+        Config config = mockConfig();
+        Files.createDirectories(postProcessedDir.toPath());
+        DeleteOriginal pp = new DeleteOriginal();
+        pp.postprocess(rec, null, config);
+
+        assertEquals(postProcessedDir, rec.getAbsoluteFile());
+        assertTrue(rec.getAbsoluteFile().exists());
+        assertFalse(originalDir.exists());
     }
 }