taskInfo.setContainer(containerInfo); return taskInfo.build();
.setType(Protos.Value.Type.SCALAR) .setScalar(Protos.Value.Scalar.newBuilder().setValue(igniteTask.disk()))) .build();
public static void setPermanentlyFailed(StateStore stateStore, Collection<Protos.TaskInfo> failedTasks) { stateStore.storeTasks(failedTasks.stream() .map(taskInfo -> taskInfo.toBuilder() .setLabels(new TaskLabelWriter(taskInfo).setPermanentlyFailed().toProto()) .build()) .collect(Collectors.toList())); } }
@Override public BackupUploadTask update(Protos.TaskState state) { return new BackupUploadTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public BackupUploadTask updateId() { return new BackupUploadTask(getBuilder().setTaskId(createId(getName())) .build()); }
@Override public RestoreSchemaTask updateId() { return new RestoreSchemaTask(getBuilder().setTaskId(createId(getName())) .build()); }
@Test public void testReadWriteZone() { Assert.assertFalse(new TaskLabelReader(getTestTaskInfo()).getZone().isPresent()); Protos.TaskInfo.Builder tb = getTestTaskInfo().toBuilder(); tb.setLabels(new TaskLabelWriter(tb).setZone(TestConstants.LOCAL_DOMAIN_INFO.getFaultDomain().getZone()).toProto()); Assert.assertTrue(new TaskLabelReader(tb.build()).getZone().isPresent()); Assert.assertEquals(TestConstants.ZONE, new TaskLabelReader(tb.build()).getZone().get()); }
@Before public void beforeEach() { MockitoAnnotations.initMocks(this); podInstance = PodTestUtils.getPodInstance(0); taskInfo = TestConstants.TASK_INFO.toBuilder().setLabels( new TaskLabelWriter(TestConstants.TASK_INFO) .setType("different-type") .setIndex(100) .toProto()) .build(); offer = OfferTestUtils.getEmptyOfferBuilder().build(); }
private static Protos.TaskInfo createTaskInfoWithZone(String taskName, String zone) { return Protos.TaskInfo.newBuilder(createTask(taskName)) .setCommand(Protos.CommandInfo.newBuilder() .setEnvironment(Protos.Environment.newBuilder() .addVariables(Protos.Environment.Variable.newBuilder() .setName(EnvConstants.ZONE_TASKENV) .setValue(zone)))) .build(); }
@Test(expected=IllegalArgumentException.class) public void testLabelConverterNoLabels() { TaskInfo task = getTask("DELETE ME", "hello-1234__uuid", "agent").toBuilder() .clearLabels() .build(); new TaskTypeLabelConverter().getTaskType(task); }
private static Protos.TaskInfo newTaskInfo(final String taskName, final Protos.TaskID taskID) { Protos.TaskInfo.Builder taskBuilder = Protos.TaskInfo.newBuilder() .setName(taskName) .setTaskId(taskID); taskBuilder.getSlaveIdBuilder().setValue("proto-field-required"); return taskBuilder.build(); }
private Protos.TaskInfo createTask(String hostname) throws InvalidProtocolBufferException { return Protos.TaskInfo.newBuilder() .setName("Test") .setTaskId(Protos.TaskID.newBuilder().setValue("TestId").build()) .setSlaveId(Protos.SlaveID.newBuilder().setValue(hostname).build()) .build(); }
public static Protos.TaskInfo getTaskInfoExternalVolume(Integer esNodeId) { return Protos.TaskInfo.newBuilder(getDefaultTaskInfo()) .setCommand(Protos.CommandInfo.newBuilder() .setValue("") .setEnvironment(Protos.Environment.newBuilder() .addVariables(Protos.Environment.Variable.newBuilder() .setName(ExecutorEnvironmentalVariables.ELASTICSEARCH_NODE_ID) .setValue(esNodeId.toString()))) .build()) .build(); }
@Test public void shouldReturnCorrectNumberOfExecutors() throws IOException { ArrayList<Protos.TaskInfo> mock = Mockito.spy(new ArrayList<>()); mock.add(ProtoTestUtil.getDefaultTaskInfo()); Protos.TaskInfo defaultTaskInfo = Protos.TaskInfo.newBuilder().mergeFrom(ProtoTestUtil.getDefaultTaskInfo()).setTaskId(Protos.TaskID.newBuilder().setValue("Task2")).build(); mock.add(defaultTaskInfo); when(state.get(contains(ESTaskStatus.STATE_KEY))).thenReturn(ProtoTestUtil.getDefaultTaskStatus(Protos.TaskState.TASK_RUNNING)); when(state.get(contains(ClusterState.STATE_LIST))).thenReturn(mock); // Be careful, the state list and state key both have the word state in them. Order is important. assertEquals(2, clusterState.getGuiTaskList().size()); clusterState.removeTask(defaultTaskInfo); assertEquals(1, clusterState.getGuiTaskList().size()); }
private static List<Protos.TaskInfo> clearTaskIds(Collection<Protos.TaskInfo> taskInfos) { List<Protos.TaskInfo> outTaskInfos = new ArrayList<>(); for (Protos.TaskInfo restartTaskInfo : taskInfos) { outTaskInfos.add( Protos.TaskInfo.newBuilder(restartTaskInfo) .setTaskId(Protos.TaskID.newBuilder().setValue("")) .build()); } return outTaskInfos; }
private Protos.TaskInfo normalizeCassandraTaskInfo(CassandraDaemonTask daemonTask) { Protos.TaskInfo daemonTaskInfo = daemonTask.getTaskInfo(); Protos.ExecutorInfo expectedExecutorInfo = Protos.ExecutorInfo.newBuilder(daemonTaskInfo.getExecutor()) .setExecutorId(Protos.ExecutorID.newBuilder().setValue("")) .build(); daemonTaskInfo = Protos.TaskInfo.newBuilder(daemonTaskInfo) .setTaskId(Protos.TaskID.newBuilder().setValue("")) .setExecutor(expectedExecutorInfo) .build(); return daemonTaskInfo; }
@Override public CleanupTask update(Protos.TaskState state) { return new CleanupTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public CassandraDaemonTask update(Protos.TaskState state) { return new CassandraDaemonTask( getBuilder() .setData(getData() .withState(state) .getBytes()).build()); }
@Override public DownloadSnapshotTask update(Protos.TaskState state) { return new DownloadSnapshotTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
private static Protos.TaskInfo getTestTaskInfo() { return Protos.TaskInfo.newBuilder() .setName(testTaskName) .setTaskId(Protos.TaskID.newBuilder().setValue(testTaskId)) .setSlaveId(Protos.SlaveID.newBuilder().setValue(testAgentId)) .build(); } }