|
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