Add test case for inference test of parent class

Fraunhofer-AISEC · Dec 12, 2024 · dedce78 · dedce78
1 parent cc094be
commit dedce78
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 7 deletions.
diff --git a/cpg-language-cxx/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/cxx/CXXInferenceTest.kt b/cpg-language-cxx/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/cxx/CXXInferenceTest.kt
@@ -32,13 +32,7 @@ import de.fraunhofer.aisec.cpg.graph.scopes.GlobalScope
 import de.fraunhofer.aisec.cpg.graph.types.BooleanType
 import de.fraunhofer.aisec.cpg.test.*
 import java.io.File
-import kotlin.test.Test
-import kotlin.test.assertContains
-import kotlin.test.assertEquals
-import kotlin.test.assertIs
-import kotlin.test.assertIsNot
-import kotlin.test.assertNotNull
-import kotlin.test.assertTrue
+import kotlin.test.*
 
 class CXXInferenceTest {
     @Test
@@ -199,4 +193,32 @@ class CXXInferenceTest {
             assertEquals(pairType, pair.returnTypes.singleOrNull())
         }
     }
+
+    @Test
+    fun testInferParentClassInNamespace() {
+        val file = File("src/test/resources/cxx/inference/parent_inference.cpp")
+        val tu =
+            analyzeAndGetFirstTU(listOf(file), file.parentFile.toPath(), true) {
+                it.registerLanguage<CPPLanguage>()
+                it.loadIncludes(false)
+                it.addIncludesToGraph(false)
+            }
+        assertNotNull(tu)
+
+        val util = tu.namespaces["ABC"]
+        assertNotNull(util)
+
+        val recordABCA = util.records["A"]
+        assertNotNull(recordABCA)
+        assertTrue(recordABCA.isInferred)
+
+        val recordA = tu.records["A"]
+        assertNotNull(recordA)
+        val funcFoo = recordA.functions["foo"]
+        assertNotNull(funcFoo)
+        assertTrue(funcFoo.isInferred)
+        val funcBar = recordA.functions["bar"]
+        assertNotNull(funcBar)
+        assertFalse(funcBar.isInferred)
+    }
 }
diff --git a/cpg-language-cxx/src/test/resources/cxx/inference/parent_inference.cpp b/cpg-language-cxx/src/test/resources/cxx/inference/parent_inference.cpp
@@ -0,0 +1,18 @@
+/*
+namespace ABC {
+struct A {
+    A();
+    void foo();
+};
+}
+*/
+
+struct A : ABC::A {
+    A() {
+        foo();
+        bar();
+    }
+    void bar() {
+
+    }
+};