enderbyendera-realtime-protection/src/test/java/com/realtime/protection/server/alertmessage/AlertMessageTest.java

package com.realtime.protection.server.alertmessage;

import com.github.xiaoymin.knife4j.annotations.Ignore;
import com.realtime.protection.configuration.entity.alert.AlertMessage;
import com.realtime.protection.configuration.entity.defense.object.ProtectObject;
import com.realtime.protection.configuration.entity.defense.template.Template;
import com.realtime.protection.configuration.entity.rule.dynamicrule.DynamicRuleObject;
import com.realtime.protection.configuration.entity.task.FiveTupleWithMask;
import com.realtime.protection.configuration.entity.task.Task;
import com.realtime.protection.configuration.exception.DorisStartException;
import com.realtime.protection.server.defense.object.ProtectObjectService;
import com.realtime.protection.server.defense.template.TemplateService;
import com.realtime.protection.server.rule.dynamicrule.DynamicRuleService;
import com.realtime.protection.server.task.TaskService;
import com.realtime.protection.server.task.status.StateChangeService;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.time.LocalDateTime;
import java.util.List;

@SpringBootTest
public class AlertMessageTest {

    private final AlertMessageService alertMessageService;
    private final ProtectObjectService protectObjectService;
    private final TemplateService templateService;
    private final DynamicRuleService dynamicRuleService;
    private final TaskService taskService;
    private final StateChangeService stateChangeService;
    @Autowired
    public AlertMessageTest(AlertMessageService alertMessageService
    ,ProtectObjectService protectObjectService,TemplateService templateService,
                            DynamicRuleService dynamicRuleService,TaskService taskService,
                            StateChangeService stateChangeService) {
        this.alertMessageService = alertMessageService;
        this.protectObjectService = protectObjectService;
        this.templateService = templateService;
        this.dynamicRuleService = dynamicRuleService;
        this.taskService = taskService;
        this.stateChangeService = stateChangeService;
    }

    @Ignore
    @Test
    void testReceiveAlertMessage() throws DorisStartException {

        List<ProtectObject> protectObject = protectObjectService.queryProtectObjects(
                null, null, null, null,
                null, null, null, null,
                null, null, 1, 1);
        List<Template> templates = templateService.queryTemplates(
                null, null, null, null, null,1, 1);

        DynamicRuleObject object = new DynamicRuleObject();
        object.setDynamicRuleName("UpdateDynamicRule2");
        object.setDynamicRuleFrequency(1);
        object.setDynamicRulePriority(1);
//        object.setDynamicRuleRange("北京");
        object.setDynamicRuleProtectLevel(1);
        object.setLogRuleId(1L);
        object.setTemplateId(templates.get(0).getTemplateId());
        object.setProtectObjectIds(List.of(new Integer[]{protectObject.get(0).getProtectObjectId()}));
        Integer dynamicRuleId = dynamicRuleService.newDynamicRuleObject(object);
        dynamicRuleService.updateAuditStatus(object.getDynamicRuleId(), 2);


        Task task = new Task();
        task.setTaskName("生产告警信息测试");
        LocalDateTime taskStartTime = LocalDateTime.now().plusMinutes(1);
        LocalDateTime taskEndTime = LocalDateTime.now().plusYears(5);
        task.setTaskStartTime(taskStartTime);
        task.setTaskEndTime(taskEndTime);
        task.setTaskAct("阻断");
        task.setTaskType(2);
        task.setTaskCreateUserId(1);
        task.setTaskCreateUsername("xxx");
        task.setTaskCreateDepart("xxx");
        task.setTaskRange("北京");
        task.setDynamicRuleIds(List.of(new Integer[]{dynamicRuleId}));
        Long taskId = taskService.newTask(task);
        System.out.println(taskId);
        //审核状态
        taskService.changeTaskAuditStatus(taskId, 2);
        //启动任务
//        stateChangeService.changeState(2, taskId, false);

//        for (int i = 0 ; i< 5; i++) {
//            AlertMessage alert = new AlertMessage();
//            FiveTupleWithMask fiveTupleWithMask = new FiveTupleWithMask();
//            fiveTupleWithMask.setSourceIP("111.1.1." + i);
//            fiveTupleWithMask.setDestinationIP("222.22.2." + i);
//            fiveTupleWithMask.setSourcePort("111");
//            fiveTupleWithMask.setDestinationPort("222");
//            fiveTupleWithMask.setProtocol("tcp");
//
//            alert.setDynamicRuleId(dynamicRuleId);
//            alert.setTaskId(taskId);
//            alert.setFiveTupleWithMask(fiveTupleWithMask);
//            alert.setContent("testcontent");
//            alert.setProtectIsSrcOrDst(1);
//            alertMessageService.processAlertMessage(alert);
//        }



    }

    @Test
    void queryAlertMessageByCommandId() {

//        String commandId = "3e2fde7c-cd91-41f3-aedf-bd9993a61737";
//
//        System.out.println(alertMessageService.queryAlarmsByCommandId(commandId));
    }
}