fix:IdentifyManager的小bug(异常值判断)

fix:引入mgmt.Exception(解决编译错误)
add:IdentifyManager单元测试
This commit is contained in:
ChessNineeee committed 2021-04-14 16:51:14 +08:00
1 parent 0dc6bb3165
commit bbee7f879e
13 files changed
+417 -13

No files matched your search

@@ -0,0 +1,330 @@
package security;
import minsecurity.Common;
import minsecurity.certificate.cert.CertUtils;
import minsecurity.certificate.cert.Certificate;
import minsecurity.crypto.sm2.SM2KeyPair;
import minsecurity.identity.Identity;
import minsecurity.identity.KeyParam;
import minsecurity.identity.TestIdentity;
import minsecurity.identity.persist.sqlite.Sqlite;
import org.junit.Test;
import org.slf4j.LoggerFactory;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
public class IdentifyManagerTest {
private static final org.slf4j.Logger logger = LoggerFactory.getLogger(TestIdentity.class);
/**
* 随机生成身份数据
* @return Identity
*/
private Identity createRandomIdentity() throws Exception{
// 测试PersistIdentity
SM2KeyPair pair = SM2KeyPair.generateKeyPair();
Identity identity = new Identity();
identity.setName("wzq"+Math.random());
KeyParam keyParam = new KeyParam();
keyParam.PublicKeyAlgorithm = 0;
keyParam.SignatureAlgorithm = 0;
identity.setKeyParam(keyParam);
identity.setPrikey(pair.getSm2PrivateKey());
identity.setPubkey(pair.getSm2PublicKey());
identity.setPasswd("2DD29CA851E7B56E4697B0E1F08507293D761A05CE4D1B628663F411A8086D99");
identity.lock("0123456789abcdef", Common.SM4ECB);
Certificate cert = new Certificate();
cert.setVersion(1);
cert.setSerialNumber(1);
cert.setPublicKey(pair.getSm2PublicKey());
cert.setSignatureAlgorithm(Common.SM3withSM2); // TODO 名字有误? SM2withSM3?
cert.setPublicKeyAlgorithm(Common.SM2);
cert.setIssueTo("root");
cert.setIssuer("root");
long timestamp = System.currentTimeMillis() / 1000;
cert.setTimestamp(timestamp); // 10bit timestamp
cert.setNotAfter(timestamp); // 10bit timestamp
cert.setNotBefore(timestamp + 1000); // 10bit timestamp
cert.setKeyUsage(Common.CertSign);
cert.setCA(true);
CertUtils.signCert(cert, pair.getSm2PrivateKey());
identity.setCert(cert);
return identity;
}
@Test
public void testIdentifyManagerInit(){
try{
// 打开数据库
// Sqlite.getInstance().openDefault();
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("默认身份:%s", manager.getDefaultIdentity().getName()));
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
logger.debug(String.format("私钥加密算法:%d", manager.getPrivateKeyEncryptionAlgorithm()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testGetIdentityByName(){
try{
IdentifyManager manager = new IdentifyManager();
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
// 查询所有manager中的用户
for (String s:
keySet) {
logger.debug(String.format("查询身份:%s,结果身份:%s", s, manager.getIdentityByName(s).getName()));
}
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testGetIdentityByName2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
for (String s:
keySet) {
// 查询不存在的名称 返回null
String sSet = s + "test";
logger.debug(String.format("查询身份:%s,结果身份:%s", sSet, manager.getIdentityByName(sSet)));
// 查询null 抛出异常 已改为返回null
sSet = null;
logger.debug(String.format("查询身份:%s,结果身份:%s", sSet, manager.getIdentityByName(sSet)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testDeleteIdentityByName2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
for (String s:
keySet) {
// 删除不存在的名称 返回false
String sSet = s + "test";
logger.debug(String.format("删除身份:%s,结果:%s", sSet, manager.deleteIdentityByName(sSet, true)));
// 查询null 抛出异常 已改为返回false
sSet = null;
logger.debug(String.format("删除身份:%s,结果:%s", sSet, manager.deleteIdentityByName(sSet, true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testDeleteIdentityByName(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
// 删除所有manager中的用户
for (String s:
keySet) {
logger.debug(String.format("删除身份:%s,结果:%s", s, manager.deleteIdentityByName(s, true)));
logger.debug(String.format("查询身份:%s,结果身份:%s", s, manager.getIdentityByName(s)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testSaveIdentity(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
// 随机生成身份并保存 重复五次
for (int i = 0; i < 5; i++) {
Identity id = createRandomIdentity();
// todo 确认是否需要强制覆盖?
logger.debug(String.format("保存身份:%s,结果:%s", id.getName(), manager.saveIdentity(id, true, true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testSaveIdentity2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
// 插入 null 重复五次
for (int i = 0; i < 5; i++) {
Identity id = null;
logger.debug(String.format("保存身份:%s,结果:%s", id, manager.saveIdentity(id, true, true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testCreateIdentityByNameAndKeyParam(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
// 随机生成名称并保存 重复五次
for (int i = 0; i < 5; i++) {
String s = "wzq"+Math.random();
KeyParam keyParam = new KeyParam();
keyParam.PublicKeyAlgorithm = 0;
keyParam.SignatureAlgorithm = 0;
// todo 确认身份在没有证书的情况下能否存储
logger.debug(String.format("保存身份:%s,结果:%s", s, manager.createIdentityByNameAndKeyParam(s, keyParam, "1234", true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testCreateIdentityByName(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
// 随机生成名称并保存 重复五次
for (int i = 0; i < 5; i++) {
String s = "wzq"+Math.random();
logger.debug(String.format("保存身份:%s,结果:%s", s, manager.createIdentityByName(s, "1234", true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testCreateIdentityByName2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
// 保存null 重复五次
for (int i = 0; i < 5; i++) {
String s = null;
// 保存失败 返回null
logger.debug(String.format("保存身份:%s,结果:%s", s, manager.createIdentityByName(s, "1234", true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testCreateIdentityByNameAndKeyParam2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
// 保存null 重复五次
for (int i = 0; i < 5; i++) {
String s = null;
// 不会插入 返回null
logger.debug(String.format("保存身份:%s,结果:%s", s, manager.createIdentityByName(s, "1234", true)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testSetDefaultIdentity(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
// 轮流设置默认身份
for (String s:
keySet) {
logger.debug(String.format("设置身份:%s,结果:%s", s, manager.setDefaultIdentity(manager.getIdentifies().get(s), true)));
logger.debug(String.format("查询默认身份:%s", manager.getDefaultIdentity()));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testSetDefaultIdentity2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
// 轮流设置默认身份 设置的身份原本不存在
for (String s:
keySet) {
Identity id = manager.getIdentifies().get(s);
String sNew = s + "test";
id.setName(sNew);
logger.debug(String.format("设置身份:%s,结果:%s", sNew, manager.setDefaultIdentity(id, true)));
logger.debug(String.format("查询默认身份:%s", manager.getDefaultIdentity()));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testExistIdentity(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
// 正常查询身份是否存在
for (String s:
keySet) {
String sSet = "test" + s;
logger.debug(String.format("查询身份:%s 是否存在, 结果:%s", s, manager.existIdentity(s)));
logger.debug(String.format("查询身份:%s 是否存在, 结果:%s", sSet, manager.existIdentity(sSet)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
@Test
public void testExistIdentity2(){
try{
IdentifyManager manager = new IdentifyManager();
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
ConcurrentMap<String, Identity> hashMap = manager.getIdentifies();
Set<String> keySet = hashMap.keySet();
// 查询 null 是否存在
for (String s:
keySet) {
logger.debug(String.format("查询身份:%s 是否存在, 结果:%s", null, manager.existIdentity(null)));
}
logger.debug(String.format("身份数量:%d", manager.getIdentifies().size()));
}catch (Exception ex){
logger.debug(ex.getMessage());
}
}
}