Serena

""" Tests for TOML language server symbol retrieval functionality. These tests focus on advanced symbol operations: - request_containing_symbol - request_document_overview - request_full_symbol_tree - request_dir_overview """ from pathlib import Path import pytest from solidlsp import SolidLanguageServer from solidlsp.ls_config import Language pytestmark = pytest.mark.toml class TestTomlSymbolRetrieval: """Test advanced symbol retrieval functionality for TOML files.""" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_request_containing_symbol_behavior(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test request_containing_symbol behavior for TOML files. Note: Taplo LSP doesn't support definition/containing symbol lookups for TOML files since TOML is a configuration format, not code. This test verifies the behavior. """ # Line 2 (0-indexed: 1) is inside the [package] table containing_symbol = language_server.request_containing_symbol("Cargo.toml", 1, 5) # Taplo doesn't support containing symbol lookup - returns None # This is expected behavior for a configuration file format assert containing_symbol is None, "TOML LSP doesn't support containing symbol lookup" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_request_document_overview_cargo(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test request_document_overview for Cargo.toml.""" overview = language_server.request_document_overview("Cargo.toml") assert overview is not None assert len(overview) > 0 # Get symbol names from overview symbol_names = {symbol.get("name") for symbol in overview if "name" in symbol} # Verify expected top-level tables appear expected_tables = {"package", "dependencies", "dev-dependencies", "features", "workspace"} assert expected_tables.issubset(symbol_names), f"Missing expected tables in overview: {expected_tables - symbol_names}" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_request_document_overview_pyproject(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test request_document_overview for pyproject.toml.""" overview = language_server.request_document_overview("pyproject.toml") assert overview is not None assert len(overview) > 0 # Get symbol names from overview symbol_names = {symbol.get("name") for symbol in overview if "name" in symbol} # Verify expected top-level tables appear assert "project" in symbol_names, "Should detect 'project' table" assert "build-system" in symbol_names, "Should detect 'build-system' table" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_request_full_symbol_tree(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test request_full_symbol_tree returns TOML files.""" symbol_tree = language_server.request_full_symbol_tree() assert symbol_tree is not None assert len(symbol_tree) > 0 # The root should be test_repo root = symbol_tree[0] assert root["name"] == "test_repo" assert "children" in root # Children should include TOML files child_names = {child["name"] for child in root.get("children", [])} # Note: File names are stripped of extension in some cases assert ( "Cargo" in child_names or "Cargo.toml" in child_names or any("cargo" in name.lower() for name in child_names) ), f"Should find Cargo.toml in tree, got: {child_names}" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_request_dir_overview(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test request_dir_overview returns symbols for TOML files.""" overview = language_server.request_dir_overview(".") assert overview is not None assert len(overview) > 0 # Should have entries for both Cargo.toml and pyproject.toml file_paths = list(overview.keys()) assert any("Cargo.toml" in path for path in file_paths), f"Should find Cargo.toml in overview, got: {file_paths}" assert any("pyproject.toml" in path for path in file_paths), f"Should find pyproject.toml in overview, got: {file_paths}" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_symbol_hierarchy_in_cargo(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test that symbol hierarchy is properly preserved in Cargo.toml.""" all_symbols, root_symbols = language_server.request_document_symbols("Cargo.toml").get_all_symbols_and_roots() # Find the 'package' table package_symbol = next((s for s in root_symbols if s.get("name") == "package"), None) assert package_symbol is not None, "Should find 'package' as root symbol" # Verify it has children (nested keys) assert "children" in package_symbol, "'package' should have children" child_names = {child.get("name") for child in package_symbol.get("children", [])} # Package should have name, version, edition at minimum assert "name" in child_names, "'package' should have 'name' child" assert "version" in child_names, "'package' should have 'version' child" assert "edition" in child_names, "'package' should have 'edition' child" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_symbol_hierarchy_in_pyproject(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test that symbol hierarchy is properly preserved in pyproject.toml.""" all_symbols, root_symbols = language_server.request_document_symbols("pyproject.toml").get_all_symbols_and_roots() # Find the 'project' table project_symbol = next((s for s in root_symbols if s.get("name") == "project"), None) assert project_symbol is not None, "Should find 'project' as root symbol" # Verify it has children assert "children" in project_symbol, "'project' should have children" child_names = {child.get("name") for child in project_symbol.get("children", [])} # Project should have name, version, dependencies at minimum assert "name" in child_names, "'project' should have 'name' child" assert "version" in child_names, "'project' should have 'version' child" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_tool_section_hierarchy(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test that tool sections in pyproject.toml are properly structured.""" all_symbols, root_symbols = language_server.request_document_symbols("pyproject.toml").get_all_symbols_and_roots() # Get all symbol names all_names = [s.get("name") for s in all_symbols] # Should detect tool.ruff, tool.mypy, or tool.pytest has_ruff = any("ruff" in name.lower() for name in all_names if name) has_mypy = any("mypy" in name.lower() for name in all_names if name) has_pytest = any("pytest" in name.lower() for name in all_names if name) assert has_ruff or has_mypy or has_pytest, f"Should detect tool sections, got names: {all_names}" @pytest.mark.parametrize("language_server", [Language.TOML], indirect=True) @pytest.mark.parametrize("repo_path", [Language.TOML], indirect=True) def test_array_of_tables_symbol(self, language_server: SolidLanguageServer, repo_path: Path) -> None: """Test that [[bin]] array of tables is detected.""" all_symbols, root_symbols = language_server.request_document_symbols("Cargo.toml").get_all_symbols_and_roots() # Get all symbol names all_names = [s.get("name") for s in all_symbols] # Should detect bin array of tables has_bin = "bin" in all_names assert has_bin, f"Should detect [[bin]] array of tables, got names: {all_names}" # Find the bin symbol and verify its structure bin_symbol = next((s for s in all_symbols if s.get("name") == "bin"), None) assert bin_symbol is not None, "Should find bin symbol" # Array of tables should be kind 18 (array) assert bin_symbol.get("kind") == 18, "[[bin]] should have kind 18 (array)" # Children of array of tables are indexed by position ('0', '1', etc.) if "children" in bin_symbol: bin_children = bin_symbol.get("children", []) assert len(bin_children) > 0, "[[bin]] should have at least one child element" # First child is index '0' first_child = bin_children[0] assert first_child.get("name") == "0", f"First array element should be named '0', got: {first_child.get('name')}" # The '0' element should contain name and path as grandchildren if "children" in first_child: grandchild_names = {gc.get("name") for gc in first_child.get("children", [])} assert "name" in grandchild_names, f"[[bin]] element should have 'name' field, got: {grandchild_names}" assert "path" in grandchild_names, f"[[bin]] element should have 'path' field, got: {grandchild_names}"