diff --git a/fastapi_template/tests/conftest.py b/fastapi_template/tests/conftest.py index 31bfa83c529c77775de013e22bc0cbcb7a64d595..b61d313581fc836f96032b62dacffaa58e84e5c6 100644 --- a/fastapi_template/tests/conftest.py +++ b/fastapi_template/tests/conftest.py @@ -41,7 +41,7 @@ def generator_start_dir() -> str: @pytest.fixture() -def default_context(project_name: str) -> None: +def default_context(project_name: str) -> BuilderContext: """ Default builder context without features. diff --git a/fastapi_template/tests/test_generator.py b/fastapi_template/tests/test_generator.py index b4c234d890eab2d833a52716854fbc52f465c10b..798ab0364d5efd58d44a1a0e6cd651b8213e9bf9 100644 --- a/fastapi_template/tests/test_generator.py +++ b/fastapi_template/tests/test_generator.py @@ -30,19 +30,19 @@ def test_default_without_db(default_context: BuilderContext): DatabaseType.mysql, ], ) -@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar]) +@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar, ORM.piccolo]) def test_default_with_db(default_context: BuilderContext, db: DatabaseType, orm: ORM): run_default_check(init_context(default_context, db, orm)) -@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar]) +@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar, ORM.piccolo]) def test_without_routers(default_context: BuilderContext, orm: ORM): context = init_context(default_context, DatabaseType.postgresql, orm) context.enable_routers = False run_default_check(context) -@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar]) +@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar, ORM.piccolo]) def test_without_migrations(default_context: BuilderContext, orm: ORM): context = init_context(default_context, DatabaseType.postgresql, orm) context.enable_migrations = False @@ -54,7 +54,7 @@ def test_with_selfhosted_swagger(default_context: BuilderContext): run_default_check(default_context) -@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar]) +@pytest.mark.parametrize("orm", [ORM.sqlalchemy, ORM.tortoise, ORM.ormar, ORM.piccolo]) def test_without_dummy(default_context: BuilderContext, orm: ORM): context = init_context(default_context, DatabaseType.postgresql, orm) context.add_dummy = False