1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
"""Fixtures for OAuth2 clients"""
import uuid
import json
import datetime
import pytest
from gn3.auth import db
from gn3.auth.authentication.users import hash_password
from gn3.auth.authentication.oauth2.models.oauth2client import OAuth2Client
@pytest.fixture(autouse=True)
def fxtr_patch_envvars(monkeypatch):
"""Fixture: patch environment variable"""
monkeypatch.setenv("AUTHLIB_INSECURE_TRANSPORT", "true")
@pytest.fixture
def fxtr_oauth2_clients(fxtr_users_with_passwords):
"""Fixture: Create the OAuth2 clients for use with tests."""
conn, users = fxtr_users_with_passwords
now = datetime.datetime.now()
clients = tuple(
OAuth2Client(str(uuid.uuid4()), f"yabadabadoo_{idx:03}", now,
now + datetime.timedelta(hours = 2),
{
"client_name": f"test_client_{idx:03}",
"scope": ["profile", "group", "role", "resource", "register-client"],
"redirect_uri": "/test_oauth2",
"token_endpoint_auth_method": [
"client_secret_post", "client_secret_basic"],
"grant_types": ["password", "authorisation_code", "refresh_token"],
"response_type": "token"
}, user)
for idx, user in enumerate(users, start=1))
with db.cursor(conn) as cursor:
cursor.executemany(
"INSERT INTO oauth2_clients VALUES (?, ?, ?, ?, ?, ?)",
((str(client.client_id), hash_password(client.client_secret),
int(client.client_id_issued_at.timestamp()),
int(client.client_secret_expires_at.timestamp()),
json.dumps(client.client_metadata), str(client.user.user_id))
for client in clients))
yield conn, clients
with db.cursor(conn) as cursor:
cursor.executemany(
"DELETE FROM oauth2_clients WHERE client_id=?",
((str(client.client_id),) for client in clients))
|
