use of com.netflix.titus.grpc.protogen.TaskAttributesUpdate in project titus-control-plane by Netflix.
the class JobManagementSpringResourceTest method testUpdateTaskAttributes.
@Test
public void testUpdateTaskAttributes() throws Exception {
TaskAttributesUpdate restRequest = TaskAttributesUpdate.newBuilder().setTaskId(TASK_ID_1).putAttributes("keyA", "valueA").build();
when(jobServiceGatewayMock.updateTaskAttributes(restRequest, JUNIT_REST_CALL_METADATA)).thenReturn(Completable.complete());
SpringMockMvcUtil.doPut(mockMvc, String.format("/api/v3/tasks/%s/attributes", TASK_ID_1), restRequest);
verify(jobServiceGatewayMock, times(1)).updateTaskAttributes(restRequest, JUNIT_REST_CALL_METADATA);
}
Aggregations