|
13 | 13 | from robotcode.core.logging import LoggingDescriptor |
14 | 14 | from robotcode.core.uri import Uri |
15 | 15 | from robotcode.jsonrpc2.protocol import rpc_method |
16 | | -from robotcode.language_server.common.lsp_types import DocumentUri, Position, Range, TextDocumentIdentifier |
| 16 | +from robotcode.language_server.common.lsp_types import ( |
| 17 | + DocumentUri, |
| 18 | + Position, |
| 19 | + Range, |
| 20 | + TextDocumentIdentifier, |
| 21 | +) |
17 | 22 | from robotcode.language_server.robotframework.configuration import RobotConfig |
18 | 23 | from robotcode.language_server.robotframework.utils.async_ast import AsyncVisitor |
19 | 24 |
|
20 | 25 | from .protocol_part import RobotLanguageServerProtocolPart |
21 | 26 |
|
22 | 27 | if TYPE_CHECKING: |
23 | | - from robotcode.language_server.robotframework.protocol import RobotLanguageServerProtocol |
| 28 | + from robotcode.language_server.robotframework.protocol import ( |
| 29 | + RobotLanguageServerProtocol, |
| 30 | + ) |
24 | 31 |
|
25 | 32 |
|
26 | 33 | @dataclass |
@@ -107,7 +114,9 @@ def _patch(self) -> None: |
107 | 114 | from robot.running import TestSuite |
108 | 115 | from robot.running.builder.builders import RobotParser, TestSuiteBuilder |
109 | 116 |
|
110 | | - from robotcode.language_server.robotframework.utils.version import get_robot_version |
| 117 | + from robotcode.language_server.robotframework.utils.version import ( |
| 118 | + get_robot_version, |
| 119 | + ) |
111 | 120 |
|
112 | 121 | def get_document_text(source: str) -> str: |
113 | 122 | check_canceled_sync() |
@@ -138,7 +147,10 @@ def my_get_model_v4(source: str, data_only: bool = False, curdir: Optional[str] |
138 | 147 | return File(source=source) |
139 | 148 |
|
140 | 149 | def my_get_model_v6( |
141 | | - source: str, data_only: bool = False, curdir: Optional[str] = None, lang: Any = None |
| 150 | + source: str, |
| 151 | + data_only: bool = False, |
| 152 | + curdir: Optional[str] = None, |
| 153 | + lang: Any = None, |
142 | 154 | ) -> Any: |
143 | 155 | check_canceled_sync() |
144 | 156 | try: |
@@ -193,7 +205,9 @@ async def get_tests_from_workspace( |
193 | 205 | from robot.running import TestCase, TestSuite |
194 | 206 | from robot.running.builder.builders import TestSuiteBuilder |
195 | 207 |
|
196 | | - from robotcode.language_server.robotframework.utils.version import get_robot_version |
| 208 | + from robotcode.language_server.robotframework.utils.version import ( |
| 209 | + get_robot_version, |
| 210 | + ) |
197 | 211 |
|
198 | 212 | async def generate(suite: TestSuite) -> TestItem: |
199 | 213 | children: List[TestItem] = [] |
@@ -280,7 +294,7 @@ def nonexisting_paths(paths: List[str]) -> Iterator[str]: |
280 | 294 |
|
281 | 295 | valid_paths = list(normalize_paths(paths)) |
282 | 296 |
|
283 | | - if get_robot_version() >= (6, 1): |
| 297 | + if get_robot_version() >= (6, 1, 0, "a", 2): |
284 | 298 | builder = TestSuiteBuilder( |
285 | 299 | included_suites=suites if suites else None, |
286 | 300 | rpa=rpa_mode, |
@@ -322,15 +336,19 @@ def nonexisting_paths(paths: List[str]) -> Iterator[str]: |
322 | 336 | ) |
323 | 337 | ] |
324 | 338 |
|
325 | | - if get_robot_version() >= (6, 1): |
| 339 | + if get_robot_version() >= (6, 1, 0, "a", 2): |
326 | 340 | builder = TestSuiteBuilder( |
327 | 341 | included_suites=suites if suites else None, |
328 | 342 | rpa=rpa_mode, |
329 | 343 | lang=languages, |
330 | 344 | parsers=parsers, |
331 | 345 | ) |
332 | 346 | elif get_robot_version() >= (6, 0): |
333 | | - builder = TestSuiteBuilder(included_suites=suites if suites else None, rpa=rpa_mode, lang=languages) |
| 347 | + builder = TestSuiteBuilder( |
| 348 | + included_suites=suites if suites else None, |
| 349 | + rpa=rpa_mode, |
| 350 | + lang=languages, |
| 351 | + ) |
334 | 352 | else: |
335 | 353 | builder = TestSuiteBuilder(included_suites=suites if suites else None, rpa=rpa_mode) |
336 | 354 | return [await generate(builder.build(str(workspace_path)))] |
@@ -359,7 +377,11 @@ def nonexisting_paths(paths: List[str]) -> Iterator[str]: |
359 | 377 | @rpc_method(name="robot/discovering/getTestsFromDocument", param_type=GetTestsParams) |
360 | 378 | @threaded() |
361 | 379 | async def get_tests_from_document( |
362 | | - self, text_document: TextDocumentIdentifier, base_name: Optional[str], *args: Any, **kwargs: Any |
| 380 | + self, |
| 381 | + text_document: TextDocumentIdentifier, |
| 382 | + base_name: Optional[str], |
| 383 | + *args: Any, |
| 384 | + **kwargs: Any, |
363 | 385 | ) -> List[TestItem]: |
364 | 386 | try: |
365 | 387 | return await FindTestCasesVisitor().get( |
|
0 commit comments