From f8d1c6e23dd5b72efbaeedd2ba6ff621f54b1968 Mon Sep 17 00:00:00 2001 From: Auto_EPMD-EDP AIAssistant Date: Thu, 24 Oct 2024 18:58:38 +0300 Subject: [PATCH] Add unit test for loadPetWithVisit in VisitController --- .../petclinic/owner/OwnerControllerTest.java | 44 +++++++++++++++++++ 1 file changed, 44 insertions(+) diff --git a/src/test/java/org/springframework/samples/petclinic/owner/OwnerControllerTest.java b/src/test/java/org/springframework/samples/petclinic/owner/OwnerControllerTest.java index c89eb6443..869c2d08a 100644 --- a/src/test/java/org/springframework/samples/petclinic/owner/OwnerControllerTest.java +++ b/src/test/java/org/springframework/samples/petclinic/owner/OwnerControllerTest.java @@ -1,7 +1,9 @@ package org.springframework.samples.petclinic.owner; +import java.util.Map; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.DisplayName; +import java.util.Map; import org.springframework.ui.Model; import org.springframework.ui.ConcurrentModel; import org.springframework.samples.petclinic.owner.Owner; @@ -205,3 +207,45 @@ class OwnerControllerTest { } } + +@ExtendWith(MockitoExtension.class) +class VisitControllerTest { + + @Mock + private OwnerRepository ownerRepository; + + @InjectMocks + private VisitController visitController; + + @Mock + private MockMvc mockMvc; + + @BeforeEach + void setup() { + mockMvc = MockMvcBuilders.standaloneSetup(visitController).build(); + } + + @Test + @DisplayName("Test loadPetWithVisit without validation errors") + void testLoadPetWithVisit() { + int ownerId = 1; + int petId = 1; + Map model = new ConcurrentModel(); + + Owner owner = new Owner(); + Pet pet = new Pet(); + Visit visit = new Visit(); + + doReturn(owner).when(ownerRepository).findById(ownerId); + doReturn(pet).when(owner).getPet(petId); + + Visit result = visitController.loadPetWithVisit(ownerId, petId, model); + + assertThat(result).isNotNull(); + assertThat(model.get("pet")).isEqualTo(pet); + assertThat(model.get("owner")).isEqualTo(owner); + verify(ownerRepository).findById(ownerId); + verify(owner).getPet(petId); + } + +}