@Test(timeout = 2000L) public void testOverlordRun() throws Exception { // basic task master lifecycle test taskMaster.start(); announcementLatch.await(); while (!taskMaster.isLeading()) { // I believe the control will never reach here and thread will never sleep but just to be on // safe side Thread.sleep(10); } Assert.assertEquals(taskMaster.getLeader(), druidNode.getHostAndPort()); // Test Overlord resource stuff overlordResource = new OverlordResource( taskMaster, new TaskStorageQueryAdapter(taskStorage), null, null, null, new AuthConfig()); Response response = overlordResource.getLeader(); Assert.assertEquals(druidNode.getHostAndPort(), response.getEntity()); final String taskId_0 = "0"; NoopTask task_0 = new NoopTask(taskId_0, 0, 0, null, null, null); response = overlordResource.taskPost(task_0, req); Assert.assertEquals(200, response.getStatus()); Assert.assertEquals(ImmutableMap.of("task", taskId_0), response.getEntity()); // Duplicate task - should fail response = overlordResource.taskPost(task_0, req); Assert.assertEquals(400, response.getStatus()); // Task payload for task_0 should be present in taskStorage response = overlordResource.getTaskPayload(taskId_0); Assert.assertEquals(task_0, ((Map) response.getEntity()).get("payload")); // Task not present in taskStorage - should fail response = overlordResource.getTaskPayload("whatever"); Assert.assertEquals(404, response.getStatus()); // Task status of the submitted task should be running response = overlordResource.getTaskStatus(taskId_0); Assert.assertEquals(taskId_0, ((Map) response.getEntity()).get("task")); Assert.assertEquals( TaskStatus.running(taskId_0).getStatusCode(), ((TaskStatus) ((Map) response.getEntity()).get("status")).getStatusCode()); // Simulate completion of task_0 taskCompletionCountDownLatches[Integer.parseInt(taskId_0)].countDown(); // Wait for taskQueue to handle success status of task_0 waitForTaskStatus(taskId_0, TaskStatus.Status.SUCCESS); // Manually insert task in taskStorage // Verifies sync from storage final String taskId_1 = "1"; NoopTask task_1 = new NoopTask(taskId_1, 0, 0, null, null, null); taskStorage.insert(task_1, TaskStatus.running(taskId_1)); // Wait for task runner to run task_1 runTaskCountDownLatches[Integer.parseInt(taskId_1)].await(); response = overlordResource.getRunningTasks(req); // 1 task that was manually inserted should be in running state Assert.assertEquals(1, (((List) response.getEntity()).size())); final OverlordResource.TaskResponseObject taskResponseObject = ((List<OverlordResource.TaskResponseObject>) response.getEntity()).get(0); Assert.assertEquals(taskId_1, taskResponseObject.toJson().get("id")); Assert.assertEquals(TASK_LOCATION, taskResponseObject.toJson().get("location")); // Simulate completion of task_1 taskCompletionCountDownLatches[Integer.parseInt(taskId_1)].countDown(); // Wait for taskQueue to handle success status of task_1 waitForTaskStatus(taskId_1, TaskStatus.Status.SUCCESS); // should return number of tasks which are not in running state response = overlordResource.getCompleteTasks(req); Assert.assertEquals(2, (((List) response.getEntity()).size())); taskMaster.stop(); Assert.assertFalse(taskMaster.isLeading()); EasyMock.verify(taskLockbox, taskActionClientFactory); }