1 files changed, 63 insertions, 0 deletions
diff --git a/container-accesslogging/src/test/java/com/yahoo/container/logging/AccessLogSamplerTest.java b/container-accesslogging/src/test/java/com/yahoo/container/logging/AccessLogSamplerTest.java
new file mode 100644
index 00000000000..35664049560
--- /dev/null
+++ b/container-accesslogging/src/test/java/com/yahoo/container/logging/AccessLogSamplerTest.java
@@ -0,0 +1,63 @@
+// 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 org.junit.Test;
+
+import java.net.URI;
+import java.util.ArrayList;
+import java.util.List;
+
+import static org.hamcrest.CoreMatchers.hasItem;
+import static org.hamcrest.CoreMatchers.hasItems;
+import static org.hamcrest.CoreMatchers.is;
+import static org.hamcrest.Matchers.contains;
+import static org.junit.Assert.assertThat;
+
+public class AccessLogSamplerTest {
+
+    private final List<String> uris = new ArrayList<>();
+    private CircularArrayAccessLogKeeper circularArrayAccessLogKeeperMock = new CircularArrayAccessLogKeeper() {
+        @Override
+        public void addUri(String uri) {
+            uris.add(uri);
+        }
+    };
+    private AccessLogSampler accessLogSampler = new AccessLogSampler(circularArrayAccessLogKeeperMock);
+
+    @Test
+    public void testAFewLines() {
+        accessLogSampler.log(createLogEntry(200, "/search/foo"));
+        accessLogSampler.log(createLogEntry(500, "/search/bar"));
+        accessLogSampler.log(createLogEntry(500, "bar"));
+        accessLogSampler.log(createLogEntry(200, "/search/what"));
+        assertThat(uris, contains("/search/foo", "/search/what"));
+    }
+
+    @Test
+    public void testSubSampling() {
+        for (int i = 0; i < CircularArrayAccessLogKeeper.SIZE; i++) {
+            accessLogSampler.log(createLogEntry(200, "/search/" + String.valueOf(i)));
+        }
+        assertThat(uris.size(), is(CircularArrayAccessLogKeeper.SIZE));
+        assertThat(uris, hasItems("/search/0", "/search/1", "/search/2",
+            "/search/" + String.valueOf(CircularArrayAccessLogKeeper.SIZE - 1)));
+        uris.clear();
+        for (int i = 0; i < CircularArrayAccessLogKeeper.SIZE; i++) {
+            accessLogSampler.log(createLogEntry(200, "/search/fuzz"));
+        }
+        assertThat(uris, hasItem("/search/fuzz"));
+        assertThat(uris.size(), is(1));
+        for (int i = 0; i < CircularArrayAccessLogKeeper.SIZE; i++) {
+            accessLogSampler.log(createLogEntry(200, "/search/ball"));
+        }
+        assertThat(uris, hasItem("/search/ball"));
+        assertThat(uris.size(), is(2));
+    }
+
+    private AccessLogEntry createLogEntry(int statusCode, String uri) {
+        AccessLogEntry accessLogEntry = new AccessLogEntry();
+        accessLogEntry.setStatusCode(statusCode);
+        accessLogEntry.setRawPath(uri);
+        return accessLogEntry;
+    }
+}