44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
# This file is part of the Open Parts Database software
|
|
# Copyright (C) 2022 Valentin Lorentz
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it under the
|
|
# terms of the GNU Affero General Public License version 3, as published by the
|
|
# Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
|
|
# PARTICULAR PURPOSE. See the GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License along with
|
|
# this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
pytest fixtures
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from opdb.db import Db, models
|
|
|
|
|
|
def iter_subclasses(cls):
|
|
"""
|
|
Recursively yields all subclasses of a class.
|
|
"""
|
|
yield cls
|
|
for subcls in cls.__subclasses__():
|
|
yield from iter_subclasses(subcls)
|
|
|
|
|
|
@pytest.fixture
|
|
def opdb_db(postgresql) -> Db:
|
|
"""
|
|
pytest fixture which yields an empty initialized OPDB database.
|
|
"""
|
|
with postgresql.cursor() as cur:
|
|
for name in dir(models):
|
|
cls = getattr(models, name)
|
|
if hasattr(cls, "TABLE"):
|
|
cur.execute(cls.db_schema())
|
|
return Db(postgresql)
|