# -*- coding: utf-8 -*- import pytest import sqlalchemy as sa from sqlalchemy.ext.declarative import declarative_base from nplusone.core import signals import nplusone.ext.sqlalchemy # noqa from tests import utils Base = declarative_base() models = utils.make_models(Base) @pytest.fixture() def session(): engine = sa.create_engine('sqlite:///:memory:') Session = sa.orm.sessionmaker(bind=engine) Base.metadata.create_all(bind=engine) return Session() @pytest.fixture() def objects(session): hobby = models.Hobby() address = models.Address() user = models.User(addresses=[address], hobbies=[hobby]) session.add(user) session.commit()
def models(db): return utils.make_models(db.Model)
import pytest import sqlalchemy as sa from sqlalchemy.ext.declarative import declarative_base from nplusone.core import signals from nplusone.core import profiler from nplusone.core import exceptions import nplusone.ext.sqlalchemy # noqa from tests import utils Base = declarative_base() models = utils.make_models(Base) @pytest.fixture() def session(): engine = sa.create_engine('sqlite:///:memory:') Session = sa.orm.sessionmaker(bind=engine) Base.metadata.create_all(bind=engine) return Session() @pytest.fixture() def objects(session): hobby = models.Hobby() address = models.Address() user = models.User(addresses=[address], hobbies=[hobby])