Add another test case and TODO

aspect-build · Oct 12, 2023 · 80d3113 · 80d3113
1 parent 6487760
commit 80d3113
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/gazelle/kotlin/parser/parser.go b/gazelle/kotlin/parser/parser.go
@@ -90,6 +90,7 @@ func (p *treeSitterParser) Parse(filePath, source string) (*ParseResult, []error
 			} else if nodeI.Type() == "package_header" {
 				if result.Package != "" {
 					fmt.Printf("Multiple package declarations found in %s\n", filePath)
+					// TODO: Don't exit here and return an error instead?
 					os.Exit(1)
 				}
 

diff --git a/gazelle/kotlin/parser/parser_test.go b/gazelle/kotlin/parser/parser_test.go
@@ -52,6 +52,20 @@ import /* fdsa */ d/* asdf */.* // w
 		pkg:      "x",
 		imports:  []string{"a", "c", "d"},
 	},
+	{
+		desc: "value class",
+		kt: `
+import a.b.C
+import c.d.E as EEE
+
+// Maybe xyz.numbers should be an import?
+@JvmInline
+value class Energy(val kwh: xyz.numbers.Double) { fun thing(): Unit {}}
+	`,
+		filename: "simple.kt",
+		pkg:      "",
+		imports:  []string{"a.b", "c.d"},
+	},
 }
 
 func TestTreesitterParser(t *testing.T) {