def test_completion_not_duplicated(workspace, cases, libspec_manager, workspace_dir):
from robocorp_ls_core.lsp import TextDocumentItem
import os.path
from robocorp_ls_core import uris
from robotframework_ls.robot_config import RobotConfig
from robotframework_ls.impl.collect_keywords import (
collect_keyword_name_to_keyword_found,
)
from robotframework_ls.impl.protocols import IKeywordFound
cases.copy_to("case_inner_keywords", workspace_dir)
workspace.set_root(workspace_dir, libspec_manager=libspec_manager)
config = RobotConfig()
config.update({"robot.pythonpath": [workspace_dir]})
libspec_manager.config = config
uri = uris.from_fs_path(os.path.join(workspace_dir, "inner", "case_inner.robot"))
workspace.ws.put_document(
TextDocumentItem(
uri,
text="""*** Settings ***
Resource case_root.robot
*** Keywords ***
Check with keyword at inner
[Arguments] ${arg1} ${arg2}
Should Be Equal ${arg1} ${arg2}
*** Test Cases ***
Testing Completion Here
Check with ke""",
)
)
doc = workspace.ws.get_document(uri, accept_from_file=False)
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl import auto_import_completions
workspace.ws.wait_for_check_done(5)
context = CompletionContext(doc, workspace=workspace.ws, config=config)
keyword_name_to_keyword_found: Dict[
str, List[IKeywordFound]
] = collect_keyword_name_to_keyword_found(context)
completions = auto_import_completions.complete(
context, keyword_name_to_keyword_found
)
# I.e.: all the related symbols are already imported and will be shown
# in the regular completion.
assert len(completions) == 0
def test_completion_with_auto_import_duplicated(workspace,
setup_case2_in_dir_doc):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl import auto_import_completions
from robotframework_ls.impl.protocols import IKeywordFound
from robotframework_ls.impl.collect_keywords import (
collect_keyword_name_to_keyword_found, )
from robocorp_ls_core.lsp import TextDocumentItem
doc = workspace.get_doc("case1.robot")
doc.source = """
*** Keywords ***
KeywordInCase1
In Lib 2
"""
# Just create a dummy entry for KeywordInCase1.
doc_foobar = workspace.ws.put_document(
TextDocumentItem(
"foobar",
text="""
*** Keywords ***
KeywordInCase1
In Lib 2
""",
))
keyword_name_to_keyword_found: Dict[
str, List[IKeywordFound]] = collect_keyword_name_to_keyword_found(
CompletionContext(doc_foobar, workspace=workspace.ws))
doc2 = setup_case2_in_dir_doc
doc2.source = """
*** Test Cases ***
User can call library
KeywordInCa"""
completions = auto_import_completions.complete(
CompletionContext(doc2, workspace=workspace.ws),
keyword_name_to_keyword_found)
assert len(completions) == 1
apply_completion(doc2, completions[0])
assert (doc2.source == """*** Settings ***
Resource ../case1.robot
*** Test Cases ***
User can call library
case1.KeywordInCase1""")
def _threaded_complete_all(self, doc_uri, line, col, monitor: IMonitor):
from robotframework_ls.impl import section_name_completions
from robotframework_ls.impl import keyword_completions
from robotframework_ls.impl import variable_completions
from robotframework_ls.impl import filesystem_section_completions
from robotframework_ls.impl import keyword_parameter_completions
from robotframework_ls.impl import auto_import_completions
from robotframework_ls.impl.collect_keywords import (
collect_keyword_name_to_keyword_found,
)
from robotframework_ls.impl import ast_utils
completion_context = self._create_completion_context(
doc_uri, line, col, monitor
)
if completion_context is None:
return []
ret = section_name_completions.complete(completion_context)
if not ret:
ret.extend(filesystem_section_completions.complete(completion_context))
if not ret:
token_info = completion_context.get_current_token()
if token_info is not None:
token = ast_utils.get_keyword_name_token(
token_info.node, token_info.token
)
if token is not None:
keyword_name_to_keyword_found: Dict[
str, List[IKeywordFound]
] = collect_keyword_name_to_keyword_found(completion_context)
ret.extend(keyword_completions.complete(completion_context))
ret.extend(
auto_import_completions.complete(
completion_context, keyword_name_to_keyword_found
)
)
return ret
if not ret:
ret.extend(variable_completions.complete(completion_context))
if not ret:
ret.extend(keyword_parameter_completions.complete(completion_context))
return ret
def _complete_from_completion_context(self, completion_context):
from robotframework_ls.impl import section_name_completions
from robotframework_ls.impl import keyword_completions
from robotframework_ls.impl import variable_completions
from robotframework_ls.impl import dictionary_completions
from robotframework_ls.impl import filesystem_section_completions
from robotframework_ls.impl import keyword_parameter_completions
from robotframework_ls.impl import auto_import_completions
from robotframework_ls.impl.collect_keywords import (
collect_keyword_name_to_keyword_found,
)
from robotframework_ls.impl import ast_utils
ret = section_name_completions.complete(completion_context)
if not ret:
ret.extend(filesystem_section_completions.complete(completion_context))
if not ret:
token_info = completion_context.get_current_token()
if token_info is not None:
token = ast_utils.get_keyword_name_token(
token_info.node, token_info.token
)
if token is not None:
keyword_name_to_keyword_found: Dict[
str, List[IKeywordFound]
] = collect_keyword_name_to_keyword_found(completion_context)
ret.extend(keyword_completions.complete(completion_context))
ret.extend(
auto_import_completions.complete(
completion_context, keyword_name_to_keyword_found
)
)
return ret
if not ret:
ret.extend(variable_completions.complete(completion_context))
if not ret:
ret.extend(dictionary_completions.complete(completion_context))
if not ret:
ret.extend(keyword_parameter_completions.complete(completion_context))
return ret
