Correctly infer return type enum fields for completions

1 files changed, 54 insertions, 0 deletions

diff --git a/tests/completions_test.odin b/tests/completions_test.odin
index bae9476..21f10a7 100644
--- a/tests/completions_test.odin
+++ b/tests/completions_test.odin
@@ -3973,3 +3973,57 @@ ast_completion_bitset_named_proc_arg_should_remove_already_used :: proc(t: ^test
 	}
 	test.expect_completion_details(t, &source, "", {"B"}, {"A"})
 }
+
+@(test)
+ast_completion_return_comp_lit_enum :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: enum {
+			A,
+			B,
+		}
+
+		Bar :: struct {
+			foo: Foo,
+		}
+
+		foo :: proc() -> Bar {
+			return {
+				foo = .{*}
+			}
+		}
+		`,
+	}
+	test.expect_completion_details(t, &source, "", {"A", "B"})
+}
+
+@(test)
+ast_completion_return_nested_comp_lit_enum :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: enum {
+			A,
+			B,
+		}
+
+		Bar :: struct {
+			foo: Foo,
+		}
+
+		Bazz :: struct {
+			bar: Bar,
+		}
+
+		foo :: proc() -> Bazz {
+			return {
+				bar = {
+					foo = .{*}
+				}
+			}
+		}
+		`,
+	}
+	test.expect_completion_details(t, &source, "", {"A", "B"})
+}