Search in sources :

Example 1 with VersionOfObjectRefStructure

use of org.rutebanken.netex.model.VersionOfObjectRefStructure in project OpenTripPlanner by opentripplanner.

the class NoticeAssignmentMapperTest method mapNoticeAssignment.

@Test
public void mapNoticeAssignment() {
    NoticeAssignment noticeAssignment = new NoticeAssignment();
    noticeAssignment.setNoticedObjectRef(new VersionOfObjectRefStructure().withRef(ROUTE_ID));
    noticeAssignment.setNotice(NOTICE);
    Route route = new Route();
    route.setId(MappingSupport.ID_FACTORY.createId(ROUTE_ID));
    EntityById<FeedScopedId, Route> routesById = new EntityById<>();
    routesById.add(route);
    NoticeAssignmentMapper noticeAssignmentMapper = new NoticeAssignmentMapper(MappingSupport.ID_FACTORY, new HierarchicalMultimap<>(), new HierarchicalMapById<>(), routesById, new EntityById<>(), new HashMap<>());
    Multimap<TransitEntity<?>, org.opentripplanner.model.Notice> noticesByElement = noticeAssignmentMapper.map(noticeAssignment);
    org.opentripplanner.model.Notice notice2 = noticesByElement.get(route).iterator().next();
    assertEquals(NOTICE_ID, notice2.getId().getId());
}
Also used : VersionOfObjectRefStructure(org.rutebanken.netex.model.VersionOfObjectRefStructure) TransitEntity(org.opentripplanner.model.TransitEntity) EntityById(org.opentripplanner.model.impl.EntityById) Notice(org.rutebanken.netex.model.Notice) NoticeAssignment(org.rutebanken.netex.model.NoticeAssignment) FeedScopedId(org.opentripplanner.model.FeedScopedId) Route(org.opentripplanner.model.Route) Test(org.junit.Test)

Example 2 with VersionOfObjectRefStructure

use of org.rutebanken.netex.model.VersionOfObjectRefStructure in project OpenTripPlanner by opentripplanner.

the class NoticeAssignmentMapperTest method mapNoticeAssignmentOnStopPoint.

@Test
public void mapNoticeAssignmentOnStopPoint() {
    HierarchicalMultimap<String, TimetabledPassingTime> passingTimeByStopPointId = new HierarchicalMultimap<>();
    HierarchicalMapById<Notice> noticesById = new HierarchicalMapById<>();
    passingTimeByStopPointId.add(STOP_POINT_ID, new TimetabledPassingTime().withId(TIMETABLED_PASSING_TIME1));
    passingTimeByStopPointId.add(STOP_POINT_ID, new TimetabledPassingTime().withId(TIMETABLED_PASSING_TIME2));
    Trip trip = new Trip();
    trip.setId(new FeedScopedId("T", "1"));
    StopTime stopTime1 = createStopTime(1, trip);
    StopTime stopTime2 = createStopTime(2, trip);
    Map<String, StopTime> stopTimesById = new HashMap<>();
    stopTimesById.put(TIMETABLED_PASSING_TIME1, stopTime1);
    stopTimesById.put(TIMETABLED_PASSING_TIME2, stopTime2);
    noticesById.add(NOTICE);
    NoticeAssignment noticeAssignment = new NoticeAssignment().withNoticedObjectRef(new VersionOfObjectRefStructure().withRef(STOP_POINT_ID)).withNoticeRef(new NoticeRefStructure().withRef(NOTICE_ID));
    NoticeAssignmentMapper noticeAssignmentMapper = new NoticeAssignmentMapper(MappingSupport.ID_FACTORY, passingTimeByStopPointId, noticesById, new EntityById<>(), new EntityById<>(), stopTimesById);
    Multimap<TransitEntity<?>, org.opentripplanner.model.Notice> noticesByElement = noticeAssignmentMapper.map(noticeAssignment);
    org.opentripplanner.model.Notice notice2a = noticesByElement.get(stopTime1.getId()).stream().findFirst().orElseThrow(IllegalStateException::new);
    org.opentripplanner.model.Notice notice2b = noticesByElement.get(stopTime2.getId()).stream().findFirst().orElseThrow(IllegalStateException::new);
    assertEquals(NOTICE_ID, notice2a.getId().getId());
    assertEquals(NOTICE_ID, notice2b.getId().getId());
}
Also used : Trip(org.opentripplanner.model.Trip) NoticeRefStructure(org.rutebanken.netex.model.NoticeRefStructure) VersionOfObjectRefStructure(org.rutebanken.netex.model.VersionOfObjectRefStructure) HashMap(java.util.HashMap) TransitEntity(org.opentripplanner.model.TransitEntity) MultilingualString(org.rutebanken.netex.model.MultilingualString) HierarchicalMultimap(org.opentripplanner.netex.loader.util.HierarchicalMultimap) HierarchicalMapById(org.opentripplanner.netex.loader.util.HierarchicalMapById) TimetabledPassingTime(org.rutebanken.netex.model.TimetabledPassingTime) Notice(org.rutebanken.netex.model.Notice) FeedScopedId(org.opentripplanner.model.FeedScopedId) NoticeAssignment(org.rutebanken.netex.model.NoticeAssignment) StopTime(org.opentripplanner.model.StopTime) Test(org.junit.Test)

Aggregations

Test (org.junit.Test)2 FeedScopedId (org.opentripplanner.model.FeedScopedId)2 TransitEntity (org.opentripplanner.model.TransitEntity)2 Notice (org.rutebanken.netex.model.Notice)2 NoticeAssignment (org.rutebanken.netex.model.NoticeAssignment)2 VersionOfObjectRefStructure (org.rutebanken.netex.model.VersionOfObjectRefStructure)2 HashMap (java.util.HashMap)1 Route (org.opentripplanner.model.Route)1 StopTime (org.opentripplanner.model.StopTime)1 Trip (org.opentripplanner.model.Trip)1 EntityById (org.opentripplanner.model.impl.EntityById)1 HierarchicalMapById (org.opentripplanner.netex.loader.util.HierarchicalMapById)1 HierarchicalMultimap (org.opentripplanner.netex.loader.util.HierarchicalMultimap)1 MultilingualString (org.rutebanken.netex.model.MultilingualString)1 NoticeRefStructure (org.rutebanken.netex.model.NoticeRefStructure)1 TimetabledPassingTime (org.rutebanken.netex.model.TimetabledPassingTime)1