mirror of
https://github.com/jaypyles/Scraperr.git
synced 2025-11-26 11:03:25 +00:00
28 lines
728 B
Python
28 lines
728 B
Python
import pytest
|
|
import logging
|
|
from unittest.mock import AsyncMock, patch, MagicMock
|
|
from api.backend.scraping import create_driver
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@patch("seleniumwire.webdriver.Chrome.get")
|
|
async def test_proxy(mock_get: AsyncMock):
|
|
# Mock the response of the requests.get call
|
|
mock_response = MagicMock()
|
|
mock_get.return_value = mock_response
|
|
|
|
driver = create_driver(proxies=["127.0.0.1:8080"])
|
|
assert driver is not None
|
|
|
|
# Simulate a request
|
|
driver.get("http://example.com")
|
|
response = driver.last_request
|
|
|
|
if response:
|
|
assert response.headers["Proxy-Connection"] == "keep-alive"
|
|
|
|
driver.quit()
|