use of org.onebusaway.transit_data_federation.impl.transit_graph.RouteCollectionEntryImpl in project onebusaway-application-modules by camsys.
the class GenerateNarrativesTaskTest method testGenerateRouteNarratives.
@Test
public void testGenerateRouteNarratives() {
RouteEntryImpl r1 = route("routeA1");
RouteEntryImpl r2 = route("routeA2");
RouteCollectionEntryImpl rc = routeCollection("routeA", r1, r2);
TripEntry t1 = trip("t1");
TripEntry t2 = trip("t2");
TripEntry t3 = trip("t3");
// r2's values should win out over r1's because it has more trips
r1.setTrips(Arrays.asList(t1));
r2.setTrips(Arrays.asList(t2, t3));
Mockito.when(_transitGraphDao.getAllRouteCollections()).thenReturn(Arrays.asList((RouteCollectionEntry) rc));
Route route1 = new Route();
route1.setId(r1.getId());
route1.setBikesAllowed(0);
route1.setColor("#000000");
route1.setDesc("Route One Desc");
route1.setLongName("Route One");
route1.setShortName("One");
route1.setTextColor("#ff0000");
route1.setType(3);
route1.setUrl("http://agency.gov/route-one");
Route route2 = new Route();
route2.setId(r2.getId());
route2.setBikesAllowed(1);
route2.setColor("#0000ff");
route2.setDesc("Route Two Desc");
route2.setLongName("Route Two");
route2.setShortName("Two");
route2.setTextColor("#000000");
route2.setType(3);
route2.setUrl("http://agency.gov/route-two");
Mockito.when(_gtfsDao.getRouteForId(r1.getId())).thenReturn(route1);
Mockito.when(_gtfsDao.getRouteForId(r2.getId())).thenReturn(route2);
_task.generateRouteNarratives(_provider);
RouteCollectionNarrative narrative = _provider.getNarrativeForRouteCollectionId(rc.getId());
assertEquals(route2.getColor(), narrative.getColor());
assertEquals(route2.getDesc(), narrative.getDescription());
assertEquals(route2.getLongName(), narrative.getLongName());
assertEquals(route2.getShortName(), narrative.getShortName());
assertEquals(route2.getTextColor(), narrative.getTextColor());
assertEquals(route2.getType(), narrative.getType());
assertEquals(route2.getUrl(), narrative.getUrl());
}
use of org.onebusaway.transit_data_federation.impl.transit_graph.RouteCollectionEntryImpl in project onebusaway-application-modules by camsys.
the class GenerateRouteCollectionSearchIndexTaskTest method testGenerateStopSearchIndex.
@Test
public void testGenerateStopSearchIndex() throws CorruptIndexException, IOException, ParseException {
RouteCollectionEntryImpl routeA = routeCollection("routeA");
RouteCollectionEntryImpl routeB = routeCollection("routeB");
RouteCollectionEntryImpl routeC = routeCollection("routeC");
RouteCollectionNarrative.Builder routeNarrativeA = RouteCollectionNarrative.builder();
routeNarrativeA.setShortName("10");
routeNarrativeA.setLongName("El Diez");
RouteCollectionNarrative.Builder routeNarrativeB = RouteCollectionNarrative.builder();
routeNarrativeB.setShortName("11");
routeNarrativeB.setLongName("El Once");
RouteCollectionNarrative.Builder routeNarrativeC = RouteCollectionNarrative.builder();
routeNarrativeC.setShortName("100");
Mockito.when(_transitGraphDao.getAllRouteCollections()).thenReturn(Arrays.asList((RouteCollectionEntry) routeA, routeB, routeC));
Mockito.when(_narrativeService.getRouteCollectionForId(routeA.getId())).thenReturn(routeNarrativeA.create());
Mockito.when(_narrativeService.getRouteCollectionForId(routeB.getId())).thenReturn(routeNarrativeB.create());
Mockito.when(_narrativeService.getRouteCollectionForId(routeC.getId())).thenReturn(routeNarrativeC.create());
_task.run();
Mockito.verify(_refreshService).refresh(RefreshableResources.ROUTE_COLLECTION_SEARCH_DATA);
RouteCollectionSearchServiceImpl searchService = new RouteCollectionSearchServiceImpl();
searchService.setBundle(_bundle);
searchService.initialize();
SearchResult<AgencyAndId> ids = searchService.searchForRoutesByName("10", 10, MIN_SCORE);
assertEquals(1, ids.size());
assertEquals(routeA.getId(), ids.getResult(0));
ids = searchService.searchForRoutesByName("el diez", 10, MIN_SCORE);
assertEquals(1, ids.size());
assertEquals(routeA.getId(), ids.getResult(0));
ids = searchService.searchForRoutesByName("diez", 10, MIN_SCORE);
assertEquals(1, ids.size());
assertEquals(routeA.getId(), ids.getResult(0));
ids = searchService.searchForRoutesByName("11", 10, MIN_SCORE);
assertEquals(1, ids.size());
assertEquals(routeB.getId(), ids.getResult(0));
ids = searchService.searchForRoutesByName("100", 10, MIN_SCORE);
assertEquals(1, ids.size());
assertEquals(routeC.getId(), ids.getResult(0));
}
Aggregations