QueryCreateHandler extends BaseManagementHandler { @Override public Result handleRequest(Request request) { Result queryResult = getReadHandler().handleRequest(request); if (queryResult.getStatus().isErrorState() || queryResult.getResultTree().getChildren().isEmpty()) { return queryResult; } Map<Resource.Type, Set<Map<String, Object>>> mapProperties; try { mapProperties = buildCreateSet(request, queryResult); } catch (IllegalArgumentException e) { return createInvalidRequestResult(e.getMessage()); } if (mapProperties.size() != 1) { return createInvalidRequestResult(mapProperties.size() == 0 ? "A minimum of one sub-resource must be specified for creation." : "Multiple sub-resource types may not be created in the same request."); } final Map.Entry<Resource.Type, Set<Map<String, Object>>> entry = mapProperties.entrySet().iterator().next(); ResourceInstance createResource = getResourceFactory().createResource( entry.getKey(), request.getResource().getKeyValueMap()); RequestBody requestBody = new RequestBody(); requestBody.setBody(request.getBody().getBody()); for (Map<String, Object> map : entry.getValue()) { requestBody.addPropertySet(new NamedPropertySet("", map)); } return persist(createResource, requestBody ); }  @Override Result handleRequest(Request request);  }
@Test public void testHandleRequest() throws Exception { final String BODY_STRING = "Body string"; Request request = createNiceMock(Request.class); RequestBody body = createNiceMock(RequestBody.class); ResourceInstance resourceInstance = createNiceMock(ResourceInstance.class); ResourceDefinition resourceDefinition = createNiceMock(ResourceDefinition.class); ResourceInstanceFactory resourceInstanceFactory = createNiceMock(ResourceInstanceFactory.class); Query query = createNiceMock(Query.class); Predicate predicate = createNiceMock(Predicate.class); Result result = createNiceMock(Result.class); ResourceInstance subResource = createNiceMock(ResourceInstance.class); ResourceDefinition subResourceDefinition = createNiceMock(ResourceDefinition.class); ClusterController controller = createNiceMock(ClusterController.class); Schema serviceSchema = createNiceMock(Schema.class); Schema componentSchema = createNiceMock(Schema.class); String resourceKeyProperty = "resourceKeyProperty"; String createKeyProperty = "createKeyProperty"; Resource resource1 = createNiceMock(Resource.class); Resource resource2 = createNiceMock(Resource.class); PersistenceManager pm = createNiceMock(PersistenceManager.class); ResourceInstance createResource = createNiceMock(ResourceInstance.class); RequestStatus status = createNiceMock(RequestStatus.class); Resource statusResource1 = createNiceMock(Resource.class); Resource statusResource2 = createNiceMock(Resource.class); RequestHandler readHandler = createStrictMock(RequestHandler.class); ResultStatus resultStatus = createNiceMock(ResultStatus.class); Capture<RequestBody> bodyCapture = EasyMock.newCapture(); Map<Resource.Type, String> mapIds = new HashMap<Resource.Type, String>(); Set<NamedPropertySet> setRequestProps = new HashSet<NamedPropertySet>(); Map<String, Object> mapProperties = new HashMap<String, Object>(); Set<Map<String, Object>> arraySet = new HashSet<Map<String, Object>>(); mapProperties.put("components", arraySet); Map<String, Object> map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "SECONDARY_NAMENODE"); arraySet.add(map); map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "HDFS_CLIENT"); arraySet.add(map); setRequestProps.add(new NamedPropertySet("", mapProperties)); Set<Map<String, Object>> setCreateProps = new HashSet<Map<String, Object>>(); Map<String, Object> map1 = new HashMap<String, Object>(); map1.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "SECONDARY_NAMENODE"); map1.put(createKeyProperty, "id1"); setCreateProps.add(map1); Map<String, Object> map2 = new HashMap<String, Object>(); map2.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "SECONDARY_NAMENODE"); map2.put(createKeyProperty, "id2"); setCreateProps.add(map2); Map<String, Object> map3 = new HashMap<String, Object>(); map3.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "HDFS_CLIENT"); map3.put(createKeyProperty, "id1"); setCreateProps.add(map3); Map<String, Object> map4 = new HashMap<String, Object>(); map4.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "HDFS_CLIENT"); map4.put(createKeyProperty, "id2"); setCreateProps.add(map4); Map<String, ResourceInstance> mapSubResources = new HashMap<String, ResourceInstance>(); mapSubResources.put("components", subResource); TreeNode<Resource> resultTree = new TreeNodeImpl<Resource>(null, null, "result"); resultTree.addChild(resource1, "resource1"); resultTree.addChild(resource2, "resource2"); Set<Resource> setStatusResources = new HashSet<Resource>(); setStatusResources.add(statusResource1); setStatusResources.add(statusResource2); expect(readHandler.handleRequest(request)).andReturn(result); expect(result.getStatus()).andReturn(resultStatus).anyTimes(); expect(resultStatus.isErrorState()).andReturn(false); expect(result.getResultTree()).andReturn(resultTree); expect(body.getBody()).andReturn(BODY_STRING).anyTimes(); expect(request.getResource()).andReturn(resourceInstance).anyTimes(); expect(request.getBody()).andReturn(body).anyTimes(); expect(body.getNamedPropertySets()).andReturn(setRequestProps).anyTimes(); expect(resourceInstance.getResourceDefinition()).andReturn(resourceDefinition).anyTimes(); expect(resourceInstance.getKeyValueMap()).andReturn(mapIds).anyTimes(); expect(resourceInstance.getSubResources()).andReturn(mapSubResources).anyTimes(); expect(resourceDefinition.getType()).andReturn(Resource.Type.Service).anyTimes(); expect(resourceDefinition.isCreatable()).andReturn(true).anyTimes(); expect(subResource.getResourceDefinition()).andReturn(subResourceDefinition).anyTimes(); expect(subResourceDefinition.getType()).andReturn(Resource.Type.Component).anyTimes(); expect(controller.getSchema(Resource.Type.Service)).andReturn(serviceSchema).anyTimes(); expect(controller.getSchema(Resource.Type.Component)).andReturn(componentSchema).anyTimes(); expect(serviceSchema.getKeyPropertyId(Resource.Type.Service)).andReturn(resourceKeyProperty).anyTimes(); expect(componentSchema.getKeyPropertyId(Resource.Type.Service)).andReturn(createKeyProperty).anyTimes(); expect(result.getResultTree()).andReturn(resultTree).anyTimes(); expect(resource1.getPropertyValue(resourceKeyProperty)).andReturn("id1").anyTimes(); expect(resource2.getPropertyValue(resourceKeyProperty)).andReturn("id2").anyTimes(); expect(resourceInstanceFactory.createResource(Resource.Type.Component, mapIds)). andReturn(createResource).anyTimes(); expect(createResource.getResourceDefinition()).andReturn(resourceDefinition).anyTimes(); expect(pm.create(same(createResource), capture(bodyCapture))).andReturn(status); expect(status.getStatus()).andReturn(RequestStatus.Status.Complete).anyTimes(); expect(status.getAssociatedResources()).andReturn(setStatusResources).anyTimes(); expect(statusResource1.getType()).andReturn(Resource.Type.Component).anyTimes(); expect(statusResource2.getType()).andReturn(Resource.Type.Component).anyTimes(); replay(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, resourceInstanceFactory, createResource, status, statusResource1, statusResource2, readHandler, resultStatus); Result testResult = new TestQueryCreateHandler(resourceInstanceFactory, controller, pm, readHandler). handleRequest(request); Collection<TreeNode<Resource>> children = testResult.getResultTree().getChild("resources").getChildren(); assertEquals(2, children.size()); boolean containsStatusResource1 = false; boolean containsStatusResource2 = false; for (TreeNode<Resource> child : children) { Resource r = child.getObject(); if (r == statusResource1) { containsStatusResource1 = true; } else if(r == statusResource2) { containsStatusResource2 = true; } } assertTrue(containsStatusResource1); assertTrue(containsStatusResource2); assertEquals(ResultStatus.STATUS.CREATED, testResult.getStatus().getStatus()); RequestBody createBody = bodyCapture.getValue(); assertEquals(BODY_STRING, createBody.getBody()); assertEquals(4, createBody.getPropertySets().size()); assertEquals(setCreateProps, createBody.getPropertySets()); verify(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, resourceInstanceFactory, createResource, status, statusResource1, statusResource2, readHandler, resultStatus); }
@Test public void tesHandleRequest_NoSubResourceNameSpecified() { Request request = createNiceMock(Request.class); RequestBody body = createNiceMock(RequestBody.class); ResourceInstance resourceInstance = createNiceMock(ResourceInstance.class); ResourceDefinition resourceDefinition = createNiceMock(ResourceDefinition.class); Query query = createNiceMock(Query.class); Predicate predicate = createNiceMock(Predicate.class); Result result = createNiceMock(Result.class); ResourceInstance subResource = createNiceMock(ResourceInstance.class); ResourceDefinition subResourceDefinition = createNiceMock(ResourceDefinition.class); ClusterController controller = createNiceMock(ClusterController.class); Schema serviceSchema = createNiceMock(Schema.class); Schema componentSchema = createNiceMock(Schema.class); String resourceKeyProperty = "resourceKeyProperty"; Resource resource1 = createNiceMock(Resource.class); Resource resource2 = createNiceMock(Resource.class); PersistenceManager pm = createNiceMock(PersistenceManager.class); ResourceInstance createResource = createNiceMock(ResourceInstance.class); RequestStatus status = createNiceMock(RequestStatus.class); Resource statusResource1 = createNiceMock(Resource.class); Resource statusResource2 = createNiceMock(Resource.class); RequestHandler readHandler = createStrictMock(RequestHandler.class); ResultStatus queryResultStatus = createNiceMock(ResultStatus.class); Map<Resource.Type, String> mapIds = new HashMap<Resource.Type, String>(); Set<NamedPropertySet> setRequestProps = new HashSet<NamedPropertySet>(); Map<String, Object> mapProperties = new HashMap<String, Object>(); Set<Map<String, Object>> arraySet = new HashSet<Map<String, Object>>(); mapProperties.put("", arraySet); Map<String, Object> map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "SECONDARY_NAMENODE"); arraySet.add(map); map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "HDFS_CLIENT"); arraySet.add(map); setRequestProps.add(new NamedPropertySet("", mapProperties)); TreeNode<Resource> resultTree = new TreeNodeImpl<Resource>(null, null, "result"); resultTree.addChild(resource1, "resource1"); resultTree.addChild(resource2, "resource2"); expect(readHandler.handleRequest(request)).andReturn(result); expect(result.getStatus()).andReturn(queryResultStatus).anyTimes(); expect(queryResultStatus.isErrorState()).andReturn(false); expect(result.getResultTree()).andReturn(resultTree); expect(request.getResource()).andReturn(resourceInstance).anyTimes(); expect(request.getBody()).andReturn(body).anyTimes(); expect(body.getNamedPropertySets()).andReturn(setRequestProps).anyTimes(); expect(resourceInstance.getResourceDefinition()).andReturn(resourceDefinition).anyTimes(); expect(resourceInstance.getKeyValueMap()).andReturn(mapIds).anyTimes(); expect(resourceDefinition.getType()).andReturn(Resource.Type.Service).anyTimes(); expect(controller.getSchema(Resource.Type.Service)).andReturn(serviceSchema).anyTimes(); expect(result.getResultTree()).andReturn(resultTree).anyTimes(); expect(resource1.getPropertyValue(resourceKeyProperty)).andReturn("id1").anyTimes(); expect(resource2.getPropertyValue(resourceKeyProperty)).andReturn("id2").anyTimes(); replay(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); Result testResult = new TestQueryCreateHandler(null, controller, pm, readHandler). handleRequest(request); ResultStatus resultStatus = testResult.getStatus(); assertEquals(ResultStatus.STATUS.BAD_REQUEST, resultStatus.getStatus()); assertEquals("Invalid Request: A sub-resource name must be supplied.", resultStatus.getMessage()); verify(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); }
@Test public void tesHandleRequest_InvalidSubResSpecified() { Request request = createNiceMock(Request.class); RequestBody body = createNiceMock(RequestBody.class); ResourceInstance resourceInstance = createNiceMock(ResourceInstance.class); ResourceDefinition resourceDefinition = createNiceMock(ResourceDefinition.class); Query query = createNiceMock(Query.class); Predicate predicate = createNiceMock(Predicate.class); Result result = createNiceMock(Result.class); ResourceInstance subResource = createNiceMock(ResourceInstance.class); ResourceDefinition subResourceDefinition = createNiceMock(ResourceDefinition.class); ClusterController controller = createNiceMock(ClusterController.class); Schema serviceSchema = createNiceMock(Schema.class); Schema componentSchema = createNiceMock(Schema.class); String resourceKeyProperty = "resourceKeyProperty"; Resource resource1 = createNiceMock(Resource.class); Resource resource2 = createNiceMock(Resource.class); PersistenceManager pm = createNiceMock(PersistenceManager.class); ResourceInstance createResource = createNiceMock(ResourceInstance.class); RequestStatus status = createNiceMock(RequestStatus.class); Resource statusResource1 = createNiceMock(Resource.class); Resource statusResource2 = createNiceMock(Resource.class); RequestHandler readHandler = createStrictMock(RequestHandler.class); ResultStatus queryResultStatus = createNiceMock(ResultStatus.class); Map<Resource.Type, String> mapIds = new HashMap<Resource.Type, String>(); Set<NamedPropertySet> setRequestProps = new HashSet<NamedPropertySet>(); Map<String, Object> mapProperties = new HashMap<String, Object>(); Set<Map<String, Object>> arraySet = new HashSet<Map<String, Object>>(); mapProperties.put("INVALID", arraySet); Map<String, Object> map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "SECONDARY_NAMENODE"); arraySet.add(map); map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "HDFS_CLIENT"); arraySet.add(map); setRequestProps.add(new NamedPropertySet("", mapProperties)); Map<String, ResourceInstance> mapSubResources = new HashMap<String, ResourceInstance>(); mapSubResources.put("components", subResource); TreeNode<Resource> resultTree = new TreeNodeImpl<Resource>(null, null, "result"); resultTree.addChild(resource1, "resource1"); resultTree.addChild(resource2, "resource2"); expect(readHandler.handleRequest(request)).andReturn(result); expect(result.getStatus()).andReturn(queryResultStatus).anyTimes(); expect(queryResultStatus.isErrorState()).andReturn(false); expect(result.getResultTree()).andReturn(resultTree); expect(request.getResource()).andReturn(resourceInstance).anyTimes(); expect(request.getBody()).andReturn(body).anyTimes(); expect(body.getNamedPropertySets()).andReturn(setRequestProps).anyTimes(); expect(resourceInstance.getResourceDefinition()).andReturn(resourceDefinition).anyTimes(); expect(resourceInstance.getKeyValueMap()).andReturn(mapIds).anyTimes(); expect(resourceInstance.getSubResources()).andReturn(mapSubResources).anyTimes(); expect(resourceDefinition.getType()).andReturn(Resource.Type.Service).anyTimes(); expect(controller.getSchema(Resource.Type.Service)).andReturn(serviceSchema).anyTimes(); expect(result.getResultTree()).andReturn(resultTree).anyTimes(); expect(resource1.getPropertyValue(resourceKeyProperty)).andReturn("id1").anyTimes(); expect(resource2.getPropertyValue(resourceKeyProperty)).andReturn("id2").anyTimes(); replay(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); Result testResult = new TestQueryCreateHandler(null, controller, pm, readHandler). handleRequest(request); ResultStatus resultStatus = testResult.getStatus(); assertEquals(ResultStatus.STATUS.BAD_REQUEST, resultStatus.getStatus()); assertEquals("Invalid Request: The specified sub-resource name is not valid: 'INVALID'.", resultStatus.getMessage()); verify(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); }
@Test public void tesHandleRequest_NoSubResourcesSpecified() { Request request = createNiceMock(Request.class); RequestBody body = createNiceMock(RequestBody.class); ResourceInstance resourceInstance = createNiceMock(ResourceInstance.class); ResourceDefinition resourceDefinition = createNiceMock(ResourceDefinition.class); Query query = createNiceMock(Query.class); Predicate predicate = createNiceMock(Predicate.class); Result result = createNiceMock(Result.class); ResourceInstance subResource = createNiceMock(ResourceInstance.class); ResourceDefinition subResourceDefinition = createNiceMock(ResourceDefinition.class); ClusterController controller = createNiceMock(ClusterController.class); Schema serviceSchema = createNiceMock(Schema.class); Schema componentSchema = createNiceMock(Schema.class); String resourceKeyProperty = "resourceKeyProperty"; Resource resource1 = createNiceMock(Resource.class); Resource resource2 = createNiceMock(Resource.class); PersistenceManager pm = createNiceMock(PersistenceManager.class); ResourceInstance createResource = createNiceMock(ResourceInstance.class); RequestStatus status = createNiceMock(RequestStatus.class); Resource statusResource1 = createNiceMock(Resource.class); Resource statusResource2 = createNiceMock(Resource.class); RequestHandler readHandler = createStrictMock(RequestHandler.class); ResultStatus queryResultStatus = createNiceMock(ResultStatus.class); Map<Resource.Type, String> mapIds = new HashMap<Resource.Type, String>(); Set<NamedPropertySet> setRequestProps = new HashSet<NamedPropertySet>(); TreeNode<Resource> resultTree = new TreeNodeImpl<Resource>(null, null, "result"); resultTree.addChild(resource1, "resource1"); resultTree.addChild(resource2, "resource2"); expect(readHandler.handleRequest(request)).andReturn(result); expect(result.getStatus()).andReturn(queryResultStatus).anyTimes(); expect(queryResultStatus.isErrorState()).andReturn(false); expect(result.getResultTree()).andReturn(resultTree); expect(request.getResource()).andReturn(resourceInstance).anyTimes(); expect(request.getBody()).andReturn(body).anyTimes(); expect(body.getNamedPropertySets()).andReturn(setRequestProps).anyTimes(); expect(resourceInstance.getResourceDefinition()).andReturn(resourceDefinition).anyTimes(); expect(resourceInstance.getKeyValueMap()).andReturn(mapIds).anyTimes(); expect(resourceDefinition.getType()).andReturn(Resource.Type.Service).anyTimes(); expect(controller.getSchema(Resource.Type.Service)).andReturn(serviceSchema).anyTimes(); expect(result.getResultTree()).andReturn(resultTree).anyTimes(); expect(resource1.getPropertyValue(resourceKeyProperty)).andReturn("id1").anyTimes(); expect(resource2.getPropertyValue(resourceKeyProperty)).andReturn("id2").anyTimes(); replay(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); Result testResult = new TestQueryCreateHandler(null, controller, pm, readHandler). handleRequest(request); ResultStatus resultStatus = testResult.getStatus(); assertEquals(ResultStatus.STATUS.BAD_REQUEST, resultStatus.getStatus()); assertEquals("Invalid Request: A minimum of one sub-resource must be specified for creation.", resultStatus.getMessage()); verify(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); }
@Test public void testHandleRequest_MultipleSubResources() throws Exception { Request request = createNiceMock(Request.class); RequestBody body = createNiceMock(RequestBody.class); ResourceInstance resourceInstance = createNiceMock(ResourceInstance.class); ResourceDefinition resourceDefinition = createNiceMock(ResourceDefinition.class); ResourceInstanceFactory resourceInstanceFactory = createNiceMock(ResourceInstanceFactory.class); Query query = createNiceMock(Query.class); Predicate predicate = createNiceMock(Predicate.class); Result result = createNiceMock(Result.class); ResourceInstance subResource1 = createNiceMock(ResourceInstance.class); ResourceInstance subResource2 = createNiceMock(ResourceInstance.class); ResourceDefinition subResourceDefinition1 = createNiceMock(ResourceDefinition.class); ResourceDefinition subResourceDefinition2 = createNiceMock(ResourceDefinition.class); ClusterController controller = createNiceMock(ClusterController.class); Schema serviceSchema = createNiceMock(Schema.class); Schema subResSchema1 = createNiceMock(Schema.class); Schema subResSchema2 = createNiceMock(Schema.class); String resourceKeyProperty = "resourceKeyProperty"; String createKeyProperty = "createKeyProperty"; Resource resource1 = createNiceMock(Resource.class); Resource resource2 = createNiceMock(Resource.class); PersistenceManager pm = createNiceMock(PersistenceManager.class); ResourceInstance createResource = createNiceMock(ResourceInstance.class); RequestStatus status = createNiceMock(RequestStatus.class); Resource statusResource1 = createNiceMock(Resource.class); Resource statusResource2 = createNiceMock(Resource.class); RequestHandler readHandler = createStrictMock(RequestHandler.class); ResultStatus queryResultStatus = createNiceMock(ResultStatus.class); Map<Resource.Type, String> mapIds = new HashMap<Resource.Type, String>(); Set<NamedPropertySet> setRequestProps = new HashSet<NamedPropertySet>(); Map<String, Object> mapProperties = new HashMap<String, Object>(); Set<Map<String, Object>> arraySet = new HashSet<Map<String, Object>>(); mapProperties.put("foo", arraySet); Map<String, Object> map = new HashMap<String, Object>(); map.put("prop", "val"); arraySet.add(map); arraySet = new HashSet<Map<String, Object>>(); mapProperties.put("bar", arraySet); map = new HashMap<String, Object>(); map.put("prop", "val"); arraySet.add(map); setRequestProps.add(new NamedPropertySet("", mapProperties)); Map<String, ResourceInstance> mapSubResources = new HashMap<String, ResourceInstance>(); mapSubResources.put("foo", subResource1); mapSubResources.put("bar", subResource2); TreeNode<Resource> resultTree = new TreeNodeImpl<Resource>(null, null, "result"); resultTree.addChild(resource1, "resource1"); resultTree.addChild(resource2, "resource2"); expect(readHandler.handleRequest(request)).andReturn(result); expect(result.getStatus()).andReturn(queryResultStatus).anyTimes(); expect(queryResultStatus.isErrorState()).andReturn(false); expect(result.getResultTree()).andReturn(resultTree); expect(request.getResource()).andReturn(resourceInstance).anyTimes(); expect(request.getBody()).andReturn(body).anyTimes(); expect(body.getNamedPropertySets()).andReturn(setRequestProps).anyTimes(); expect(resourceInstance.getResourceDefinition()).andReturn(resourceDefinition).anyTimes(); expect(resourceInstance.getKeyValueMap()).andReturn(mapIds).anyTimes(); expect(resourceInstance.getSubResources()).andReturn(mapSubResources).anyTimes(); expect(resourceDefinition.getType()).andReturn(Resource.Type.Service).anyTimes(); expect(subResource1.getResourceDefinition()).andReturn(subResourceDefinition1).anyTimes(); expect(subResourceDefinition1.getType()).andReturn(Resource.Type.Component).anyTimes(); expect(subResource2.getResourceDefinition()).andReturn(subResourceDefinition2).anyTimes(); expect(subResourceDefinition2.getType()).andReturn(Resource.Type.HostComponent).anyTimes(); expect(controller.getSchema(Resource.Type.Service)).andReturn(serviceSchema).anyTimes(); expect(controller.getSchema(Resource.Type.Component)).andReturn(subResSchema1).anyTimes(); expect(controller.getSchema(Resource.Type.HostComponent)).andReturn(subResSchema2).anyTimes(); expect(serviceSchema.getKeyPropertyId(Resource.Type.Service)).andReturn(resourceKeyProperty).anyTimes(); expect(subResSchema1.getKeyPropertyId(Resource.Type.Service)).andReturn(createKeyProperty).anyTimes(); expect(subResSchema2.getKeyPropertyId(Resource.Type.Service)).andReturn(createKeyProperty).anyTimes(); expect(result.getResultTree()).andReturn(resultTree).anyTimes(); expect(resource1.getPropertyValue(resourceKeyProperty)).andReturn("id1").anyTimes(); expect(resource2.getPropertyValue(resourceKeyProperty)).andReturn("id2").anyTimes(); replay(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource1, subResource2, subResourceDefinition1, subResourceDefinition2, controller, serviceSchema, subResSchema1, subResSchema2, resource1, resource2, pm, resourceInstanceFactory, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); Result testResult = new TestQueryCreateHandler(resourceInstanceFactory, controller, pm, readHandler). handleRequest(request); ResultStatus resultStatus = testResult.getStatus(); assertEquals(ResultStatus.STATUS.BAD_REQUEST, resultStatus.getStatus()); assertEquals("Invalid Request: Multiple sub-resource types may not be created in the same request.", resultStatus.getMessage()); verify(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource1, subResource2, subResourceDefinition1, subResourceDefinition2, controller, serviceSchema, subResSchema1, subResSchema2, resource1, resource2, pm, resourceInstanceFactory, createResource, status, statusResource1, statusResource2, readHandler, queryResultStatus); }
@Test public void testHandleRequest_AuthorizationFailure() throws Exception { final String BODY_STRING = "Body string"; Request request = createMock(Request.class); RequestBody body = createMock(RequestBody.class); ResourceInstance resourceInstance = createMock(ResourceInstance.class); ResourceDefinition resourceDefinition = createMock(ResourceDefinition.class); ResourceInstanceFactory resourceInstanceFactory = createMock(ResourceInstanceFactory.class); Query query = createMock(Query.class); Predicate predicate = createMock(Predicate.class); Result result = createMock(Result.class); ResourceInstance subResource = createMock(ResourceInstance.class); ResourceDefinition subResourceDefinition = createMock(ResourceDefinition.class); ClusterController controller = createMock(ClusterController.class); Schema serviceSchema = createMock(Schema.class); Schema componentSchema = createMock(Schema.class); String resourceKeyProperty = "resourceKeyProperty"; String createKeyProperty = "createKeyProperty"; Resource resource1 = createMock(Resource.class); Resource resource2 = createMock(Resource.class); PersistenceManager pm = createMock(PersistenceManager.class); ResourceInstance createResource = createMock(ResourceInstance.class); RequestHandler readHandler = createStrictMock(RequestHandler.class); ResultStatus resultStatus = createMock(ResultStatus.class); Map<Resource.Type, String> mapIds = new HashMap<Resource.Type, String>(); Set<NamedPropertySet> setRequestProps = new HashSet<NamedPropertySet>(); Map<String, Object> mapProperties = new HashMap<String, Object>(); Set<Map<String, Object>> arraySet = new HashSet<Map<String, Object>>(); mapProperties.put("components", arraySet); Map<String, Object> map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "SECONDARY_NAMENODE"); arraySet.add(map); map = new HashMap<String, Object>(); map.put(PropertyHelper.getPropertyId("ServiceComponentInfo", "component_name"), "HDFS_CLIENT"); arraySet.add(map); setRequestProps.add(new NamedPropertySet("", mapProperties)); Map<String, ResourceInstance> mapSubResources = new HashMap<String, ResourceInstance>(); mapSubResources.put("components", subResource); TreeNode<Resource> resultTree = new TreeNodeImpl<Resource>(null, null, "result"); resultTree.addChild(resource1, "resource1"); resultTree.addChild(resource2, "resource2"); expect(readHandler.handleRequest(request)).andReturn(result).atLeastOnce(); expect(result.getStatus()).andReturn(resultStatus).atLeastOnce(); expect(resultStatus.isErrorState()).andReturn(false).atLeastOnce(); expect(body.getBody()).andReturn(BODY_STRING).atLeastOnce(); expect(request.getResource()).andReturn(resourceInstance).atLeastOnce(); expect(request.getBody()).andReturn(body).atLeastOnce(); expect(body.getNamedPropertySets()).andReturn(setRequestProps).atLeastOnce(); expect(resourceInstance.getResourceDefinition()).andReturn(resourceDefinition).atLeastOnce(); expect(resourceInstance.getKeyValueMap()).andReturn(mapIds).atLeastOnce(); expect(resourceInstance.getSubResources()).andReturn(mapSubResources).atLeastOnce(); expect(resourceDefinition.getType()).andReturn(Resource.Type.Service).atLeastOnce(); expect(subResource.getResourceDefinition()).andReturn(subResourceDefinition).atLeastOnce(); expect(subResourceDefinition.getType()).andReturn(Resource.Type.Component).atLeastOnce(); expect(controller.getSchema(Resource.Type.Service)).andReturn(serviceSchema).atLeastOnce(); expect(controller.getSchema(Resource.Type.Component)).andReturn(componentSchema).atLeastOnce(); expect(serviceSchema.getKeyPropertyId(Resource.Type.Service)).andReturn(resourceKeyProperty).atLeastOnce(); expect(componentSchema.getKeyPropertyId(Resource.Type.Service)).andReturn(createKeyProperty).atLeastOnce(); expect(result.getResultTree()).andReturn(resultTree).atLeastOnce(); expect(resource1.getPropertyValue(resourceKeyProperty)).andReturn("id1").atLeastOnce(); expect(resource2.getPropertyValue(resourceKeyProperty)).andReturn("id2").atLeastOnce(); expect(resourceInstanceFactory.createResource(Resource.Type.Component, mapIds)). andReturn(createResource).atLeastOnce(); expect(pm.create(anyObject(ResourceInstance.class), anyObject(RequestBody.class))).andThrow(new AuthorizationException()); replay(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, resourceInstanceFactory, createResource, readHandler, resultStatus); Result testResult = new TestQueryCreateHandler(resourceInstanceFactory, controller, pm, readHandler). handleRequest(request); assertEquals(ResultStatus.STATUS.FORBIDDEN, testResult.getStatus().getStatus()); verify(request, body, resourceInstance, resourceDefinition, query, predicate, result, subResource, subResourceDefinition, controller, serviceSchema, componentSchema, resource1, resource2, pm, resourceInstanceFactory, createResource, readHandler, resultStatus); }