From 973aedab4e744a41b5be874247d296d72ebeacb3 Mon Sep 17 00:00:00 2001
From: ricekot <git@ricekot.com>
Date: Wed, 15 Jan 2025 18:47:31 +0530
Subject: [PATCH] [javasrc2cpg] Parse enums correctly when keepTypeArguments is
 enabled

Signed-off-by: ricekot <git@ricekot.com>
---
 .../io/joern/javasrc2cpg/querying/VarDeclTests.scala  | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/joern-cli/frontends/javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/querying/VarDeclTests.scala b/joern-cli/frontends/javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/querying/VarDeclTests.scala
index a14cfa4e4781..9ff6e174b751 100644
--- a/joern-cli/frontends/javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/querying/VarDeclTests.scala
+++ b/joern-cli/frontends/javasrc2cpg/src/test/scala/io/joern/javasrc2cpg/querying/VarDeclTests.scala
@@ -206,6 +206,17 @@ class VarDeclTests extends JavaSrcCode2CpgFixture {
         .head shouldBe "org.apache.flink.streaming.connectors.kafka.FlinkKafkaProducer.<init>:<unresolvedSignature>(3)"
     }
 
+    "parse enums correctly" in {
+      val cpg = code("""
+          |public enum Color {
+          |  RED,
+          |  GREEN,
+          |  BLUE
+          |}
+          |""".stripMargin).withConfig(Config().withKeepTypeArguments(true))
+
+      cpg.typeDecl("Color").code.head shouldBe "public enum Color"
+    }
   }
 
 }