first commit

WangHan

2024-09-12 d5855a4926926698b740bc6c7ba489de47adb68b

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package tech.powerjob.worker.persistence;
 
import com.google.common.collect.Lists;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import tech.powerjob.worker.common.constants.StoreStrategy;
import tech.powerjob.worker.common.constants.TaskStatus;
import tech.powerjob.worker.core.processor.TaskResult;
import tech.powerjob.worker.persistence.db.ConnectionFactory;
import tech.powerjob.worker.persistence.db.SimpleTaskQuery;
import tech.powerjob.worker.persistence.db.TaskDAO;
import tech.powerjob.worker.persistence.db.TaskDAOImpl;
 
import java.sql.SQLIntegrityConstraintViolationException;
import java.util.List;
import java.util.Map;
 
import static org.junit.jupiter.api.Assertions.*;
 
/**
 * 任务持久化接口测试
 *
 * @author tjq
 * @since 2022/10/23
 */
@Slf4j
class TaskDAOTest extends AbstractTaskDAOTest {
 
    private static TaskDAO taskDAO;
 
    @BeforeAll
    static void initDAO() throws Exception {
 
        // 1. 创建对象
        ConnectionFactory connectionFactory = new ConnectionFactory();
        connectionFactory.initDatasource(StoreStrategy.DISK);
        taskDAO = new TaskDAOImpl(connectionFactory);
 
        // 2. 初始化表
        taskDAO.initTable();
    }
 
    @Test
    @SneakyThrows
    void testUniqueKey() {
 
        TaskDO taskDO = buildTaskDO("2", 200000000000000L, TaskStatus.WORKER_PROCESS_FAILED);
        boolean firstSaveRet = taskDAO.save(taskDO);
        assert firstSaveRet;
        log.info("[testUniqueKey] first save result: {}", firstSaveRet);
        assertThrows(SQLIntegrityConstraintViolationException.class, () -> {
            taskDAO.save(taskDO);
        });
    }
 
    @Test
    @SneakyThrows
    void testCRUD() {
        TaskDO oneTask = buildTaskDO("1", 1L, TaskStatus.WAITING_DISPATCH);
        TaskDO twoTask = buildTaskDO("2", 1L, TaskStatus.WAITING_DISPATCH);
        TaskDO threeTask = buildTaskDO("99", 1L, TaskStatus.WAITING_DISPATCH);
 
        boolean batchSave = taskDAO.batchSave(Lists.newArrayList(oneTask, twoTask, threeTask));
        log.info("[testCRUD] batchSave result: {}", batchSave);
        assert batchSave;
 
        SimpleTaskQuery query = new SimpleTaskQuery();
        query.setInstanceId(1L);
        List<TaskDO> simpleQueryRet = taskDAO.simpleQuery(query);
        log.info("[testCRUD] simple query by instanceId's result: {}", simpleQueryRet);
        assert simpleQueryRet.size() == 3;
 
        SimpleTaskQuery deleteQuery = new SimpleTaskQuery();
        deleteQuery.setTaskId("99");
        deleteQuery.setInstanceId(1L);
        boolean simpleDelete = taskDAO.simpleDelete(deleteQuery);
        log.info("[testCRUD] simpleDelete result: {}", simpleDelete);
        assert simpleDelete;
 
        query.setQueryContent("status, result");
        List<Map<String, Object>> simpleQueryPlusRet = taskDAO.simpleQueryPlus(query);
        log.info("[testCRUD] simple query plus by instanceId's result: {}", simpleQueryPlusRet);
        assert simpleQueryPlusRet.size() == 2;
        assert simpleQueryPlusRet.get(0).get("status") != null;
        assert simpleQueryPlusRet.get(0).get("instanceId") == null;
 
        boolean updateToSuccessRet = taskDAO.updateTaskStatus(1L, "1", TaskStatus.WORKER_PROCESS_SUCCESS.getValue(), System.currentTimeMillis(), "UPDATE_TO_SUCCESS");
        boolean updateToFailedRet = taskDAO.updateTaskStatus(1L, "2", TaskStatus.WORKER_PROCESS_FAILED.getValue(), System.currentTimeMillis(), "UPDATE_TO_FAILED");
        assert updateToSuccessRet;
        assert updateToFailedRet;
 
        List<TaskResult> allTaskResult = taskDAO.getAllTaskResult(1L, 1L);
        log.info("[testCRUD] allTaskResult: {}", allTaskResult);
        assert allTaskResult.size() == 2;
    }
 
}