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()