def test_list_blocked_ips_rdns_failure(get_ip_status_urlopen_mock,
socket_error_mock):
rdns_return = sndslib.list_blocked_ips_rdns(['0.0.0.1', '0.0.0.1'])
expected_return = [{
'ip': '0.0.0.1',
'rdns': 'NXDOMAIN'
}, {
'ip': '0.0.0.1',
'rdns': 'NXDOMAIN'
}]
assert rdns_return == expected_return
def test_list_blocked_ips_rdns_success(get_ip_status_urlopen_mock,
socket_mock):
resp = sndslib.get_ip_status('test')
blocked_ips = sndslib.list_blocked_ips(resp)
rdns_return = sndslib.list_blocked_ips_rdns(blocked_ips)
expected_return = [
{
'ip': '1.1.1.0',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.1.1',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.1.3',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.1.254',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.1.255',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.2.0',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.2.1',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.1.255.255',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.2.0.0',
'rdns': 'rdns.mock.com'
},
{
'ip': '1.255.255.255',
'rdns': 'rdns.mock.com'
},
{
'ip': '2.0.0.0',
'rdns': 'rdns.mock.com'
},
]
assert rdns_return == expected_return
def list_blocked_ips_rdns(self):
_rdns = sndslib.list_blocked_ips_rdns(self.blocked_ips)
self._print_list_blocked_ips_rdns(_rdns)
def blocked_ips_rdns_mock(get_ip_status_urlopen_mock, socket_mock):
resp = sndslib.get_ip_status('test')
blocked_ips = sndslib.list_blocked_ips(resp)
blocked_ips_with_rdns = sndslib.list_blocked_ips_rdns(blocked_ips)
return blocked_ips_with_rdns
def test_list_blocked_ips_rdns_empty_list(get_ip_status_urlopen_mock):
rdns_return = sndslib.list_blocked_ips_rdns([])
assert rdns_return == []
def test_list_blocked_ips_rdns_failure_single_ip(get_ip_status_urlopen_mock,
socket_error_mock):
rdns_return = sndslib.list_blocked_ips_rdns('0.0.0.1')
assert rdns_return == [{'ip': '0.0.0.1', 'rdns': 'NXDOMAIN'}]
def test_list_blocked_ips_rdns_success_single_ip(get_ip_status_urlopen_mock,
socket_mock):
rdns_return = sndslib.list_blocked_ips_rdns('1.1.1.0')
expected_return = [{'ip': '1.1.1.0', 'rdns': 'rdns.mock.com'}]
assert rdns_return == expected_return