test: add tests for the new feature

This commit is contained in:
Karim shoair
2026-02-07 01:28:58 +02:00
parent 66476d42be
commit 5f557a7f9f
2 changed files with 173 additions and 1 deletions
+22
View File
@@ -77,3 +77,25 @@ class TestValidators:
config = validate(params, StealthConfig)
assert config.timeout == 60000 # Should be increased
def test_playwright_config_blocked_domains(self):
"""Test PlaywrightConfig with blocked_domains"""
params = {"blocked_domains": {"ads.example.com", "tracker.io"}}
config = validate(params, PlaywrightConfig)
assert config.blocked_domains == {"ads.example.com", "tracker.io"}
def test_playwright_config_blocked_domains_default_none(self):
"""Test PlaywrightConfig blocked_domains defaults to None"""
config = validate({}, PlaywrightConfig)
assert config.blocked_domains is None
def test_stealth_config_blocked_domains(self):
"""Test StealthConfig inherits blocked_domains"""
params = {"blocked_domains": {"ads.example.com"}}
config = validate(params, StealthConfig)
assert config.blocked_domains == {"ads.example.com"}