feat: Reintroduce of updating the tests when typing

Author: d-biehl

.vscode/launch.json

@@ -38,7 +38,8 @@
 				// "--",
 				// "--help"
 				// "run"
-				"--format", "toml",
+				// "--format",
+				// "toml",
 				"--no-color",
 				"--no-pager",
 				"--default-path",
@@ -48,7 +49,8 @@
 				"all",
 				"--norpa",
 				// "--by-longname", "asd:dasd",
-				// "--suite",
+				"--suite",
+				"Data.Tests.Versions.Rf61.a suite with a custom Name"
 				// "Robotframework.Atest.Testdata.Variables",
 				//"."
 			]

packages/plugin/src/robotcode/plugin/__init__.py

@@ -73,7 +73,7 @@ def verbose(
         message: Union[str, Callable[[], Any], None],
         file: Optional[IO[AnyStr]] = None,
         nl: bool = True,
-        err: bool = False,
+        err: bool = True,
     ) -> None:
         if self.config.verbose:
             click.secho(

packages/runner/src/robotcode/runner/cli/discover/discover.py

@@ -3,7 +3,7 @@
 import sys
 from dataclasses import dataclass
 from pathlib import Path
-from typing import Any, Dict, Iterable, List, Optional, Tuple
+from typing import Any, Dict, Iterable, List, Optional, Tuple, Union
 
 import click
 import robot.running.model as running_model
@@ -14,6 +14,7 @@
 from robot.output import LOGGER, Message
 from robot.running.builder import TestSuiteBuilder
 from robot.running.builder.builders import SuiteStructureParser
+from robotcode.core.dataclasses import from_json
 from robotcode.core.lsp.types import Diagnostic, DiagnosticSeverity, DocumentUri, Position, Range
 from robotcode.core.uri import Uri
 from robotcode.plugin import Application, OutputFormat, UnknownError, pass_application
@@ -31,13 +32,18 @@ def __init__(self, *args: Any, error_message: str, **kwargs: Any) -> None:
 __patched = False
 
 
+_stdin_data: Optional[Dict[str, str]] = None
+
+
 def _patch() -> None:
     global __patched
     if __patched:
         return
     __patched = True
 
     if get_robot_version() <= (6, 1, 0, "a", 1, None):
+        from robot.running.builder.parsers import RobotParser
+
         if get_robot_version() > (5, 0) and get_robot_version() < (6, 0, 0) or get_robot_version() < (5, 0):
             from robot.running.builder.testsettings import TestDefaults  # pyright: ignore[reportMissingImports]
         else:
@@ -75,8 +81,20 @@ def build_suite(self: SuiteStructureParser, structure: Any) -> Tuple[TestSuite,
 
         SuiteStructureParser._build_suite = build_suite
 
+        old_get_source = RobotParser._get_source
+
+        def _get_source(self: RobotParser, path: Path) -> Union[Path, str]:
+            if _stdin_data is not None and (data := _stdin_data.get(str(path))) is not None:
+                if data is not None:
+                    return data
+
+            return old_get_source(self, path)  # type: ignore
+
+        RobotParser._get_source = _get_source
+
     elif get_robot_version() >= (6, 1, 0, "a", 1, None):
         from robot.parsing.suitestructure import SuiteDirectory, SuiteFile
+        from robot.running.builder.parsers import RobotParser
         from robot.running.builder.settings import TestDefaults  # pyright: ignore[reportMissingImports]
 
         old_validate_not_empty = TestSuiteBuilder._validate_not_empty
@@ -117,6 +135,17 @@ def build_suite_directory(
 
         SuiteStructureParser._build_suite_directory = build_suite_directory
 
+        old_get_source = RobotParser._get_source
+
+        def _get_source(self: RobotParser, path: Path) -> Union[Path, str]:
+            if _stdin_data is not None and (data := _stdin_data.get(str(path))) is not None:
+                if data is not None:
+                    return data
+
+            return old_get_source(self, path)  # type: ignore
+
+        RobotParser._get_source = _get_source
+
 
 @dataclass
 class TestItem:
@@ -217,8 +246,9 @@ def visit_test(self, test: TestCase) -> None:
 
 
 @click.group(invoke_without_command=False)
+@click.option("--read-from-stdin", is_flag=True, help="Read file contents from stdin. This is an internal option.")
 @pass_application
-def discover(app: Application) -> None:
+def discover(app: Application, read_from_stdin: bool) -> None:
     """\
     Commands to discover informations about the current project.
 
@@ -229,6 +259,10 @@ def discover(app: Application) -> None:
     robotcode --profile regression discover tests
     ```
     """
+    if read_from_stdin:
+        global _stdin_data
+        _stdin_data = from_json(sys.stdin.buffer.read(), Dict[str, str], strict=True)
+        app.verbose(f"Read data from stdin: {repr(_stdin_data)}")
 
 
 RE_IN_FILE_LINE_MATCHER = re.compile(r".+\sin\sfile\s'(?P<file>.*)'\son\sline\s(?P<line>\d+):(?P<message>.*)")

tests/robotcode/language_server/robotframework/parts/data/robot.toml

@@ -2,8 +2,9 @@ default-profiles = ["devel"]
 paths = ["."]
 python-path = ["./lib", "./resources"]
 output-dir = { expr = "f'results/run-{datetime.now().strftime(\"%d-%m-%Y-%H.%M.%S\")}'" }
-rpa = false
 
+
+rpa = false
 # [parsers]
 # myparser = ["asd"]
 

tests/robotcode/language_server/robotframework/parts/data/tests/versions/rf61/__init__.robot

@@ -0,0 +1,2 @@
+*** Settings ***
+# Name    Test Suite for 61

vscode-client/pythonmanger.ts

@@ -181,8 +181,12 @@ export class PythonManager {
   public async executeRobotCode(
     folder: vscode.WorkspaceFolder,
     args: string[],
+    stdioData?: string,
     token?: vscode.CancellationToken
   ): Promise<unknown> {
+    const config = vscode.workspace.getConfiguration(CONFIG_SECTION, folder);
+    const robotCodeExtraArgs = config.get<string[]>("extraArgs", []);
+
     const pythonCommand = this.getPythonCommand(folder);
     if (pythonCommand === undefined) throw new Error("Can't find python executable.");
 
@@ -191,6 +195,7 @@ export class PythonManager {
       "-X",
       "utf8",
       this.robotCodeMain,
+      ...robotCodeExtraArgs,
       "--format",
       "json",
       "--no-color",
@@ -220,12 +225,18 @@ export class PythonManager {
 
       process.stdout.setEncoding("utf8");
       process.stderr.setEncoding("utf8");
+      if (stdioData !== undefined) {
+        process.stdin.cork();
+        process.stdin.write(stdioData, "utf8");
+        process.stdin.end();
+      }
 
       process.stdout.on("data", (data) => {
         stdout += data;
       });
 
       process.stderr.on("data", (data) => {
+        this.outputChannel.append(data as string);
         stderr += data;
       });
 

vscode-client/testcontrollermanager.ts

@@ -511,6 +511,7 @@ export class TestControllerManager {
     folder: vscode.WorkspaceFolder,
     discoverArgs: string[],
     extraArgs: string[],
+    stdioData?: string,
     token?: vscode.CancellationToken
   ): Promise<RobotCodeDiscoverResult> {
     const config = vscode.workspace.getConfiguration(CONFIG_SECTION, folder);
@@ -544,6 +545,7 @@ export class TestControllerManager {
         ...robotArgs,
         ...extraArgs,
       ],
+      stdioData,
       token
     )) as RobotCodeDiscoverResult;
   }
@@ -553,7 +555,24 @@ export class TestControllerManager {
     token?: vscode.CancellationToken
   ): Promise<RobotTestItem[] | undefined> {
     try {
-      const result = await this.discoverTests(folder, ["discover", "all"], [], token);
+      const o: { [key: string]: string } = {};
+
+      for (const document of vscode.workspace.textDocuments) {
+        if (
+          SUPPORTED_LANGUAGES.includes(document.languageId) &&
+          vscode.workspace.getWorkspaceFolder(document.uri) === folder
+        ) {
+          o[document.fileName] = document.getText();
+        }
+      }
+
+      const result = await this.discoverTests(
+        folder,
+        ["discover", "--read-from-stdin", "all"],
+        [],
+        JSON.stringify(o),
+        token
+      );
 
       this.diagnosticCollection.forEach((uri, _diagnostics, collection) => {
         if (vscode.workspace.getWorkspaceFolder(uri) === folder) {
@@ -605,7 +624,16 @@ export class TestControllerManager {
     if (!folder) return undefined;
 
     try {
-      const result = await this.discoverTests(folder, ["discover", "tests"], ["--suite", testItem?.longname], token);
+      const o: { [key: string]: string } = {};
+      o[document.fileName] = document.getText();
+
+      const result = await this.discoverTests(
+        folder,
+        ["discover", "--read-from-stdin", "tests"],
+        ["--suite", testItem?.longname],
+        JSON.stringify(o),
+        token
+      );
 
       return result?.items;
     } catch (e) {