def test_assert_valid_domain_success(domain): trusted_domains = ["https://example.com", "https://test.com"] sws_m = SWSMessage(trusted_domains=trusted_domains) sws_m._assert_valid_domain(urlsplit(domain))
def test_assert_valid_domain_fail(domain): valid_domains = ["https://example.com", "https://test.com"] sws_m = SWSMessage(trusted_domains=valid_domains) with pytest.raises(UntrustedDomainException): sws_m._assert_valid_domain(urlsplit(domain))