def is_payload_collection_blacklisted(url):
"""
Return whether the payload should be collected according to the blacklisted
urls list in the Trace.
:param url: url string
:return: True if URL is blacklisted, else False
"""
url = urllib.parse.urlparse(url).netloc
if trace_factory.get_trace():
trace_blacklist_urls = trace_factory.get_trace().url_patterns_to_ignore
else:
trace_blacklist_urls = tuple()
return any(blacklist_url in url for blacklist_url in trace_blacklist_urls)
def test_trace_blacklist():
"""
Validate trace URL Blacklist mechanism.
:return: None
"""
trace_factory.get_trace().url_patterns_to_ignore = set(
('test.net', 'test2.net'))
assert epsagon.http_filters.is_payload_collection_blacklisted(
'http://www.test.net')
assert epsagon.http_filters.is_payload_collection_blacklisted(
'http://www.bla.test.net')
assert not epsagon.http_filters.is_payload_collection_blacklisted(
'http://www.test.new.net')
trace_factory.get_trace().url_patterns_to_ignore = set()
assert not epsagon.http_filters.is_payload_collection_blacklisted(
'http://www.test.net')
assert not epsagon.http_filters.is_payload_collection_blacklisted(
'http://www.bla.test.net')