GelfConfigurationTest.java

/*
 * Copyright 2014 TORCH GmbH
 *
 * 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 org.graylog2.gelfclient;

import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.io.File;
import java.net.InetSocketAddress;

import static org.testng.Assert.assertFalse;
import static org.testng.Assert.assertNull;
import static org.testng.Assert.assertTrue;
import static org.testng.AssertJUnit.assertEquals;

public class GelfConfigurationTest {
    private GelfConfiguration config;

    @BeforeMethod
    public void setup() {
        this.config = new GelfConfiguration();
    }

    @Test(expectedExceptions = IllegalArgumentException.class)
    public void testConstructorNullHostname() throws Exception {
        new GelfConfiguration(null, 12201);
    }

    @Test(expectedExceptions = IllegalArgumentException.class)
    public void testConstructorEmptyHostname() throws Exception {
        new GelfConfiguration("  ", 12201);
    }

    @Test
    public void testConstructorMinPort() throws Exception {
        new GelfConfiguration("127.0.0.1", 1);
    }

    @Test
    public void testConstructorMaxPort() throws Exception {
        new GelfConfiguration("127.0.0.1", 65535);
    }

    @Test(expectedExceptions = IllegalArgumentException.class)
    public void testConstructorSmallerMinPort() throws Exception {
        new GelfConfiguration("127.0.0.1", 0);
    }

    @Test(expectedExceptions = IllegalArgumentException.class)
    public void testConstructorLargerMaxPort() throws Exception {
        new GelfConfiguration("127.0.0.1", 65536);
    }

    @Test
    public void testQueueSize() {
        // Check default value.
        assertEquals(512, config.getQueueSize());

        config.queueSize(124);

        assertEquals(124, config.getQueueSize());
    }

    @Test
    public void testTls() {
        assertFalse(config.isTlsEnabled());

        config.enableTls();

        assertTrue(config.isTlsEnabled());

        config.disableTls();

        assertFalse(config.isTlsEnabled());
    }

    @Test
    public void testTlsTrustCertChainFile() {
        assertNull(config.getTlsTrustCertChainFile());

        config.tlsTrustCertChainFile(new File("/tmp/cert.pem"));

        assertEquals(config.getTlsTrustCertChainFile(), new File("/tmp/cert.pem"));
    }

    @Test
    public void testTlsVerifyCert() throws Exception {
        assertTrue(config.isTlsCertVerificationEnabled());

        config.disableTlsCertVerification();

        assertFalse(config.isTlsCertVerificationEnabled());

        config.enableTlsCertVerification();

        assertTrue(config.isTlsCertVerificationEnabled());
    }

    @Test
    public void testRemoteAddress() {
        assertEquals(new InetSocketAddress("127.0.0.1", 12201), config.getRemoteAddress());
        assertEquals(new InetSocketAddress("10.0.0.1", 12345),
                new GelfConfiguration(new InetSocketAddress("10.0.0.1", 12345)).getRemoteAddress());
    }

    @Test
    public void testPort() {
        assertEquals(12201, config.getPort());
        assertEquals(10000, new GelfConfiguration(10000).getPort());
    }

    @Test
    public void testHostName() {
        assertEquals("127.0.0.1", config.getHostname());
        assertEquals("example.com", new GelfConfiguration("example.com").getHostname());
    }

    @Test
    public void testTransport() {
        assertEquals(GelfTransports.TCP, config.getTransport());
        config.transport(GelfTransports.UDP);
        assertEquals(GelfTransports.UDP, config.getTransport());
    }

    @Test
    public void testReconnectDelay() {
        // Check default value.
        assertEquals(500, config.getReconnectDelay());

        config.reconnectDelay(5000);

        assertEquals(5000, config.getReconnectDelay());
    }

    @Test
    public void testConnectTimeout() {
        // Check default value.
        assertEquals(1000, config.getConnectTimeout());

        config.connectTimeout(10000);

        assertEquals(10000, config.getConnectTimeout());
    }

    @Test
    public void testtcpNoDelay() {
        // Check default value.
        assertFalse(config.isTcpNoDelay());

        config.tcpNoDelay(true);

        assertTrue(config.isTcpNoDelay());
    }

    @Test
    public void testtcpKeepAlive() {
        // Check default value.
        assertFalse(config.isTcpKeepAlive());

        config.tcpKeepAlive(true);

        assertTrue(config.isTcpKeepAlive());
    }

    @Test
    public void testSendBufferSize() {
        // Check default value.
        assertEquals(-1, config.getSendBufferSize());

        config.sendBufferSize(32768);

        assertEquals(32768, config.getSendBufferSize());
    }

    @Test
    public void testMaxInflightSends() {
        assertEquals(512, config.getMaxInflightSends());
        assertEquals(config.maxInflightSends(1024), config);
        assertEquals(1024, config.getMaxInflightSends());
    }
}