Fix unity build (#1986)

* Add test case for unity build * Fix unity file generation
Fraunhofer-AISEC · Jan 25, 2025 · c4037bb · c4037bb
1 parent d0d7243
commit c4037bb
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 1 deletion.
diff --git a/cpg-core/src/main/kotlin/de/fraunhofer/aisec/cpg/TranslationManager.kt b/cpg-core/src/main/kotlin/de/fraunhofer/aisec/cpg/TranslationManager.kt
@@ -221,7 +221,7 @@ private constructor(
                                         .trimIndent()
                                 )
                             }
-                                ?: {
+                                ?: run {
                                     writer.write(
                                         """
 #include "${file.absolutePath}"

diff --git a/cpg-language-cxx/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/cxx/CXXIncludeTest.kt b/cpg-language-cxx/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/cxx/CXXIncludeTest.kt
@@ -260,4 +260,47 @@ internal class CXXIncludeTest : BaseTest() {
         assertEquals(2, tu.methods.size)
         assertEquals(1, tu.methods.filterIsInstance<ConstructorDeclaration>().size)
     }
+
+    @Test
+    @Throws(Exception::class)
+    fun testUnityBuild() {
+        val file = File("src/test/resources/include.cpp")
+        val tus =
+            analyzeWithBuilder(
+                TranslationConfiguration.builder()
+                    .sourceLocations(listOf(file))
+                    .loadIncludes(true)
+                    .useUnityBuild(true)
+                    .debugParser(true)
+                    .registerLanguage<CPPLanguage>()
+                    .failOnError(true)
+            )
+        assertNotNull(tus)
+
+        val tu = tus.firstOrNull()
+        assertNotNull(tu)
+        assertFalse(tu.records.isEmpty())
+    }
+
+    @Test
+    @Throws(Exception::class)
+    fun testUnityBuildWithComponent() {
+        val file = File("src/test/resources/include.cpp")
+        val tus =
+            analyzeWithBuilder(
+                TranslationConfiguration.builder()
+                    .sourceLocations(listOf(file))
+                    .topLevel(file.parentFile)
+                    .loadIncludes(true)
+                    .useUnityBuild(true)
+                    .debugParser(true)
+                    .registerLanguage<CPPLanguage>()
+                    .failOnError(true)
+            )
+        assertNotNull(tus)
+
+        val tu = tus.firstOrNull()
+        assertNotNull(tu)
+        assertFalse(tu.records.isEmpty())
+    }
 }