def test_discharge(self):
jar = requests.cookies.RequestsCookieJar()
with HTTMock(first_407_then_200):
with HTTMock(discharge_200):
resp = requests.get(ID_PATH,
cookies=jar,
auth=httpbakery.BakeryAuth(cookies=jar))
resp.raise_for_status()
assert 'macaroon-test' in jar.keys()
def test_cookie_with_port(self):
jar = requests.cookies.RequestsCookieJar()
with HTTMock(first_407_then_200_with_port):
with HTTMock(discharge_200):
resp = requests.get('http://example.com:8000/someprotecteurl',
cookies=jar,
auth=httpbakery.BakeryAuth(cookies=jar))
resp.raise_for_status()
assert 'macaroon-test' in jar.keys()
def test_407_then_401_on_discharge(self, mock_open):
jar = requests.cookies.RequestsCookieJar()
with HTTMock(first_407_then_200):
with HTTMock(discharge_401):
with HTTMock(wait_after_401):
resp = requests.get(
ID_PATH, auth=httpbakery.BakeryAuth(cookies=jar))
resp.raise_for_status()
mock_open.assert_called_once_with(u'http://example.com/visit', new=1)
assert 'macaroon-test' in jar.keys()