From a0dda8322cad101dbd64e9f7109cbdd997a7f37d Mon Sep 17 00:00:00 2001
From: HuangXin <huangxin@cmhi.chinamobile.com>
Date: Sat, 9 May 2020 18:24:44 +0800
Subject: [PATCH] =?UTF-8?q?OCT=20REM:=201.=20=E6=B7=BB=E5=8A=A0TaskService?=
=?UTF-8?q?=E5=8D=95=E5=85=83=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
.../com/dispose/service/TaskServiceTest.java | 62 +++++++++++++++++--
1 file changed, 58 insertions(+), 4 deletions(-)
diff --git a/src/test/java/com/dispose/service/TaskServiceTest.java b/src/test/java/com/dispose/service/TaskServiceTest.java
index 15c9d6b9..bc2fe115 100644
--- a/src/test/java/com/dispose/service/TaskServiceTest.java
+++ b/src/test/java/com/dispose/service/TaskServiceTest.java
@@ -1,7 +1,12 @@
package com.dispose.service;
import com.dispose.Global.InitTestEnvironment;
+import com.dispose.common.DeviceCapacity;
+import com.dispose.common.DisposeTaskStatus;
import com.dispose.common.ErrorCode;
+import com.dispose.common.FlowDirection;
+import com.dispose.mapper.DisposeTaskMapper;
+import com.dispose.pojo.vo.common.TaskInfoDetail;
import lombok.extern.slf4j.Slf4j;
import org.junit.Assert;
import org.junit.FixMethodOrder;
@@ -9,27 +14,76 @@ import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.MethodSorters;
import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Resource;
+/**
+ * The type Task service test.
+ */
@RunWith(SpringRunner.class)
@SpringBootTest
@Slf4j
+@Transactional
+@Rollback
@FixMethodOrder(MethodSorters.NAME_ASCENDING)
public class TaskServiceTest extends InitTestEnvironment {
+ /**
+ * The Task service.
+ */
@Resource
private TaskService taskService;
- @Override
- public void userLogin() {
+ /**
+ * The Dispose task mapper.
+ */
+ @Resource
+ private DisposeTaskMapper disposeTaskMapper;
+ /**
+ * User login.
+ */
+ @Override
+ public void userLogin() throws Exception {
+ super.userLogin();
+
+ TaskInfoDetail taskData = TaskInfoDetail.builder()
+ .id(-1L)
+ .deviceId(1L)
+ .accountId(1L)
+ .type(DeviceCapacity.CLEANUP.getCode())
+ .disposeIp("192.168.5.3")
+ .attackType("0")
+ .flowDirection(FlowDirection.DIRECTION_TWOWAY.getCode())
+ .currentStatus(DisposeTaskStatus.TASK_NEW.getCode())
+ .planEndTime("60")
+ .build();
+
+ taskService.createTask(taskData);
}
+ /**
+ * T 2 start task test.
+ */
+ @Test
+ public void t2_startTaskTest() {
+ disposeTaskMapper.getAllTaskByStatus(DisposeTaskStatus.TASK_NEW.getCode()).forEach(v -> {
+ ErrorCode err = taskService.startTask(v.getId());
+ Assert.assertEquals(err, ErrorCode.ERR_OK);
+ });
+ }
+
+ /**
+ * T 1 stop task test.
+ */
@Test
public void t1_stopTaskTest() {
- ErrorCode err = taskService.stopTask(-1L);
+ disposeTaskMapper.getAllTaskByStatus(DisposeTaskStatus.TASK_RUNNING.getCode()).forEach(v -> {
+ ErrorCode err = taskService.stopTask(v.getId());
+ Assert.assertEquals(err, ErrorCode.ERR_OK);
+ });
- Assert.assertNotEquals(err, ErrorCode.ERR_OK);
}
}