Add R language support (#574)

test/solidlsp/r/test_r_basic.py

@@ -0,0 +1,97 @@
+"""
+Basic tests for R Language Server integration
+"""
+
+import os
+from pathlib import Path
+
+import pytest
+
+from solidlsp import SolidLanguageServer
+from solidlsp.ls_config import Language
+
+
+@pytest.mark.r
+class TestRLanguageServer:
+    """Test basic functionality of the R language server."""
+
+    @pytest.mark.parametrize("language_server", [Language.R], indirect=True)
+    @pytest.mark.parametrize("repo_path", [Language.R], indirect=True)
+    def test_server_initialization(self, language_server: SolidLanguageServer, repo_path: Path):
+        """Test that the R language server initializes properly."""
+        assert language_server is not None
+        assert language_server.language_id == "r"
+        assert language_server.is_running()
+        assert Path(language_server.language_server.repository_root_path).resolve() == repo_path.resolve()
+
+    @pytest.mark.parametrize("language_server", [Language.R], indirect=True)
+    def test_symbol_retrieval(self, language_server: SolidLanguageServer):
+        """Test R document symbol extraction."""
+        all_symbols, root_symbols = language_server.request_document_symbols(os.path.join("R", "utils.R"))
+
+        # Should find the three exported functions
+        function_symbols = [s for s in all_symbols if s.get("kind") == 12]  # Function kind
+        assert len(function_symbols) >= 3
+
+        # Check that we found the expected functions
+        function_names = {s.get("name") for s in function_symbols}
+        expected_functions = {"calculate_mean", "process_data", "create_data_frame"}
+        assert expected_functions.issubset(function_names), f"Expected functions {expected_functions} but found {function_names}"
+
+    @pytest.mark.parametrize("language_server", [Language.R], indirect=True)
+    def test_find_definition_across_files(self, language_server: SolidLanguageServer):
+        """Test finding function definitions across files."""
+        analysis_file = os.path.join("examples", "analysis.R")
+
+        # In analysis.R line 7: create_data_frame(n = 50)
+        # The function create_data_frame is defined in R/utils.R
+        # Find definition of create_data_frame function call (0-indexed: line 6)
+        definition_location_list = language_server.request_definition(analysis_file, 6, 17)  # cursor on 'create_data_frame'
+
+        assert definition_location_list, f"Expected non-empty definition_location_list but got {definition_location_list=}"
+        assert len(definition_location_list) >= 1
+        definition_location = definition_location_list[0]
+        assert definition_location["uri"].endswith("utils.R")
+        # Definition should be around line 37 (0-indexed: 36) where create_data_frame is defined
+        assert definition_location["range"]["start"]["line"] >= 35
+
+    @pytest.mark.parametrize("language_server", [Language.R], indirect=True)
+    def test_find_references_across_files(self, language_server: SolidLanguageServer):
+        """Test finding function references across files."""
+        analysis_file = os.path.join("examples", "analysis.R")
+
+        # Test from usage side: find references to calculate_mean from its usage in analysis.R
+        # In analysis.R line 13: calculate_mean(clean_data$value)
+        # calculate_mean function call is at line 13 (0-indexed: line 12)
+        references = language_server.request_references(analysis_file, 12, 15)  # cursor on 'calculate_mean'
+
+        assert references, f"Expected non-empty references for calculate_mean but got {references=}"
+
+        # Must find the definition in utils.R (cross-file reference)
+        reference_files = [ref["uri"] for ref in references]
+        assert any(uri.endswith("utils.R") for uri in reference_files), "Cross-file reference to definition in utils.R not found"
+
+        # Verify we actually found the right location in utils.R
+        utils_refs = [ref for ref in references if ref["uri"].endswith("utils.R")]
+        assert len(utils_refs) >= 1, "Should find at least one reference in utils.R"
+        utils_ref = utils_refs[0]
+        # Should be around line 6 where calculate_mean is defined (0-indexed: line 5)
+        assert (
+            utils_ref["range"]["start"]["line"] == 5
+        ), f"Expected reference at line 5 in utils.R, got line {utils_ref['range']['start']['line']}"
+
+    def test_file_matching(self):
+        """Test that R files are properly matched."""
+        from solidlsp.ls_config import Language
+
+        matcher = Language.R.get_source_fn_matcher()
+
+        assert matcher.is_relevant_filename("script.R")
+        assert matcher.is_relevant_filename("analysis.r")
+        assert not matcher.is_relevant_filename("script.py")
+        assert not matcher.is_relevant_filename("README.md")
+
+    def test_r_language_enum(self):
+        """Test R language enum value."""
+        assert Language.R == "r"
+        assert str(Language.R) == "r"