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")