Add functionality for retrieving logs

1 files changed, 36 insertions, 0 deletions

diff --git a/container-core/src/test/java/com/yahoo/container/handler/LogReaderTest.java b/container-core/src/test/java/com/yahoo/container/handler/LogReaderTest.java
new file mode 100644
index 00000000000..ff6ea74a411
--- /dev/null
+++ b/container-core/src/test/java/com/yahoo/container/handler/LogReaderTest.java
@@ -0,0 +1,36 @@
+package com.yahoo.container.handler;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.ByteArrayOutputStream;
+
+import static org.junit.Assert.*;
+
+public class LogReaderTest {
+
+    ByteArrayOutputStream outputStream;
+
+    @Before
+    public void setup() {
+        outputStream = new ByteArrayOutputStream();
+    }
+
+    @Test
+    public void testThatFilesAreWrittenCorrectlyToOutputStream() throws Exception{
+        String logDirectory = "src/test/resources/logfolder/";
+        LogReader.writeToOutputStream(logDirectory, outputStream);
+        String expected = "{\"subfolder\":{\"log2.log\":\"VGhpcyBpcyBhbm90aGVyIGxvZyBmaWxl\"},\"log1.log\":\"VGhpcyBpcyBvbmUgbG9nIGZpbGU=\"}";
+        String actual = new String(outputStream.toByteArray());
+        assertEquals(expected, actual);
+    }
+
+    @Test
+    public void testNothingISWrittenToOutputStreamWithEmptyLogFolder() throws Exception {
+        String logDirectory = "src/test/resources/emptylogfolder/";
+        LogReader.writeToOutputStream(logDirectory, outputStream);
+        String expected = "{}";
+        String actual = new String(outputStream.toByteArray());
+        assertEquals(expected, actual);
+    }
+}
\ No newline at end of file