public class QueryCreateHandlerTest {
@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 = new Capture<RequestBody>();
// test request body
// {
// "components" : [
// { "ServiceComponentInfo" : {
// "component_name" : "SECONDARY_NAMENODE"
// }
// },
// { "ServiceComponentInfo" : {
// "component_name" : "HDFS_CLIENT"
// }
// }
// ]
// }
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);
//expectations
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();