From de0d29a3240ac2f87a466ea2a7c0906bd619ed58 Mon Sep 17 00:00:00 2001 From: Danny van Bruggen Date: Wed, 17 Jan 2018 21:10:22 +0100 Subject: Add deducation of the name of the primary type --- .../com/github/javaparser/ast/CompilationUnitTest.java | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'javaparser-testing/src') diff --git a/javaparser-testing/src/test/java/com/github/javaparser/ast/CompilationUnitTest.java b/javaparser-testing/src/test/java/com/github/javaparser/ast/CompilationUnitTest.java index b29be112c..ec17e39b7 100644 --- a/javaparser-testing/src/test/java/com/github/javaparser/ast/CompilationUnitTest.java +++ b/javaparser-testing/src/test/java/com/github/javaparser/ast/CompilationUnitTest.java @@ -87,4 +87,22 @@ public class CompilationUnitTest { assertEquals(false, cu.getPrimaryTypeName().isPresent()); } + @Test + public void testGetPrimaryType() throws IOException { + Path sourceRoot = mavenModuleRoot(CompilationUnitTest.class).resolve(Paths.get("src", "test", "resources")).normalize(); + Path testFile = sourceRoot.resolve(Paths.get("com", "github", "javaparser", "storage", "PrimaryType.java")); + CompilationUnit cu = JavaParser.parse(testFile); + + assertEquals("PrimaryType", cu.getPrimaryType().get().getNameAsString()); + } + + @Test + public void testNoPrimaryType() throws IOException { + Path sourceRoot = mavenModuleRoot(CompilationUnitTest.class).resolve(Paths.get("src", "test", "resources")).normalize(); + Path testFile = sourceRoot.resolve(Paths.get("com", "github", "javaparser", "storage", "PrimaryType2.java")); + CompilationUnit cu = JavaParser.parse(testFile); + + assertEquals(false, cu.getPrimaryType().isPresent()); + } + } -- cgit v1.2.3