Scalable factory pattern with Flask, Pytest using SqlAlchemy - python-3.x

Question

Goal:
I want a scalable way to add factories that represent my models so I can manipulate them at run time and have a test DB from test set up to tear down.
Issue:
cls = <class 'tests.factories.analysis.Analysis'>
model_class = <class 'src.api.someModel'>, session = <function dbsession at 0xffffb8e6a5e0>, args = ()
kwargs = {'analysis_id': 365, 'created_by': 619, 'file_extension': 'Stop.', 'original_file_name': 'Share.', ...}
#classmethod
def _save(cls, model_class, session, args, kwargs):
session_persistence = cls._meta.sqlalchemy_session_persistence
obj = model_class(*args, **kwargs)
> session.add(obj)
E AttributeError: 'function' object has no attribute 'add'
What this means to me ^ my Session object is not behaving the way it should, why i dont know.
conftest.py ( got this code snippet from here https://gist.github.com/kissgyorgy/e2365f25a213de44b9a2?permalink_comment_id=3496042 )
import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import Session
from domain.declarative import Base
from sqlalchemy.orm import sessionmaker
TEST_DATABASE = "postgresql://postgres:postgres#localhost:5432/testDB"
#pytest.fixture(scope="session")
def engine():
return create_engine(TEST_DATABASE)
#pytest.fixture(scope="session")
def tables(engine):
Base.metadata.create_all(engine)
yield
Base.metadata.drop_all(engine)
#pytest.fixture
def dbsession(engine, tables):
"""Returns an sqlalchemy session, and after the test tears down everything properly."""
connection = engine.connect()
# begin the nested transaction
transaction = connection.begin()
# use the connection with the already started transaction
session = Session(bind=connection)
yield session
session.close()
# roll back the broader transaction
transaction.rollback()
# release connection
connection.close()
Base_factory.py
import factory.alchemy
from sqlalchemy.orm import scoped_session, sessionmaker
from the.code.above import engine, dbsession
session = scoped_session(sessionmaker(bind=engine))
class BaseFactory(factory.alchemy.SQLAlchemyModelFactory):
class Meta:
abstract = True
sqlalchemy_session = dbsession
sqlalchemy_session_persistence = "commit"
Child Factory
import random
from datetime import datetime
from faker import Faker
from domain.declarative import TheModel
from tests.factories.base_factory import BaseFactory
faker = Faker()
class ChildFactory(BaseFactory):
class Meta:
model = TheModel
analysis_id = random.randrange(1, 1000)
created_by = random.randrange(1, 1000)
file_extension = faker.text(max_nb_chars=10)
original_file_name = faker.text(max_nb_chars=10)
upload_date = faker.date_between_dates(
date_start=datetime(2000, 1, 1), date_end=datetime(2019, 12, 31)
)
some test
class TestSimple(unittest.TestCase):
def test_important_stuff(self):
f = ChildFactory()
So this looks like to me the Session object is not being instantiated properly and is of a type function? it's calling add here at somepoint and we get an atrribute error because of that where am I going wrong?
session.add(obj)
E AttributeError: 'function' object has no attribute 'add'