Impelement language agnostic files() method for ProjectFiles

hadii-tech · Dec 14, 2022 · b8a99c7 · b8a99c7
1 parent 98aaa52
commit b8a99c7
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 1 deletion.
diff --git a/pom.xml b/pom.xml
@@ -3,7 +3,7 @@
 
 	<groupId>com.github.hadii-tech</groupId>
 	<artifactId>clarpse</artifactId>
-	<version>7.1.0</version>
+	<version>7.1.1</version>
 	<packaging>jar</packaging>
 
 	<name>${project.groupId}:${project.artifactId}</name>

diff --git a/src/main/java/com/hadii/clarpse/compiler/ProjectFiles.java b/src/main/java/com/hadii/clarpse/compiler/ProjectFiles.java
@@ -173,6 +173,12 @@ public final Collection<ProjectFile> files(Lang language) {
         return this.langToFilesMap.getOrDefault(language, new ArrayList<>());
     }
 
+    public final Collection<ProjectFile> files() {
+        Set<ProjectFile> allFiles = new HashSet<>();
+        this.langToFilesMap.forEach((lang, files) -> allFiles.addAll(files));
+        return allFiles;
+    }
+
     public String projectDir() {
         if (this.projectDir != null && !this.projectDir.isEmpty()) {
             return this.projectDir;

diff --git a/src/test/java/com/hadii/test/ProjectFilesTest.java b/src/test/java/com/hadii/test/ProjectFilesTest.java
@@ -136,6 +136,47 @@ public void testProjectFilesSize() {
         assertEquals(2, pfs.size());
     }
 
+    @Test
+    public void testGetAllFiles() {
+        ProjectFiles pfs = new ProjectFiles();
+        pfs.insertFile(new ProjectFile("/test.java", "{}"));
+        pfs.insertFile(new ProjectFile("/tester.go", "{}"));
+        assertEquals(2, pfs.files().size());
+    }
+
+    @Test
+    public void testGetAllGoFiles() {
+        ProjectFiles pfs = new ProjectFiles();
+        pfs.insertFile(new ProjectFile("/test.java", "{}"));
+        pfs.insertFile(new ProjectFile("/tester.go", "{}"));
+        assertEquals(1, pfs.files(Lang.GOLANG).size());
+    }
+
+    @Test
+    public void testGetAllFilesForNonExistentLang() {
+        ProjectFiles pfs = new ProjectFiles();
+        pfs.insertFile(new ProjectFile("/test.java", "{}"));
+        pfs.insertFile(new ProjectFile("/tester.go", "{}"));
+        assertEquals(0, pfs.files(Lang.JAVASCRIPT).size());
+    }
+
+    @Test
+    public void testGetAllJavaFiles() {
+        ProjectFiles pfs = new ProjectFiles();
+        pfs.insertFile(new ProjectFile("/test.java", "{}"));
+        pfs.insertFile(new ProjectFile("/tester.go", "{}"));
+        assertEquals(1, pfs.files(Lang.JAVA).size());
+    }
+
+    @Test
+    public void testGetAllJSFiles() {
+        ProjectFiles pfs = new ProjectFiles();
+        pfs.insertFile(new ProjectFile("/test.js", "{}"));
+        pfs.insertFile(new ProjectFile("/tester.go", "{}"));
+        assertEquals(1, pfs.files(Lang.JAVASCRIPT).size());
+    }
+
+
     @Test
     public void testEmptyProjectFilesSize() {
         ProjectFiles pfs = new ProjectFiles();