Examples with Task com.tencent.angel.worker.task.Task used on opensource projects

Example 1 with Task

use of com.tencent.angel.worker.task.Task in project angel by Tencent.

the class WorkerTest method testWorkerTaskManager.

@Test
public void testWorkerTaskManager() throws Exception {
    try {
        LOG.info("===========================testWorkerInitAndStart===============================");
        localWorker = LocalClusterContext.get().getWorker(worker0Attempt0Id);
        worker = localWorker.getWorker();
        // test worker getActiveTaskNum
        assertEquals(2, worker.getActiveTaskNum());
        // test worker getTaskNum
        assertEquals(2, worker.getTaskNum());
        // test worker getTaskManager
        TaskManager taskManager = worker.getTaskManager();
        assertTrue(taskManager != null);
        assertEquals(2, taskManager.getTaskCount());
        Task task_0 = taskManager.getRunningTask().get(task0Id);
        assertTrue(task_0 != null);
        Task task_1 = taskManager.getRunningTask().get(task1Id);
        assertTrue(task_1 != null);
        assertTrue(taskManager.isAllTaskRunning());
    } catch (Exception x) {
        LOG.error("run testWorkerTaskManager failed ", x);
        throw x;
    }
}