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)