def test_db(): """Setup a test database fixture Yields: DataStore: a sqlite backed datastore with all test data """ from comet_core.data_store import DataStore from tests.utils import get_all_test_messages data_store = DataStore("sqlite://") for event in get_all_test_messages(parsed=True): data_store.add_record(event.get_record()) yield data_store
def test_data_store(): data_store = comet_core.data_store.DataStore("sqlite://") for event in get_all_test_messages(): data_store.add_record(event.get_record())