unit_tests.py

import unittest
import main
from main import app
from fastapi import FastAPI
from fastapi.testclient import TestClient

client = TestClient(app)

class TestStringMethods(unittest.TestCase):

 def test_root(self):
 root_value = main.read_root()
 self.assertEqual(root_value['Hello'], 'World')

 def test_get_quote(self):
 first_quote = main.read_item(1)
 self.assertEqual(first_quote.id, 1)
 self.assertEqual(first_quote.author, 'Lemmy Kilmister')

 def test_insert_quote(self):
 response = client.post(
 "/items/",
 headers={"X-Token": "coneofsilence"},
 json={"id": 100, "quote":"New quote", "author":"New author"},
 )
 assert response.status_code == 200
 assert response.json() == {"id": 100, "quote":"New quote", "author":"New author"}


if __name__ == '__main__':
 unittest.main()