BluefloodServiceStarterTest.java

package com.rackspacecloud.blueflood.service;

import com.codahale.metrics.MetricRegistry;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PowerMockIgnore;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import java.io.IOException;

import static junit.framework.Assert.*;
import static org.powermock.api.mockito.PowerMockito.mock;

@RunWith(PowerMockRunner.class)
@PrepareForTest(BluefloodServiceStarter.class)
@PowerMockIgnore("javax.management.*")
public class BluefloodServiceStarterTest {

    @Before
    public void setUp() throws Exception {

        Configuration config = Configuration.getInstance();

        config.init();

        config.setProperty(CoreConfig.ZOOKEEPER_CLUSTER, "NONE");

        config.setProperty(CoreConfig.INGEST_MODE, "false");
        config.setProperty(CoreConfig.ROLLUP_MODE, "false");
        config.setProperty(CoreConfig.QUERY_MODE, "false");

        PowerMockito.spy(BluefloodServiceStarter.class);
        PowerMockito.doReturn(mock(MetricRegistry.class)).when(BluefloodServiceStarter.class, "getRegistry");
    }

    @Test
    public void testAllModesDisabled() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.INGEST_MODE, "false");
        config.setProperty(CoreConfig.QUERY_MODE, "false");
        config.setProperty(CoreConfig.ROLLUP_MODE, "false");

        // when
        BluefloodServiceStarter.run();

        // then
        // no exception was thrown
    }

    @Test(expected = BluefloodServiceStarterException.class)
    public void testNoCassandraHostsFailsValidation() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.CASSANDRA_HOSTS, "");

        // when
        BluefloodServiceStarter.validateCassandraHosts();

        // then
        // exception was thrown
    }

    @Test(expected = BluefloodServiceStarterException.class)
    public void testInvalidCassandraHostsFailsValidation() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.CASSANDRA_HOSTS, "something");

        // when
        BluefloodServiceStarter.validateCassandraHosts();

        // then
        // exception was thrown
    }

    @Test(expected = BluefloodServiceStarterException.class)
    public void testCassandraHostWithoutPortFailsValidation() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.CASSANDRA_HOSTS, "127.0.0.1");

        // when
        BluefloodServiceStarter.validateCassandraHosts();

        // then
        // exception was thrown
    }

    @Test
    public void testCassandraHostWithHyphenSucceedsValidation() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.CASSANDRA_HOSTS, "my-hostname:9160");

        // when
        BluefloodServiceStarter.validateCassandraHosts();

        // then
        // no exception was thrown
    }

    @Test
    public void testCassandraHostWithUnderscoreSucceedsValidation() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.CASSANDRA_HOSTS, "my_hostname:9160");

        // when
        BluefloodServiceStarter.validateCassandraHosts();

        // then
        // no exception was thrown
    }

    @Test(expected = BluefloodServiceStarterException.class)
    public void testIngestModeEnabledWithoutModules() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.INGEST_MODE, "true");
        config.setProperty(CoreConfig.INGESTION_MODULES, "");

        // when
        BluefloodServiceStarter.run();

        // then
        // exception was thrown
    }

    @Test
    public void testIngestModeEnabledWithModules() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.INGEST_MODE, "true");
        config.setProperty(CoreConfig.INGESTION_MODULES, "com.rackspacecloud.blueflood.service.DummyIngestionService");

        // when
        BluefloodServiceStarter.run();

        // then
        assertNotNull(DummyIngestionService.getInstances());
        assertEquals(1, DummyIngestionService.getInstances().size());
        assertNotNull(DummyIngestionService.getMostRecentInstance());

        assertTrue(DummyIngestionService.getMostRecentInstance().getStartServiceCalled());
        assertFalse(DummyIngestionService.getMostRecentInstance().getShutdownServiceCalled());
    }

    @Test(expected = BluefloodServiceStarterException.class)
    public void testQueryModeEnabledWithoutModules() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.QUERY_MODE, "true");
        config.setProperty(CoreConfig.QUERY_MODULES, "");

        // when
        BluefloodServiceStarter.run();

        // then
        // exception was thrown
    }

    @Test
    public void testQueryModeEnabledWithDummyModule() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.QUERY_MODE, "true");
        config.setProperty(CoreConfig.QUERY_MODULES, "com.rackspacecloud.blueflood.service.DummyQueryService");

        // when
        BluefloodServiceStarter.run();

        // then
        assertNotNull(DummyQueryService.getInstances());
        assertEquals(1, DummyQueryService.getInstances().size());
        assertNotNull(DummyQueryService.getMostRecentInstance());

        assertTrue(DummyQueryService.getMostRecentInstance().getStartServiceCalled());
    }

    @Test
    public void testRollupModeEnabled() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.ROLLUP_MODE, "true");

        // when
        BluefloodServiceStarter.run();

        // then
        // no exception thrown
    }

    @Test
    public void testEventListenerServiceEnabledWithDummyModule() {

        // given
        Configuration config = Configuration.getInstance();
        config.setProperty(CoreConfig.EVENT_LISTENER_MODULES, "com.rackspacecloud.blueflood.service.DummyEventListenerService");

        // when
        BluefloodServiceStarter.run();

        // then
        assertNotNull(DummyEventListenerService.getInstances());
        assertEquals(1, DummyEventListenerService.getInstances().size());
        assertNotNull(DummyEventListenerService.getMostRecentInstance());

        assertTrue(DummyEventListenerService.getMostRecentInstance().getStartServiceCalled());
    }

    @After
    public void tearDown() throws IOException {
        Configuration config = Configuration.getInstance();
        config.init();
    }
}