Mockito 使用不同的集合参数模拟相同的方法调用

Question

我尝试使用不同的集合参数模拟相同的方法调用。

我的问题是我没有从 Mocked-Call 中获得正确的模拟答案作为输入。

测试类：

@ExtendWith(SpringExtension.class)
public class CollectionTest {

    @MockBean
    private Controller c;
    
    @BeforeEach
    public void init() {
        Collection<String> a = Mockito.anyCollection();
        a.add("a");
        Mockito.when(c.run(a)).thenReturn("a");
        
        Collection<String> b = Mockito.anyCollection();
        b.add("b");
        Mockito.when(c.run(b)).thenReturn("b");
    }

    @Test
    public void test() {
        assertEquals("a", c.run(Lists.newArrayList("a"))); // DOESNT'WORK!!! Returns "b" but should "a"
        assertEquals("b", c.run(Lists.newArrayList("b"))); // 
    }
}

控制器类：

@Service
public class Controller{
    public String run(Collection<String> c) {
        return "not-mocked";
    }   
}

我不知道为什么它不返回“a”。我试图将集合更改为字符串，但行为相同。

要执行哪些步骤才能获得以下行为？

@Test
public void test() {
    assertEquals("a", c.run(Lists.newArrayList("a"))); // should return "a"
    assertEquals("b", c.run(Lists.newArrayList("b"))); // should return "b"
}

我正在使用 Java Mockito "3.1" 和 Spring，但我认为 Mockito 是这里的重要信息。

Answer 1

你的第二个电话 - Mockito.when(c.run(b)).thenReturn("b"); 否决了我们的第一次调用，因此 Mockito 将始终返回“b”。 如果您需要来自同一呼叫的多个答案，则可以使用 varags 变体：

when(c.run(anyCollection())).thenReturn("a", "b");

现在对控制器的 run 方法的第一次调用将返回“a”，所有后续调用都将返回“b”。您可以根据需要提供任意数量的返回结果，并且从那时起将重复最后一个作为答案。

Answer 2

编写两个测试将显示您所期望的结果。 您正在向同一个控制器添加两个不同的结果，所以您只得到最后一个：Mockito.when(c.run(b)).thenReturn("b"); 普通的。您的 setUp() 中最后一个模拟的预期结果将保留在内存中。

之前的回答是： 您可以使用 junit 和 mockito 之类的东西来测试您的 spring-web-mvc 应用程序。 看起来是这样的：

@WebMvcTest(controllers = UserController.class)
@ActiveProfiles("test")
class UserControllerTest {
   
    @Autowired                           
    private MockMvc mockMvc;  
                                                 
    @MockBean                           
    private UserService userService; 
                                               
    private List<User> userList;       
                                            
    @BeforeEach                           
    void setUp() {                               
       this.userList = new ArrayList<>();
       this.userList.add(new User(1L, "user1@gmail.com", "pwd1","User1"));
       this.userList.add(new User(2L, "user2@gmail.com", "pwd2","User2"));
       this.userList.add(new User(3L, "user3@gmail.com", "pwd3","User3"));                                                       
    }
}

举个例子：

@Test
void shouldFetchAllUsers() throws Exception {
    given(userService.findAllUsers()).willReturn(userList);
    this.mockMvc.perform(get("/api/users"))
        .andExpect(status().isOk())
        .andExpect(jsonPath("$.size()", is(userList.size() )));
}

来自@see https://medium.com/backend-habit/integrate-junit-and-mockito-unit-testing-for-controller-layer-91bb4099c2a5的示例