diff options
author | Harald Musum <musum@yahooinc.com> | 2023-02-03 14:36:12 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2023-02-03 14:36:12 +0100 |
commit | 83bd7804654aff81692e322325592002a68c2949 (patch) | |
tree | 2027fbb3111c13f1ac0ebe316ea6e685ae2dfc5d /container-search/src/test/java/com/yahoo | |
parent | facd90e5c4a4f17273f8b368b78baf0a1d5a294c (diff) |
Add flag deciding if we should keep success files
flag is false by default
Diffstat (limited to 'container-search/src/test/java/com/yahoo')
-rw-r--r-- | container-search/src/test/java/com/yahoo/search/logging/SpoolerTest.java | 19 |
1 files changed, 16 insertions, 3 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/logging/SpoolerTest.java b/container-search/src/test/java/com/yahoo/search/logging/SpoolerTest.java index bfcf3b5e3e5..eb5d7f85481 100644 --- a/container-search/src/test/java/com/yahoo/search/logging/SpoolerTest.java +++ b/container-search/src/test/java/com/yahoo/search/logging/SpoolerTest.java @@ -28,7 +28,7 @@ public class SpoolerTest { Path spoolDir = tempDir.resolve("spool"); int maxEntriesPerFile = 1; - Spooler spooler = new Spooler(spoolDir, maxEntriesPerFile, clock); + Spooler spooler = new Spooler(spoolDir, maxEntriesPerFile, clock, true); TestLogger logger = new TestLogger(spooler); assertTrue(sendEntry(logger, "Yo entry")); @@ -60,7 +60,7 @@ public class SpoolerTest { Path spoolDir = tempDir.resolve("spool"); int maxEntriesPerFile = 2; - Spooler spooler = new Spooler(spoolDir, maxEntriesPerFile, clock); + Spooler spooler = new Spooler(spoolDir, maxEntriesPerFile, clock, true); TestLogger logger = new TestLogger(spooler); assertTrue(sendEntry(logger, "Yo entry")); @@ -111,7 +111,7 @@ public class SpoolerTest { @Test public void failingToTransportIsRetried() throws IOException { Path spoolDir = tempDir.resolve("spool"); - Spooler spooler = new Spooler(spoolDir, 1, clock); + Spooler spooler = new Spooler(spoolDir, 1, clock, true); FailingToTransportSecondEntryLogger logger = new FailingToTransportSecondEntryLogger(spooler); assertTrue(sendEntry(logger, "Yo entry")); @@ -126,6 +126,19 @@ public class SpoolerTest { assertEquals(2, spooler.listFilesInPath(spooler.successesPath()).size()); } + @Test + public void noSuccessFiles() throws IOException { + Path spoolDir = tempDir.resolve("spool"); + boolean keepSuccessFiles = false; + Spooler spooler = new Spooler(spoolDir, 1, clock, keepSuccessFiles); + FailingToTransportSecondEntryLogger logger = new FailingToTransportSecondEntryLogger(spooler); + + assertTrue(sendEntry(logger, "Yo entry")); + logger.manualRun(); // Success for first message + assertEquals(0, spooler.listFilesInPath(spooler.successesPath()).size()); + } + + private boolean sendEntry(Logger logger, String x) { return logger.newEntry() .blob(x.getBytes()) |