update tests/integration/test_grabber_integration.py

tests/integration/test_grabber_integration.py

@@ -0,0 +1,59 @@
+import pytest
+import os
+from app.grabber.github_grabber import GitHubGrabber
+from app.models.source import SourceConfig, SourceType
+
+
+class TestGrabberIntegration:
+    @pytest.mark.asyncio
+    async def test_extract_from_fixture_http(self):
+        fixture_path = os.path.join(
+            os.path.dirname(__file__), "../fixtures/github_raw_http.txt"
+        )
+
+        with open(fixture_path, "r") as f:
+            content = f.read()
+
+        grabber = GitHubGrabber()
+
+        proxies = await grabber.parse_content(content, SourceType.GITHUB_RAW)
+
+        assert len(proxies) >= 4
+        assert all(p.protocol == "http" for p in proxies)
+        assert all(1 <= p.port <= 65535 for p in proxies)
+
+    @pytest.mark.asyncio
+    async def test_extract_from_fixture_mixed(self):
+        fixture_path = os.path.join(
+            os.path.dirname(__file__), "../fixtures/github_raw_mixed.txt"
+        )
+
+        with open(fixture_path, "r") as f:
+            content = f.read()
+
+        grabber = GitHubGrabber()
+        proxies = await grabber.parse_content(content, SourceType.GENERIC_TEXT)
+
+        protocols = {p.protocol for p in proxies}
+
+        assert "http" in protocols
+        assert len(proxies) >= 3
+
+    @pytest.mark.asyncio
+    async def test_base64_subscription_fixture(self):
+        fixture_path = os.path.join(
+            os.path.dirname(__file__), "../fixtures/subscription_vmess.txt"
+        )
+
+        with open(fixture_path, "r") as f:
+            content = f.read().strip()
+
+        grabber = GitHubGrabber()
+        proxies = await grabber.parse_content(content, SourceType.SUBSCRIPTION_BASE64)
+
+        assert len(proxies) >= 1
+
+    @pytest.mark.integration
+    @pytest.mark.asyncio
+    async def test_real_github_url_skip(self):
+        pytest.skip("Real URL test - run with: pytest -m integration")