Improve test: Use PackageDeclaration in conjunction with CU

javaparser · Sep 1, 2018 · 9c82d65 · 9c82d65
1 parent 85ac387
commit 9c82d65
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/...ser-symbol-solver-testing/src/test/java/com/github/javaparser/symbolsolver/Issue1814.java b/...ser-symbol-solver-testing/src/test/java/com/github/javaparser/symbolsolver/Issue1814.java
@@ -31,12 +31,12 @@ public class Issue1814 extends AbstractResolutionTest {
 
     @Before
     public void setup() {
+        final CompilationUnit compilationUnit = new CompilationUnit();
+        compilationUnit.setPackageDeclaration("java.lang");
         // construct a fake java.lang.Object class with only one method (java.lang.Object#equals(java.lang.Object)
-        ClassOrInterfaceDeclaration fakeObject = new ClassOrInterfaceDeclaration();
-        fakeObject.setName(new SimpleName("java.lang.Object"));
-
-        final MethodDeclaration equals = fakeObject.addMethod("equals", Modifier.PUBLIC);
-        equals.addParameter("java.lang.Object", "obj");
+        final ClassOrInterfaceDeclaration clazz = compilationUnit.addClass("Object", Modifier.PUBLIC);
+        final MethodDeclaration equals = clazz.addMethod("equals", Modifier.PUBLIC);
+        equals.addParameter("Object", "obj");
         final BlockStmt body = new BlockStmt();
         body.addStatement("return this == obj;");
         equals.setBody(body);
@@ -55,7 +55,7 @@ public void setParent(TypeSolver parent) {
             public SymbolReference<ResolvedReferenceTypeDeclaration> tryToSolveType(String name) {
                 if ("java.lang.Object".equals(name)) {
                     // custom handling
-                    return SymbolReference.solved(new JavaParserClassDeclaration(fakeObject, this));
+                    return SymbolReference.solved(new JavaParserClassDeclaration(clazz, this));
                 }
 
                 return SymbolReference.unsolved(ResolvedReferenceTypeDeclaration.class);