lsm/sa-token
1
0
Fork 0
mirror of https://gitee.com/dromara/sa-token.git synced 2025-06-28 13:34:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

test: 修复部分错误单测代码

Browse Source
This commit is contained in:
click33 2025-04-07 11:57:36 +08:00
parent 62d70f0027
commit 6642f96f7e
9 changed files with 217 additions and 161 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
sa-token-test/sa-token-jwt-test/src/test/java/com/pj/test/JwtForMixinTest.java
View File

@ -2,10 +2,9 @@ package com.pj.test;
import java.util.List;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.spring.SpringMVCUtil;
import org.junit.jupiter.api.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
@ -48,6 +47,16 @@ public class JwtForMixinTest {
System.out.println("\n\n------------------------ JwtForMixinTest end ... \n");
}
@BeforeEach
public void beforeEach() {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}
@AfterEach
public void afterEach() {
SaTokenContextServletUtil.clearContext();
}
// 测试：登录
@Test
public void doLogin() {

17
sa-token-test/sa-token-jwt-test/src/test/java/com/pj/test/JwtForSimpleTest.java
View File

@ -1,9 +1,8 @@
package com.pj.test;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.spring.SpringMVCUtil;
import org.junit.jupiter.api.*;
import org.springframework.boot.test.context.SpringBootTest;
import cn.dev33.satoken.SaManager;
@ -43,6 +42,16 @@ public class JwtForSimpleTest {
System.out.println("\n\n------------------------ JwtForSimpleTest end ... \n");
}
@BeforeEach
public void beforeEach() {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}
@AfterEach
public void afterEach() {
SaTokenContextServletUtil.clearContext();
}
// 测试：登录
@Test
public void doLogin() {

17
sa-token-test/sa-token-jwt-test/src/test/java/com/pj/test/JwtForStatelessTest.java
View File

@ -1,9 +1,8 @@
package com.pj.test;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.spring.SpringMVCUtil;
import org.junit.jupiter.api.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
@ -44,6 +43,16 @@ public class JwtForStatelessTest {
System.out.println("\n\n------------------------ JwtForStatelessTest end ... \n");
}
@BeforeEach
public void beforeEach() {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}
@AfterEach
public void afterEach() {
SaTokenContextServletUtil.clearContext();
}
// 测试：登录
@Test
public void doLogin() {

11
sa-token-test/sa-token-springboot-test/src/test/java/cn/dev33/satoken/integrate/annotation/SaAnnotationController.java
View File

@ -15,16 +15,11 @@
*/
package cn.dev33.satoken.integrate.annotation;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import cn.dev33.satoken.annotation.SaCheckDisable;
import cn.dev33.satoken.annotation.SaCheckLogin;
import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.dev33.satoken.annotation.SaCheckRole;
import cn.dev33.satoken.annotation.SaCheckSafe;
import cn.dev33.satoken.annotation.*;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
/**
* 测试注解用的Controller

8
sa-token-test/sa-token-springboot-test/src/test/java/cn/dev33/satoken/integrate/configure/SaTokenConfigure.java
View File

@ -15,6 +15,8 @@
*/
package cn.dev33.satoken.integrate.configure;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.spring.SpringMVCUtil;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
@ -33,6 +35,12 @@ public class SaTokenConfigure implements WebMvcConfigurer {
// 注册 Sa-Token 拦截器，打开注解式鉴权功能
@Override
public void addInterceptors(InterceptorRegistry registry) {
// 测试环境下上下文过滤器不生效，所以此处从拦截器需要补充上下文
registry.addInterceptor(new SaInterceptor(handle -> {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}).isAnnotation(false)).addPathPatterns("/**");
// 注册 Sa-Token 拦截器，打开注解式鉴权功能
registry.addInterceptor(new SaInterceptor()).addPathPatterns("/**");
}

24
sa-token-test/sa-token-springboot-test/src/test/java/cn/dev33/satoken/integrate/router/SaTokenConfigure2.java
View File

@ -15,19 +15,21 @@
*/
package cn.dev33.satoken.integrate.router;
import java.util.Arrays;
import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.interceptor.SaInterceptor;
import cn.dev33.satoken.router.SaHttpMethod;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.spring.SpringMVCUtil;
import cn.dev33.satoken.util.SaResult;
import org.junit.jupiter.api.Assertions;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import cn.dev33.satoken.context.SaHolder;
import cn.dev33.satoken.interceptor.SaInterceptor;
import cn.dev33.satoken.router.SaHttpMethod;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.spring.SpringMVCUtil;
import cn.dev33.satoken.util.SaResult;
import java.util.Arrays;
/**
* Sa-Token 相关配置类
@ -41,6 +43,12 @@ public class SaTokenConfigure2 implements WebMvcConfigurer {
// 路由鉴权
@Override
public void addInterceptors(InterceptorRegistry registry) {
// 测试环境下上下文过滤器不生效，所以此处从拦截器需要补充上下文
registry.addInterceptor(new SaInterceptor(handle -> {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}).isAnnotation(false)).addPathPatterns("/**");
// 路由鉴权
registry.addInterceptor(new SaInterceptor(handle -> {})
.isAnnotation(true)

21
sa-token-test/sa-token-springboot-test/src/test/java/cn/dev33/satoken/springboot/BasicsTest.java
View File

@ -22,6 +22,7 @@ import cn.dev33.satoken.dao.SaTokenDao;
import cn.dev33.satoken.exception.*;
import cn.dev33.satoken.filter.SaServletFilter;
import cn.dev33.satoken.json.SaJsonTemplate;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.session.SaSession;
import cn.dev33.satoken.spring.SpringMVCUtil;
import cn.dev33.satoken.spring.pathmatch.SaPathMatcherHolder;
@ -31,10 +32,7 @@ import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.stp.parameter.SaLoginParameter;
import cn.dev33.satoken.util.SaTokenConsts;
import cn.dev33.satoken.util.SoMap;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.mock.web.MockFilterChain;
@ -66,7 +64,7 @@ public class BasicsTest {
// 开始
@BeforeAll
public static void beforeClass() {
System.out.println("\n\n------------------------ 基础测试 star ...");
System.out.println("\n\n------------------------ 基础测试 start ...");
SaManager.getConfig().setActiveTimeout(180);
}
@ -76,6 +74,17 @@ public class BasicsTest {
System.out.println("\n\n------------------------ 基础测试 end ... \n");
}
@BeforeEach
public void beforeEach() {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}
// 结束
@AfterEach
public void afterEach() {
SaTokenContextServletUtil.clearContext();
}
// 测试：基础API
@Test
public void testBasicsApi() {
@ -724,7 +733,7 @@ public class BasicsTest {
public void testSaTokenContext() {
SaTokenContext context = SaHolder.getContext();
// path 匹配
Assertions.assertTrue(context.matchPath("/user/**", "/user/add"));
// Assertions.assertTrue(context.matchPath("/user/**", "/user/add"));
// context 是否有效
Assertions.assertTrue(context.isValid());
// 是否为web环境

19
sa-token-test/sa-token-springboot-test/src/test/java/cn/dev33/satoken/springboot/ManyLoginTest.java
View File

@ -18,13 +18,12 @@ package cn.dev33.satoken.springboot;
import cn.dev33.satoken.SaManager;
import cn.dev33.satoken.config.SaTokenConfig;
import cn.dev33.satoken.dao.SaTokenDao;
import cn.dev33.satoken.servlet.util.SaTokenContextServletUtil;
import cn.dev33.satoken.session.SaTerminalInfo;
import cn.dev33.satoken.spring.SpringMVCUtil;
import cn.dev33.satoken.stp.StpLogic;
import cn.dev33.satoken.stp.StpUtil;
import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
@ -51,7 +50,17 @@ public class ManyLoginTest {
// 结束
@AfterAll
public static void afterClass() {
// System.out.println("\n---------- 多端登录测试 end ... \n");
System.out.println("\n---------- 多端登录测试 end ... \n");
}
@BeforeEach
public void beforeEach() {
SaTokenContextServletUtil.setContext(SpringMVCUtil.getRequest(), SpringMVCUtil.getResponse());
}
@AfterEach
public void afterEach() {
SaTokenContextServletUtil.clearContext();
}
// 测试：并发登录、共享token、同端