|
| 1 | +package org.apache.dolphinscheduler.dao.entity; |
| 2 | + |
| 3 | +import static org.junit.jupiter.api.Assertions.assertEquals; |
| 4 | + |
| 5 | +import org.apache.dolphinscheduler.common.enums.CommandType; |
| 6 | +import org.apache.dolphinscheduler.common.enums.FailureStrategy; |
| 7 | +import org.apache.dolphinscheduler.common.enums.Flag; |
| 8 | +import org.apache.dolphinscheduler.common.enums.Priority; |
| 9 | +import org.apache.dolphinscheduler.common.enums.TaskDependType; |
| 10 | +import org.apache.dolphinscheduler.common.enums.WarningType; |
| 11 | + |
| 12 | +import java.util.Date; |
| 13 | + |
| 14 | +import org.junit.jupiter.api.Test; |
| 15 | + |
| 16 | +public class ErrorCommandTest { |
| 17 | + |
| 18 | + @Test |
| 19 | + public void testConstructor() { |
| 20 | + Command command = new Command(); |
| 21 | + command.setId(1); |
| 22 | + command.setCommandType(CommandType.PAUSE); |
| 23 | + command.setExecutorId(1); |
| 24 | + command.setProcessDefinitionCode(123); |
| 25 | + command.setProcessDefinitionVersion(1); |
| 26 | + command.setProcessInstanceId(1); |
| 27 | + command.setCommandParam("param"); |
| 28 | + command.setTaskDependType(TaskDependType.TASK_POST); |
| 29 | + command.setFailureStrategy(FailureStrategy.CONTINUE); |
| 30 | + command.setWarningType(WarningType.ALL); |
| 31 | + command.setWarningGroupId(1); |
| 32 | + command.setScheduleTime(new Date()); |
| 33 | + command.setStartTime(new Date()); |
| 34 | + command.setUpdateTime(new Date()); |
| 35 | + command.setProcessInstancePriority(Priority.HIGHEST); |
| 36 | + command.setWorkerGroup("default"); |
| 37 | + command.setTenantCode("root"); |
| 38 | + command.setEnvironmentCode(1L); |
| 39 | + command.setDryRun(1); |
| 40 | + command.setTestFlag(Flag.NO.getCode()); |
| 41 | + |
| 42 | + ErrorCommand errorCommand = new ErrorCommand(command, "test"); |
| 43 | + assertEquals(command.getCommandType(), errorCommand.getCommandType()); |
| 44 | + assertEquals(command.getExecutorId(), errorCommand.getExecutorId()); |
| 45 | + assertEquals(command.getProcessDefinitionCode(), errorCommand.getProcessDefinitionCode()); |
| 46 | + assertEquals(command.getProcessDefinitionVersion(), errorCommand.getProcessDefinitionVersion()); |
| 47 | + assertEquals(command.getProcessInstanceId(), errorCommand.getProcessInstanceId()); |
| 48 | + assertEquals(command.getCommandParam(), errorCommand.getCommandParam()); |
| 49 | + assertEquals(command.getTaskDependType(), errorCommand.getTaskDependType()); |
| 50 | + assertEquals(command.getFailureStrategy(), errorCommand.getFailureStrategy()); |
| 51 | + assertEquals(command.getWarningType(), errorCommand.getWarningType()); |
| 52 | + assertEquals(command.getWarningGroupId(), errorCommand.getWarningGroupId()); |
| 53 | + assertEquals(command.getScheduleTime(), errorCommand.getScheduleTime()); |
| 54 | + assertEquals(command.getStartTime(), errorCommand.getStartTime()); |
| 55 | + assertEquals(command.getUpdateTime(), errorCommand.getUpdateTime()); |
| 56 | + assertEquals(command.getProcessInstancePriority(), errorCommand.getProcessInstancePriority()); |
| 57 | + assertEquals(command.getWorkerGroup(), errorCommand.getWorkerGroup()); |
| 58 | + assertEquals(command.getTenantCode(), errorCommand.getTenantCode()); |
| 59 | + assertEquals(command.getEnvironmentCode(), errorCommand.getEnvironmentCode()); |
| 60 | + assertEquals(command.getDryRun(), errorCommand.getDryRun()); |
| 61 | + assertEquals(command.getTestFlag(), errorCommand.getTestFlag()); |
| 62 | + assertEquals("test", errorCommand.getMessage()); |
| 63 | + } |
| 64 | + |
| 65 | +} |
0 commit comments