From 876bea653d2db5e0df75f6549da02f089e1f727b Mon Sep 17 00:00:00 2001
From: Tamas Vajk <tamasvajk@github.com>
Date: Tue, 4 Oct 2022 13:29:15 +0200
Subject: [PATCH] Kotlin: Add test case for missing enum constants

---
 java/ql/test/kotlin/library-tests/enum/enumUser.kt   | 2 ++
 java/ql/test/kotlin/library-tests/enum/test.expected | 2 ++
 java/ql/test/kotlin/library-tests/enum/test.ql       | 2 ++
 3 files changed, 6 insertions(+)

diff --git a/java/ql/test/kotlin/library-tests/enum/enumUser.kt b/java/ql/test/kotlin/library-tests/enum/enumUser.kt
index a0b938cbb79..c684d2d1998 100644
--- a/java/ql/test/kotlin/library-tests/enum/enumUser.kt
+++ b/java/ql/test/kotlin/library-tests/enum/enumUser.kt
@@ -1 +1,3 @@
 fun usesEnum(e: Enum<*>) = e.ordinal.toString() + e.name
+
+enum class E { A, B, C }
diff --git a/java/ql/test/kotlin/library-tests/enum/test.expected b/java/ql/test/kotlin/library-tests/enum/test.expected
index c410e68ddd2..51d983047a2 100644
--- a/java/ql/test/kotlin/library-tests/enum/test.expected
+++ b/java/ql/test/kotlin/library-tests/enum/test.expected
@@ -1,3 +1,5 @@
+enumConstants
+#select
 | addAll |
 | addRange |
 | allOf |
diff --git a/java/ql/test/kotlin/library-tests/enum/test.ql b/java/ql/test/kotlin/library-tests/enum/test.ql
index b255cc61bd6..8d83314e5c3 100644
--- a/java/ql/test/kotlin/library-tests/enum/test.ql
+++ b/java/ql/test/kotlin/library-tests/enum/test.ql
@@ -3,3 +3,5 @@ import java
 from Method m
 where m.getDeclaringType().getName().matches("Enum%")
 select m.getName()
+
+query predicate enumConstants(EnumConstant ec) { ec.fromSource() }