Karim shoair
127 lines
4.4 KiB
Python
127 lines
4.4 KiB
Python
import pytest
|
|
import pytest_httpbin
|
|
|
|
from scrapling.fetchers import AsyncFetcher
|
|
|
|
AsyncFetcher.auto_match = True
|
|
|
|
|
|
@pytest_httpbin.use_class_based_httpbin
|
|
@pytest.mark.asyncio
|
|
class TestAsyncFetcher:
|
|
@pytest.fixture(scope="class")
|
|
def fetcher(self):
|
|
return AsyncFetcher
|
|
|
|
@pytest.fixture(scope="class")
|
|
def urls(self, httpbin):
|
|
return {
|
|
"status_200": f"{httpbin.url}/status/200",
|
|
"status_404": f"{httpbin.url}/status/404",
|
|
"status_501": f"{httpbin.url}/status/501",
|
|
"basic_url": f"{httpbin.url}/get",
|
|
"post_url": f"{httpbin.url}/post",
|
|
"put_url": f"{httpbin.url}/put",
|
|
"delete_url": f"{httpbin.url}/delete",
|
|
"html_url": f"{httpbin.url}/html",
|
|
}
|
|
|
|
async def test_basic_get(self, fetcher, urls):
|
|
"""Test doing basic get request with multiple statuses"""
|
|
assert (await fetcher.get(urls["status_200"])).status == 200
|
|
assert (await fetcher.get(urls["status_404"])).status == 404
|
|
assert (await fetcher.get(urls["status_501"])).status == 501
|
|
|
|
async def test_get_properties(self, fetcher, urls):
|
|
"""Test if different arguments with GET request breaks the code or not"""
|
|
assert (
|
|
await fetcher.get(urls["status_200"], stealthy_headers=True)
|
|
).status == 200
|
|
assert (
|
|
await fetcher.get(urls["status_200"], follow_redirects=True)
|
|
).status == 200
|
|
assert (await fetcher.get(urls["status_200"], timeout=None)).status == 200
|
|
assert (
|
|
await fetcher.get(
|
|
urls["status_200"],
|
|
stealthy_headers=True,
|
|
follow_redirects=True,
|
|
timeout=None,
|
|
)
|
|
).status == 200
|
|
|
|
async def test_post_properties(self, fetcher, urls):
|
|
"""Test if different arguments with POST request breaks the code or not"""
|
|
assert (
|
|
await fetcher.post(urls["post_url"], data={"key": "value"})
|
|
).status == 200
|
|
assert (
|
|
await fetcher.post(
|
|
urls["post_url"], data={"key": "value"}, stealthy_headers=True
|
|
)
|
|
).status == 200
|
|
assert (
|
|
await fetcher.post(
|
|
urls["post_url"], data={"key": "value"}, follow_redirects=True
|
|
)
|
|
).status == 200
|
|
assert (
|
|
await fetcher.post(urls["post_url"], data={"key": "value"}, timeout=None)
|
|
).status == 200
|
|
assert (
|
|
await fetcher.post(
|
|
urls["post_url"],
|
|
data={"key": "value"},
|
|
stealthy_headers=True,
|
|
follow_redirects=True,
|
|
timeout=None,
|
|
)
|
|
).status == 200
|
|
|
|
async def test_put_properties(self, fetcher, urls):
|
|
"""Test if different arguments with PUT request breaks the code or not"""
|
|
assert (await fetcher.put(urls["put_url"], data={"key": "value"})).status in [
|
|
200,
|
|
405,
|
|
]
|
|
assert (
|
|
await fetcher.put(
|
|
urls["put_url"], data={"key": "value"}, stealthy_headers=True
|
|
)
|
|
).status in [200, 405]
|
|
assert (
|
|
await fetcher.put(
|
|
urls["put_url"], data={"key": "value"}, follow_redirects=True
|
|
)
|
|
).status in [200, 405]
|
|
assert (
|
|
await fetcher.put(urls["put_url"], data={"key": "value"}, timeout=None)
|
|
).status in [200, 405]
|
|
assert (
|
|
await fetcher.put(
|
|
urls["put_url"],
|
|
data={"key": "value"},
|
|
stealthy_headers=True,
|
|
follow_redirects=True,
|
|
timeout=None,
|
|
)
|
|
).status in [200, 405]
|
|
|
|
async def test_delete_properties(self, fetcher, urls):
|
|
"""Test if different arguments with DELETE request breaks the code or not"""
|
|
assert (
|
|
await fetcher.delete(urls["delete_url"], stealthy_headers=True)
|
|
).status == 200
|
|
assert (
|
|
await fetcher.delete(urls["delete_url"], follow_redirects=True)
|
|
).status == 200
|
|
assert (await fetcher.delete(urls["delete_url"], timeout=None)).status == 200
|
|
assert (
|
|
await fetcher.delete(
|
|
urls["delete_url"],
|
|
stealthy_headers=True,
|
|
follow_redirects=True,
|
|
timeout=None,
|
|
)
|
|
).status == 200
|