private static NicSecurityGroupsTaskState buildStartState( BaseModelTest test, NicSecurityGroupsRequest.OperationRequestType requestType, InstanceAdapterTestTypes instanceAdapterType) throws Throwable { NicSecurityGroupsTaskState state = new NicSecurityGroupsTaskState(); state.requestType = requestType; state.networkInterfaceLink = createNetworkInterface(test).documentSelfLink; state.securityGroupLinks = Arrays .asList(createSecurityGroup(test, UUID.randomUUID().toString()).documentSelfLink); switch (instanceAdapterType) { case SUCCESS: state.adapterReference = UriUtils .buildUri(test.getHost(), MockAdapter.MockNetworkInterfaceSecurityGroupsSuccessAdapter.SELF_LINK); break; case FAILURE: state.adapterReference = UriUtils .buildUri(test.getHost(), MockAdapter.MockNetworkInterfaceSecurityGroupsFailureAdapter.SELF_LINK); break; default: state.adapterReference = null; } return state; }
@Override public void handleRequest(Operation op) { if (!op.hasBody()) { op.fail(new IllegalArgumentException("body is required")); return; } switch (op.getAction()) { case PATCH: NicSecurityGroupsRequest request = op .getBody(NicSecurityGroupsRequest.class); NicSecurityGroupsTaskState taskState = new NicSecurityGroupsTaskState(); taskState.taskInfo = new TaskState(); taskState.taskInfo.stage = TaskState.TaskStage.FINISHED; sendRequest(Operation.createPatch( request.taskReference).setBody( taskState)); op.complete(); break; default: super.handleRequest(op); } } }
@Override public void handleRequest(Operation op) { if (!op.hasBody()) { op.fail(new IllegalArgumentException("body is required")); return; } switch (op.getAction()) { case PATCH: NicSecurityGroupsRequest request = op .getBody(NicSecurityGroupsRequest.class); NicSecurityGroupsTaskState taskState = new NicSecurityGroupsTaskState(); taskState.taskInfo = createFailedTaskInfo(); sendRequest(Operation.createPatch( request.taskReference).setBody( taskState)); op.complete(); break; default: super.handleRequest(op); } } }
private void sendSelfPatch(TaskState.TaskStage stage, Throwable e) { NicSecurityGroupsTaskState body = new NicSecurityGroupsTaskState(); body.taskInfo = new TaskState(); if (e == null) { body.taskInfo.stage = stage; } else { body.taskInfo.stage = TaskState.TaskStage.FAILED; body.taskInfo.failure = Utils.toServiceErrorResponse(e); logWarning(() -> String.format("Patching to failed: %s", Utils.toString(e))); } sendSelfPatch(body); } }