def test_process_true(self, mock_get, mock_process):
response = MagicMock()
response.json = MagicMock(return_value='response_json')
mock_get.return_value = response
mock_process.return_value = 'processed_data'
expected = 'processed_data'
actual = rdap.get('1.1.1.1')
self.assertEqual(expected, actual)
def retrieve_ip_data(shared, ip):
result = {
'ip': ip,
'geoip': None,
'rdap': [],
'ip_rdap': []
}
geo_ip_info = geoip.get(ip)
if geo_ip_info:
result['geoip'] = geo_ip_info
'''
index = 'geoip'
added = wh.write(index, geo_ip_info)
if added:
write_stats[index]['added'] += 1
else:
write_stats[index]['skipped'] += 1
'''
rdap_info = rdap.get(ip)
if rdap_info:
for datum in rdap_info:
if 'handle' not in datum:
continue
result['rdap'].append(datum)
'''
index = 'rdap'
added = wh.write(index, datum)
if added:
write_stats[index]['added'] += 1
else:
write_stats[index]['skipped'] += 1
'''
result['ip_rdap'].append({'ip': ip, 'handle': datum['handle']})
'''
index = 'ip_rdap'
added = wh.write(index, {'ip': ip, 'handle': datum['handle']})
if added:
write_stats[index]['added'] += 1
else:
write_stats[index]['skipped'] += 1
'''
d = shared[0]
d['i'] += 1
new_p = round(d['i'] / d['num'] * 100)
if new_p != d['p']:
print(f'{new_p}%')
d['p'] = new_p
shared[0] = d
return result
def test_not_an_ip(self):
with self.assertRaises(Exception):
rdap.get('not_an_ip')
def test_none_ip(self):
with self.assertRaises(Exception):
rdap.get(None)