def test_type_visitor(self):
cu = parse(TypeVisitor.java_source)
tv = TypeVisitor()
tv.visit(cu, None)
assertTrue(ReferenceType in [type(n) for n in tv.getNodes()])
names = [str(node) for node in tv.getNodes()]
assertTrue("VoidVisitorAdapter" in names)
assertTrue("PyObject" in names)
assertTrue("WildcardType" in names)
def test_annotation_visitor(self):
cu = parse(TypeVisitor.java_source)
av = AnnotationVisitor()
av.visit(cu, None)
assertTrue(str(av.getNodes()[0]) == "@Override")
