1.完成的excel批量新增、按id删除、批量删除、按id查询、分页查询、按id新增、修改审核状态
(excel批量新增还未http请求测试) (批量新增、批量删除、修改审核状态已复用sy代码) 2、查询ip是否存在于白名单功能初步实现,仍不完善
This commit is contained in:
@@ -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);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user