def test_check_networks_both(self, gethostbyname_ex): allowed_networks = ['15/8', '74.125/16'] gethostbyname_ex.return_value = ( 'example.com', [], [ '74.125.224.64', '74.125.224.67', '74.125.224.68', '74.125.224.70', ] ) self.assertTrue(validators.check_networks( 'example.com', allowed_networks)) self.assertTrue(validators.check_networks_strict( 'example.com', allowed_networks)) gethostbyname_ex.return_value = ('example.com', [], ['12.2.2.2']) self.assertFalse(validators.check_networks( 'example.com', allowed_networks)) gethostbyname_ex.return_value = ( 'example.com', [], [ '15.8.2.2', '15.8.2.1', '16.1.1.1', ] ) self.assertFalse(validators.check_networks_strict( 'example.com', allowed_networks))
def test_check_networks_strict_exception(self, gethostbyname_ex): gethostbyname_ex.side_effect = socket.gaierror() self.assertFalse( validators.check_networks_strict('mock', ['mock']), )