- OperationResult operationResult = mock(OperationResult.class);
- when(aaiClient.getResource(any(), any(), any())).thenReturn(operationResult);
- when(operationResult.getResultCode()).thenReturn(Response.Status.BAD_REQUEST.getStatusCode());
-
- checkRollback(Collections.singletonList(new ModelArtifact()));
- }
-
- @Test
- public void testNullResourceModelResult() {
- when(config.getAaiBaseUrl()).thenReturn("");
- when(config.getAaiModelUrl(any())).thenReturn("");
- when(aaiClient.getResource(any(), any(), any())).thenReturn(null);
+ ResponseEntity operationResult = mock(ResponseEntity.class);
+ when(aaiClient.getResource(any(), any(), any(), any())).thenReturn(operationResult);
+ when(operationResult.getStatusCode()).thenReturn(HttpStatus.BAD_REQUEST);