30 lines
718 B
Python
30 lines
718 B
Python
from flask import url_for
|
|
|
|
from app.models import ApiToCookieToken, ApiKey
|
|
from tests.utils import create_new_user
|
|
|
|
|
|
def test_get_cookie(flask_client):
|
|
user = create_new_user()
|
|
api_key = ApiKey.create(
|
|
user_id=user.id,
|
|
commit=True,
|
|
)
|
|
token = ApiToCookieToken.create(
|
|
user_id=user.id,
|
|
api_key_id=api_key.id,
|
|
commit=True,
|
|
)
|
|
token_code = token.code
|
|
token_id = token.id
|
|
|
|
r = flask_client.get(
|
|
url_for(
|
|
"auth.api_to_cookie", token=token_code, next=url_for("dashboard.setting")
|
|
),
|
|
follow_redirects=True,
|
|
)
|
|
|
|
assert ApiToCookieToken.get(token_id) is None
|
|
assert r.headers.getlist("Set-Cookie") is not None
|