Examples with ServiceIdOverlapCache org.onebusaway.transit_data_federation.bundle.tasks.transit_graph.ServiceIdOverlapCache used on opensource projects

Example 1 with ServiceIdOverlapCache

use of org.onebusaway.transit_data_federation.bundle.tasks.transit_graph.ServiceIdOverlapCache in project onebusaway-application-modules by camsys.

the class ServiceIdOverlapCacheTest method test.

@Test
public void test() {
    CalendarService calendarService = Mockito.mock(CalendarService.class);
    TimeZone tz = TimeZone.getDefault();
    AgencyAndId serviceIdA = new AgencyAndId("1", "serviceIdA");
    AgencyAndId serviceIdB = new AgencyAndId("1", "serviceIdB");
    AgencyAndId serviceIdC = new AgencyAndId("1", "serviceIdC");
    AgencyAndId serviceIdD = new AgencyAndId("1", "serviceIdD");
    LocalizedServiceId lsidA = new LocalizedServiceId(serviceIdA, tz);
    LocalizedServiceId lsidB = new LocalizedServiceId(serviceIdB, tz);
    LocalizedServiceId lsidC = new LocalizedServiceId(serviceIdC, tz);
    LocalizedServiceId lsidD = new LocalizedServiceId(serviceIdD, tz);
    Set<ServiceDate> serviceDatesA = set(new ServiceDate(2010, 9, 10), new ServiceDate(2010, 9, 11));
    Set<ServiceDate> serviceDatesB = set(new ServiceDate(2010, 9, 11), new ServiceDate(2010, 9, 12));
    Set<ServiceDate> serviceDatesC = set(new ServiceDate(2010, 9, 12), new ServiceDate(2010, 9, 13));
    Set<ServiceDate> serviceDatesD = set(new ServiceDate(2010, 9, 13));
    Mockito.when(calendarService.getServiceDatesForServiceId(serviceIdA)).thenReturn(serviceDatesA);
    Mockito.when(calendarService.getServiceDatesForServiceId(serviceIdB)).thenReturn(serviceDatesB);
    Mockito.when(calendarService.getServiceDatesForServiceId(serviceIdC)).thenReturn(serviceDatesC);
    Mockito.when(calendarService.getServiceDatesForServiceId(serviceIdD)).thenReturn(serviceDatesD);
    ServiceIdOverlapCache cache = new ServiceIdOverlapCache();
    cache.setCalendarService(calendarService);
    List<ServiceIdActivation> combinations = cache.getOverlappingServiceIdCombinations(set(lsidA, lsidB, lsidC, lsidD));
    assertEquals(4, combinations.size());
    ServiceIdActivation combo = combinations.get(0);
    assertEquals(Arrays.asList(lsidA, lsidB), combo.getActiveServiceIds());
    assertEquals(Arrays.asList(), combo.getInactiveServiceIds());
    combo = combinations.get(1);
    assertEquals(Arrays.asList(lsidB, lsidC), combo.getActiveServiceIds());
    assertEquals(Arrays.asList(), combo.getInactiveServiceIds());
    combo = combinations.get(2);
    assertEquals(Arrays.asList(lsidC, lsidD), combo.getActiveServiceIds());
    assertEquals(Arrays.asList(), combo.getInactiveServiceIds());
    combo = combinations.get(3);
    assertEquals(Arrays.asList(lsidA), combo.getActiveServiceIds());
    assertEquals(Arrays.asList(lsidB), combo.getInactiveServiceIds());
    combinations = cache.getOverlappingServiceIdCombinations(set(lsidA, lsidB, lsidC, lsidD));
    Mockito.verify(calendarService, Mockito.times(1)).getServiceDatesForServiceId(serviceIdA);
    Mockito.verify(calendarService, Mockito.times(1)).getServiceDatesForServiceId(serviceIdB);
    Mockito.verify(calendarService, Mockito.times(1)).getServiceDatesForServiceId(serviceIdC);
    Mockito.verify(calendarService, Mockito.times(1)).getServiceDatesForServiceId(serviceIdD);
}