Added unit tests

Author: jimbethancourt

codebase-graph-builder/src/test/java/org/hjug/graphbuilder/visitor/JavaNewClassVisitorFullTest.java

@@ -0,0 +1,69 @@
+package org.hjug.graphbuilder.visitor;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.List;
+import java.util.stream.Collectors;
+import org.jgrapht.Graph;
+import org.jgrapht.graph.DefaultWeightedEdge;
+import org.jgrapht.graph.SimpleDirectedWeightedGraph;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.openrewrite.ExecutionContext;
+import org.openrewrite.InMemoryExecutionContext;
+import org.openrewrite.java.JavaParser;
+
+public class JavaNewClassVisitorFullTest {
+
+    @Test
+    void visitNewClass() throws IOException {
+
+        File srcDirectory = new File("src/test/java/org/hjug/graphbuilder/visitor/testclasses/newClass");
+
+        JavaParser javaParser = JavaParser.fromJavaVersion().build();
+        ExecutionContext ctx = new InMemoryExecutionContext(Throwable::printStackTrace);
+
+        Graph<String, DefaultWeightedEdge> classReferencesGraph =
+                new SimpleDirectedWeightedGraph<>(DefaultWeightedEdge.class);
+
+        Graph<String, DefaultWeightedEdge> packageReferencesGraph =
+                new SimpleDirectedWeightedGraph<>(DefaultWeightedEdge.class);
+
+        final JavaVisitor<ExecutionContext> javaVisitor =
+                new JavaVisitor<>(classReferencesGraph, packageReferencesGraph);
+        final JavaVariableTypeVisitor<ExecutionContext> javaVariableTypeVisitor =
+                new JavaVariableTypeVisitor<>(classReferencesGraph, packageReferencesGraph);
+        final JavaMethodDeclarationVisitor<ExecutionContext> javaMethodDeclarationVisitor =
+                new JavaMethodDeclarationVisitor<>(classReferencesGraph, packageReferencesGraph);
+
+        List<Path> list = Files.walk(Paths.get(srcDirectory.getAbsolutePath())).collect(Collectors.toList());
+
+        // Parse sources with all visitors, not only
+        javaParser.parse(list, Paths.get(srcDirectory.getAbsolutePath()), ctx).forEach(cu -> {
+            javaVisitor.visit(cu, ctx);
+            javaVariableTypeVisitor.visit(cu, ctx);
+            javaMethodDeclarationVisitor.visit(cu, ctx);
+        });
+
+        Graph<String, DefaultWeightedEdge> graph = javaVisitor.getClassReferencesGraph();
+        Assertions.assertTrue(graph.containsVertex("org.hjug.graphbuilder.visitor.testclasses.newClass.A"));
+        Assertions.assertTrue(graph.containsVertex("org.hjug.graphbuilder.visitor.testclasses.newClass.B"));
+        Assertions.assertTrue(graph.containsVertex("org.hjug.graphbuilder.visitor.testclasses.newClass.C"));
+
+        // capturing counts of all types
+        Assertions.assertEquals(
+                6,
+                graph.getEdgeWeight(graph.getEdge(
+                        "org.hjug.graphbuilder.visitor.testclasses.newClass.A",
+                        "org.hjug.graphbuilder.visitor.testclasses.newClass.B")));
+
+        Assertions.assertEquals(
+                3,
+                graph.getEdgeWeight(graph.getEdge(
+                        "org.hjug.graphbuilder.visitor.testclasses.newClass.A",
+                        "org.hjug.graphbuilder.visitor.testclasses.newClass.C")));
+    }
+}

codebase-graph-builder/src/test/java/org/hjug/graphbuilder/visitor/JavaNewClassVisitorTest.java

@@ -48,7 +48,13 @@ void visitNewClass() throws IOException {
         Assertions.assertTrue(graph.containsVertex("org.hjug.graphbuilder.visitor.testclasses.newClass.B"));
         Assertions.assertTrue(graph.containsVertex("org.hjug.graphbuilder.visitor.testclasses.newClass.C"));
 
-        // TODO: Investigate further to confirm correctness
+        // only looking for what was visited by classDeclarationVisitor and variableTypeVisitor
+        Assertions.assertEquals(
+                5,
+                graph.getEdgeWeight(graph.getEdge(
+                        "org.hjug.graphbuilder.visitor.testclasses.newClass.A",
+                        "org.hjug.graphbuilder.visitor.testclasses.newClass.B")));
+
         Assertions.assertEquals(
                 3,
                 graph.getEdgeWeight(graph.getEdge(

codebase-graph-builder/src/test/java/org/hjug/graphbuilder/visitor/testclasses/newClass/A.java

@@ -1,11 +1,19 @@
 package org.hjug.graphbuilder.visitor.testclasses.newClass;
 
+import java.util.ArrayList;
+import java.util.List;
+
 public class A {
 
     B newClassMethod() {
         new C();
         C c = new C();
 
+        // var treated like "B", counts as 2
+        var b = new B(null);
+        // <> treated like <B>, counts as 2
+        List<B> listB = new ArrayList<>();
+
         // TODO: add visitor for J.ReturnType
         return new B(c);
     }