add junit test

Author: xlorne

springboot-starter-security-jwt/src/test/java/com/codingapi/springboot/security/SecurityJwtApplication.java

@@ -1,12 +1,41 @@
 package com.codingapi.springboot.security;
 
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Bean;
+import org.springframework.security.core.GrantedAuthority;
+import org.springframework.security.core.authority.SimpleGrantedAuthority;
+import org.springframework.security.core.userdetails.User;
+import org.springframework.security.core.userdetails.UserDetailsService;
+import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
+import org.springframework.security.crypto.password.PasswordEncoder;
+import org.springframework.security.provisioning.InMemoryUserDetailsManager;
 
+import java.util.ArrayList;
+import java.util.List;
+
+@Slf4j
 @SpringBootApplication
 public class SecurityJwtApplication {
 
     public static void main(String[] args) {
         SpringApplication.run(SecurityJwtApplication.class, args);
     }
+
+    @Bean
+    public UserDetailsService myUserDetailsService(PasswordEncoder passwordEncoder) {
+        InMemoryUserDetailsManager inMemoryUserDetailsManager = new InMemoryUserDetailsManager();
+        List<GrantedAuthority> authorities = new ArrayList<>();
+        authorities.add(new SimpleGrantedAuthority("ADMIN"));
+        inMemoryUserDetailsManager.createUser(new User("admin", passwordEncoder.encode("123456"),authorities));
+        return inMemoryUserDetailsManager;
+    }
+
+
+    @Bean
+    public PasswordEncoder passwordEncoder() {
+        return new BCryptPasswordEncoder();
+    }
+
 }

springboot-starter-security-jwt/src/test/java/com/codingapi/springboot/security/SecurityJwtApplicationTest.java

@@ -1,15 +1,61 @@
 package com.codingapi.springboot.security;
 
+import com.alibaba.fastjson.JSONObject;
+import lombok.extern.slf4j.Slf4j;
 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.context.SpringBootTest;
+import org.springframework.http.MediaType;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.MvcResult;
 
+import java.nio.charset.StandardCharsets;
+
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+import static org.junit.jupiter.api.Assertions.*;
+
+@AutoConfigureMockMvc
 @SpringBootTest
+@Slf4j
 public class SecurityJwtApplicationTest {
 
+    @Autowired
+    private MockMvc mockMvc;
+
+    @Test
+    void login() throws Exception {
+        JSONObject json = new JSONObject();
+        json.put("username","admin");
+        json.put("password","123456");
+        mockMvc.perform(post("/user/login").content(json.toJSONString().getBytes(StandardCharsets.UTF_8)).contentType(MediaType.APPLICATION_JSON)).andExpect(status().isOk());
+    }
 
     @Test
-    void test() {
+    void noToken() throws Exception {
+        mockMvc.perform(get("/api/hello").contentType(MediaType.APPLICATION_JSON)).andExpect(result -> {
+            String body = result.getResponse().getContentAsString();
+            JSONObject jsonObject = JSONObject.parseObject(body);
+            log.info("body:{}",jsonObject);
+            assertEquals(jsonObject.getString("errCode"),"token.error","token authentication error");
+        });
+    }
+
+    @Test
+    void haveToken() throws Exception {
+
+        JSONObject json = new JSONObject();
+        json.put("username","admin");
+        json.put("password","123456");
+        MvcResult mvcResult =  mockMvc.perform(post("/user/login").content(json.toJSONString().getBytes(StandardCharsets.UTF_8)).contentType(MediaType.APPLICATION_JSON)).andReturn();
+        JSONObject loginData = JSONObject.parseObject(mvcResult.getResponse().getContentAsString());
 
+        mockMvc.perform(get("/api/hello").header("Authorization",loginData.getJSONObject("data").getString("token")).contentType(MediaType.APPLICATION_JSON)).andExpect(result -> {
+            String body = result.getResponse().getContentAsString();
+            assertEquals(body,"hello","token authentication error");
+        });
     }
 
 }

springboot-starter-security-jwt/src/test/java/com/codingapi/springboot/security/controller/DemoController.java

@@ -0,0 +1,15 @@
+package com.codingapi.springboot.security.controller;
+
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RestController
+@RequestMapping("/api")
+public class DemoController {
+
+    @GetMapping("/hello")
+    public String hello(){
+        return "hello";
+    }
+}