66 lines
1.1 KiB
Python
66 lines
1.1 KiB
Python
import uuid
|
|
import redis
|
|
|
|
from utils.http_client import http_request
|
|
from utils.yaml_loader import load_config
|
|
from utils.template import render
|
|
|
|
|
|
config = load_config()
|
|
|
|
BASE_URL = config["base_url"]
|
|
flow = config["flows"]["auth"]
|
|
|
|
redis_client = redis.Redis(
|
|
host="redis",
|
|
port=6379,
|
|
decode_responses=True
|
|
)
|
|
|
|
|
|
def test_auth_flow():
|
|
|
|
context = {}
|
|
|
|
|
|
# -------- register --------
|
|
|
|
login = flow["login"]
|
|
|
|
body = render(login["body"], context)
|
|
|
|
res = http_request(
|
|
login["method"],
|
|
BASE_URL + login["path"],
|
|
authenticated=False,
|
|
json=body
|
|
)
|
|
print(res)
|
|
|
|
assert res["status"] == login["expected_status"]
|
|
|
|
token_field = login["extract"]["token"]
|
|
|
|
token = res["data"][token_field]
|
|
|
|
assert token is not None
|
|
|
|
context["token"] = token
|
|
|
|
# -------- redis store --------
|
|
|
|
redis_cfg = login["store_redis"]
|
|
|
|
key = render(redis_cfg["key"], context)
|
|
|
|
redis_client.set(
|
|
key,
|
|
token,
|
|
ex=redis_cfg["ttl"]
|
|
)
|
|
|
|
saved_token = redis_client.get(key)
|
|
|
|
assert saved_token == token
|
|
|