import pytest
import pytest_httpbin

from scrapling import StealthyFetcher

StealthyFetcher.auto_match = True


@pytest_httpbin.use_class_based_httpbin
class TestStealthyFetcher:
    @pytest.fixture(scope="class")
    def fetcher(self):
        """Fixture to create a StealthyFetcher instance for the entire test class"""
        return StealthyFetcher

    @pytest.fixture(autouse=True)
    def setup_urls(self, httpbin):
        """Fixture to set up URLs for testing"""
        self.status_200 = f'{httpbin.url}/status/200'
        self.status_404 = f'{httpbin.url}/status/404'
        self.status_501 = f'{httpbin.url}/status/501'
        self.basic_url = f'{httpbin.url}/get'
        self.html_url = f'{httpbin.url}/html'
        self.delayed_url = f'{httpbin.url}/delay/10'  # 10 Seconds delay response
        self.cookies_url = f"{httpbin.url}/cookies/set/test/value"

    def test_basic_fetch(self, fetcher):
        """Test doing basic fetch request with multiple statuses"""
        assert fetcher.fetch(self.status_200).status == 200
        assert fetcher.fetch(self.status_404).status == 404
        assert fetcher.fetch(self.status_501).status == 501

    def test_networkidle(self, fetcher):
        """Test if waiting for `networkidle` make page does not finish loading or not"""
        assert fetcher.fetch(self.basic_url, network_idle=True).status == 200

    def test_blocking_resources(self, fetcher):
        """Test if blocking resources make page does not finish loading or not"""
        assert fetcher.fetch(self.basic_url, block_images=True).status == 200
        assert fetcher.fetch(self.basic_url, disable_resources=True).status == 200

    def test_waiting_selector(self, fetcher):
        """Test if waiting for a selector make page does not finish loading or not"""
        assert fetcher.fetch(self.html_url, wait_selector='h1').status == 200
        assert fetcher.fetch(self.html_url, wait_selector='h1', wait_selector_state='visible').status == 200

    def test_cookies_loading(self, fetcher):
        """Test if cookies are set after the request"""
        assert fetcher.fetch(self.cookies_url).cookies == {'test': 'value'}

    def test_automation(self, fetcher):
        """Test if automation break the code or not"""
        def scroll_page(page):
            page.mouse.wheel(10, 0)
            page.mouse.move(100, 400)
            page.mouse.up()
            return page

        assert fetcher.fetch(self.html_url, page_action=scroll_page).status == 200

    def test_properties(self, fetcher):
        """Test if different arguments breaks the code or not"""
        assert fetcher.fetch(self.html_url, block_webrtc=True, allow_webgl=True).status == 200
        assert fetcher.fetch(self.html_url, block_webrtc=False, allow_webgl=True).status == 200
        assert fetcher.fetch(self.html_url, block_webrtc=True, allow_webgl=False).status == 200
        assert fetcher.fetch(self.html_url, extra_headers={'ayo': ''}, os_randomize=True).status == 200

    def test_infinite_timeout(self, fetcher):
        """Test if infinite timeout breaks the code or not"""
        assert fetcher.fetch(self.delayed_url, timeout=None).status == 200