Add extension tests

tamasvajk · igfoo · commit 6dbf2782694a · 2022-05-10T19:51:08.000+01:00
diff --git a/java/ql/test/kotlin/library-tests/extensions/extensions.kt b/java/ql/test/kotlin/library-tests/extensions/extensions.kt
@@ -0,0 +1,16 @@
+
+class SomeClass {
+    fun someClassMethod() {}
+}
+class AnotherClass {
+    fun anotherClassMethod() {}
+}
+
+fun SomeClass.someFun() {}
+fun AnotherClass.anotherFun() {}
+
+fun SomeClass.bothFun() {}
+fun AnotherClass.bothFun() {}
+
+fun SomeClass.bothFunDiffTypes(): Int { return 5 }
+fun AnotherClass.bothFunDiffTypes(): String { return "Foo" }
diff --git a/java/ql/test/kotlin/library-tests/extensions/methods.expected b/java/ql/test/kotlin/library-tests/extensions/methods.expected
@@ -0,0 +1,16 @@
+| extensions.kt:2:1:4:1 | <obinit> | file://<external>/Unit.class:0:0:0:0 | Unit |
+| extensions.kt:2:1:4:1 | equals | file://:0:0:0:0 | boolean |
+| extensions.kt:2:1:4:1 | hashCode | file://:0:0:0:0 | int |
+| extensions.kt:2:1:4:1 | toString | file://<external>/String.class:0:0:0:0 | String |
+| extensions.kt:3:5:3:28 | someClassMethod | file://:0:0:0:0 | void |
+| extensions.kt:5:1:7:1 | <obinit> | file://<external>/Unit.class:0:0:0:0 | Unit |
+| extensions.kt:5:1:7:1 | equals | file://:0:0:0:0 | boolean |
+| extensions.kt:5:1:7:1 | hashCode | file://:0:0:0:0 | int |
+| extensions.kt:5:1:7:1 | toString | file://<external>/String.class:0:0:0:0 | String |
+| extensions.kt:6:5:6:31 | anotherClassMethod | file://:0:0:0:0 | void |
+| extensions.kt:9:1:9:26 | someFun | file://:0:0:0:0 | void |
+| extensions.kt:10:1:10:32 | anotherFun | file://:0:0:0:0 | void |
+| extensions.kt:12:1:12:26 | bothFun | file://:0:0:0:0 | void |
+| extensions.kt:13:1:13:29 | bothFun | file://:0:0:0:0 | void |
+| extensions.kt:15:1:15:50 | bothFunDiffTypes | file://:0:0:0:0 | int |
+| extensions.kt:16:1:16:60 | bothFunDiffTypes | file://<external>/String.class:0:0:0:0 | String |
diff --git a/java/ql/test/kotlin/library-tests/extensions/methods.ql b/java/ql/test/kotlin/library-tests/extensions/methods.ql
@@ -0,0 +1,16 @@
+import java
+
+// Stop external filepaths from appearing in the results
+class ClassLocation extends Class {
+  override predicate hasLocationInfo(string path, int sl, int sc, int el, int ec) {
+    exists(string fullPath |
+           super.hasLocationInfo(fullPath, sl, sc, el, ec) |
+           if exists(this.getFile().getRelativePath())
+           then path = fullPath
+           else path = fullPath.regexpReplaceAll(".*/", "<external>/"))
+  }
+}
+
+from Method m
+where m.fromSource()
+select m, m.getReturnType()
