def test_custom_actor_class(self, user_class):
manager = VersioningManager(actor_cls=user_class)
manager.init(declarative_base())
sa.orm.configure_mappers()
assert isinstance(
manager.activity_cls.actor_id.property.columns[0].type, sa.Integer)
assert manager.activity_cls.actor
manager.remove_listeners()
def test_custom_string_actor_class(self):
base = declarative_base()
class User(base):
__tablename__ = 'user'
id = sa.Column(sa.Integer, primary_key=True)
User()
manager = VersioningManager(actor_cls='User')
manager.init(base)
sa.orm.configure_mappers()
assert isinstance(
manager.activity_cls.actor_id.property.columns[0].type, sa.Integer)
assert manager.activity_cls.actor
manager.remove_listeners()
def versioning_manager(base, schema_name):
vm = VersioningManager(schema_name=schema_name)
vm.init(base)
yield vm
vm.remove_listeners()
def versioning_manager(base):
vm = VersioningManager()
vm.init(base)
yield vm
vm.remove_listeners()
def versioning_manager(schema_name):
return VersioningManager(schema_name=schema_name)
