|
|
@@ -11,7 +11,6 @@ import org.springframework.boot.test.context.SpringBootTest;
|
|
|
import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
|
|
|
import javax.annotation.Resource;
|
|
|
-import javax.servlet.http.HttpServletRequest;
|
|
|
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
import static org.mockito.ArgumentMatchers.*;
|
|
|
@@ -306,15 +305,14 @@ public class AuthControllerTest {
|
|
|
|
|
|
@Test
|
|
|
public void getInfo_Success() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(1L);
|
|
|
+ Long userId = 1L;
|
|
|
|
|
|
User mockUser = new User();
|
|
|
mockUser.setId(1L);
|
|
|
mockUser.setNickname("测试用户");
|
|
|
when(userService.getUserInfo(1L)).thenReturn(mockUser);
|
|
|
|
|
|
- Result<User> result = controller.getInfo(request);
|
|
|
+ Result<User> result = controller.getInfo(userId);
|
|
|
assertEquals(200, result.getCode());
|
|
|
assertNotNull(result.getData());
|
|
|
// Password should be cleared
|
|
|
@@ -323,11 +321,9 @@ public class AuthControllerTest {
|
|
|
|
|
|
@Test
|
|
|
public void getInfo_NotLoggedIn() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(null);
|
|
|
- when(request.getHeader("X-User-Id")).thenReturn(null);
|
|
|
+ Long userId = null;
|
|
|
|
|
|
- Result<User> result = controller.getInfo(request);
|
|
|
+ Result<User> result = controller.getInfo(userId);
|
|
|
assertEquals(500, result.getCode());
|
|
|
assertEquals("未登录", result.getMessage());
|
|
|
}
|
|
|
@@ -336,15 +332,14 @@ public class AuthControllerTest {
|
|
|
|
|
|
@Test
|
|
|
public void setPassword_Success() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(1L);
|
|
|
+ Long userId = 1L;
|
|
|
|
|
|
SetPasswordDTO dto = new SetPasswordDTO();
|
|
|
dto.setPassword("newPass123");
|
|
|
|
|
|
when(userService.setPassword(1L, "newPass123")).thenReturn(true);
|
|
|
|
|
|
- Result<Boolean> result = controller.setPassword(request, dto);
|
|
|
+ Result<Boolean> result = controller.setPassword(dto, userId);
|
|
|
assertEquals(200, result.getCode());
|
|
|
assertTrue(result.getData());
|
|
|
}
|
|
|
@@ -353,29 +348,27 @@ public class AuthControllerTest {
|
|
|
|
|
|
@Test
|
|
|
public void verifyPassword_Correct() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(1L);
|
|
|
+ Long userId = 1L;
|
|
|
|
|
|
VerifyPasswordDTO dto = new VerifyPasswordDTO();
|
|
|
dto.setPassword("correctPass");
|
|
|
|
|
|
when(userService.verifyPassword(1L, "correctPass")).thenReturn(true);
|
|
|
|
|
|
- Result<Boolean> result = controller.verifyPassword(request, dto);
|
|
|
+ Result<Boolean> result = controller.verifyPassword(dto, userId);
|
|
|
assertTrue(result.getData());
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void verifyPassword_Wrong() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(1L);
|
|
|
+ Long userId = 1L;
|
|
|
|
|
|
VerifyPasswordDTO dto = new VerifyPasswordDTO();
|
|
|
dto.setPassword("wrongPass");
|
|
|
|
|
|
when(userService.verifyPassword(1L, "wrongPass")).thenReturn(false);
|
|
|
|
|
|
- Result<Boolean> result = controller.verifyPassword(request, dto);
|
|
|
+ Result<Boolean> result = controller.verifyPassword(dto, userId);
|
|
|
assertFalse(result.getData());
|
|
|
}
|
|
|
|
|
|
@@ -541,8 +534,7 @@ public class AuthControllerTest {
|
|
|
|
|
|
@Test
|
|
|
public void switchRole_Success() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(1L);
|
|
|
+ Long userId = 1L;
|
|
|
|
|
|
SwitchRoleDTO dto = new SwitchRoleDTO();
|
|
|
dto.setRole("teacher");
|
|
|
@@ -552,21 +544,19 @@ public class AuthControllerTest {
|
|
|
loginResult.setRole("teacher");
|
|
|
when(userService.switchRoleForLogin(1L, "teacher")).thenReturn(loginResult);
|
|
|
|
|
|
- Result<LoginResultDTO> result = controller.switchRole(request, dto);
|
|
|
+ Result<LoginResultDTO> result = controller.switchRole(dto, userId);
|
|
|
assertEquals(200, result.getCode());
|
|
|
assertEquals("teacher", result.getData().getRole());
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void switchRole_NotLoggedIn() {
|
|
|
- HttpServletRequest request = mock(HttpServletRequest.class);
|
|
|
- when(request.getAttribute("userId")).thenReturn(null);
|
|
|
- when(request.getHeader("X-User-Id")).thenReturn(null);
|
|
|
+ Long userId = null;
|
|
|
|
|
|
SwitchRoleDTO dto = new SwitchRoleDTO();
|
|
|
dto.setRole("teacher");
|
|
|
|
|
|
- Result<LoginResultDTO> result = controller.switchRole(request, dto);
|
|
|
+ Result<LoginResultDTO> result = controller.switchRole(dto, userId);
|
|
|
assertEquals(500, result.getCode());
|
|
|
assertEquals("未登录", result.getMessage());
|
|
|
}
|