From 506639c5d8cef6187e1b2eba8a846427c1cc1796 Mon Sep 17 00:00:00 2001 From: Amanda Date: Sat, 24 Feb 2018 23:40:52 -0500 Subject: [PATCH] Addition to VisitController tests --- .../samples/petclinic/owner/VisitControllerTests.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/test/java/org/springframework/samples/petclinic/owner/VisitControllerTests.java b/src/test/java/org/springframework/samples/petclinic/owner/VisitControllerTests.java index 08d61360e..ff02bce95 100644 --- a/src/test/java/org/springframework/samples/petclinic/owner/VisitControllerTests.java +++ b/src/test/java/org/springframework/samples/petclinic/owner/VisitControllerTests.java @@ -16,10 +16,15 @@ import org.springframework.boot.test.mock.mockito.MockBean; import org.springframework.samples.petclinic.owner.Pet; import org.springframework.samples.petclinic.owner.PetRepository; import org.springframework.samples.petclinic.owner.VisitController; +import org.springframework.samples.petclinic.visit.Visit; import org.springframework.samples.petclinic.visit.VisitRepository; import org.springframework.test.context.junit4.SpringRunner; import org.springframework.test.web.servlet.MockMvc; +import static org.mockito.Mockito.verify; +import static org.mockito.Matchers.any; +import static org.mockito.Matchers.anyInt; + /** * Test class for {@link VisitController} * @@ -50,6 +55,7 @@ public class VisitControllerTests { mockMvc.perform(get("/owners/*/pets/{petId}/visits/new", TEST_PET_ID)) .andExpect(status().isOk()) .andExpect(view().name("pets/createOrUpdateVisitForm")); + verify(pets).findById(anyInt()); //Ensure that a pet gets loaded per visit } @Test @@ -60,6 +66,8 @@ public class VisitControllerTests { ) .andExpect(status().is3xxRedirection()) .andExpect(view().name("redirect:/owners/{ownerId}")); + verify(pets).findById(anyInt()); //Ensure that a pet gets loaded per visit + verify(visits).save(any(Visit.class)); //Ensure that upon success, the visit gets saved } @Test @@ -70,6 +78,6 @@ public class VisitControllerTests { .andExpect(model().attributeHasErrors("visit")) .andExpect(status().isOk()) .andExpect(view().name("pets/createOrUpdateVisitForm")); + verify(pets).findById(anyInt()); //Ensure that a pet gets loaded per visit } - }