Examples with CallMetadataAuthentication com.netflix.titus.runtime.endpoint.metadata.spring.CallMetadataAuthentication used on opensource projects

Example 1 with CallMetadataAuthentication

use of com.netflix.titus.runtime.endpoint.metadata.spring.CallMetadataAuthentication in project titus-control-plane by Netflix.

the class AppAutoScalingCallbackSpringResourceTest method testResponseEntity.

@Test
public void testResponseEntity() {
    AppAutoScalingCallbackService appAutoScalingCallbackService = mock(AppAutoScalingCallbackService.class);
    String jobId = "job-1";
    ScalableTargetResourceInfo scalableTargetResourceInfo = ScalableTargetResourceInfo.newBuilder().actualCapacity(4).desiredCapacity(6).scalingStatus(DefaultAppAutoScalingCallbackService.ScalingStatus.Pending.name()).build();
    CallMetadata callMetadata = CallMetadata.newBuilder().withCallerId("unit-testing").build();
    CallMetadataAuthentication callMetadataAuthentication = mock(CallMetadataAuthentication.class);
    when(callMetadataAuthentication.getCallMetadata()).thenReturn(callMetadata);
    when(appAutoScalingCallbackService.setScalableTargetResourceInfo(jobId, scalableTargetResourceInfo, callMetadata)).thenReturn(Observable.just(scalableTargetResourceInfo));
    AppAutoScalingCallbackSpringResource springResource = new AppAutoScalingCallbackSpringResource(appAutoScalingCallbackService);
    ResponseEntity<ScalableTargetResourceInfo> resp = springResource.setScalableTargetResourceInfo(jobId, scalableTargetResourceInfo, callMetadataAuthentication);
    assertThat(resp).isNotNull();
    assertThat(resp.getStatusCodeValue()).isEqualTo(200);
    assertThat(resp.getBody()).isEqualTo(scalableTargetResourceInfo);
    scalableTargetResourceInfo.setDesiredCapacity(-1);
    ResponseEntity<ScalableTargetResourceInfo> resp2 = springResource.setScalableTargetResourceInfo(jobId, scalableTargetResourceInfo, callMetadataAuthentication);
    assertThat(resp2).isNotNull();
    assertThat(resp2.getStatusCodeValue()).isEqualTo(400);
}