Преглед изворни кода

test(backend-testing-01): add pattern_manager parsing tests

- parse_theta_rho_file tests (valid, comments, empty lines, not found)
- parse_theta_rho_file edge cases (invalid lines, whitespace, scientific notation, negative)
- list_theta_rho_files tests (basic, subdirectories, skip cached_images)
- get_status tests (idle, running, paused, playlist)
- is_clear_pattern tests (standard, mini, pro, regular patterns)

22 tests total, all passing
tuanchris пре 1 недеља
родитељ
комит
398f015ecb
1 измењених фајлова са 352 додато и 0 уклоњено
  1. 352 0
      tests/unit/test_pattern_manager.py

+ 352 - 0
tests/unit/test_pattern_manager.py

@@ -0,0 +1,352 @@
+"""
+Unit tests for pattern_manager parsing logic.
+
+Tests the core pattern file operations:
+- Parsing theta-rho files
+- Handling comments and empty lines
+- Error handling for invalid files
+- Listing pattern files
+"""
+import os
+import pytest
+from unittest.mock import patch, MagicMock
+
+
+class TestParseTheTaRhoFile:
+    """Tests for parse_theta_rho_file function."""
+
+    def test_parse_theta_rho_file_valid(self, tmp_path):
+        """Test parsing a valid theta-rho file."""
+        # Create test file
+        test_file = tmp_path / "valid.thr"
+        test_file.write_text("0.0 0.5\n1.57 0.8\n3.14 0.3\n")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert len(coordinates) == 3
+        assert coordinates[0] == (0.0, 0.5)
+        assert coordinates[1] == (1.57, 0.8)
+        assert coordinates[2] == (3.14, 0.3)
+
+    def test_parse_theta_rho_file_with_comments(self, tmp_path):
+        """Test parsing handles # comments correctly."""
+        test_file = tmp_path / "commented.thr"
+        test_file.write_text("""# This is a header comment
+0.0 0.5
+# Another comment in the middle
+1.0 0.6
+# Trailing comment
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert len(coordinates) == 2
+        assert coordinates[0] == (0.0, 0.5)
+        assert coordinates[1] == (1.0, 0.6)
+
+    def test_parse_theta_rho_file_empty_lines(self, tmp_path):
+        """Test parsing handles empty lines correctly."""
+        test_file = tmp_path / "spaced.thr"
+        test_file.write_text("""0.0 0.5
+
+1.0 0.6
+
+2.0 0.7
+
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert len(coordinates) == 3
+        assert coordinates[0] == (0.0, 0.5)
+        assert coordinates[1] == (1.0, 0.6)
+        assert coordinates[2] == (2.0, 0.7)
+
+    def test_parse_theta_rho_file_not_found(self, tmp_path):
+        """Test parsing a non-existent file returns empty list."""
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(tmp_path / "nonexistent.thr"))
+
+        assert coordinates == []
+
+    def test_parse_theta_rho_file_invalid_lines(self, tmp_path):
+        """Test parsing skips invalid lines (non-numeric values)."""
+        test_file = tmp_path / "invalid.thr"
+        test_file.write_text("""0.0 0.5
+invalid line
+1.0 0.6
+not a number here
+2.0 0.7
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        # Should only get the valid lines
+        assert len(coordinates) == 3
+        assert coordinates[0] == (0.0, 0.5)
+        assert coordinates[1] == (1.0, 0.6)
+        assert coordinates[2] == (2.0, 0.7)
+
+    def test_parse_theta_rho_file_whitespace_handling(self, tmp_path):
+        """Test parsing handles various whitespace correctly."""
+        test_file = tmp_path / "whitespace.thr"
+        test_file.write_text("""  0.0 0.5
+	1.0 0.6
+0.0    0.5
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert len(coordinates) == 3
+
+    def test_parse_theta_rho_file_scientific_notation(self, tmp_path):
+        """Test parsing handles scientific notation."""
+        test_file = tmp_path / "scientific.thr"
+        test_file.write_text("""1.5e-3 0.5
+3.14159 1.0e0
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert len(coordinates) == 2
+        assert coordinates[0][0] == pytest.approx(0.0015)
+        assert coordinates[1][1] == pytest.approx(1.0)
+
+    def test_parse_theta_rho_file_negative_values(self, tmp_path):
+        """Test parsing handles negative values."""
+        test_file = tmp_path / "negative.thr"
+        test_file.write_text("""-3.14 0.5
+0.0 -0.5
+-1.0 -0.3
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert len(coordinates) == 3
+        assert coordinates[0] == (-3.14, 0.5)
+        assert coordinates[1] == (0.0, -0.5)
+        assert coordinates[2] == (-1.0, -0.3)
+
+    def test_parse_theta_rho_file_only_comments(self, tmp_path):
+        """Test parsing a file with only comments returns empty list."""
+        test_file = tmp_path / "comments_only.thr"
+        test_file.write_text("""# This file only has comments
+# No actual coordinates
+# Just documentation
+""")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert coordinates == []
+
+    def test_parse_theta_rho_file_empty_file(self, tmp_path):
+        """Test parsing an empty file returns empty list."""
+        test_file = tmp_path / "empty.thr"
+        test_file.write_text("")
+
+        from modules.core.pattern_manager import parse_theta_rho_file
+
+        coordinates = parse_theta_rho_file(str(test_file))
+
+        assert coordinates == []
+
+
+class TestListThetaRhoFiles:
+    """Tests for list_theta_rho_files function."""
+
+    def test_list_theta_rho_files_basic(self, tmp_path):
+        """Test listing pattern files in directory."""
+        # Create test pattern files
+        patterns_dir = tmp_path / "patterns"
+        patterns_dir.mkdir()
+        (patterns_dir / "circle.thr").write_text("0 0.5")
+        (patterns_dir / "spiral.thr").write_text("0 0.5")
+        (patterns_dir / "readme.txt").write_text("not a pattern")
+
+        with patch("modules.core.pattern_manager.THETA_RHO_DIR", str(patterns_dir)):
+            from modules.core.pattern_manager import list_theta_rho_files
+
+            files = list_theta_rho_files()
+
+        # Should only list .thr files
+        assert len(files) == 2
+        assert "circle.thr" in files
+        assert "spiral.thr" in files
+
+    def test_list_theta_rho_files_subdirectories(self, tmp_path):
+        """Test listing pattern files in subdirectories."""
+        patterns_dir = tmp_path / "patterns"
+        patterns_dir.mkdir()
+
+        # Create subdirectory with patterns
+        subdir = patterns_dir / "custom"
+        subdir.mkdir()
+        (subdir / "custom_pattern.thr").write_text("0 0.5")
+        (patterns_dir / "root_pattern.thr").write_text("0 0.5")
+
+        with patch("modules.core.pattern_manager.THETA_RHO_DIR", str(patterns_dir)):
+            from modules.core.pattern_manager import list_theta_rho_files
+
+            files = list_theta_rho_files()
+
+        assert len(files) == 2
+        assert "root_pattern.thr" in files
+        # Subdirectory patterns should include relative path
+        assert "custom/custom_pattern.thr" in files
+
+    def test_list_theta_rho_files_skips_cached_images(self, tmp_path):
+        """Test that cached_images directories are skipped."""
+        patterns_dir = tmp_path / "patterns"
+        patterns_dir.mkdir()
+
+        # Create cached_images directory with files
+        cache_dir = patterns_dir / "cached_images"
+        cache_dir.mkdir()
+        (cache_dir / "preview.thr").write_text("should be skipped")
+
+        (patterns_dir / "real_pattern.thr").write_text("0 0.5")
+
+        with patch("modules.core.pattern_manager.THETA_RHO_DIR", str(patterns_dir)):
+            from modules.core.pattern_manager import list_theta_rho_files
+
+            files = list_theta_rho_files()
+
+        # Should only list the real pattern, not cached files
+        assert len(files) == 1
+        assert "real_pattern.thr" in files
+
+    def test_list_theta_rho_files_empty_directory(self, tmp_path):
+        """Test listing from empty directory returns empty list."""
+        patterns_dir = tmp_path / "patterns"
+        patterns_dir.mkdir()
+
+        with patch("modules.core.pattern_manager.THETA_RHO_DIR", str(patterns_dir)):
+            from modules.core.pattern_manager import list_theta_rho_files
+
+            files = list_theta_rho_files()
+
+        assert files == []
+
+
+class TestGetStatus:
+    """Tests for get_status function."""
+
+    def test_get_status_idle(self, mock_state):
+        """Test get_status returns expected fields when idle."""
+        with patch("modules.core.pattern_manager.state", mock_state):
+            from modules.core.pattern_manager import get_status
+
+            status = get_status()
+
+        assert "current_file" in status
+        assert "is_paused" in status
+        assert "is_running" in status
+        assert "is_homing" in status
+        assert "progress" in status
+        assert "playlist" in status
+        assert "speed" in status
+        assert "connection_status" in status
+        assert status["is_running"] is False
+        assert status["current_file"] is None
+
+    def test_get_status_running_pattern(self, mock_state):
+        """Test get_status reflects running pattern."""
+        mock_state.current_playing_file = "test_pattern.thr"
+        mock_state.stop_requested = False
+        mock_state.execution_progress = (50, 100, 30.5, 60.0)
+
+        with patch("modules.core.pattern_manager.state", mock_state):
+            from modules.core.pattern_manager import get_status
+
+            status = get_status()
+
+        assert status["is_running"] is True
+        assert status["current_file"] == "test_pattern.thr"
+        assert status["progress"] is not None
+        assert status["progress"]["current"] == 50
+        assert status["progress"]["total"] == 100
+        assert status["progress"]["percentage"] == 50.0
+
+    def test_get_status_paused(self, mock_state):
+        """Test get_status reflects paused state."""
+        mock_state.pause_requested = True
+
+        with patch("modules.core.pattern_manager.state", mock_state):
+            with patch("modules.core.pattern_manager.is_in_scheduled_pause_period", return_value=False):
+                from modules.core.pattern_manager import get_status
+
+                status = get_status()
+
+        assert status["is_paused"] is True
+        assert status["manual_pause"] is True
+
+    def test_get_status_with_playlist(self, mock_state):
+        """Test get_status includes playlist info when running."""
+        mock_state.current_playlist = ["a.thr", "b.thr", "c.thr"]
+        mock_state.current_playlist_name = "test_playlist"
+        mock_state.current_playlist_index = 1
+        mock_state.playlist_mode = "indefinite"
+
+        with patch("modules.core.pattern_manager.state", mock_state):
+            from modules.core.pattern_manager import get_status
+
+            status = get_status()
+
+        assert status["playlist"] is not None
+        assert status["playlist"]["current_index"] == 1
+        assert status["playlist"]["total_files"] == 3
+        assert status["playlist"]["mode"] == "indefinite"
+        assert status["playlist"]["name"] == "test_playlist"
+
+
+class TestIsClearPattern:
+    """Tests for is_clear_pattern function."""
+
+    def test_is_clear_pattern_matches_standard(self):
+        """Test identifying standard clear patterns."""
+        from modules.core.pattern_manager import is_clear_pattern
+
+        assert is_clear_pattern("./patterns/clear_from_out.thr") is True
+        assert is_clear_pattern("./patterns/clear_from_in.thr") is True
+        assert is_clear_pattern("./patterns/clear_sideway.thr") is True
+
+    def test_is_clear_pattern_matches_mini(self):
+        """Test identifying mini table clear patterns."""
+        from modules.core.pattern_manager import is_clear_pattern
+
+        assert is_clear_pattern("./patterns/clear_from_out_mini.thr") is True
+        assert is_clear_pattern("./patterns/clear_from_in_mini.thr") is True
+        assert is_clear_pattern("./patterns/clear_sideway_mini.thr") is True
+
+    def test_is_clear_pattern_matches_pro(self):
+        """Test identifying pro table clear patterns."""
+        from modules.core.pattern_manager import is_clear_pattern
+
+        assert is_clear_pattern("./patterns/clear_from_out_pro.thr") is True
+        assert is_clear_pattern("./patterns/clear_from_in_pro.thr") is True
+        assert is_clear_pattern("./patterns/clear_sideway_pro.thr") is True
+
+    def test_is_clear_pattern_rejects_regular_patterns(self):
+        """Test that regular patterns are not identified as clear patterns."""
+        from modules.core.pattern_manager import is_clear_pattern
+
+        assert is_clear_pattern("./patterns/circle.thr") is False
+        assert is_clear_pattern("./patterns/spiral.thr") is False
+        assert is_clear_pattern("./patterns/custom/my_pattern.thr") is False