def test_as_json_w_geo():
p = Proxy('8.8.8.8', '3128')
p._runtimes = [1, 3, 3]
p.types.update({'HTTP': 'Anonymous', 'HTTPS': None})
json_tpl = {
'host':
'8.8.8.8',
'port':
3128,
'geo': {
'country': {
'code': 'US',
'name': 'United States',
},
},
'types': [
{
'type': 'HTTP',
'level': 'Anonymous'
},
{
'type': 'HTTPS',
'level': ''
},
],
'avg_resp_time':
2.33,
'error_rate':
0,
}
assert p.as_json() == json_tpl
def test_as_json_wo_geo():
p = Proxy('127.0.0.1', '80')
p.log('MSG', time.time(), ProxyConnError)
p.stat['requests'] = 4
json_tpl = {
'host': '127.0.0.1',
'port': 80,
'geo': {
'country': {'code': '--', 'name': 'Unknown'},
'region': {'code': 'Unknown', 'name': 'Unknown'},
'city': 'Unknown',
},
'types': [],
'avg_resp_time': 0,
'error_rate': 0.25,
}
assert p.as_json() == json_tpl