backend/crappy_test.py

import uuid

import pytest
from fastapi.testclient import TestClient

import backend

client = TestClient(backend.app)


@pytest.mark.asyncio
async def test_register_new_key():
    data = {"openai_key": "new_key"}
    response = client.post("/register", json=data)
    assert response.status_code == 200
    assert "uid" in response.json()
    uid = response.json()["uid"]

    data = {"openai_key": "new_key"}
    response = client.post("/register", json=data)
    assert response.status_code == 200
    assert "uid" in response.json()
    assert response.json()["uid"] == uid


@pytest.mark.asyncio
async def test_register_existing_key():
    data = {"openai_key": "existing_key"}
    response = client.post("/register", json=data)
    assert response.status_code == 200
    assert "uid" in response.json()
    uid = response.json()["uid"]

    data = {"openai_key": "existing_key"}
    response = client.post("/register", json=data)
    assert response.status_code == 200
    assert "uid" in response.json()
    assert response.json()["uid"] == uid


@pytest.mark.asyncio
async def test_send_event():
    data = {
        "uid": str(uuid.uuid4()),
        "event": "test event",
    }
    client.post("/register", json=data)


@pytest.mark.asyncio
async def test_assist():
    data = {
        "uid": str(uuid.uuid4()),
        "prompt": "test prompt",
        "version": "davinci-fake",
    }
    client.post("/assist", json=data)