45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
|
import threading
|
||
|
|
import uvicorn
|
||
|
|
from main import app
|
||
|
|
from httpx import Client
|
||
|
|
|
||
|
|
server_thread = threading.Thread(target=uvicorn.run, args=(app,), daemon=True,
|
||
|
|
kwargs={"host": "127.0.0.1", "port": 8000, "reload": False})
|
||
|
|
server_thread.start()
|
||
|
|
|
||
|
|
client = Client(base_url="http://127.0.0.1:8000", timeout=60 * 60 * 60)
|
||
|
|
|
||
|
|
|
||
|
|
def test__message_send():
|
||
|
|
# Complete authorization
|
||
|
|
resp = client.post('/api/v1/auth', json={'username': 1})
|
||
|
|
assert resp.status_code == 200
|
||
|
|
token = resp.json()['access_token']
|
||
|
|
chat_id = client.get('/api/v1/chat', headers={'Authorization': f'Bearer {token}'}).json()[0]["id"]
|
||
|
|
|
||
|
|
resp = client.post('/api/v1/message', json={
|
||
|
|
"chat_id": chat_id,
|
||
|
|
"content": "test__message_send"}, headers={'Authorization': f'Bearer {token}'})
|
||
|
|
assert resp.status_code == 201
|
||
|
|
assert resp.json()["content"] == "test__message_send"
|
||
|
|
assert resp.json()["id"] > 0
|
||
|
|
|
||
|
|
|
||
|
|
def test__message_send_without_auth():
|
||
|
|
resp = client.post('/api/v1/message', json={
|
||
|
|
"chat_id": 70,
|
||
|
|
"content": "test__message_send_without_auth"}, headers={'Authorization': f'Bearer 123'})
|
||
|
|
assert resp.status_code == 401
|
||
|
|
|
||
|
|
|
||
|
|
def test__message_send_without_chat():
|
||
|
|
# Complete authorization
|
||
|
|
resp = client.post('/api/v1/auth', json={'username': 1})
|
||
|
|
assert resp.status_code == 200
|
||
|
|
token = resp.json()['access_token']
|
||
|
|
|
||
|
|
resp = client.post('/api/v1/message', json={
|
||
|
|
"chat_id": -1,
|
||
|
|
"content": "test__message_send_without_chat"}, headers={'Authorization': f'Bearer {token}'})
|
||
|
|
assert resp.status_code == 404
|