PacketHandlerImplTest.java

package dk.dma.ais.abnormal.analyzer;

import com.google.inject.Injector;
import dk.dma.ais.abnormal.analyzer.analysis.Analysis;
import dk.dma.ais.abnormal.analyzer.analysis.CloseEncounterAnalysis;
import dk.dma.ais.abnormal.analyzer.analysis.CourseOverGroundAnalysis;
import dk.dma.ais.abnormal.analyzer.analysis.DriftAnalysis;
import dk.dma.ais.abnormal.analyzer.analysis.FreeFlowAnalysis;
import dk.dma.ais.abnormal.analyzer.analysis.ShipTypeAndSizeAnalysis;
import dk.dma.ais.abnormal.analyzer.analysis.SpeedOverGroundAnalysis;
import dk.dma.ais.abnormal.analyzer.analysis.SuddenSpeedChangeAnalysis;
import dk.dma.ais.abnormal.analyzer.services.SafetyZoneService;
import dk.dma.ais.tracker.eventEmittingTracker.EventEmittingTracker;
import org.apache.commons.configuration.PropertiesConfiguration;
import org.jmock.Expectations;
import org.jmock.integration.junit4.JUnit4Mockery;
import org.junit.Test;

import java.util.Set;

import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_CLOSEENCOUNTER_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_COG_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_DRIFT_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_FREEFLOW_BBOX;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_FREEFLOW_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_SOG_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_SUDDENSPEEDCHANGE_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_ANALYSIS_TYPESIZE_ENABLED;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_BEHIND;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_BREADTH;
import static dk.dma.ais.abnormal.analyzer.config.Configuration.CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_LENGTH;
import static org.junit.Assert.assertEquals;

public class PacketHandlerImplTest {

    @Test
    public void initNoAnalyses() throws Exception {
        PropertiesConfiguration configuration = new PropertiesConfiguration();
        configuration.addProperty(CONFKEY_ANALYSIS_COG_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_SOG_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_TYPESIZE_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_DRIFT_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_SUDDENSPEEDCHANGE_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_CLOSEENCOUNTER_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_FREEFLOW_ENABLED, false);

        final JUnit4Mockery context = new JUnit4Mockery();
        Injector injectorMock = context.mock(Injector.class);

        context.checking(new Expectations() {{
        }});

        PacketHandlerImpl sut = new PacketHandlerImpl(configuration, injectorMock, null, null, null, null);
        Set<Analysis> analyses = sut.getAnalyses();

        assertEquals(0, analyses.size());
    }

