Examples with StopValidator edu.usf.cutr.gtfsrtvalidator.lib.validation.rules.StopValidator used on opensource projects

Example 1 with StopValidator

use of edu.usf.cutr.gtfsrtvalidator.lib.validation.rules.StopValidator in project gtfs-realtime-validator by CUTR-at-USF.

the class StopValidatorTest method testE015.

/**
 * E015 - All stop_ids referenced in GTFS-rt feeds must have the location_type = 0
 * <p>
 * testagency2.zip has:
 * stop_id A with location_type = 0
 * stop_id B with location_type = 1
 */
@Test
public void testE015() {
    StopValidator locationValidator = new StopValidator();
    Map<ValidationRule, Integer> expected = new HashMap<>();
    GtfsRealtime.TripDescriptor.Builder tripDescriptorBuilder = GtfsRealtime.TripDescriptor.newBuilder();
    GtfsRealtime.TripUpdate.StopTimeUpdate.Builder stopTimeUpdateBuilder = GtfsRealtime.TripUpdate.StopTimeUpdate.newBuilder();
    GtfsRealtime.EntitySelector.Builder entitySelectorBuilder = GtfsRealtime.EntitySelector.newBuilder();
    // tripDescriptor is a required field in tripUpdate
    tripUpdateBuilder.setTrip(tripDescriptorBuilder.build());
    feedEntityBuilder.setTripUpdate(tripUpdateBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    // setting stop id = "A" (which has location_type = 0) in all the three feeds
    stopTimeUpdateBuilder.setStopId("A");
    tripUpdateBuilder.addStopTimeUpdate(stopTimeUpdateBuilder.build());
    feedEntityBuilder.setTripUpdate(tripUpdateBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    vehiclePositionBuilder.setStopId("A");
    feedEntityBuilder.setVehicle(vehiclePositionBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    entitySelectorBuilder.setStopId("A");
    alertBuilder.addInformedEntity(entitySelectorBuilder.build());
    feedEntityBuilder.setAlert(alertBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    // All the feeds have valid stop id with location_type = 0. So, returns 0 results
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData2, gtfsData2Metadata, feedMessageBuilder.build(), null, null);
    expected.clear();
    TestUtils.assertResults(expected, results);
    // Change to use stop_id "B", which has location-type 1
    stopTimeUpdateBuilder.setStopId("B");
    tripUpdateBuilder.addStopTimeUpdate(stopTimeUpdateBuilder.build());
    feedEntityBuilder.setTripUpdate(tripUpdateBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData2, gtfsData2Metadata, feedMessageBuilder.build(), null, null);
    // one error from TripUpdate feed stop_id = "B", which is location_type 1
    expected.put(E015, 1);
    TestUtils.assertResults(expected, results);
    vehiclePositionBuilder.setStopId("B");
    feedEntityBuilder.setVehicle(vehiclePositionBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData2, gtfsData2Metadata, feedMessageBuilder.build(), null, null);
    // 2 results from TripUpdate and VehiclePosition feeds stop_id = "B", which is location_type 1
    expected.put(E015, 2);
    TestUtils.assertResults(expected, results);
    entitySelectorBuilder.setStopId("B");
    alertBuilder.addInformedEntity(entitySelectorBuilder.build());
    feedEntityBuilder.setAlert(alertBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData2, gtfsData2Metadata, feedMessageBuilder.build(), null, null);
    // 3 results from TripUpdate, VehiclePosition and alert feeds stop_id = "B", which is location_type 1
    expected.put(E015, 3);
    TestUtils.assertResults(expected, results);
    clearAndInitRequiredFeedFields();
}

Example 2 with StopValidator

use of edu.usf.cutr.gtfsrtvalidator.lib.validation.rules.StopValidator in project gtfs-realtime-validator by CUTR-at-USF.

the class StopValidatorTest method testE011.

/**
 * E011 - All stop_ids referenced in GTFS-rt feed must appear in the GTFS feed
 */
@Test
public void testE011() {
    StopValidator locationValidator = new StopValidator();
    Map<ValidationRule, Integer> expected = new HashMap<>();
    GtfsRealtime.TripDescriptor.Builder tripDescriptorBuilder = GtfsRealtime.TripDescriptor.newBuilder();
    GtfsRealtime.TripUpdate.StopTimeUpdate.Builder stopTimeUpdateBuilder = GtfsRealtime.TripUpdate.StopTimeUpdate.newBuilder();
    GtfsRealtime.EntitySelector.Builder entitySelectorBuilder = GtfsRealtime.EntitySelector.newBuilder();
    GtfsRealtime.VehicleDescriptor.Builder vehicleDescriptorBuilder = GtfsRealtime.VehicleDescriptor.newBuilder();
    vehicleDescriptorBuilder.setId("1");
    // tripDescriptor is a required field in tripUpdate
    tripUpdateBuilder.setTrip(tripDescriptorBuilder.build());
    feedEntityBuilder.setTripUpdate(tripUpdateBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    // setting stop id = "A" in all the three feeds that matches the stop id in static Gtfs data
    stopTimeUpdateBuilder.setStopId("A");
    tripUpdateBuilder.addStopTimeUpdate(stopTimeUpdateBuilder.build());
    feedEntityBuilder.setTripUpdate(tripUpdateBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    vehiclePositionBuilder.setStopId("A");
    vehiclePositionBuilder.setVehicle(vehicleDescriptorBuilder.build());
    feedEntityBuilder.setVehicle(vehiclePositionBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    entitySelectorBuilder.setStopId("A");
    alertBuilder.addInformedEntity(entitySelectorBuilder.build());
    feedEntityBuilder.setAlert(alertBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    // all the feeds have valid stop id matching that in static Gtfs data. So, returns 0 results
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData, gtfsDataMetadata, feedMessageBuilder.build(), null, null);
    expected.clear();
    TestUtils.assertResults(expected, results);
    // setting stop id = "DUMMY" in TripUpdate feed that does not match with any stop id in static Gtfs data
    stopTimeUpdateBuilder.setStopId("DUMMY");
    tripUpdateBuilder.addStopTimeUpdate(stopTimeUpdateBuilder.build());
    feedEntityBuilder.setTripUpdate(tripUpdateBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData, gtfsDataMetadata, feedMessageBuilder.build(), null, null);
    // one error from TripUpdate feed stop_id = "DUMMY". VehiclePosition and Alert feeds have valid stop id = "A"
    expected.put(E011, 1);
    TestUtils.assertResults(expected, results);
    vehiclePositionBuilder.setStopId("DUMMY");
    feedEntityBuilder.setVehicle(vehiclePositionBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData, gtfsDataMetadata, feedMessageBuilder.build(), null, null);
    // 2 results from TripUpdate and VehiclePosition feeds stop_id="DUMMY". Alert feed have valid stop id ="A"
    expected.put(E011, 2);
    TestUtils.assertResults(expected, results);
    entitySelectorBuilder.setStopId("DUMMY");
    alertBuilder.addInformedEntity(entitySelectorBuilder.build());
    feedEntityBuilder.setAlert(alertBuilder.build());
    feedMessageBuilder.setEntity(0, feedEntityBuilder.build());
    results = locationValidator.validate(TimestampUtils.MIN_POSIX_TIME, gtfsData, gtfsDataMetadata, feedMessageBuilder.build(), null, null);
    // 3 results from TripUpdate, VehiclePosition and alert feeds stop_id="DUMMY"
    expected.put(E011, 3);
    TestUtils.assertResults(expected, results);
    clearAndInitRequiredFeedFields();
}