Improve resolving symbols on struct field container types

1 files changed, 79 insertions, 9 deletions

diff --git a/tests/references_test.odin b/tests/references_test.odin
index 2464924..c698771 100644
--- a/tests/references_test.odin
+++ b/tests/references_test.odin
@@ -977,26 +977,47 @@ ast_reference_struct_field_enumerated_array :: proc(t: ^testing.T) {
 }
 
 @(test)
-ast_reference_struct_field_map :: proc(t: ^testing.T) {
+ast_reference_struct_field_map_key :: proc(t: ^testing.T) {
 	source := test.Source {
 		main = `package test
 
-		Foo :: enum {
-			A,
-			B,
+		Foo :: int
+
+		Bar :: struct {}
+
+		Bazz :: struct {
+			bars: map[Fo{*}o]Bar
 		}
+		`,
+	}
 
-		Bar :: struct {
-			foos: map[F{*}oo]Bazz
+	locations := []common.Location {
+		{range = {start = {line = 2, character = 2}, end = {line = 2, character = 5}}},
+		{range = {start = {line = 7, character = 13}, end = {line = 7, character = 16}}},
+	}
+
+	test.expect_reference_locations(t, &source, locations[:])
+}
+
+@(test)
+ast_reference_struct_field_map_value :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: int
+
+		Bar :: struct {}
+
+		Bazz :: struct {
+			bars: map[Foo]B{*}ar
 		}
 
-		Bazz :: struct {}
 		`,
 	}
 
 	locations := []common.Location {
-		{range = {start = {line = 2, character = 2}, end = {line = 2, character = 5}}},
-		{range = {start = {line = 8, character = 13}, end = {line = 8, character = 16}}},
+		{range = {start = {line = 4, character = 2}, end = {line = 4, character = 5}}},
+		{range = {start = {line = 7, character = 17}, end = {line = 7, character = 20}}},
 	}
 
 	test.expect_reference_locations(t, &source, locations[:])
@@ -1262,3 +1283,52 @@ ast_references_switch_cases_binary_expr :: proc(t: ^testing.T) {
 
 	test.expect_reference_locations(t, &source, locations[:])
 }
+
+@(test)
+ast_reference_struct_field_matrix_row :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: int
+
+		Bar :: struct {}
+
+		Bazz :: struct {
+			bars: matrix[Fo{*}o, 2]Bar
+		}
+
+		`,
+	}
+
+	locations := []common.Location {
+		{range = {start = {line = 2, character = 2}, end = {line = 2, character = 5}}},
+		{range = {start = {line = 7, character = 16}, end = {line = 7, character = 19}}},
+	}
+
+	test.expect_reference_locations(t, &source, locations[:])
+}
+
+@(test)
+ast_reference_struct_field_bitfield_backing_type :: proc(t: ^testing.T) {
+	source := test.Source {
+		main = `package test
+
+		Foo :: int
+
+		Bar :: struct {}
+
+		Bazz :: struct {
+			bars: bit_field Fo{*}o {
+			}
+		}
+
+		`,
+	}
+
+	locations := []common.Location {
+		{range = {start = {line = 2, character = 2}, end = {line = 2, character = 5}}},
+		{range = {start = {line = 7, character = 19}, end = {line = 7, character = 22}}},
+	}
+
+	test.expect_reference_locations(t, &source, locations[:])
+}