Fix the unit tests for 'stop-point' object_code

hove-io · Oct 8, 2019 · 157f54c · 157f54c
1 parent 883ffe7
commit 157f54c
Showing 1 changed file with 12 additions and 4 deletions.
diff --git 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);
         });
     }