Source code for florist.tests.integration.api.test_client

import requests
import uvicorn

from florist.api.auth.token import Token, DEFAULT_USERNAME, DEFAULT_PASSWORD, _simple_hash
from florist.tests.integration.api.utils import TestUvicornServer, change_default_password, use_test_database




def test_authentication_success(use_test_database):
    host = "localhost"
    port = 8001
    new_password = "new_password"
    service_config = uvicorn.Config("florist.api.client:app", host=host, port=port, log_level="debug")
    service = TestUvicornServer(config=service_config)

    with service.run_in_thread():
        response = requests.post(
            f"http://{host}:{port}/api/client/auth/token",
            data={
                "grant_type": "password",
                "username": DEFAULT_USERNAME,
                "password": _simple_hash(DEFAULT_PASSWORD),
            }
        )
        token = Token(**response.json())
        assert token.should_change_password == True

        token = change_default_password(f"{host}:{port}", _simple_hash(new_password), "client")
        assert token.should_change_password == False

        response = requests.get(
            f"http://{host}:{port}/api/client/connect",
            headers={"Authorization": f"Bearer {token.access_token}"}
        )
        assert response.status_code == 200





def test_authentication_failure_default_password(use_test_database):
    host = "localhost"
    port = 8001
    service_config = uvicorn.Config("florist.api.client:app", host=host, port=port, log_level="debug")
    service = TestUvicornServer(config=service_config)

    with service.run_in_thread():
        response = requests.post(
            f"http://{host}:{port}/api/client/auth/token",
            data={
                "grant_type": "password",
                "username": DEFAULT_USERNAME,
                "password": _simple_hash(DEFAULT_PASSWORD),
            }
        )
        token = Token(**response.json())
        assert token.should_change_password == True

        response = requests.get(
            f"http://{host}:{port}/api/client/connect",
            headers={"Authorization": f"Bearer {token.access_token}"}
        )
        assert response.status_code == 401





def test_authentication_failure_invalid_token(use_test_database):
    host = "localhost"
    port = 8001
    service_config = uvicorn.Config("florist.api.client:app", host=host, port=port, log_level="debug")
    service = TestUvicornServer(config=service_config)

    with service.run_in_thread():
        response = requests.get(
            f"http://{host}:{port}/api/client/connect",
            headers={"Authorization": f"Bearer notavalidtoken"}
        )
        assert response.status_code == 401