1.完成的excel批量新增、按id删除、批量删除、按id查询、分页查询、按id新增、修改审核状态

(excel批量新增还未http请求测试)
(批量新增、批量删除、修改审核状态已复用sy代码)
2、查询ip是否存在于白名单功能初步实现,仍不完善
This commit is contained in:
Hao Miao
2024-01-08 00:22:14 +08:00
parent db02907f0a
commit 1e9fe37d0d
10 changed files with 524 additions and 21 deletions

View File

@@ -1,22 +1,37 @@
package com.realtime.protection.server.whitelist;
import com.realtime.protection.configuration.entity.whitelist.WhiteListObject;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import java.util.ArrayList;
import java.util.List;
import static org.junit.jupiter.api.Assertions.*;
@SpringBootTest
class WhiteListServiceTest {
private final WhiteListService whiteListService;
private WhiteListObject whiteListObject;
@Autowired
WhiteListServiceTest(WhiteListService whiteListService) {
this.whiteListService = whiteListService;
}
@BeforeEach
void setUp() {
whiteListObject = new WhiteListObject();
whiteListObject.setWhiteListName("test");
whiteListObject.setWhiteListSystemName("china");
whiteListObject.setWhiteListIP("128.1.1.1");
whiteListObject.setWhiteListPort(80);
whiteListObject.setWhiteListUrl("www.baidu.com");
whiteListObject.setWhiteListProtocol("TCP");
}
@Test
void testNewWhiteList() {
WhiteListObject object = new WhiteListObject();
@@ -25,4 +40,35 @@ class WhiteListServiceTest {
Integer objectId = whiteListService.newWhiteListObject(object);
assertTrue(objectId > 0);
}
@Test
void newProtectObjects() {
List<WhiteListObject> whiteListObjects = new ArrayList<>();
for (int i = 0; i < 1000; i++) {
whiteListObjects.add(whiteListObject);
}
Boolean success = whiteListService.newWhiteListObjects(whiteListObjects);
assertTrue(success);
}
@Test
void testUpdateWhiteList() {
WhiteListObject object = new WhiteListObject();
object.setWhiteListId(7);
object.setWhiteListName("test_update");
whiteListService.updateWhiteListObject(object);
}
@Test
void testExistWhiteList() {
List<Integer> ruleIds = List.of(6,7,8);
List<String> ip_list = whiteListService.existWhiteListObject(ruleIds);
System.out.println(ip_list);
}
@Test
void testUpdateWhiteListAuditStatus() {
whiteListService.updateWhiteListObjectAuditStatus(7, 1);
}
}