Fixing Assessment Unit Tests

CDOT-CV · Jun 13, 2024 · 7c36b12 · 7c36b12
1 parent 859706e
commit 7c36b12
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jpo-conflictvisualizer-api/src/test/java/us/dot/its/jpo/ode/api/AssessmentTests.java b/jpo-conflictvisualizer-api/src/test/java/us/dot/its/jpo/ode/api/AssessmentTests.java
@@ -59,6 +59,7 @@ public void testLaneDirectionOfTravelAssessment() {
 
 
         Query query = laneDirectionOfTravelAssessmentRepo.getQuery(assessment.getIntersectionID(), assessment.getAssessmentGeneratedAt()-1, assessment.getAssessmentGeneratedAt() + 1, false);
+
         when(laneDirectionOfTravelAssessmentRepo.find(query)).thenReturn(assessments);
 
         ResponseEntity<List<LaneDirectionOfTravelAssessment>> result = controller.findLaneDirectionOfTravelAssessment(assessment.getRoadRegulatorID(), assessment.getIntersectionID(), assessment.getAssessmentGeneratedAt() - 1, assessment.getAssessmentGeneratedAt() + 1, false, false);
@@ -81,7 +82,7 @@ public void testConnectionOfTravelAssessment() {
         Query query = connectionOfTravelAssessmentRepo.getQuery(assessment.getIntersectionID(), assessment.getAssessmentGeneratedAt()-1, assessment.getAssessmentGeneratedAt() + 1, false);
         when(connectionOfTravelAssessmentRepo.find(query)).thenReturn(assessments);
 
-        ResponseEntity<List<ConnectionOfTravelAssessment>> result = controller.findConnectionOfTravelAssessment(assessment.getRoadRegulatorID(), assessment.getIntersectionID(), assessment.getAssessmentGeneratedAt() - 1, assessment.getAssessmentGeneratedAt() + 1, false, true, false);
+        ResponseEntity<List<ConnectionOfTravelAssessment>> result = controller.findConnectionOfTravelAssessment(assessment.getRoadRegulatorID(), assessment.getIntersectionID(), assessment.getAssessmentGeneratedAt() - 1, assessment.getAssessmentGeneratedAt() + 1, false, false);
         assertThat(result.getStatusCode()).isEqualTo(HttpStatus.OK);
         // assertThat(result.getHeaders().getContentType()).isEqualTo(MediaType.APPLICATION_JSON);
         assertThat(result.getBody()).isEqualTo(assessments);