test: Add tests for workspace discovery

Author: d-biehl

robotcode/language_server/robotframework/parts/discovering.py

@@ -116,8 +116,8 @@ async def get_config(self, workspace_uri: str) -> Optional[RobotConfig]:
     async def get_tests_from_workspace(
         self,
         workspace_folder: str,
-        paths: Optional[List[str]],
-        suites: Optional[List[str]],
+        paths: Optional[List[str]] = None,
+        suites: Optional[List[str]] = None,
         *args: Any,
         **kwargs: Any,
     ) -> List[TestItem]:

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

@@ -0,0 +1,44 @@
+import dataclasses
+from pathlib import Path
+
+import pytest
+from pytest_regressions.data_regression import DataRegressionFixture
+
+from robotcode.language_server.robotframework.protocol import (
+    RobotLanguageServerProtocol,
+)
+
+
+@pytest.mark.usefixtures("protocol")
+@pytest.mark.asyncio
+async def test_workspace_discovery(
+    data_regression: DataRegressionFixture,
+    protocol: RobotLanguageServerProtocol,
+) -> None:
+    from robotcode.language_server.robotframework.parts.discovering import TestItem
+
+    def split(item: TestItem) -> TestItem:
+        return dataclasses.replace(
+            item,
+            id=item.id.split(";", 1)[1] if ";" in item.id else item.id,
+            uri="/".join(item.uri.split("/")[-2:]) if item.uri else item.uri,
+            children=sorted(
+                [split(v) for v in item.children],
+                key=lambda v: (
+                    v.uri,
+                    v.range.start if v.range is not None else None,
+                    v.range.end if v.range is not None else None,
+                ),
+            )
+            if item.children
+            else item.children,
+        )
+
+    result = await protocol.robot_discovering.get_tests_from_workspace(Path(Path(__file__).parent, "data").as_uri())
+    data_regression.check(
+        {
+            "result": sorted((split(v) for v in result), key=lambda v: (v.uri, v.range.start, v.range.end))
+            if result
+            else result,
+        }
+    )