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