// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.container.logging;

import java.net.URI;


/**
 * @author frodelu
 */
public class JSONLogTestCase extends junit.framework.TestCase {

    private static String ipAddress = "152.200.54.243";
    private static final String EMPTY_REFERRER = "";
    private static final String EMPTY_USERAGENT = "";

    private AccessLogEntry newAccessLogEntry(final String query) {
        final AccessLogEntry entry = new AccessLogEntry();
        entry.setURI(newQueryUri(query));
        entry.setIpV4Address(ipAddress);
        entry.setHttpMethod("GET");
        entry.setHttpVersion("HTTP/1.1");
        entry.setUserAgent("Mozilla/4.05 [en] (Win95; I)");
        entry.setHitCounts(new HitCounts(0, 10, 1234, 0, 10));
        entry.setHostString("localhost");
        entry.setStatusCode(200);
        entry.setTimeStamp(920880005123L);
        entry.setDurationBetweenRequestResponse(122);
        entry.setReturnedContentSize(9875);

        return entry;
    }

    private static URI newQueryUri(final String query) {
        return URI.create("http://localhost?query=" + query);
    }

    public void test_json_log_entry() throws Exception {
        AccessLogEntry entry = newAccessLogEntry("test");

         String expectedOutput =
            "{\"ip\":\"152.200.54.243\"," +
            "\"time\":\"920880005.123\"," +
            "\"duration\":122," +
            "\"size\":9875," +
            "\"code\":200," +
            "\"totalhits\":1234," +
            "\"hits\":0," +
            "\"method\":\"GET\"," +
            "\"uri\":\"?query=test\"," +
            "\"version\":\"HTTP/1.1\"," +
            "\"agent\":\"Mozilla/4.05 [en] (Win95; I)\"," +
            "\"host\":\"localhost\"" +
             "}";

        assertEquals(expectedOutput, new JSONFormatter(entry).format());
    }

    public void test_with_keyvalues() throws Exception {
        AccessLogEntry entry = newAccessLogEntry("test");
        entry.addKeyValue("singlevalue", "value1");
        entry.addKeyValue("multivalue", "value2");
        entry.addKeyValue("multivalue", "value3");

        String expectedOutput =
            "{\"ip\":\"152.200.54.243\"," +
            "\"time\":\"920880005.123\"," +
            "\"duration\":122," +
            "\"size\":9875," +
            "\"code\":200," +
            "\"totalhits\":1234," +
            "\"hits\":0," +
            "\"method\":\"GET\"," +
            "\"uri\":\"?query=test\"," +
            "\"version\":\"HTTP/1.1\"," +
            "\"agent\":\"Mozilla/4.05 [en] (Win95; I)\"," +
            "\"host\":\"localhost\"," +
            "\"attributes\":{" +
            "\"singlevalue\":\"value1\"," +
            "\"multivalue\":[\"value2\",\"value3\"]}" +
            "}";

        assertEquals(expectedOutput, new JSONFormatter(entry).format());

    }

    public void test_with_remoteaddrport() throws Exception {
        AccessLogEntry entry = newAccessLogEntry("test");

        // First test with only remote address and not port set
        entry.setRemoteAddress("FE80:0000:0000:0000:0202:B3FF:FE1E:8329");

        String expectedOutput =
            "{\"ip\":\"152.200.54.243\"," +
            "\"time\":\"920880005.123\"," +
            "\"duration\":122," +
            "\"size\":9875," +
            "\"code\":200," +
            "\"totalhits\":1234," +
            "\"hits\":0," +
            "\"method\":\"GET\"," +
            "\"uri\":\"?query=test\"," +
            "\"version\":\"HTTP/1.1\"," +
            "\"agent\":\"Mozilla/4.05 [en] (Win95; I)\"," +
            "\"host\":\"localhost\"," +
            "\"remoteaddr\":\"FE80:0000:0000:0000:0202:B3FF:FE1E:8329\"" +
            "}";

        assertEquals(expectedOutput, new JSONFormatter(entry).format());

        // Add remote port and verify
        entry.setRemotePort(1234);

        expectedOutput =
            "{\"ip\":\"152.200.54.243\"," +
            "\"time\":\"920880005.123\"," +
            "\"duration\":122," +
            "\"size\":9875," +
            "\"code\":200," +
            "\"totalhits\":1234," +
            "\"hits\":0," +
            "\"method\":\"GET\"," +
            "\"uri\":\"?query=test\"," +
            "\"version\":\"HTTP/1.1\"," +
            "\"agent\":\"Mozilla/4.05 [en] (Win95; I)\"," +
            "\"host\":\"localhost\"," +
            "\"remoteaddr\":\"FE80:0000:0000:0000:0202:B3FF:FE1E:8329\"," +
            "\"remoteport\":1234" +
            "}";

        assertEquals(expectedOutput, new JSONFormatter(entry).format());
    }

    public void test_remote_address_same_as_ip_address() throws Exception {
        AccessLogEntry entry = newAccessLogEntry("test");
        AccessLogEntry entrywithremote = newAccessLogEntry("test");
        entrywithremote.setRemoteAddress(entry.getIpV4Address());

        assertEquals(new JSONFormatter(entry).format(), new JSONFormatter(entrywithremote).format());
    }

    public void test_useragent_with_quotes() throws Exception {
        final AccessLogEntry entry = new AccessLogEntry();
        entry.setURI(newQueryUri("test"));
        entry.setIpV4Address(ipAddress);
        entry.setHttpMethod("GET");
        entry.setHttpVersion("HTTP/1.1");
        entry.setUserAgent("Mozilla/4.05 [en] (Win95; I; \"Best Browser Ever\")");
        entry.setHitCounts(new HitCounts(0, 10, 1234, 0, 10));
        entry.setHostString("localhost");
        entry.setStatusCode(200);
        entry.setTimeStamp(920880005123L);
        entry.setDurationBetweenRequestResponse(122);
        entry.setReturnedContentSize(9875);

        String expectedOutput =
            "{\"ip\":\"152.200.54.243\"," +
            "\"time\":\"920880005.123\"," +
            "\"duration\":122," +
            "\"size\":9875," +
            "\"code\":200," +
            "\"totalhits\":1234," +
            "\"hits\":0," +
            "\"method\":\"GET\"," +
            "\"uri\":\"?query=test\"," +
            "\"version\":\"HTTP/1.1\"," +
            "\"agent\":\"Mozilla/4.05 [en] (Win95; I; \\\"Best Browser Ever\\\")\"," +
            "\"host\":\"localhost\"" +
            "}";

        assertEquals(expectedOutput, new JSONFormatter(entry).format());
    }

}