    @Test
    public void initAllAnalyses() throws Exception {
        PropertiesConfiguration configuration = new PropertiesConfiguration();
        configuration.addProperty(CONFKEY_ANALYSIS_COG_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_SOG_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_TYPESIZE_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_DRIFT_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_SUDDENSPEEDCHANGE_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_CLOSEENCOUNTER_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_FREEFLOW_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_FREEFLOW_BBOX, new Integer[] {1, 2, 3, 4});
        configuration.addProperty(CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_LENGTH, 2.0);
        configuration.addProperty(CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_BREADTH, 3.0);
        configuration.addProperty(CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_BEHIND, 0.25);

        final JUnit4Mockery context = new JUnit4Mockery();
        Injector injectorMock = context.mock(Injector.class);
        EventEmittingTracker trackingServiceMock = context.mock(EventEmittingTracker.class);

        SafetyZoneService safetyZoneService = new SafetyZoneService(configuration);

        context.checking(new Expectations() {{
            ignoring(trackingServiceMock);
            oneOf(injectorMock).getInstance(with(CourseOverGroundAnalysis.class)); will(returnValue(new CourseOverGroundAnalysis(configuration, null, null, trackingServiceMock, null, null)));
            oneOf(injectorMock).getInstance(with(SpeedOverGroundAnalysis.class)); will(returnValue(new SpeedOverGroundAnalysis(configuration, null, null, trackingServiceMock, null, null)));
            oneOf(injectorMock).getInstance(with(ShipTypeAndSizeAnalysis.class)); will(returnValue(new ShipTypeAndSizeAnalysis(configuration, null, null, trackingServiceMock, null, null)));
            oneOf(injectorMock).getInstance(with(DriftAnalysis.class)); will(returnValue(new DriftAnalysis(configuration, null, trackingServiceMock, null)));
            oneOf(injectorMock).getInstance(with(SuddenSpeedChangeAnalysis.class)); will(returnValue(new SuddenSpeedChangeAnalysis(configuration, null, trackingServiceMock, null)));
            oneOf(injectorMock).getInstance(with(CloseEncounterAnalysis.class)); will(returnValue(new CloseEncounterAnalysis(configuration, null, trackingServiceMock, null, safetyZoneService)));
            oneOf(injectorMock).getInstance(with(FreeFlowAnalysis.class)); will(returnValue(new FreeFlowAnalysis(configuration, null, trackingServiceMock, null)));
        }});

        PacketHandlerImpl sut = new PacketHandlerImpl(configuration, injectorMock, null, null, null, null);

        Set<Analysis> analyses = sut.getAnalyses();
        assertEquals(7, analyses.size());
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof CourseOverGroundAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof SpeedOverGroundAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof ShipTypeAndSizeAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof DriftAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof SuddenSpeedChangeAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof CloseEncounterAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof FreeFlowAnalysis));
    }

    @Test
    public void initSelectedAnalyses() throws Exception {
        PropertiesConfiguration configuration = new PropertiesConfiguration();
        configuration.addProperty(CONFKEY_ANALYSIS_COG_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_SOG_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_TYPESIZE_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_DRIFT_ENABLED, false);
        configuration.addProperty(CONFKEY_ANALYSIS_SUDDENSPEEDCHANGE_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_CLOSEENCOUNTER_ENABLED, true);
        configuration.addProperty(CONFKEY_ANALYSIS_FREEFLOW_ENABLED, false);
        configuration.addProperty(CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_LENGTH, 2.0);
        configuration.addProperty(CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_BREADTH, 3.0);
        configuration.addProperty(CONFKEY_SAFETYZONES_SAFETY_ELLIPSE_BEHIND, 0.25);

        final JUnit4Mockery context = new JUnit4Mockery();
        Injector injectorMock = context.mock(Injector.class);
        EventEmittingTracker trackingServiceMock = context.mock(EventEmittingTracker.class);

        SafetyZoneService safetyZoneService = new SafetyZoneService(configuration);

        context.checking(new Expectations() {{
            ignoring(trackingServiceMock);
            oneOf(injectorMock).getInstance(with(SuddenSpeedChangeAnalysis.class)); will(returnValue(new SuddenSpeedChangeAnalysis(configuration, null, trackingServiceMock, null)));
            oneOf(injectorMock).getInstance(with(CloseEncounterAnalysis.class)); will(returnValue(new CloseEncounterAnalysis(configuration, null, trackingServiceMock, null, safetyZoneService)));
        }});

        PacketHandlerImpl sut = new PacketHandlerImpl(configuration, injectorMock, null, null, null, null);

        Set<Analysis> analyses = sut.getAnalyses();
        assertEquals(2, analyses.size());
        assertEquals(false, analyses.stream().anyMatch(a -> a instanceof CourseOverGroundAnalysis));
        assertEquals(false, analyses.stream().anyMatch(a -> a instanceof SpeedOverGroundAnalysis));
        assertEquals(false, analyses.stream().anyMatch(a -> a instanceof ShipTypeAndSizeAnalysis));
        assertEquals(false, analyses.stream().anyMatch(a -> a instanceof DriftAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof SuddenSpeedChangeAnalysis));
        assertEquals(true, analyses.stream().anyMatch(a -> a instanceof CloseEncounterAnalysis));
        assertEquals(false, analyses.stream().anyMatch(a -> a instanceof FreeFlowAnalysis));
    }

}