1 files changed, 213 insertions, 0 deletions
diff --git a/container-accesslogging/src/test/java/com/yahoo/container/logging/LogFileHandlerTestCase.java b/container-accesslogging/src/test/java/com/yahoo/container/logging/LogFileHandlerTestCase.java
new file mode 100644
index 00000000000..0330a76951d
--- /dev/null
+++ b/container-accesslogging/src/test/java/com/yahoo/container/logging/LogFileHandlerTestCase.java
@@ -0,0 +1,213 @@
+// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.container.logging;
+
+import com.yahoo.io.IOUtils;
+import org.junit.Test;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.logging.Formatter;
+import java.util.logging.Level;
+import java.util.logging.LogRecord;
+import java.util.logging.SimpleFormatter;
+import java.util.zip.GZIPInputStream;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+/**
+ * @author <a href="mailto:travisb@yahoo-inc.com">Bob Travis</a>
+ */
+// TODO: Make these tests wait until the right things happen rather than waiting for a predetermined time
+// These tests take too long, and are not cleaning up properly. See how this should be done in YApacheLogTestCase
+public class LogFileHandlerTestCase {
+
+    /**
+     * The scenario
+     */
+    @Test
+    public void testIt() {
+        LogFileHandler h = new LogFileHandler();
+        h.setFilePattern("./logfilehandlertest.%Y%m%d%H%M%S");
+        h.setFormatter(new Formatter() {
+                public String format(LogRecord r) {
+                    DateFormat df = new SimpleDateFormat("yyyy.MM.dd:HH:mm:ss.SSS");
+                    String timeStamp = df.format(new Date(r.getMillis()));
+                    return ("["+timeStamp+"]" + " " + formatMessage(r) + "\n");
+                }
+            } );
+        long now = System.currentTimeMillis();
+        long millisPerDay = 60*60*24*1000;
+        long tomorrowDays = (now / millisPerDay) +1;
+        long tomorrowMillis = tomorrowDays * millisPerDay;
+        assertEquals (tomorrowMillis, h.getNextRotationTime(now));
+        long[] rTimes = {1000, 2000, 10000};
+        h.setRotationTimes(rTimes);
+        assertEquals (tomorrowMillis+1000, h.getNextRotationTime(tomorrowMillis));
+        assertEquals (tomorrowMillis+10000, h.getNextRotationTime(tomorrowMillis+3000));
+        boolean okToWrite = false; // don't want regular unit tests to create tiles....
+        if (okToWrite) {
+            LogRecord lr = new LogRecord(Level.INFO, "test");
+            h.publish(lr);
+            h.publish(new LogRecord(Level.INFO, "another test"));
+            h.rotateNow();
+            h.publish(lr);
+            h.flush();
+        }
+    }
+
+    private boolean delete(String fileOrDir) {
+      File file = new File(fileOrDir);
+      return file.delete();
+    }
+
+    private void deleteOnExit(String fileOrDir) {
+        new File(fileOrDir).deleteOnExit();
+    }
+
+    private static void deleteRecursive(String directory) {
+       IOUtils.recursiveDeleteDir(new File(directory));
+    }
+
+    @Test
+    public void testSimpleLogging() {
+      String logFilePattern = "./testLogFileG1.txt";
+
+      //create logfilehandler
+      LogFileHandler h = new LogFileHandler();
+      h.setFilePattern(logFilePattern);
+      h.setFormatter(new SimpleFormatter());
+      h.setRotationTimes("0 5 ...");
+
+      //write log
+      LogRecord lr = new LogRecord(Level.INFO, "testDeleteFileFirst1");
+      h.publish(lr);
+      h.flush();
+
+      new File(logFilePattern).deleteOnExit();
+    }
+
+    @Test
+    public void testDeleteFileDuringLogging() {
+      String logFilePattern = "./testLogFileG2.txt";
+
+      //create logfilehandler
+      LogFileHandler h = new LogFileHandler();
+      h.setFilePattern(logFilePattern);
+      h.setFormatter(new SimpleFormatter());
+      h.setRotationTimes("0 5 ...");
+
+      //write log
+      LogRecord lr = new LogRecord(Level.INFO, "testDeleteFileDuringLogging1");
+      h.publish(lr);
+      h.flush();
+
+      //delete log file
+      delete(logFilePattern);
+
+      //write log again
+      lr = new LogRecord(Level.INFO, "testDeleteFileDuringLogging2");
+      h.publish(lr);
+      h.flush();
+
+      new File(logFilePattern).deleteOnExit();
+    }
+
+    @Test
+    public void testSymlink() {
+        LogFileHandler h = new LogFileHandler();
+        h.setFilePattern("./testlogforsymlinkchecking/logfilehandlertest.%Y%m%d%H%M%S%s");
+        h.setFormatter(new Formatter() {
+            public String format(LogRecord r) {
+                DateFormat df = new SimpleDateFormat("yyyy.MM.dd:HH:mm:ss.SSS");
+                String timeStamp = df.format(new Date(r.getMillis()));
+                return ("["+timeStamp+"]" + " " + formatMessage(r) + "\n");
+            }
+        } );
+        h.setSymlinkName("symlink");
+        LogRecord lr = new LogRecord(Level.INFO, "test");
+        h.publish(lr);
+        String f1 = h.getFileName();
+        String f2 = null;
+        try {
+            while (f1 == null) {
+                Thread.sleep(1);
+                f1 = h.getFileName();
+            }
+            h.rotateNow();
+            Thread.sleep(1);
+            f2 = h.getFileName();
+            while (f1.equals(f2)) {
+                Thread.sleep(1);
+                f2 = h.getFileName();
+            }
+            lr = new LogRecord(Level.INFO, "string which is way longer than the word test");
+            h.publish(lr);
+            Thread.sleep(1000);
+            File f = new File(f1);
+            long first = f.length();
+            f = new File(f2);
+            long second = f.length();
+            final long secondLength = 72;
+            for (int n = 0; n < 20 && second != secondLength; ++n) {
+                Thread.sleep(1000);
+                second = f.length();
+            }
+            f = new File("./testlogforsymlinkchecking", "symlink");
+            long link = f.length();
+            assertEquals(secondLength, link);
+            assertEquals(31, first);
+            assertEquals(secondLength, second);
+        } catch (InterruptedException e) {
+            // just let the test pass
+        }
+        deleteOnExit("./testlogforsymlinkchecking");
+        deleteOnExit("./testlogforsymlinkchecking/symlink");
+        deleteOnExit(f1);
+        if (f2 != null)
+            deleteOnExit(f2);
+    }
+
+    @Test
+    public void testcompression() throws InterruptedException, IOException {
+        IOUtils.recursiveDeleteDir(new File("./testcompression"));
+        LogFileHandler h = new LogFileHandler(true);
+        h.setFilePattern("./testcompression/logfilehandlertest.%Y%m%d%H%M%S%s");
+        h.setFormatter(new Formatter() {
+            public String format(LogRecord r) {
+                DateFormat df = new SimpleDateFormat("yyyy.MM.dd:HH:mm:ss.SSS");
+                String timeStamp = df.format(new Date(r.getMillis()));
+                return ("["+timeStamp+"]" + " " + formatMessage(r) + "\n");
+            }
+        } );
+        for (int i=0; i < 10000; i++) {
+            LogRecord lr = new LogRecord(Level.INFO, "test");
+            h.publish(lr);
+        }
+        String f1 = h.getFileName();
+        assertTrue(f1.startsWith("./testcompression/logfilehandlertest."));
+        File uncompressed = new File(f1);
+        File compressed = new File(f1 + ".gz");
+        h.waitDrained();
+        assertTrue(uncompressed.exists());
+        assertFalse(compressed.exists());
+        String content = IOUtils.readFile(uncompressed);
+        assertEquals(310000, content.length());
+        h.rotateNow();
+        while (uncompressed.exists()) {
+            Thread.sleep(10);
+        }
+        assertTrue(compressed.exists());
+        String unzipped = IOUtils.readAll(new InputStreamReader(new GZIPInputStream(new FileInputStream(compressed))));
+        assertEquals(content, unzipped);
+
+        IOUtils.recursiveDeleteDir(new File("./testcompression"));
+    }
+
+}