def create_repo(stored_repo_class, temp_file_name): if stored_repo_class == CassandraStoredEventRepository: setup_cassandra_connection(*get_cassandra_setup_params()) repo = CassandraStoredEventRepository( always_check_expected_version=True, always_write_entity_version=True, ) elif stored_repo_class == SQLAlchemyStoredEventRepository: uri = 'sqlite:///' + temp_file_name scoped_session_facade = get_scoped_session_facade(uri) repo = SQLAlchemyStoredEventRepository(scoped_session_facade, always_check_expected_version=True, always_write_entity_version=True, ) elif stored_repo_class == PythonObjectsStoredEventRepository: repo = PythonObjectsStoredEventRepository( always_check_expected_version=True, always_write_entity_version=True, ) else: raise Exception("Stored repo class not yet supported in test: {}".format(stored_repo_class)) return repo
def setup_cassandra_connection(*args): setup_cassandra_connection(*get_cassandra_setup_params(*args))
def setUp(self): setup_cassandra_connection(*get_cassandra_setup_params()) super(CassandraStoredEventRepoTestCase, self).setUp()