使用Mockito和Spring MVC的Junit测试用例

@RunWith(MockitoJUnitRunner.class)
public class UserControllerTest {

    @InjectMocks
    private UserController userController;

    @Mock
    private UserService userService;

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
    }

    @Test
    public void testGetUserById() {
        // 模拟userService的行为
        User mockUser = new User("1", "John");
        Mockito.when(userService.getUserById("1")).thenReturn(mockUser);

        // 调用控制器方法
        ResponseEntity<User> response = userController.getUserById("1");

        // 验证返回结果
        Assert.assertEquals(HttpStatus.OK, response.getStatusCode());
        Assert.assertEquals(mockUser, response.getBody());
    }

    @Test
    public void testCreateUser() {
        // 模拟userService的行为
        User mockUser = new User("1", "John");
        Mockito.when(userService.createUser(Mockito.any(User.class))).thenReturn(mockUser);

        // 调用控制器方法
        ResponseEntity<User> response = userController.createUser(mockUser);

        // 验证返回结果
        Assert.assertEquals(HttpStatus.CREATED, response.getStatusCode());
        Assert.assertEquals(mockUser, response.getBody());
    }
}