def test_mount_disabled(self): sess = advocate.Session() self.assertRaises( MountDisabledException, sess.mount, "foo://", None, )
def test_passing_session(self): try: FuturesSession(session=requests.Session()) assert False except NotImplementedError: pass sess = FuturesSession() try: sess.session = requests.Session() assert False except NotImplementedError: pass sess.session = advocate.Session()
def advocate_session(): kwargs = {} if os.environ.get('AUCTUS_REQUEST_WHITELIST', '').strip(): kwargs['ip_whitelist'] = { ipaddress.ip_network(info[4][0] + '/32') for host in os.environ['AUCTUS_REQUEST_WHITELIST'].split(',') for info in socket.getaddrinfo( host, 80, 0, socket.SOCK_STREAM, ) } validator = advocate.AddrValidator(**kwargs) return advocate.Session(validator=validator)