Add ČSFD Anubis bypass, drop legacy preset tags, rename Země → Země původu

tests/test_csfd.py

@@ -16,9 +16,19 @@ from src.core.csfd import (
     _extract_genres,
     _extract_origin_info,
     _check_dependencies,
+    _solve_anubis_pow,
 )
 
 
+def _mock_session(mock_requests):
+    """Wire ``mock_requests`` so ``requests.Session()`` (also as a context
+    manager) yields a single configurable session mock and return it."""
+    session = MagicMock()
+    session.__enter__.return_value = session
+    mock_requests.Session.return_value = session
+    return session
+
+
 # Sample HTML for testing
 SAMPLE_JSON_LD = """
 {
@@ -219,7 +229,8 @@ class TestFetchMovie:
         mock_response = MagicMock()
         mock_response.text = SAMPLE_HTML
         mock_response.raise_for_status = MagicMock()
-        mock_requests.get.return_value = mock_response
+        session = _mock_session(mock_requests)
+        session.get.return_value = mock_response
 
         movie = fetch_movie("https://www.csfd.cz/film/123-test/")
 
@@ -227,13 +238,14 @@ class TestFetchMovie:
         assert movie.csfd_id == 123
         assert movie.rating == 86
         assert "Drama" in movie.genres
-        mock_requests.get.assert_called_once()
+        session.get.assert_called_once()
 
     @patch("src.core.csfd.requests")
     def test_fetch_movie_network_error(self, mock_requests):
         """Test network error handling."""
         import requests as real_requests
-        mock_requests.get.side_effect = real_requests.RequestException("Network error")
+        session = _mock_session(mock_requests)
+        session.get.side_effect = real_requests.RequestException("Network error")
 
         with pytest.raises(real_requests.RequestException):
             fetch_movie("https://www.csfd.cz/film/123/")
@@ -254,7 +266,8 @@ class TestSearchMovies:
         mock_response = MagicMock()
         mock_response.text = search_html
         mock_response.raise_for_status = MagicMock()
-        mock_requests.get.return_value = mock_response
+        session = _mock_session(mock_requests)
+        session.get.return_value = mock_response
         mock_requests.utils.quote = lambda x: x
 
         results = search_movies("test", limit=10)
@@ -277,6 +290,24 @@ class TestFetchMovieById:
         assert movie.title == "Test"
 
 
+class TestAnubisPoW:
+    """Tests for the Anubis proof-of-work solver."""
+
+    def test_solve_pow_difficulty_one(self):
+        """Difficulty 1 requires a single leading zero nibble in the hash."""
+        import hashlib
+
+        random_data = "abc123"
+        hash_hex, nonce, _ = _solve_anubis_pow(random_data, difficulty=1)
+        assert hash_hex[0] == "0"
+        assert hashlib.sha256(f"{random_data}{nonce}".encode()).hexdigest() == hash_hex
+
+    def test_solve_pow_difficulty_two(self):
+        """Difficulty 2 requires two leading zero nibbles (one zero byte)."""
+        hash_hex, _, _ = _solve_anubis_pow("seed", difficulty=2)
+        assert hash_hex[:2] == "00"
+
+
 class TestDependencyCheck:
     """Tests for dependency checking."""