Java Code Examples for org.eclipse.xtext.nodemodel.ICompositeNode#getAsTreeIterable()

@Test
public void testAllGrammarElementsUnique() throws Exception {
	ContentAssistFragmentTestLanguageRoot result = parseHelper.parse("newArrayList()");
	XtextResource res = (XtextResource) result.eResource();
	ICompositeNode root = res.getParseResult().getRootNode();
	new InvariantChecker().checkInvariant(root);
	Set<EObject> set = new HashSet<>();
	for (INode node : root.getAsTreeIterable()) {
		if (node instanceof ICompositeNode) {
			ICompositeNode compositeNode = (ICompositeNode) node;
			if (compositeNode.getGrammarElement() != null) {
				Assert.assertTrue(compositeNode.getGrammarElement().toString(),
						set.add(compositeNode.getGrammarElement()));
			} else {
				Assert.fail("node without grammar element");
			}
		}
	}
}
 

@Test
public void testAllGrammarElementsUniqueAfterReparse() throws Exception {
	ContentAssistFragmentTestLanguageRoot result = parseHelper.parse("");
	XtextResource res = (XtextResource) result.eResource();
	res.update(0, 0, "newArrayList()");
	ICompositeNode root = res.getParseResult().getRootNode();
	Set<EObject> set = new HashSet<>();
	for (INode node : root.getAsTreeIterable()) {
		if (node instanceof ICompositeNode) {
			ICompositeNode compositeNode = (ICompositeNode) node;
			if (compositeNode.getGrammarElement() != null) {
				Assert.assertTrue(compositeNode.getGrammarElement().toString(),
						set.add(compositeNode.getGrammarElement()));
			} else {
				Assert.fail("node without grammar element");
			}
		}
	}
}
 

@Test 
public void testAllGrammarElementsUnique() throws Exception {
	XtendClass clazz = clazz("class Foo { def m() { newArrayList() } }");
	XtextResource resource = (XtextResource) clazz.eResource();
	ICompositeNode root = resource.getParseResult().getRootNode();
	new InvariantChecker().checkInvariant(root);
	assertSame(root, root.getRootNode());
	Set<EObject> grammarElements = Sets.newHashSet();
	for(INode node: root.getAsTreeIterable()) {
		if (node instanceof ICompositeNode) {
			if (node.getGrammarElement() == null) {
				fail("node without grammar element");
			}
			if (!grammarElements.add(node.getGrammarElement())) {
				fail(node.getGrammarElement().toString());
			}
		}
	}
}
 

@Test 
public void testAllGrammarElementsUniqueAfterReparse() throws Exception {
	String text = "class Foo { def m() { newArrayList() } }";
	XtendClass clazz = clazz(text);
	XtextResource resource = (XtextResource) clazz.eResource();
	resource.update(text.indexOf('m'), 0, "m");
	ICompositeNode root = resource.getParseResult().getRootNode();
	assertSame(root, root.getRootNode());
	Set<EObject> grammarElements = Sets.newHashSet();
	for(INode node: root.getAsTreeIterable()) {
		if (node instanceof ICompositeNode) {
			if (node.getGrammarElement() == null) {
				fail("node without grammar element");
			}
			if (!grammarElements.add(node.getGrammarElement())) {
				fail(node.getGrammarElement().toString());
			}
		}
	}
}
 

/**
 * Retrieves a comment for the given object.
 * 
 * @param object
 *          an object
 * @return the comment corresponding to the closest JavaDoc-like comment
 */
protected String findComment(final EObject object) {
  String returnValue = null;
  ICompositeNode node = NodeModelUtils.getNode(object);
  if (node != null) {
    // get the last multi line comment before a non hidden leaf node
    for (INode abstractNode : node.getAsTreeIterable()) {
      String comment = getJavaDocComment(abstractNode);
      if (comment != null) {
        returnValue = comment;
        break;
      }
    }
  }
  return returnValue;
}
 

protected Set<INode> getLeadingCommentsIncludingWhitespace(ICompositeNode node) {
	for (INode child : node.getAsTreeIterable()) {
		if (child instanceof ILeafNode && !((ILeafNode) child).isHidden()) {
			return getLeadingCommentsIncludingWhitespace((ILeafNode) child);
		}
	}
	return Sets.newHashSet();
}