test: introduce timeout/wait_for for langserver tests

Author: d-biehl

tests/robotcode/language_server/robotframework/parts/test_discovering.py

@@ -1,3 +1,4 @@
+import asyncio
 import dataclasses
 from pathlib import Path
 
@@ -37,8 +38,11 @@ def split(item: TestItem) -> TestItem:
             else item.children,
         )
 
-    result = await run_coroutine_in_thread(
-        protocol.robot_discovering.get_tests_from_workspace, Path(Path(__file__).parent, "data").as_uri()
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(
+            protocol.robot_discovering.get_tests_from_workspace, Path(Path(__file__).parent, "data").as_uri()
+        ),
+        120,
     )
     regtest.write(
         yaml.dump(

tests/robotcode/language_server/robotframework/parts/test_document_highlight.py

@@ -1,3 +1,4 @@
+import asyncio
 from pathlib import Path
 
 import pytest
@@ -33,11 +34,14 @@ async def test(
     test_document: TextDocument,
     data: GeneratedTestData,
 ) -> None:
-    result = result = await run_coroutine_in_thread(
-        protocol.robot_document_highlight.collect,
-        protocol.robot_document_highlight,
-        test_document,
-        Position(line=data.line, character=data.character),
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(
+            protocol.robot_document_highlight.collect,
+            protocol.robot_document_highlight,
+            test_document,
+            Position(line=data.line, character=data.character),
+        ),
+        60,
     )
     regtest.write(
         yaml.dump(

tests/robotcode/language_server/robotframework/parts/test_foldingrange.py

@@ -1,3 +1,4 @@
+import asyncio
 from pathlib import Path
 from typing import Any, Iterable, Iterator, Tuple, Union
 
@@ -70,8 +71,8 @@ async def test(
     test_document: TextDocument,
     data: GeneratedTestData,
 ) -> None:
-    result = await run_coroutine_in_thread(
-        protocol.robot_folding_ranges.collect, protocol.robot_folding_ranges, test_document
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(protocol.robot_folding_ranges.collect, protocol.robot_folding_ranges, test_document), 60
     )
 
     regtest.write(yaml.dump({"data": data, "result": result}))

tests/robotcode/language_server/robotframework/parts/test_goto_definition.py

@@ -1,3 +1,4 @@
+import asyncio
 from pathlib import Path
 
 import pytest
@@ -34,11 +35,14 @@ async def test_definition(
     test_document: TextDocument,
     data: GeneratedTestData,
 ) -> None:
-    result = await run_coroutine_in_thread(
-        protocol.robot_goto.collect_definition,
-        protocol.robot_goto,
-        test_document,
-        Position(line=data.line, character=data.character),
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(
+            protocol.robot_goto.collect_definition,
+            protocol.robot_goto,
+            test_document,
+            Position(line=data.line, character=data.character),
+        ),
+        60,
     )
 
     regtest.write(yaml.dump({"data": data, "result": split(result)}))

tests/robotcode/language_server/robotframework/parts/test_goto_implementation.py

@@ -1,3 +1,4 @@
+import asyncio
 from pathlib import Path
 from typing import List, Union, cast
 
@@ -58,11 +59,14 @@ async def test_implementation(
     test_document: TextDocument,
     data: GeneratedTestData,
 ) -> None:
-    result = await run_coroutine_in_thread(
-        protocol.robot_goto.collect_implementation,
-        protocol.robot_goto,
-        test_document,
-        Position(line=data.line, character=data.character),
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(
+            protocol.robot_goto.collect_implementation,
+            protocol.robot_goto,
+            test_document,
+            Position(line=data.line, character=data.character),
+        ),
+        60,
     )
 
     regtest.write(yaml.dump({"data": data, "result": split(result)}))

tests/robotcode/language_server/robotframework/parts/test_hover.py

@@ -1,3 +1,4 @@
+import asyncio
 from pathlib import Path
 from typing import Optional
 
@@ -19,6 +20,17 @@
 )
 
 
+def split(hover: Optional[Hover]) -> Optional[Hover]:
+    if hover is None:
+        return None
+    if isinstance(hover.contents, MarkupContent):
+        return Hover(
+            MarkupContent(hover.contents.kind, hover.contents.value.splitlines()[0].split("=")[0].strip()),
+            hover.range,
+        )
+    return hover
+
+
 @pytest.mark.parametrize(
     ("test_document", "data"),
     generate_tests_from_source_document(Path(Path(__file__).parent, "data/tests/hover.robot")),
@@ -34,18 +46,14 @@ async def test(
     test_document: TextDocument,
     data: GeneratedTestData,
 ) -> None:
-    result = await run_coroutine_in_thread(
-        protocol.robot_hover.collect, protocol.hover, test_document, Position(line=data.line, character=data.character)
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(
+            protocol.robot_hover.collect,
+            protocol.hover,
+            test_document,
+            Position(line=data.line, character=data.character),
+        ),
+        60,
     )
 
-    def split(hover: Optional[Hover]) -> Optional[Hover]:
-        if hover is None:
-            return None
-        if isinstance(hover.contents, MarkupContent):
-            return Hover(
-                MarkupContent(hover.contents.kind, hover.contents.value.splitlines()[0].split("=")[0].strip()),
-                hover.range,
-            )
-        return hover
-
     regtest.write(yaml.dump({"data": data, "result": split(result)}))

tests/robotcode/language_server/robotframework/parts/test_references.py

@@ -1,3 +1,4 @@
+import asyncio
 from pathlib import Path
 
 import pytest
@@ -40,12 +41,15 @@ async def test(
     def split(location: Location) -> Location:
         return Location("/".join(location.uri.split("/")[-2:]), location.range)
 
-    result = await run_coroutine_in_thread(
-        protocol.robot_references.collect,
-        protocol.robot_references,
-        test_document,
-        Position(line=data.line, character=data.character),
-        ReferenceContext(include_declaration=True),
+    result = await asyncio.wait_for(
+        run_coroutine_in_thread(
+            protocol.robot_references.collect,
+            protocol.robot_references,
+            test_document,
+            Position(line=data.line, character=data.character),
+            ReferenceContext(include_declaration=True),
+        ),
+        60,
     )
     regtest.write(
         yaml.dump(