fix: correctly mock stuff

Author: reecelikesramen

tests/test_transpiler.py

@@ -1,3 +1,4 @@
+import textwrap
 import pytest
 from pywire_language_server.transpiler import Transpiler
 
@@ -12,14 +13,14 @@ def test_transpile_simple_interpolation():
     # This assertion is vague, we'll refine it as we implement.
 
 def test_transpile_python_section():
-    source = """
----
-x = 1
-def foo():
-    pass
----
-<h1>Hi</h1>
-"""
+    source = textwrap.dedent("""
+        ---
+        x = 1
+        def foo():
+            pass
+        ---
+        <h1>Hi</h1>
+    """).strip()
     transpiler = Transpiler(source)
     code, _ = transpiler.transpile()
 
@@ -36,22 +37,24 @@ def test_transpile_directive():
     assert "'/home'" in code
 
 def test_transpile_multiline_interpolation():
-    source = """<div class={
-    'active' if True
-    else 'inactive'
-}></div>"""
+    source = textwrap.dedent("""
+        <div class={
+            'active' if True
+            else 'inactive'
+        }></div>
+    """).strip()
     transpiler = Transpiler(source)
     code, _ = transpiler.transpile()
 
     assert "'active' if True" in code
     assert "else 'inactive'" in code
 
 def test_transpile_wrappers():
-    source = """
-    <div $if={x > 1}></div>
-    <div $for={i in items}></div>
-    <div @click={do_something()}></div>
-    """
+    source = textwrap.dedent("""
+        <div $if={x > 1}></div>
+        <div $for={i in items}></div>
+        <div @click={do_something()}></div>
+    """).strip()
     transpiler = Transpiler(source)
     code, _ = transpiler.transpile()
 
@@ -63,16 +66,16 @@ def test_transpile_wrappers():
 
 def test_explicit_property_mapping():
     """Test that {count.value} maps 'count' correctly."""
-    source = """
----
-count = wire(0)
----
-<p>{count.value}</p>
-"""
+    source = textwrap.dedent("""
+        ---
+        count = wire(0)
+        ---
+        <p>{count.value}</p>
+    """).strip()
     transpiler = Transpiler(source)
     code, source_map = transpiler.transpile()
 
-    usage_line = 4
+    usage_line = 3
     usage_col_start = 4 # { is at 3, count at 4
 
     gen_loc = source_map.to_generated(usage_line, usage_col_start)
@@ -87,19 +90,19 @@ def test_explicit_property_mapping():
 
 def test_event_handler_mapping():
     """Test @click={count.value += 1} mapping."""
-    source = """
----
-count = wire(0)
----
-<button @click={count.value += 1}>Inc</button>
-"""
+    source = textwrap.dedent("""
+        ---
+        count = wire(0)
+        ---
+        <button @click={count.value += 1}>Inc</button>
+    """).strip()
     transpiler = Transpiler(source)
     code, source_map = transpiler.transpile()
 
     # Usage: {count.value}
     # <button @click={count...
     # c is at 16.
-    usage_line = 4
+    usage_line = 3
     usage_col_start = 16
 
     gen_loc = source_map.to_generated(usage_line, usage_col_start)

tests/test_ty_integration.py

@@ -42,6 +42,8 @@ async def test_completion_trigger_kind(mock_ls, mock_ty_client):
     server_module.virtual_manager = Mock()
     server_module.virtual_manager.root_path = "/workspace"
     server_module.virtual_manager.get_shadow_uri.return_value = "file:///shadow.py"
+    server_module.virtual_manager.get_stub_uri.return_value = None
+    server_module.virtual_manager._uri_to_path.return_value = "/workspace/test.wire"
 
     # Mock document mapping
     uri = "file:///test.wire"
@@ -81,6 +83,8 @@ async def test_hover_formatting(mock_ls, mock_ty_client):
     server_module.virtual_manager = Mock()
     server_module.virtual_manager.root_path = "/workspace"
     server_module.virtual_manager.get_shadow_uri.return_value = "file:///shadow.py"
+    server_module.virtual_manager.get_stub_uri.return_value = None
+    server_module.virtual_manager._uri_to_path.return_value = "/workspace/test.wire"
 
     uri = "file:///test.wire"
     text = """
@@ -123,6 +127,8 @@ async def test_hover_docstring_separation(mock_ls, mock_ty_client):
     server_module.virtual_manager = Mock()
     server_module.virtual_manager.root_path = "/workspace"
     server_module.virtual_manager.get_shadow_uri.return_value = "file:///shadow.py"
+    server_module.virtual_manager.get_stub_uri.return_value = None
+    server_module.virtual_manager._uri_to_path.return_value = "/workspace/test.wire"
 
     # Test that we separate signature from docstring
     uri = "file:///test.wire"
@@ -269,6 +275,8 @@ async def test_definition_reference_fallback(mock_ls, mock_ty_client):
     server_module.virtual_manager = Mock()
     server_module.virtual_manager.root_path = "/workspace"
     server_module.virtual_manager.get_shadow_uri.return_value = "file:///shadow.py"
+    server_module.virtual_manager.get_stub_uri.return_value = None
+    server_module.virtual_manager._uri_to_path.return_value = "/workspace/test.wire"
 
     uri = "file:///test.wire"
     text = """