def test_pop_next_url(): auth = authcode.Auth(SECRET_KEY) session = {auth.redirect_key: '/abc/'} next_url = pop_next_url(auth, request, session) assert next_url == '/abc/' auth.sign_in_redirect = '/test/' next_url = pop_next_url(auth, request, {}) assert next_url == auth.sign_in_redirect auth.sign_in_redirect = lambda request: '/dynamic/' next_url = pop_next_url(auth, request, {}) assert next_url == '/dynamic/' auth.sign_in_redirect = None next_url = pop_next_url(auth, request, {}) assert next_url == '/'
def test_pop_next_url_save(): auth = authcode.Auth(SECRET_KEY) session = Session({auth.redirect_key: '/abc/'}) next_url = pop_next_url(auth, request, session) assert next_url == '/abc/' assert session.saved == 1