diff --git a/src/test/java/org/springframework/samples/petclinic/teamkogan/VisitControllerTest.java b/src/test/java/org/springframework/samples/petclinic/teamkogan/VisitControllerTest.java index 7a1c7b94e..45f39bbe5 100644 --- a/src/test/java/org/springframework/samples/petclinic/teamkogan/VisitControllerTest.java +++ b/src/test/java/org/springframework/samples/petclinic/teamkogan/VisitControllerTest.java @@ -17,22 +17,22 @@ public class VisitControllerTest @Test public void testPetVisitDate() { - // Define random pet ID - final Integer PET_ID = 7; - Pet pet7 = new Pet(); - - // Mock dependencies - VisitRepository mockVisitRepo = mock(VisitRepository.class); - PetRepository mockPetRepo = mock(PetRepository.class); - VisitController visitController = new VisitController(mockVisitRepo, mockPetRepo); - - // Define mock behaviour - when(mockPetRepo.findById(PET_ID)).thenReturn(pet7); - - // Call method under inspection - Visit visit = visitController.loadPetWithVisit(PET_ID, new HashMap<>()); - - // Confirm that the same visit date was assigned to the pet - assertEquals(visit.getDate(), pet7.getVisits().get(0).getDate()); + // Define random pet ID + final Integer PET_ID = 7; + Pet pet7 = new Pet(); + + // Mock dependencies + VisitRepository mockVisitRepo = mock(VisitRepository.class); + PetRepository mockPetRepo = mock(PetRepository.class); + VisitController visitController = new VisitController(mockVisitRepo, mockPetRepo); + + // Define mock behaviour + when(mockPetRepo.findById(PET_ID)).thenReturn(pet7); + + // Call method under inspection + Visit visit = visitController.loadPetWithVisit(PET_ID, new HashMap<>()); + + // Confirm that the same visit date was assigned to the pet + assertEquals(visit.getDate(), pet7.getVisits().get(0).getDate()); } }