11package org .opentripplanner .gtfs .mapping ;
22
3+ import static com .google .common .truth .Truth .assertThat ;
34import static org .junit .jupiter .api .Assertions .assertEquals ;
4- import static org .junit .jupiter .api .Assertions .assertTrue ;
55import static org .onebusaway .gtfs .model .AgencyAndIdFactory .obaId ;
66
77import java .util .List ;
@@ -111,13 +111,14 @@ void mapNoticeAssignmentOnTrip() {
111111 void missingNoticeLogsIssue () {
112112 var issues = new DefaultDataImportIssueStore ();
113113 var noticeMapper = new NoticeMapper (ID_FACTORY );
114+ var routeMapper = createRouteMapper ();
114115
115116 var mapper = new NoticeAssignmentMapper (
116117 ID_FACTORY ,
117118 issues ,
118119 noticeMapper ,
119- createTripMapper (createRouteMapper () ),
120- createRouteMapper ()
120+ createTripMapper (routeMapper ),
121+ routeMapper
121122 );
122123
123124 var assignment = new NoticeAssignment ();
@@ -127,10 +128,9 @@ void missingNoticeLogsIssue() {
127128
128129 var result = mapper .map (List .of (assignment ));
129130
130- assertTrue (result .isEmpty ());
131- assertEquals (
132- List .of ("NoticeAssignmentWithoutNotice" ),
133- issues .listIssues ().stream ().map (DataImportIssue ::getType ).toList ()
131+ assertThat (result ).isEmpty ();
132+ assertThat (issues .listIssues ().stream ().map (DataImportIssue ::getType )).containsExactly (
133+ "NoticeAssignmentWithoutNotice"
134134 );
135135 }
136136
@@ -139,13 +139,14 @@ void missingRouteEntityLogsIssue() {
139139 var issues = new DefaultDataImportIssueStore ();
140140 var noticeMapper = new NoticeMapper (ID_FACTORY );
141141 noticeMapper .map (GTFS_NOTICE );
142+ var routeMapper = createRouteMapper ();
142143
143144 var mapper = new NoticeAssignmentMapper (
144145 ID_FACTORY ,
145146 issues ,
146147 noticeMapper ,
147- createTripMapper (createRouteMapper () ),
148- createRouteMapper ()
148+ createTripMapper (routeMapper ),
149+ routeMapper
149150 );
150151
151152 var assignment = new NoticeAssignment ();
@@ -155,10 +156,9 @@ void missingRouteEntityLogsIssue() {
155156
156157 var result = mapper .map (List .of (assignment ));
157158
158- assertTrue (result .isEmpty ());
159- assertEquals (
160- List .of ("NoticeAssignmentWithUnknownEntity" ),
161- issues .listIssues ().stream ().map (DataImportIssue ::getType ).toList ()
159+ assertThat (result ).isEmpty ();
160+ assertThat (issues .listIssues ().stream ().map (DataImportIssue ::getType ).toList ()).containsExactly (
161+ "NoticeAssignmentWithUnknownEntity"
162162 );
163163 }
164164
@@ -167,13 +167,14 @@ void missingTripEntityLogsIssue() {
167167 var issues = new DefaultDataImportIssueStore ();
168168 var noticeMapper = new NoticeMapper (ID_FACTORY );
169169 noticeMapper .map (GTFS_NOTICE );
170+ var routeMapper = createRouteMapper ();
170171
171172 var mapper = new NoticeAssignmentMapper (
172173 ID_FACTORY ,
173174 issues ,
174175 noticeMapper ,
175- createTripMapper (createRouteMapper () ),
176- createRouteMapper ()
176+ createTripMapper (routeMapper ),
177+ routeMapper
177178 );
178179
179180 var assignment = new NoticeAssignment ();
@@ -183,10 +184,9 @@ void missingTripEntityLogsIssue() {
183184
184185 var result = mapper .map (List .of (assignment ));
185186
186- assertTrue (result .isEmpty ());
187- assertEquals (
188- List .of ("NoticeAssignmentWithUnknownEntity" ),
189- issues .listIssues ().stream ().map (DataImportIssue ::getType ).toList ()
187+ assertThat (result ).isEmpty ();
188+ assertThat (issues .listIssues ().stream ().map (DataImportIssue ::getType ).toList ()).containsExactly (
189+ "NoticeAssignmentWithUnknownEntity"
190190 );
191191 }
192192
0 commit comments