重构 sa-token-jwt 模块

This commit is contained in:
click33
2022-05-01 04:46:28 +08:00
parent d45d404fa0
commit a7ef71737e
16 changed files with 513 additions and 378 deletions

View File

@@ -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 整合 jwtmix 模式 测试
* Sa-Token 整合 jwtmixin 模式 测试
*
* @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);
}
}

View File

@@ -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 整合 jwtStyle 模式 测试
* Sa-Token 整合 jwtSimple 模式 测试
*
* @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);
}
}

View File

@@ -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);
}
}