def test_semantic_highlighting_keyword(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """*** Keywords ***
Some Keyword
[Arguments] ${arg1}
Call Keyword ${arg1}
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Keywords ***", "header"),
("Some Keyword", "keywordNameDefinition"),
("[", "variableOperator"),
("Arguments", "setting"),
("]", "variableOperator"),
("${", "variableOperator"),
("arg1", "variable"),
("}", "variableOperator"),
("Call Keyword", "keywordNameCall"),
("${", "variableOperator"),
("arg1", "variable"),
("}", "variableOperator"),
],
)
def test_semantic_highlighting_arguments(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """
*** Test Cases ***
Some Test
Clear All Highlights formatter=some ${arg1} other
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Test Cases ***", "header"),
("Some Test", "testCaseName"),
("Clear All Highlights", "keywordNameCall"),
("formatter", "parameterName"),
("=", "variableOperator"),
("some ", "argumentValue"),
("${", "variableOperator"),
("arg1", "variable"),
("}", "variableOperator"),
(" other", "argumentValue"),
],
)
def m_text_document__semantic_tokens__full(self, textDocument=None):
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
doc_uri = textDocument["uri"]
context = self._create_completion_context(doc_uri, -1, -1, None)
if context is None:
return []
return {"resultId": None, "data": semantic_tokens_full(context)}
def threaded_semantic_tokens_full(
self, textDocument: TextDocumentTypedDict, monitor: Optional[IMonitor] = None
):
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
doc_uri = textDocument["uri"]
context = self._create_completion_context(doc_uri, -1, -1, monitor)
if context is None:
return {"resultId": None, "data": []}
return {"resultId": None, "data": semantic_tokens_full(context)}
def test_semantic_highlighting_task_name(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """*** Task ***
Some Task
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[("*** Task ***", "header"), ("Some Task", "testCaseName")],
)
def test_semantic_highlighting_base(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """*** Settings ***
Library my.lib
*** Keywords ***
Some Keyword
[Arguments] Some ${arg1} Another ${arg2}
Clear All Highlights ${arg1} ${arg2}
""".replace(
"\r\n", "\n"
).replace(
"\r", "\n"
)
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Settings ***", "header"),
("Library", "setting"),
("my.lib", "name"),
("*** Keywords ***", "header"),
("Some Keyword", "keywordNameDefinition"),
("[", "variableOperator"),
("Arguments", "setting"),
("]", "variableOperator"),
("Some ", "parameter"),
("${", "variableOperator"),
("arg1", "variable"),
("}", "variableOperator"),
("Another ", "parameter"),
("${", "variableOperator"),
("arg2", "variable"),
("}", "variableOperator"),
("Clear All Highlights", "keywordNameCall"),
("${", "variableOperator"),
("arg1", "variable"),
("}", "variableOperator"),
("${", "variableOperator"),
("arg2", "variable"),
("}", "variableOperator"),
],
)
def test_semantic_highlighting_comments(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """*** Comments ***
Comment part 1
Comment part 2
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Comments ***", "header"),
("Comment part 1", "comment"),
("Comment part 2", "comment"),
],
)
def test_semantic_highlighting_arguments_in_doc(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """
*** Settings ***
Documentation Some = eq
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Settings ***", "header"),
("Documentation", "setting"),
("Some = eq", "argumentValue"),
],
)
def test_semantic_highlighting_catenate(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """*** Test Case ***
Test Case
Catenate FOO
... Check = 22
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Test Case ***", "header"),
("Test Case", "testCaseName"),
("Catenate", "keywordNameCall"),
("FOO", "argumentValue"),
("Check = 22", "argumentValue"),
],
)
def test_semantic_highlighting_for_if(workspace):
from robotframework_ls.impl.completion_context import CompletionContext
from robotframework_ls.impl.semantic_tokens import semantic_tokens_full
workspace.set_root("case1")
doc = workspace.get_doc("case1.robot")
doc.source = """*** Keywords ***
Some keyword
FOR ${element} IN @{LIST}
IF ${random} == ${NUMBER_TO_PASS_ON}
Pass Execution "${random} == ${NUMBER_TO_PASS_ON}"
ELSE IF ${random} > ${NUMBER_TO_PASS_ON}
Log To Console Too high.
ELSE
Log To Console Too low.
END
END
""".replace("\r\n", "\n").replace("\r", "\n")
context = CompletionContext(doc, workspace=workspace.ws)
semantic_tokens = semantic_tokens_full(context)
check(
(semantic_tokens, doc),
[
("*** Keywords ***", "header"),
("Some keyword", "keywordNameDefinition"),
("FOR", "control"),
("${", "variableOperator"),
("element", "variable"),
("}", "variableOperator"),
("IN", "control"),
("@{", "variableOperator"),
("LIST", "variable"),
("}", "variableOperator"),
("IF", "control"),
("${", "variableOperator"),
("random", "variable"),
("}", "variableOperator"),
(" == ", "argumentValue"),
("${", "variableOperator"),
("NUMBER_TO_PASS_ON", "variable"),
("}", "variableOperator"),
("Pass Execution", "keywordNameCall"),
('"', "argumentValue"),
("${", "variableOperator"),
("random", "variable"),
("}", "variableOperator"),
(" == ", "argumentValue"),
("${", "variableOperator"),
("NUMBER_TO_PASS_ON", "variable"),
("}", "variableOperator"),
('"', "argumentValue"),
("ELSE IF", "control"),
("${", "variableOperator"),
("random", "variable"),
("}", "variableOperator"),
(" > ", "argumentValue"),
("${", "variableOperator"),
("NUMBER_TO_PASS_ON", "variable"),
("}", "variableOperator"),
("Log To Console", "keywordNameCall"),
("Too high.", "argumentValue"),
("ELSE", "control"),
("Log To Console", "keywordNameCall"),
("Too low.", "argumentValue"),
("END", "control"),
("END", "control"),
],
)