diff --git a/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/tests/test_dummy.py b/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/tests/test_dummy.py index 655e2ce26d461d9d3a2aeb00100186795826dff3..52f8cd96126da27751b7ee43e85499558a51f77f 100644 --- a/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/tests/test_dummy.py +++ b/fastapi_template/template/{{cookiecutter.project_name}}/{{cookiecutter.project_name}}/tests/test_dummy.py @@ -24,7 +24,7 @@ async def test_creation( assert response.status_code == status.HTTP_200_OK {%- if cookiecutter.orm == "sqlalchemy" %} dao = DummyDAO(dbsession) - {%- elif cookiecutter.orm in ["tortoise", "ormar"] %} + {%- elif cookiecutter.orm in ["tortoise", "ormar", "piccolo"] %} dao = DummyDAO() {%- endif %} instances = await dao.filter(name=test_name) @@ -42,7 +42,7 @@ async def test_getting( """Tests dummy instance retrieval.""" {%- if cookiecutter.orm == "sqlalchemy" %} dao = DummyDAO(dbsession) - {%- elif cookiecutter.orm in ["tortoise", "ormar"] %} + {%- elif cookiecutter.orm in ["tortoise", "ormar", "piccolo"] %} dao = DummyDAO() {%- endif %} test_name = uuid.uuid4().hex