-
Notifications
You must be signed in to change notification settings - Fork 115
feat: Add shapes.txt validator #2123
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
cswilson252
wants to merge
39
commits into
MobilityData:master
Choose a base branch
from
cswilson252:noShapesNoDrt
base: master
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Open
Changes from all commits
Commits
Show all changes
39 commits
Select commit
Hold shift + click to select a range
0fcaf9e
push for diff
cswilson252 ee49572
add to own file
cswilson252 ffbcca5
add validator functionality
cswilson252 d078d3b
half baked test +some lint
cswilson252 abadac7
push up actual notice tests
cswilson252 5caa4e8
fix to asserttrue?
cswilson252 e457ddc
assertThat?
cswilson252 4ca9741
make the non-errors assertFalse
cswilson252 13e3475
assertThat
cswilson252 32a2cfd
add import
cswilson252 4d5f495
other assertThat import
cswilson252 95a83c6
get rid of assertFalse
cswilson252 337b917
isFalse
cswilson252 0e93ab1
Merge branch 'master' into noShapesNoDrt
cswilson252 6923baf
Merge branch 'master' into noShapesNoDrt
cswilson252 a1c4524
copilot feedback
cswilson252 19a3143
missing import
cswilson252 f5e6311
comparisonfailure
cswilson252 d5196e4
we have to pass another arg, luckily its nullable
cswilson252 98c14d0
remove 1f check in favour of 1
cswilson252 ce9c224
isAtLeast assertation check
cswilson252 2447088
not 1l?
cswilson252 528b409
fix semicolon
cswilson252 8329286
Merge branch 'master' into noShapesNoDrt
cswilson252 dfe1600
correct tests and validator loop logic
cswilson252 22b978b
return, don't break
cswilson252 69af53e
lint
cswilson252 04c0638
fully remove arguments
cswilson252 3763437
0 rows
cswilson252 6962ced
-1 rows
cswilson252 516dab9
break if rows are -1
cswilson252 b86fa31
try returning
cswilson252 d7de502
[Skip CI] push method signature WIP
cswilson252 e4e72a8
rely on FeedMetadata in validator method
cswilson252 7d7564d
stringify filename
cswilson252 86e77ef
pass single feedContainer
cswilson252 ccaedfd
lint
cswilson252 52b330e
add createFeedContainer()
cswilson252 049e820
lint
cswilson252 File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
51 changes: 51 additions & 0 deletions
51
main/src/main/java/org/mobilitydata/gtfsvalidator/validator/MissingShapesFileValidator.java
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,51 @@ | ||
| package org.mobilitydata.gtfsvalidator.validator; | ||
|
|
||
| import javax.inject.Inject; | ||
| import org.mobilitydata.gtfsvalidator.annotation.GtfsValidator; | ||
| import org.mobilitydata.gtfsvalidator.notice.MissingRecommendedFileNotice; | ||
| import org.mobilitydata.gtfsvalidator.notice.NoticeContainer; | ||
| import org.mobilitydata.gtfsvalidator.reportsummary.model.FeedMetadata; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsFeedContainer; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsLocationGroupsTableContainer; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsShapeTableContainer; | ||
|
|
||
| /** | ||
| * Validates that the feed has either a `shapes.txt` file, or uses zone-based DRT or fixed-stops | ||
| * DRT. | ||
| * | ||
| * <p>Generated notice: {@link MissingRecommendedFileNotice}. | ||
| */ | ||
| @GtfsValidator | ||
| public class MissingShapesFileValidator extends FileValidator { | ||
| private final GtfsShapeTableContainer shapeTable; | ||
| private final GtfsLocationGroupsTableContainer locationGroups; | ||
| private final GtfsFeedContainer feedContainer; | ||
|
|
||
| @Inject | ||
| MissingShapesFileValidator( | ||
| GtfsShapeTableContainer shapeTable, | ||
| GtfsLocationGroupsTableContainer locationGroups, | ||
| GtfsFeedContainer feedContainer) { | ||
| this.shapeTable = shapeTable; | ||
| this.locationGroups = locationGroups; | ||
| this.feedContainer = feedContainer; | ||
| } | ||
|
|
||
| @Override | ||
| public void validate(NoticeContainer noticeContainer) { | ||
|
|
||
| Boolean missingShapes = shapeTable == null || shapeTable.isMissingFile(); | ||
| boolean hasZoneBasedDrt = FeedMetadata.hasAtLeastOneTripWithOnlyLocationId(feedContainer); | ||
| boolean hasFixedStopsDrt = | ||
| FeedMetadata.hasAtLeastOneRecordInFile(feedContainer, "location_groups.txt") | ||
| && FeedMetadata.hasAtLeastOneTripWithOnlyLocationGroupId(feedContainer); | ||
|
|
||
| // Do we NOT have: a shapes.txt file and the required fields for Zone-Based DRT, | ||
| // and also the required fields for Fixed-Stop DRT? | ||
| if (missingShapes && !hasZoneBasedDrt && !hasFixedStopsDrt) { | ||
| noticeContainer.addValidationNotice(new MissingRecommendedFileNotice("shapes.txt")); | ||
| // This is a feed-level warning; emit it at most once. | ||
| return; | ||
| } | ||
| } | ||
| } | ||
104 changes: 104 additions & 0 deletions
104
...rc/test/java/org/mobilitydata/gtfsvalidator/validator/MissingShapesFileValidatorTest.java
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,104 @@ | ||
| package org.mobilitydata.gtfsvalidator.validator; | ||
|
|
||
| import static com.google.common.truth.Truth.assertThat; | ||
|
|
||
| import com.google.common.collect.ImmutableList; | ||
| import java.util.ArrayList; | ||
| import java.util.List; | ||
| import org.junit.Test; | ||
| import org.mobilitydata.gtfsvalidator.notice.MissingRecommendedFileNotice; | ||
| import org.mobilitydata.gtfsvalidator.notice.NoticeContainer; | ||
| import org.mobilitydata.gtfsvalidator.notice.ValidationNotice; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsFeedContainer; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsLocationGroups; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsLocationGroupsTableContainer; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsShape; | ||
| import org.mobilitydata.gtfsvalidator.table.GtfsShapeTableContainer; | ||
|
|
||
| public class MissingShapesFileValidatorTest { | ||
|
|
||
| private static GtfsFeedContainer createFeedContainer( | ||
| List<GtfsShape> shapes, List<GtfsLocationGroups> locationGroups) { | ||
| NoticeContainer noticeContainer = new NoticeContainer(); | ||
| return new GtfsFeedContainer( | ||
| ImmutableList.of( | ||
| GtfsShapeTableContainer.forEntities(shapes, noticeContainer), | ||
| GtfsLocationGroupsTableContainer.forEntities(locationGroups, noticeContainer))); | ||
| } | ||
|
|
||
| private static List<GtfsShape> createShapeTable(int rows) { | ||
| ArrayList<GtfsShape> shapes = new ArrayList<>(); | ||
| for (int i = 0; i < rows; i++) { | ||
| if (rows == -1) { | ||
| return null; | ||
| } | ||
| shapes.add(new GtfsShape.Builder().setCsvRowNumber(i + 1).setShapeId("s" + i).build()); | ||
| } | ||
| return shapes; | ||
| } | ||
|
|
||
| private static List<GtfsLocationGroups> createLocationGroupsTable( | ||
| int rows, String groupId, String groupName) { | ||
| ArrayList<GtfsLocationGroups> locationGroups = new ArrayList<>(); | ||
| for (int i = 0; i < rows; i++) { | ||
| locationGroups.add( | ||
| new GtfsLocationGroups.Builder() | ||
| .setCsvRowNumber(i + 1) | ||
| .setLocationGroupId(groupId) | ||
| .setLocationGroupName(groupName) | ||
| .build()); | ||
| } | ||
| return locationGroups; | ||
| } | ||
|
|
||
| @Test | ||
| public void testShapesFileAndFixedDrtPresent() { | ||
| List<ValidationNotice> notices = | ||
| generateNotices( | ||
| createShapeTable(1), | ||
| createLocationGroupsTable(1, "b", "testgroup"), | ||
| createFeedContainer( | ||
| createShapeTable(1), createLocationGroupsTable(1, "b", "testgroup"))); | ||
| boolean found = | ||
| notices.stream().anyMatch(notice -> notice instanceof MissingRecommendedFileNotice); | ||
| assertThat(found).isFalse(); | ||
| } | ||
|
|
||
| @Test | ||
| public void testShapesFileAndZoneBasedDrtPresent() { | ||
| List<ValidationNotice> notices = | ||
| generateNotices( | ||
| createShapeTable(1), | ||
| createLocationGroupsTable(1, "d", "t3stgroup"), | ||
| createFeedContainer( | ||
| createShapeTable(1), createLocationGroupsTable(1, "d", "t3stgroup"))); | ||
| boolean found = | ||
| notices.stream().anyMatch(notice -> notice instanceof MissingRecommendedFileNotice); | ||
| assertThat(found).isFalse(); | ||
| } | ||
|
|
||
| @Test | ||
| public void testNoShapesFileAndNoDrtPresent() { | ||
| List<ValidationNotice> notices = | ||
| generateNotices( | ||
| createShapeTable(-1), | ||
| createLocationGroupsTable(0, null, null), | ||
| createFeedContainer(createShapeTable(-1), createLocationGroupsTable(0, null, null))); | ||
| long missingRecommendedFileNoticesCount = | ||
| notices.stream().filter(notice -> notice instanceof MissingRecommendedFileNotice).count(); | ||
| assertThat(missingRecommendedFileNoticesCount).isAtLeast(1); | ||
|
Contributor
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. There is now an assertion error here :( probably related to my implementation of |
||
| } | ||
|
|
||
| private static List<ValidationNotice> generateNotices( | ||
|
cswilson252 marked this conversation as resolved.
|
||
| List<GtfsShape> shapes, | ||
| List<GtfsLocationGroups> locationGroups, | ||
| GtfsFeedContainer feedContainer) { | ||
| NoticeContainer noticeContainer = new NoticeContainer(); | ||
| new MissingShapesFileValidator( | ||
| GtfsShapeTableContainer.forEntities(shapes, noticeContainer), | ||
| GtfsLocationGroupsTableContainer.forEntities(locationGroups, noticeContainer), | ||
| feedContainer) | ||
| .validate(noticeContainer); | ||
| return noticeContainer.getValidationNotices(); | ||
| } | ||
| } | ||
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Uh oh!
There was an error while loading. Please reload this page.