notifications-api/tests/__init__.py

import uuid

from flask import current_app
from sqlalchemy import select

from app import db
from app.dao.api_key_dao import save_model_api_key
from app.dao.services_dao import dao_fetch_service_by_id
from app.enums import KeyType
from app.models import ApiKey
from notifications_python_client.authentication import create_jwt_token


def create_service_authorization_header(service_id, key_type=KeyType.NORMAL):
    client_id = str(service_id)
    secrets = (
        db.session.execute(
            select(ApiKey).where(
                ApiKey.service_id == service_id, ApiKey.key_type == key_type
            )
        )
        .scalars()
        .all()
    )

    if secrets:
        secret = secrets[0].secret
    else:
        service = dao_fetch_service_by_id(service_id)
        data = {
            "service": service,
            "name": uuid.uuid4(),
            "created_by": service.created_by,
            "key_type": key_type,
        }
        api_key = ApiKey(**data)
        save_model_api_key(api_key)
        secret = api_key.secret

    token = create_jwt_token(secret=secret, client_id=client_id)
    return "Authorization", "Bearer {}".format(token)


def create_admin_authorization_header():
    client_id = current_app.config["ADMIN_CLIENT_ID"]
    return create_internal_authorization_header(client_id)


def create_internal_authorization_header(client_id):
    secret = current_app.config["INTERNAL_CLIENT_API_KEYS"][client_id][0]
    token = create_jwt_token(secret=secret, client_id=client_id)
    return "Authorization", f"Bearer {token}"


def unwrap_function(fn):
    """
    Given a function, returns its undecorated original.
    """
    while hasattr(fn, "__wrapped__"):
        fn = fn.__wrapped__
    return fn