@Test
public void testSetJobDisruptionBudget() throws Exception {

    JobDisruptionBudget restRequest = toGrpcDisruptionBudget(DisruptionBudgetGenerator.budgetSelfManagedBasic());
    JobDisruptionBudgetUpdate forwardedRequest = JobDisruptionBudgetUpdate.newBuilder()
            .setJobId(JOB_ID_1)
            .setDisruptionBudget(restRequest)
            .build();
    when(jobServiceGatewayMock.updateJobDisruptionBudget(forwardedRequest, JUNIT_REST_CALL_METADATA)).thenReturn(Mono.empty());
    SpringMockMvcUtil.doPut(mockMvc, String.format("/api/v3/jobs/%s/disruptionBudget", JOB_ID_1), restRequest);
    verify(jobServiceGatewayMock, times(1)).updateJobDisruptionBudget(forwardedRequest, JUNIT_REST_CALL_METADATA);
}