Initial version of stealthii
An aiohttp/httpx-shaped async HTTP client backed by a single, persistent, stealth-patched Chromium instance via Playwright. get/post/put/patch/ delete/head/download use context.request (real Chrome TLS/HTTP2 fingerprint, no tab needed); render()/screenshot()/page() escalate to full JS-executing page rendering for targets that need a JS challenge solved. Supports named sessions for cookie persistence/warm-up, and `async with stealth as page` to claim a tab directly, task-safe for concurrent use on one shared instance. Context/browser recycling bounds cookie and on-disk cache growth in a long-lived process. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_0116i7dYLNZTXNR7Lqf439bV
This commit is contained in:
@@ -0,0 +1,75 @@
|
||||
# retoor <retoor@molodetz.nl>
|
||||
import pytest
|
||||
|
||||
from stealthii import Stealthii, StealthResponse
|
||||
from stealthii.client import _ContextSlot
|
||||
|
||||
|
||||
def test_construction_is_lazy_and_cheap():
|
||||
s = Stealthii()
|
||||
assert s._browser is None
|
||||
assert s._playwright is None
|
||||
assert s._contexts == {}
|
||||
|
||||
|
||||
def test_invalid_fingerprint_rejected():
|
||||
with pytest.raises(ValueError):
|
||||
Stealthii(fingerprint='ja5')
|
||||
|
||||
|
||||
def test_defaults():
|
||||
s = Stealthii()
|
||||
assert s.fingerprint == 'ja4'
|
||||
assert s.max_tabs == 4
|
||||
assert s.max_retries == 3
|
||||
assert s.recycle_context_after_count == 500
|
||||
assert s.recycle_context_after_s == 1800.0
|
||||
|
||||
|
||||
def test_build_body_json():
|
||||
headers = {}
|
||||
kwargs = Stealthii._build_body(headers, data=None, json={'a': 1}, files=None)
|
||||
assert kwargs == {'data': '{"a": 1}'}
|
||||
assert headers['Content-Type'] == 'application/json'
|
||||
|
||||
|
||||
def test_build_body_form_dict():
|
||||
kwargs = Stealthii._build_body({}, data={'x': '1'}, json=None, files=None)
|
||||
assert kwargs == {'form': {'x': '1'}}
|
||||
|
||||
|
||||
def test_build_body_raw_bytes():
|
||||
kwargs = Stealthii._build_body({}, data=b'raw', json=None, files=None)
|
||||
assert kwargs == {'data': b'raw'}
|
||||
|
||||
|
||||
def test_build_body_none():
|
||||
assert Stealthii._build_body({}, data=None, json=None, files=None) == {}
|
||||
|
||||
|
||||
def test_build_body_multipart_with_files_and_fields():
|
||||
kwargs = Stealthii._build_body(
|
||||
{}, data={'field': 'value'}, json=None,
|
||||
files={'upload': ('name.txt', b'content')},
|
||||
)
|
||||
multipart = kwargs['multipart']
|
||||
assert multipart['field'] == 'value'
|
||||
assert multipart['upload']['name'] == 'name.txt'
|
||||
assert multipart['upload']['buffer'] == b'content'
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stealth_response_shapes_like_aiohttp():
|
||||
resp = StealthResponse(200, {'Content-Type': 'application/json'}, b'{"ok": true}', 'https://x/')
|
||||
assert resp.status == 200
|
||||
assert await resp.text() == '{"ok": true}'
|
||||
assert await resp.json() == {'ok': True}
|
||||
assert await resp.read() == b'{"ok": true}'
|
||||
async with resp as r:
|
||||
assert r is resp
|
||||
|
||||
|
||||
def test_context_slot_starts_fresh():
|
||||
slot = _ContextSlot(context=object())
|
||||
assert slot.count == 0
|
||||
assert slot.patched is False
|
||||
Reference in New Issue
Block a user