36 lines
947 B
Python
36 lines
947 B
Python
import json
|
|
from typing import Any, Tuple
|
|
|
|
from flask import Flask
|
|
from flask.testing import FlaskClient
|
|
|
|
|
|
class ApiTestCaseMixin:
|
|
@staticmethod
|
|
def get_test_client_and_auth_token(
|
|
app: Flask, as_admin: bool = False
|
|
) -> Tuple[FlaskClient, str]:
|
|
client = app.test_client()
|
|
resp_login = client.post(
|
|
'/api/auth/login',
|
|
data=json.dumps(
|
|
dict(
|
|
email='admin@example.com' if as_admin else 'test@test.com',
|
|
password='12345678',
|
|
)
|
|
),
|
|
content_type='application/json',
|
|
)
|
|
auth_token = json.loads(resp_login.data.decode())['auth_token']
|
|
return client, auth_token
|
|
|
|
|
|
class CallArgsMixin:
|
|
@staticmethod
|
|
def get_args(call_args: Any) -> Any:
|
|
if len(call_args) == 2:
|
|
args, _ = call_args
|
|
else:
|
|
_, args, _ = call_args
|
|
return args
|