gfwleak/enderbyendera-realtime-protection
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1. 新增数据源oracle，已在application.yml中配置

Browse Source 
2. 新增任务状态类，未来将在切换任务状态中使用
3. 新增ProtectLevel实体类，用来存储Template对应的三种防护等级数据
4. Task实体类中删除protectObjectIds，因为MySQL表结构发生修改
5. TaskController新增audit和delete路由，用以审核和删除Task
6. TemplateMapper新增newProtectLevel方法
7.
This commit is contained in:
EnderByEndera
2024-01-08 20:01:20 +08:00
parent 1e9fe37d0d
commit 2b04a7d6ce
26 changed files with 392 additions and 204 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
src/test/java/com/realtime/protection/server/task/TaskServiceTest.java
View File

@@ -38,13 +38,13 @@ class TaskServiceTest {
task.setDynamicRuleIds(List.of());
task.setTaskCreateUserId(1);
task.setTaskCreateUsername("xxx");
task.setTaskCreateDepart("xxx");
task.setProtectObjectIds(List.of(1));
task.setTaskCreateDepart("xxx");;
}
@Test
void testNewTaskSuccess() {
assertDoesNotThrow(() -> {Integer taskId = taskService.newTask(task); assertTrue(taskId > 0);});
assertTrue(task.getTaskId() > 0);
}
@Test
@@ -58,8 +58,36 @@ class TaskServiceTest {
@Test
void testQueryTasks() {
List<Task> tasks = taskService.queryTasks(null, null, null, null, 1, 5);
assertEquals(5, tasks.size());
List<Task> tasks = taskService.queryTasks(null, null, null, null, 1, 10);
assertTrue(tasks.get(0).getTaskId() > 0);
}
@Test
void testUpdateTasks() {
task.setStaticRuleIds(List.of(6, 7, 8));
task.setTaskId(26);
task.setTaskName("修改测试");
assertTrue(taskService.updateTask(task));
assertEquals("修改测试", taskService.queryTask(26).getTaskName());
}
@Test
void testDeleteTask() {
int testNum = taskService.queryTasks(null, null, null, null, 1, 10)
.get(0).getTaskId();
assertTrue(taskService.deleteTask(testNum));
assertFalse(taskService.deleteTask(235235));
}
@Test
void testChangeAuditStatus() {
int testNum = taskService.queryTasks(null, null, null, null, 1, 1)
.get(0).getTaskId();
assertTrue(taskService.changeTaskAuditStatus(testNum, 2));
assertFalse(taskService.changeTaskAuditStatus(testNum, 0));
assertFalse(taskService.changeTaskAuditStatus(testNum, 1));
}
}