diff --git a/src/gtfs/read.rs b/src/gtfs/read.rs index d591ab1b2..be6368132 100644 --- a/src/gtfs/read.rs +++ b/src/gtfs/read.rs @@ -1224,18 +1224,26 @@ mod tests { //validate stop_point code assert_eq!(1, stop_points.len()); let stop_point = stop_points.iter().next().unwrap().1; - assert_eq!(1, stop_point.codes.len()); - let code = stop_point.codes.iter().next().unwrap(); + assert_eq!(2, stop_point.codes.len()); + let mut codes_iterator = stop_point.codes.iter(); + let code = codes_iterator.next().unwrap(); assert_eq!("gtfs_stop_code", code.0); assert_eq!("1234", code.1); + let code = codes_iterator.next().unwrap(); + assert_eq!("source", code.0); + assert_eq!("stoppoint_id", code.1); //validate stop_area code assert_eq!(1, stop_areas.len()); let stop_area = stop_areas.iter().next().unwrap().1; - assert_eq!(1, stop_area.codes.len()); - let code = stop_area.codes.iter().next().unwrap(); + assert_eq!(2, stop_area.codes.len()); + let mut codes_iterator = stop_area.codes.iter(); + let code = codes_iterator.next().unwrap(); assert_eq!("gtfs_stop_code", code.0); assert_eq!("5678", code.1); + let code = codes_iterator.next().unwrap(); + assert_eq!("source", code.0); + assert_eq!("stoparea_id", code.1); }); }