forked from Github-Mirrors/canaille
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
import datetime
|
|
from canaille.commands import clean
|
|
from canaille.models import AuthorizationCode, Token
|
|
from werkzeug.security import gen_salt
|
|
|
|
|
|
def test_clean_command(testclient, slapd_connection, client, user):
|
|
AuthorizationCode.ocs_by_name(slapd_connection)
|
|
code = AuthorizationCode(
|
|
oauthCode="my-code",
|
|
oauthClient=client.dn,
|
|
oauthSubject=user.dn,
|
|
oauthRedirectURI="https://foo.bar/callback",
|
|
oauthResponseType="code",
|
|
oauthScope="openid profile",
|
|
oauthNonce="nonce",
|
|
oauthAuthorizationDate=(
|
|
datetime.datetime.now() - datetime.timedelta(days=1)
|
|
).strftime("%Y%m%d%H%M%SZ"),
|
|
oauthAuthorizationLifetime="3600",
|
|
oauthCodeChallenge="challenge",
|
|
oauthCodeChallengeMethod="method",
|
|
oauthRevokation="",
|
|
)
|
|
code.save(slapd_connection)
|
|
|
|
Token.ocs_by_name(slapd_connection)
|
|
token = Token(
|
|
oauthAccessToken="my-token",
|
|
oauthClient=client.dn,
|
|
oauthSubject=user.dn,
|
|
oauthTokenType=None,
|
|
oauthRefreshToken=gen_salt(48),
|
|
oauthScope="openid profile",
|
|
oauthIssueDate=(datetime.datetime.now() - datetime.timedelta(days=1)).strftime(
|
|
"%Y%m%d%H%M%SZ"
|
|
),
|
|
oauthTokenLifetime=str(3600),
|
|
)
|
|
token.save(slapd_connection)
|
|
|
|
assert AuthorizationCode.get("my-code", conn=slapd_connection)
|
|
assert Token.get("my-token", conn=slapd_connection)
|
|
assert code.is_expired()
|
|
assert token.is_expired()
|
|
|
|
runner = testclient.app.test_cli_runner()
|
|
res = runner.invoke(clean)
|
|
|
|
assert not AuthorizationCode.get("my-code", conn=slapd_connection)
|
|
assert not Token.get("my-token", conn=slapd_connection)
|