def hp_hosts_get(url):
"""Get mock representing a response for a GET request.
:param url: a request address
:returns: a Mock instance representing response object expected
by HpHosts
"""
query_string = urlparse(url).query
query_data = parse_qs(query_string)
content = 'Not Listed'
host = query_data['s'][0]
if host in listed_hosts:
content = 'Listed,{}'.format(class_str)
response = Mock()
response.text = content
return response
def hp_hosts_get(url):
"""Get mock representing a response for a GET request.
:param url: a request address
:returns: a Mock instance representing response object expected
by HpHosts
"""
query_string = urlparse(url).query
query_data = parse_qs(query_string)
content = 'Not Listed'
host = query_data['s'][0]
if host in listed_hosts:
content = 'Listed,{}'.format(class_str)
response = Mock()
response.text = content
return response
def post(_, body):
"""Get mock of a response to a POST query to GSB Lookup API.
:param body: a request body
:returns: a Mock instance representing the response. Properties
of the object depend on external values provided by the creator
of the method: expected_401, spam_urls and classification
"""
response = Mock()
if expected_401:
response.status_code = 401
response.raise_for_status.side_effect = HTTPError
else:
urls = body.splitlines()[1:]
classes = ['ok' if u not in spam_urls else
','.join(classification) for u in urls]
response.text = '\n'.join(classes)
code = 200 if spam_urls else 204
response.status_code = code
return response
def post(_, body):
"""Get mock of a response to a POST query to GSB Lookup API.
:param body: a request body
:returns: a Mock instance representing the response. Properties
of the object depend on external values provided by the creator
of the method: expected_401, spam_urls and classification
"""
response = Mock()
if expected_401:
response.status_code = 401
response.raise_for_status.side_effect = HTTPError
else:
urls = body.splitlines()[1:]
classes = [
'ok' if u not in spam_urls else ','.join(classification)
for u in urls
]
response.text = '\n'.join(classes)
code = 200 if spam_urls else 204
response.status_code = code
return response