From 6fa6c63c85639cf1f72e65c5127b86a1c4880f60 Mon Sep 17 00:00:00 2001
From: 0xb00bface <0xboobface@gmail.com>
Date: Sat, 26 Sep 2020 16:07:11 +0200
Subject: [PATCH] Add test for DeleteTooShort pp

---
 .../postprocessing/DeleteTooShortTest.java    | 98 +++++++++++++++++++
 1 file changed, 98 insertions(+)
 create mode 100644 common/src/test/java/ctbrec/recorder/postprocessing/DeleteTooShortTest.java

diff --git a/common/src/test/java/ctbrec/recorder/postprocessing/DeleteTooShortTest.java b/common/src/test/java/ctbrec/recorder/postprocessing/DeleteTooShortTest.java
new file mode 100644
index 00000000..6c3a5e08
--- /dev/null
+++ b/common/src/test/java/ctbrec/recorder/postprocessing/DeleteTooShortTest.java
@@ -0,0 +1,98 @@
+package ctbrec.recorder.postprocessing;
+
+import static org.junit.Assert.*;
+import static org.mockito.Mockito.*;
+
+import java.io.File;
+import java.io.IOException;
+import java.time.Duration;
+import java.util.Collections;
+
+import org.junit.Test;
+
+import ctbrec.Config;
+import ctbrec.Recording;
+import ctbrec.recorder.RecordingManager;
+import ctbrec.recorder.download.Download;
+
+public class DeleteTooShortTest extends AbstractPpTest {
+
+    @Test
+    public void tooShortSingleFileRecShouldBeDeleted() throws IOException, InterruptedException {
+        testProcess(original);
+    }
+
+    @Test
+    public void tooShortDirectoryRecShouldBeDeleted() throws IOException, InterruptedException {
+        testProcess(originalDir);
+    }
+
+    private void testProcess(File original) throws IOException {
+        Recording rec = createRec(original);
+        Config config = mockConfig();
+        RecordingManager recordingManager = new RecordingManager(config, Collections.emptyList());
+        recordingManager.add(rec);
+
+        assertEquals(1, recordingManager.getAll().size());
+
+        DeleteTooShort pp = new DeleteTooShort();
+        pp.getConfig().put(DeleteTooShort.MIN_LEN_IN_SECS, "10");
+        pp.postprocess(rec, recordingManager, config);
+
+        assertFalse(rec.getAbsoluteFile().exists());
+        assertFalse(original.exists());
+        assertEquals(0, recordingManager.getAll().size());
+    }
+
+    @Test
+    public void testGetName() {
+        assertEquals("delete too short", new DeleteTooShort().getName());
+    }
+
+    @Test
+    public void testDisabledWithSingleFile() throws IOException, InterruptedException {
+        Recording rec = createRec(original);
+        Config config = mockConfig();
+        RecordingManager recordingManager = new RecordingManager(config, Collections.emptyList());
+        recordingManager.add(rec);
+        assertEquals(1, recordingManager.getAll().size());
+
+        DeleteTooShort pp = new DeleteTooShort();
+        pp.getConfig().put(DeleteTooShort.MIN_LEN_IN_SECS, "0");
+        pp.postprocess(rec, recordingManager, config);
+
+        assertTrue(rec.getAbsoluteFile().exists());
+        assertTrue(original.exists());
+        assertEquals(1, recordingManager.getAll().size());
+    }
+
+    @Test
+    public void longEnoughVideoShouldStay() throws IOException, InterruptedException {
+        Recording rec = createRec(original);
+        Config config = mockConfig();
+        RecordingManager recordingManager = new RecordingManager(config, Collections.emptyList());
+        recordingManager.add(rec);
+        assertEquals(1, recordingManager.getAll().size());
+
+        DeleteTooShort pp = new DeleteTooShort();
+        pp.getConfig().put(DeleteTooShort.MIN_LEN_IN_SECS, "1");
+        pp.postprocess(rec, recordingManager, config);
+
+        assertTrue(rec.getAbsoluteFile().exists());
+        assertTrue(original.exists());
+        assertEquals(1, recordingManager.getAll().size());
+    }
+
+    private Recording createRec(File original) {
+        Download download = mock(Download.class);
+        when(download.getLength()).thenReturn(Duration.ofSeconds(5));
+        Recording rec = new Recording();
+        rec.setModel(mockModel());
+        rec.setAbsoluteFile(original);
+        rec.setPostProcessedFile(original);
+        rec.setStartDate(now);
+        rec.setSingleFile(true);
+        rec.setDownload(download);
+        return rec;
+    }
+}