mirror of
https://gitee.com/dromara/sa-token.git
synced 2025-09-19 10:08:07 +08:00
重构 sa-token-jwt 模块
This commit is contained in:
@@ -14,21 +14,22 @@ import cn.dev33.satoken.dao.SaTokenDao;
|
||||
import cn.dev33.satoken.exception.ApiDisabledException;
|
||||
import cn.dev33.satoken.exception.DisableLoginException;
|
||||
import cn.dev33.satoken.jwt.SaJwtUtil;
|
||||
import cn.dev33.satoken.jwt.StpLogicJwtForMix;
|
||||
import cn.dev33.satoken.jwt.StpLogicJwtForMixin;
|
||||
import cn.dev33.satoken.session.SaSession;
|
||||
import cn.dev33.satoken.stp.SaLoginConfig;
|
||||
import cn.dev33.satoken.stp.StpUtil;
|
||||
import cn.dev33.satoken.util.SaTokenConsts;
|
||||
import cn.hutool.json.JSONObject;
|
||||
import cn.hutool.jwt.JWT;
|
||||
|
||||
/**
|
||||
* Sa-Token 整合 jwt:mix 模式 测试
|
||||
* Sa-Token 整合 jwt:mixin 模式 测试
|
||||
*
|
||||
* @author kong
|
||||
*
|
||||
*/
|
||||
@SpringBootTest(classes = StartUpApplication.class)
|
||||
public class JwtForMixTest {
|
||||
public class JwtForMixinTest {
|
||||
|
||||
// 持久化Bean
|
||||
@Autowired(required = false)
|
||||
@@ -38,7 +39,7 @@ public class JwtForMixTest {
|
||||
@BeforeAll
|
||||
public static void beforeClass() {
|
||||
System.out.println("\n\n------------------------ JwtForMixTest star ...");
|
||||
StpUtil.setStpLogic(new StpLogicJwtForMix());
|
||||
StpUtil.setStpLogic(new StpLogicJwtForMixin());
|
||||
}
|
||||
|
||||
// 结束
|
||||
@@ -254,5 +255,17 @@ public class JwtForMixTest {
|
||||
Assertions.assertTrue(list.size() >= 5);
|
||||
});
|
||||
}
|
||||
|
||||
// 测试:getExtra
|
||||
@Test
|
||||
public void getExtra() {
|
||||
// 登录
|
||||
StpUtil.login(10001, SaLoginConfig.setExtra("name", "zhangsan"));
|
||||
|
||||
// 可以取到
|
||||
Assertions.assertEquals(StpUtil.getExtra("name"), "zhangsan");
|
||||
// 取不到
|
||||
Assertions.assertEquals(StpUtil.getExtra("name2"), null);
|
||||
}
|
||||
|
||||
}
|
@@ -8,22 +8,23 @@ import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
import cn.dev33.satoken.SaManager;
|
||||
import cn.dev33.satoken.dao.SaTokenDao;
|
||||
import cn.dev33.satoken.jwt.StpLogicJwtForStyle;
|
||||
import cn.dev33.satoken.jwt.StpLogicJwtForSimple;
|
||||
import cn.dev33.satoken.session.SaSession;
|
||||
import cn.dev33.satoken.stp.SaLoginConfig;
|
||||
import cn.dev33.satoken.stp.StpUtil;
|
||||
import cn.dev33.satoken.util.SaTokenConsts;
|
||||
import cn.hutool.json.JSONObject;
|
||||
import cn.hutool.jwt.JWT;
|
||||
|
||||
/**
|
||||
* Sa-Token 整合 jwt:Style 模式 测试
|
||||
* Sa-Token 整合 jwt:Simple 模式 测试
|
||||
*
|
||||
* @author kong
|
||||
*
|
||||
*/
|
||||
//@RunWith(SpringRunner.class)
|
||||
@SpringBootTest(classes = StartUpApplication.class)
|
||||
public class JwtForStyleTest {
|
||||
public class JwtForSimpleTest {
|
||||
|
||||
// 持久化Bean
|
||||
static SaTokenDao dao;
|
||||
@@ -33,7 +34,7 @@ public class JwtForStyleTest {
|
||||
public static void beforeClass() {
|
||||
System.out.println("\n\n------------------------ JwtForStyleTest star ...");
|
||||
dao = SaManager.getSaTokenDao();
|
||||
StpUtil.setStpLogic(new StpLogicJwtForStyle());
|
||||
StpUtil.setStpLogic(new StpLogicJwtForSimple());
|
||||
}
|
||||
|
||||
// 结束
|
||||
@@ -69,5 +70,17 @@ public class JwtForStyleTest {
|
||||
Assertions.assertEquals(session.getId(), "satoken:login:session:" + 10001);
|
||||
Assertions.assertTrue(session.getTokenSignList().size() >= 1);
|
||||
}
|
||||
|
||||
// 测试:getExtra
|
||||
@Test
|
||||
public void getExtra() {
|
||||
// 登录
|
||||
StpUtil.login(10001, SaLoginConfig.setExtra("name", "zhangsan"));
|
||||
|
||||
// 可以取到
|
||||
Assertions.assertEquals(StpUtil.getExtra("name"), "zhangsan");
|
||||
// 取不到
|
||||
Assertions.assertEquals(StpUtil.getExtra("name2"), null);
|
||||
}
|
||||
|
||||
}
|
@@ -12,6 +12,7 @@ import cn.dev33.satoken.dao.SaTokenDao;
|
||||
import cn.dev33.satoken.exception.ApiDisabledException;
|
||||
import cn.dev33.satoken.jwt.SaJwtUtil;
|
||||
import cn.dev33.satoken.jwt.StpLogicJwtForStateless;
|
||||
import cn.dev33.satoken.stp.SaLoginConfig;
|
||||
import cn.dev33.satoken.stp.StpUtil;
|
||||
import cn.dev33.satoken.util.SaTokenConsts;
|
||||
import cn.hutool.json.JSONObject;
|
||||
@@ -164,4 +165,16 @@ public class JwtForStatelessTest {
|
||||
});
|
||||
}
|
||||
|
||||
// 测试:getExtra
|
||||
@Test
|
||||
public void getExtra() {
|
||||
// 登录
|
||||
StpUtil.login(10001, SaLoginConfig.setExtra("name", "zhangsan"));
|
||||
|
||||
// 可以取到
|
||||
Assertions.assertEquals(StpUtil.getExtra("name"), "zhangsan");
|
||||
// 取不到
|
||||
Assertions.assertEquals(StpUtil.getExtra("name2"), null);
|
||||
}
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user