test: 新增conftest.py

tests/test_crud/conftest.py

@@ -0,0 +1,11 @@
+import pytest
+
+from tests.test_crud.main import app
+
+
+@pytest.fixture(scope='session', autouse=True)
+def startup():
+    import asyncio
+    # asyncio.run(app.router.startup())
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(app.router.startup())

tests/test_crud/main.py

@@ -21,5 +21,3 @@ async def startup():
         await conn.run_sync(SQLModel.metadata.drop_all)
         await conn.run_sync(SQLModel.metadata.create_all)
 
-
-asyncio.run(startup())

tests/test_crud/test_crud.py

@@ -1,76 +1,75 @@
-from unittest import TestCase
 from fastapi.testclient import TestClient
 from tests.test_crud.main import app
 
 client = TestClient(app)
 
 
-class TestSQLModelCrud(TestCase):
+def test_register_crud():
+    response = client.get('/openapi.json')
 
-    def test_register_crud(self):
-        response = client.get('/openapi.json')
+    # test paths
+    paths = response.json()['paths']
+    assert '/category/list' in paths
+    assert '/category/item' in paths
+    assert '/category/item/{item_id}' in paths
+    assert '/tag/list' in paths
+    assert '/tag/item' in paths
+    assert '/tag/item/{item_id}' in paths
 
-        # test paths 
-        paths = response.json()['paths']
-        assert '/category/list' in paths
-        assert '/category/item' in paths
-        assert '/category/item/{item_id}' in paths
-        assert '/tag/list' in paths
-        assert '/tag/item' in paths
-        assert '/tag/item/{item_id}' in paths
+    # test schemas
+    schemas = response.json()['components']['schemas']
+    assert 'Category' in schemas
+    assert 'CategoryFilter' in schemas
+    assert 'CategoryList' in schemas
+    assert 'CategoryUpdate' in schemas
+    assert 'ItemListSchema_CategoryList_' in schemas
+    assert 'TagFilter' in schemas
+    assert 'TagList' in schemas
+    assert 'TagUpdate' in schemas
 
-        # test schemas
-        schemas = response.json()['components']['schemas']
-        assert 'Category' in schemas
-        assert 'CategoryFilter' in schemas
-        assert 'CategoryList' in schemas
-        assert 'CategoryUpdate' in schemas
-        assert 'ItemListSchema_CategoryList_' in schemas
-        assert 'TagFilter' in schemas
-        assert 'TagList' in schemas
-        assert 'TagUpdate' in schemas
 
-    def test_crud_one(self):
-        # create one
-        res = client.post('/category/item', json={"name": 'category_name', "description": "description"})
-        category = res.json().get('data')
-        assert category['id'] > 0, res.json()
-        assert category['name'] == 'category_name', res.json()
-        # update one
-        res = client.put(f'/category/item/{category["id"]}', json={"name": "category_name1"})
-        count = res.json()['data']
-        assert count == 1
-        # read one
-        res = client.get(f'/category/item/{category["id"]}')
-        category_new = res.json()['data']
-        assert category_new['id'] == category["id"]
-        assert category_new['name'] == "category_name1"
-        # delete one
-        res = client.delete(f'/category/item/{category["id"]}')
-        count = res.json()['data']
-        assert count == 1, res.text
+def test_crud_one():
+    # create one
+    res = client.post('/category/item', json={"name": 'category_name', "description": "description"})
+    category = res.json().get('data')
+    assert category['id'] > 0, res.json()
+    assert category['name'] == 'category_name', res.json()
+    # update one
+    res = client.put(f'/category/item/{category["id"]}', json={"name": "category_name1"})
+    count = res.json()['data']
+    assert count == 1
+    # read one
+    res = client.get(f'/category/item/{category["id"]}')
+    category_new = res.json()['data']
+    assert category_new['id'] == category["id"]
+    assert category_new['name'] == "category_name1"
+    # delete one
+    res = client.delete(f'/category/item/{category["id"]}')
+    count = res.json()['data']
+    assert count == 1, res.text
 
-    def test_crud_bulk(self):
-        # create one
-        category_name = 'category_name'
-        count = 2
-        # create bulk
-        categorys = [{'id': i + 1, "name": f'{category_name}_{i}', "description": "description"} for i in range(count)]
-        res = client.post('/category/item', json=categorys)
-        assert res.json()['data'] == count, res.json()
-        # update bulk
-        item_ids = ','.join([str(category['id']) for category in categorys])
-        res = client.put(f'/category/item/{item_ids}', json={"description": "description2"})
-        assert res.json()['data'] == count, res.json()
-        # read bulk
-        res = client.get(f'/category/item/{item_ids}')
-        categorys_new = res.json()['data']
-        assert len(categorys_new) == count, res.json()
-        assert categorys_new[0]['description'] == "description2"
-        # list
-        res = client.post('/category/list')
-        categorys_new2 = res.json()['data']['items']
-        assert len(categorys_new2) == count, res.json()
-        # delete one
-        res = client.delete(f'/category/item/{item_ids}')
-        assert res.json()['data'] == count, res.json()
+
+def test_crud_bulk():
+    # create one
+    category_name = 'category_name'
+    count = 2
+    # create bulk
+    categorys = [{'id': i + 1, "name": f'{category_name}_{i}', "description": "description"} for i in range(count)]
+    res = client.post('/category/item', json=categorys)
+    assert res.json()['data'] == count, res.json()
+    # update bulk
+    item_ids = ','.join([str(category['id']) for category in categorys])
+    res = client.put(f'/category/item/{item_ids}', json={"description": "description2"})
+    assert res.json()['data'] == count, res.json()
+    # read bulk
+    res = client.get(f'/category/item/{item_ids}')
+    categorys_new = res.json()['data']
+    assert len(categorys_new) == count, res.json()
+    assert categorys_new[0]['description'] == "description2"
+    # list
+    res = client.post('/category/list')
+    categorys_new2 = res.json()['data']['items']
+    assert len(categorys_new2) == count, res.json()
+    # delete one
+    res = client.delete(f'/category/item/{item_ids}')
+    assert res.json()['data'] == count, res.json()