refactor(vscode): improve lsp tests (#4342)

Author: benfdking

sqlmesh/lsp/context.py

@@ -16,4 +16,5 @@ def __init__(self, context: Context) -> None:
             if model._path is not None:
                 path = Path(model._path).resolve()
                 map[f"file://{path.as_posix()}"].append(model.name)
+
         self.map = map

tests/lsp/test_reference.py

@@ -25,26 +25,7 @@ def test_reference() -> None:
     path = active_customers_uri.removeprefix("file://")
     read_file = open(path, "r").readlines()
     # Get the string range in the read file
-    reference_range = references[0].range
-    start_line = reference_range.start.line
-    end_line = reference_range.end.line
-    start_character = reference_range.start.character
-    end_character = reference_range.end.character
-    # Get the string from the file
-
-    # If the reference spans multiple lines, handle it accordingly
-    if start_line == end_line:
-        # Reference is on a single line
-        line_content = read_file[start_line]
-        referenced_text = line_content[start_character:end_character]
-    else:
-        # Reference spans multiple lines
-        referenced_text = read_file[start_line][
-            start_character:
-        ]  # First line from start_character to end
-        for line_num in range(start_line + 1, end_line):  # Middle lines (if any)
-            referenced_text += read_file[line_num]
-        referenced_text += read_file[end_line][:end_character]  # Last line up to end_character
+    referenced_text = get_string_from_range(read_file, references[0].range)
     assert referenced_text == "sushi.customers"
 
 
@@ -60,6 +41,32 @@ def test_reference_with_alias() -> None:
     references = get_model_definitions_for_a_path(lsp_context, waiter_revenue_by_day_uri)
     assert len(references) == 3
 
+    path = waiter_revenue_by_day_uri.removeprefix("file://")
+    read_file = open(path, "r").readlines()
+
     assert references[0].uri.endswith("orders.py")
+    assert get_string_from_range(read_file, references[0].range) == "sushi.orders"
     assert references[1].uri.endswith("order_items.py")
+    assert get_string_from_range(read_file, references[1].range) == "sushi.order_items"
     assert references[2].uri.endswith("items.py")
+    assert get_string_from_range(read_file, references[2].range) == "sushi.items"
+
+
+def get_string_from_range(file_lines, range_obj) -> str:
+    start_line = range_obj.start.line
+    end_line = range_obj.end.line
+    start_character = range_obj.start.character
+    end_character = range_obj.end.character
+
+    # If the reference spans multiple lines, handle it accordingly
+    if start_line == end_line:
+        # Reference is on a single line
+        line_content = file_lines[start_line]
+        return line_content[start_character:end_character]
+
+    # Reference spans multiple lines
+    result = file_lines[start_line][start_character:]  # First line from start_character to end
+    for line_num in range(start_line + 1, end_line):  # Middle lines (if any)
+        result += file_lines[line_num]
+    result += file_lines[end_line][:end_character]  # Last line up to end_character
+    return result