LoggingFilterHttpServerAdapterTests.java

/*
 * Copyright 2011 Sonian Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.sonian.elasticsearch.http.filter.logging;

import com.sonian.elasticsearch.http.filter.FilterHttpServerTransport;
import com.sonian.elasticsearch.http.filter.FilterHttpServerTransportModule;
import com.sonian.elasticsearch.http.jetty.AbstractJettyHttpServerTests;
import com.sonian.elasticsearch.http.jetty.HttpClientResponse;
import org.elasticsearch.common.collect.MapBuilder;
import org.elasticsearch.common.logging.ESLogger;
import org.elasticsearch.common.logging.ESLoggerFactory;
import org.elasticsearch.common.logging.log4j.Log4jESLoggerFactory;
import org.elasticsearch.common.settings.ImmutableSettings;
import org.elasticsearch.http.HttpServerTransport;
import org.elasticsearch.node.internal.InternalNode;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.util.Map;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;

import static com.sonian.elasticsearch.http.filter.logging.RequestLoggingLevel.Level.*;
import static org.elasticsearch.rest.RestRequest.Method.GET;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.*;

/**
 * @author imotov
 */
public class LoggingFilterHttpServerAdapterTests extends AbstractJettyHttpServerTests {

    private MockESLoggerFactory mockESLoggerFactory;

    @BeforeMethod
    public void setup() {
        mockESLoggerFactory = new MockESLoggerFactory("INFO", "com.sonian.elasticsearch.http.filter.logging");
        ESLoggerFactory.setDefaultFactory(mockESLoggerFactory);
        putDefaultSettings(ImmutableSettings.settingsBuilder()
                .put("http.type", FilterHttpServerTransportModule.class.getName())
        );
    }

    @AfterMethod
    public void closeNodes() {
        closeAllNodes();
        ESLoggerFactory.setDefaultFactory(mockESLoggerFactory.getRealFactory());
    }

    @Test
    public void testClusterHealth() throws Exception {
        startNode("server1");
        createTestIndex();
        // Shouldn't log cluster health call
        HttpClientResponse response = httpClient("server1").request("_cluster/health");
        assertThat((String) response.get("status"), equalTo("green"));

        Map<String, Object> data = createSearchQuery("user:kimchy");
        httpClient("server1").request("POST", "_search", data);
        // Should start with logging for the POST /_search request
        String logMessage = mockESLoggerFactory.getMessage();
        assertThat(logMessage, startsWith("INFO:[server1] POST /_search - 200 OK"));
        // should contain request body
        assertThat(logMessage, containsString("user:kimchy"));
    }

    @Test
    public void testLoggingSettings() throws Exception {
        startNode("server1");
        FilterHttpServerTransport filterHttpServerTransport = (FilterHttpServerTransport)
                ((InternalNode) node("server1")).injector().getInstance(HttpServerTransport.class);
        LoggingFilterHttpServerAdapter logging = (LoggingFilterHttpServerAdapter) filterHttpServerTransport.filter("logging");
        assertThat(logging.requestLoggingLevelSettings().getLoggingLevel(GET, "/_bulk").logBody(), equalTo(false));
        assertThat(logging.requestLoggingLevelSettings().getLoggingLevel(GET, "/idx/_bulk").logBody(), equalTo(false));
        assertThat(logging.requestLoggingLevelSettings().getLoggingLevel(GET, "/_cluster/health").logLevel(), equalTo(TRACE));
    }

    @Test
    public void testEmptyLoggingSettings() throws Exception {
        // Replace logger with custom settings with logger2 with default settings
        putDefaultSettings(ImmutableSettings.settingsBuilder()
                .putArray("sonian.elasticsearch.http.filter.http_filter_chain", "timeout", "logging2")
                .put("sonian.elasticsearch.http.filter.http_filter.logging2.type",
                        "com.sonian.elasticsearch.http.filter.logging.LoggingFilterHttpServerAdapter")
        );
        startNode("server1");
        createTestIndex();

        // Should log cluster health call
        HttpClientResponse response = httpClient("server1").request("_cluster/health");
        assertThat((String) response.get("status"), equalTo("green"));
        assertThat(mockESLoggerFactory.getMessage(), startsWith("INFO:[server1] GET /_cluster/health - 200 OK"));

        Map<String, Object> data = createSearchQuery("user:kimchy");
        httpClient("server1").request("POST", "_search", data);
        String logMessage = mockESLoggerFactory.getMessage();
        // Should start with logging for the POST /_search request
        assertThat(logMessage, startsWith("INFO:[server1] POST /_search - 200 OK"));
        // shouldn't contain request body
        assertThat(logMessage, not(containsString("user:kimchy")));
    }

}