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

1. 添加doris数据源

Browse Source 
2. 添加StateMapper，准备进行oracle数据库对接
This commit is contained in:
EnderByEndera
2024-01-09 09:20:13 +08:00
parent 2b04a7d6ce
commit a04b83e4c3
15 changed files with 134 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/test/java/com/realtime/protection/server/defense/template/TemplateServiceTest.java
View File

@@ -53,22 +53,31 @@ class TemplateServiceTest {
@Test
void testQueryTemplate() {
List<Template> templates = templateService.queryTemplates("DDOS", 1, 5);
System.out.println(templates);
assertEquals(5, templates.size());
for (Template template : templates) {
assertTrue(template.getTemplateId() > 0);
}
}
@Test
void testUpdateTemplateSuccess() {
template.setTemplateName("洪泛型DDOS攻击");
assertTrue(templateService.updateTemplate(3, template));
List<Template> templates = templateService.queryTemplates("反射", 1, 6);
Template testTemplate = templates.get(0);
testTemplate.setTemplateName("洪泛型DDOS攻击");
assertTrue(templateService.updateTemplate(testTemplate.getTemplateId(), testTemplate));
}
@Test
void testAddTemplateUsedTimes() {
List<Template> templates = templateService.queryTemplates("洪泛型DDOS攻击", 1, 1);
template.setTemplateName("add test");
templateService.newTemplate(template);
List<Template> templates = templateService.queryTemplates("add", 1, 1);
Template originalTemplate = templates.get(0);
Boolean success = templateService.addTemplateUsedTimes(3, 4);
templates = templateService.queryTemplates("洪泛型DDOS攻击", 1, 1);
assertEquals(templates.get(0).getTemplateUsedTimes(), originalTemplate.getTemplateUsedTimes() + 4);
Boolean success = templateService.addTemplateUsedTimes(originalTemplate.getTemplateId(), 4);
templates = templateService.queryTemplates("add", 1, 1);
assertEquals(originalTemplate.getTemplateUsedTimes() + 4, templates.get(0).getTemplateUsedTimes());
assertTrue(success);
}
}