test_api.py

from fastapi.testclient import TestClient
from api import app


#
# Some Unit Tests
client = TestClient(app)


def test_list_news():
    response = client.get("/news")
    assert response.status_code == 200
    assert type(response.json()) == list   # list object is returned


def test_list_news_no_empty_results():
    response = client.get("/news")
    assert len([obj for obj in response.json() if not obj]) == 0   # response has no empty objects/ {}


def test_list_news_required_fields():
    response = client.get("/news")
    objects_with_missing_fields = []
    for obj in response.json():
        print(obj.keys())
        if any(field not in obj.keys() for field in ["title", "link", "source"]):
            objects_with_missing_fields.append(obj)
    assert len(objects_with_missing_fields) == 0   # make sure 3 essential fields are present