示例#1
0
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 == '/'
示例#2
0
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