@Override public void expect(ClusterState state, SchedulerDriver mockDriver) throws AssertionError { Optional<Protos.TaskInfo> task = new StateStore(persisterWithTasks).fetchTask(taskName); Assert.assertTrue(String.format("Task %s not found", taskName), task.isPresent()); Assert.assertEquals(String.format("Expected zero resources, got: %s", task.get().getResourcesList()), 0, task.get().getResourcesCount()); } }
@Test public void testCreateCassandraContainer() throws Exception { CassandraContainer container = getTestCassandraContainer(); Collection<Protos.TaskInfo> taskInfos = container.getTaskInfos(); Assert.assertEquals(2, taskInfos.size()); Iterator<Protos.TaskInfo> iter = taskInfos.iterator(); Protos.TaskInfo daemonTaskInfo = iter.next(); Protos.TaskInfo clusterTemplateTaskInfo = iter.next(); validateDaemonTaskInfo(daemonTaskInfo); Assert.assertEquals(CassandraTemplateTask.toTemplateTaskName(daemonTaskInfo.getName()), clusterTemplateTaskInfo.getName()); Assert.assertEquals(2, clusterTemplateTaskInfo.getResourcesCount()); Assert.assertTrue(clusterTemplateTaskInfo.getTaskId().getValue().isEmpty()); for (Protos.Resource resource : clusterTemplateTaskInfo.getResourcesList()) { Assert.assertTrue(ResourceUtils.getResourceId(resource).isEmpty()); } }
Assert.assertEquals(recommendations.toString(), 2, launchTask.getResourcesCount()); Resource launchResource = launchTask.getResources(1);
Assert.assertEquals(recommendations.toString(), 2, launchTask.getResourcesCount()); Resource launchResource = launchTask.getResources(1);
private void validateDaemonTaskInfo(Protos.TaskInfo daemonTaskInfo) throws TaskException { Assert.assertEquals(testDaemonName, daemonTaskInfo.getName()); Assert.assertEquals(4, daemonTaskInfo.getResourcesCount()); Assert.assertEquals(testDaemonName, TaskUtils.toTaskName(daemonTaskInfo.getTaskId())); Assert.assertTrue(daemonTaskInfo.getSlaveId().getValue().isEmpty()); for (Protos.Resource resource : daemonTaskInfo.getResourcesList()) { Assert.assertTrue(ResourceUtils.getResourceId(resource).isEmpty()); } }