@Test public void testAddChildTypesToQuery() { ClassType parentType = mock(ClassType.class); when(parentType.getName()).thenReturn("Section"); ClassType childType = mock(ClassType.class); when(childType.getName()).thenReturn("StudentSectionAssociation"); ClassType nonChildType = mock(ClassType.class); when(nonChildType.getName()).thenReturn("StudentParentAssociation"); SelectorQuery childQuery = new SelectorQuery(); childQuery.put(childType, new SelectorQueryPlan()); List<Object> childQueryPlans = new ArrayList<Object>(); childQueryPlans.add(childQuery); SelectorQueryPlan plan = new SelectorQueryPlan(); plan.setChildQueryPlans(childQueryPlans); NeutralQuery query = new NeutralQuery(); defaultSelectorDocument.addChildTypesToQuery(parentType, plan, query); assertEquals("Should match", 1, query.getEmbeddedFields().size()); assertEquals("Should match", "studentSectionAssociation", query.getEmbeddedFields().get(0)); query = new NeutralQuery(); defaultSelectorDocument.addChildTypesToQuery(parentType, new SelectorQueryPlan(), query); assertEquals("Should match", 0, query.getEmbeddedFields().size()); childQuery = new SelectorQuery(); childQuery.put(nonChildType, new SelectorQueryPlan()); childQueryPlans = new ArrayList<Object>(); childQueryPlans.add(childQuery); plan = new SelectorQueryPlan(); plan.setChildQueryPlans(childQueryPlans); query = new NeutralQuery(); defaultSelectorDocument.addChildTypesToQuery(parentType, new SelectorQueryPlan(), query); assertEquals("Should match", 0, query.getEmbeddedFields().size()); }
protected void addChildTypesToQuery(Type currentType, SelectorQueryPlan selectorQueryPlan, NeutralQuery neutralQuery) { List<Object> childQueries = selectorQueryPlan.getChildQueryPlans(); List<String> embeddedFields = new ArrayList<String>(); for (Object plan : childQueries) { SelectorQuery query = (SelectorQuery) plan; for (Type key : query.keySet()) { String childType = StringUtils.uncapitalise(key.getName()); if (currentType.getName().equalsIgnoreCase(EmbeddedDocumentRelations.getParentEntityType(childType))) { embeddedFields.add(childType); } } } neutralQuery.setEmbeddedFields(embeddedFields); }
DefaultSelectorDocument implements SelectorDocument { protected void addChildTypesToQuery(Type currentType, SelectorQueryPlan selectorQueryPlan, NeutralQuery neutralQuery) { List<Object> childQueries = selectorQueryPlan.getChildQueryPlans(); List<String> embeddedFields = new ArrayList<String>(); for (Object plan : childQueries) { SelectorQuery query = (SelectorQuery) plan; for (Type key : query.keySet()) { String childType = StringUtils.uncapitalise(key.getName()); if (currentType.getName().equalsIgnoreCase(EmbeddedDocumentRelations.getParentEntityType(childType))) { embeddedFields.add(childType); } } } neutralQuery.setEmbeddedFields(embeddedFields); } }
DefaultSelectorDocument implements SelectorDocument { protected void addChildTypesToQuery(Type currentType, SelectorQueryPlan selectorQueryPlan, NeutralQuery neutralQuery) { List<Object> childQueries = selectorQueryPlan.getChildQueryPlans(); List<String> embeddedFields = new ArrayList<String>(); for (Object plan : childQueries) { SelectorQuery query = (SelectorQuery) plan; for (Type key : query.keySet()) { String childType = StringUtils.uncapitalise(key.getName()); if (currentType.getName().equalsIgnoreCase(EmbeddedDocumentRelations.getParentEntityType(childType))) { embeddedFields.add(childType); } } } neutralQuery.setEmbeddedFields(embeddedFields); }  }
DefaultSelectorDocument implements SelectorDocument { protected void addChildTypesToQuery(Type currentType, SelectorQueryPlan selectorQueryPlan, NeutralQuery neutralQuery) { List<Object> childQueries = selectorQueryPlan.getChildQueryPlans(); List<String> embeddedFields = new ArrayList<String>(); for (Object plan : childQueries) { SelectorQuery query = (SelectorQuery) plan; for (Type key : query.keySet()) { String childType = StringUtils.uncapitalise(key.getName()); if (currentType.getName().equalsIgnoreCase(EmbeddedDocumentRelations.getParentEntityType(childType))) { embeddedFields.add(childType); } } } neutralQuery.setEmbeddedFields(embeddedFields); }  @Override List<EntityBody> aggregate(SelectorQuery selectorQuery, final NeutralQuery constraint); List<EntityBody> getEntitySubList(List<EntityBody> list, String field, String value); }
DefaultSelectorDocument implements SelectorDocument { protected void addChildTypesToQuery(Type currentType, SelectorQueryPlan selectorQueryPlan, NeutralQuery neutralQuery) { List<Object> childQueries = selectorQueryPlan.getChildQueryPlans(); List<String> embeddedFields = new ArrayList<String>(); for (Object plan : childQueries) { SelectorQuery query = (SelectorQuery) plan; for (Type key : query.keySet()) { String childType = StringUtils.uncapitalise(key.getName()); if (currentType.getName().equalsIgnoreCase(EmbeddedDocumentRelations.getParentEntityType(childType))) { embeddedFields.add(childType); } } } neutralQuery.setEmbeddedFields(embeddedFields); }  @Override List<EntityBody> aggregate(SelectorQuery selectorQuery, final NeutralQuery constraint); List<EntityBody> getEntitySubList(List<EntityBody> list, String field, String value); static final int EMBEDDED_DOCUMENT_LIMIT; }