def stored_event_repo(self):
try:
return self._stored_event_repo
except AttributeError:
self.temp_file = NamedTemporaryFile('a')
uri = 'sqlite:///' + self.temp_file.name
scoped_session_facade = get_scoped_session_facade(uri)
stored_event_repo = SQLAlchemyStoredEventRepository(scoped_session_facade)
self._stored_event_repo = stored_event_repo
return self._stored_event_repo
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 create_db_session(uri):
return get_scoped_session_facade(uri)
def create_db_session(uri):
return get_scoped_session_facade(uri)
