mirror of
https://github.com/nagisa77/OpenIsle.git
synced 2026-02-18 21:10:57 +08:00
69 lines
2.6 KiB
Java
69 lines
2.6 KiB
Java
package com.openisle.controller;
|
|
|
|
import org.junit.jupiter.api.Test;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
|
|
import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
|
|
import org.springframework.context.annotation.Import;
|
|
import org.springframework.test.web.servlet.MockMvc;
|
|
import org.springframework.boot.test.mock.mockito.MockBean;
|
|
import com.openisle.config.CustomAccessDeniedHandler;
|
|
import com.openisle.config.SecurityConfig;
|
|
import com.openisle.service.JwtService;
|
|
import com.openisle.repository.UserRepository;
|
|
import com.openisle.service.UserVisitService;
|
|
import com.openisle.model.Role;
|
|
import com.openisle.model.User;
|
|
import java.util.Optional;
|
|
import org.mockito.Mockito;
|
|
|
|
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
|
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
|
|
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
|
|
|
|
@WebMvcTest(HelloController.class)
|
|
@AutoConfigureMockMvc
|
|
@Import({SecurityConfig.class, CustomAccessDeniedHandler.class})
|
|
class HelloControllerTest {
|
|
@Autowired
|
|
private MockMvc mockMvc;
|
|
|
|
@MockBean
|
|
private JwtService jwtService;
|
|
@MockBean
|
|
private UserRepository userRepository;
|
|
@MockBean
|
|
private UserVisitService userVisitService;
|
|
|
|
@Test
|
|
void helloReturnsMessage() throws Exception {
|
|
Mockito.when(jwtService.validateAndGetSubject("token")).thenReturn("user");
|
|
User user = new User();
|
|
user.setUsername("user");
|
|
user.setPassword("p");
|
|
user.setEmail("u@example.com");
|
|
user.setRole(Role.USER);
|
|
Mockito.when(userRepository.findByUsername("user")).thenReturn(Optional.of(user));
|
|
|
|
mockMvc.perform(get("/api/hello").header("Authorization", "Bearer token"))
|
|
.andExpect(status().isOk())
|
|
.andExpect(jsonPath("$.message").value("Hello, Authenticated User"));
|
|
}
|
|
|
|
@Test
|
|
void helloMissingToken() throws Exception {
|
|
mockMvc.perform(get("/api/hello"))
|
|
.andExpect(status().isUnauthorized())
|
|
.andExpect(jsonPath("$.error").value("Missing token"));
|
|
}
|
|
|
|
@Test
|
|
void helloInvalidToken() throws Exception {
|
|
Mockito.when(jwtService.validateAndGetSubject("bad")).thenThrow(new RuntimeException());
|
|
|
|
mockMvc.perform(get("/api/hello").header("Authorization", "Bearer bad"))
|
|
.andExpect(status().isUnauthorized())
|
|
.andExpect(jsonPath("$.error").value("Invalid or expired token"));
|
|
}
|
|
}
|