test_app.py

import pytest

from fastapi.testclient import TestClient
from main import app


@pytest.fixture
def client():
    return TestClient(app)


def test_base_endpoint_get(client):
    """ test base endpoint """

    response = client.get('/')
    assert response.status_code == 404


def test_base_endpoint_post(client):
    response = client.post('/')
    assert response.status_code == 404


def test_predict_endpoint_get(client):
    """ test predict endpoint """

    # test get
    response = client.get('/predict')
    assert response.status_code == 405


def test_predict_endpoint_post_nojson(client):
    # test invalid post
    response = client.post('/predict')
    assert response.status_code == 422


def test_predict_endpoint_post_valid_json(client):
    # test valid post
    response = client.post('/predict', json={
        "text": "it's a beautiful world"
    })
    assert response.status_code == 200
    assert response.json()["output"] in ["positive", "negative"]