def test_login(): with requests_mock.mock() as m: m.get('https://databasin.org/', cookies={'csrftoken': 'abcd'}) m.post(LOGIN_URL, cookies={'sessionid': 'asdf'}) c = Client() c.login('foo', 'bar') assert m.call_count == 2
def test_login_no_redirect(): with requests_mock.mock() as m: m.get('https://databasin.org/redirect/') m.get('https://databasin.org/', cookies={'csrftoken': 'abcd'}) m.get(LOGIN_URL, cookies={'csrftoken': 'abcd'}) m.post( LOGIN_URL, headers={'Location': 'https://databasin.org/'}, cookies={'sessionid': 'asdf'}, status_code=302 ) c = Client() c.login('foo', 'bar') assert m.call_count == 2 assert not any(r.url for r in m.request_history if r.url == 'https://databasin.org/redirect/')
def test_login_no_redirect(): with requests_mock.mock() as m: m.get('https://databasin.org/redirect/') m.get('https://databasin.org/', cookies={'csrftoken': 'abcd'}) m.get(LOGIN_URL, cookies={'csrftoken': 'abcd'}) m.post(LOGIN_URL, headers={'Location': 'https://databasin.org/'}, cookies={'sessionid': 'asdf'}, status_code=302) c = Client() c.login('foo', 'bar') assert m.call_count == 2 assert not any(r.url for r in m.request_history if r.url == 'https://databasin.org/redirect/')