def test_prepare_doesnt_upgrade_schema_if_already_initialized():
db = MockDB(is_initialized=True)
settings = MockSettings(db=db)
with spy_on(db.upgrade_schema) as spy:
with pytest.raises(DBInitializedError):
prepare(settings)
assert not spy.called
def test_migrate_doesnt_upgrade_schema_if_not_initialized():
db = MockDB(is_initialized=False)
settings = MockSettings(db=db)
with spy_on(db.upgrade_schema) as spy:
with pytest.raises(DBNotInitializedError):
migrate(settings)
assert not spy.called
def test_service_keeps_css_classes_if_writing_to_fs(mock_template_store,
tmp_path):
mail = FsMail("", logging.create_dev_logger(), tmp_path)
db = MockDB(is_initialized=True)
settings = MockSettings(worker=MockWorker(), mail=mail, db=db)
service(settings, Mock())
assert mock_template_store.call_args.kwargs["keep_css_classes"] is True
def test_prepare_upgrades_schema_and_sets_position():
source = MockSource(feed_end=50)
worker = MockWorker()
db = MockDB(is_initialized=False)
settings = MockSettings(source=source, worker=worker, db=db)
with spy_on(db.upgrade_schema), spy_on(worker.set_initial_position):
prepare(settings)
assert db.upgrade_schema.called
assert worker.set_initial_position.calls[0].args[1] == 50
def test_migrate_upgrades_schema():
db = MockDB(is_initialized=True)
settings = MockSettings(db=db)
with spy_on(db.upgrade_schema) as spy:
migrate(settings)
assert spy.called
def test_service_reads_css(mock_template_store):
db = MockDB(is_initialized=True)
settings = MockSettings(worker=MockWorker(), db=db)
service(settings, Mock())
assert ".event-content" in mock_template_store.call_args.args[1]
def test_service_throws_error_if_db_not_initialized():
settings = MockSettings(db=MockDB(is_initialized=False))
with pytest.raises(DBNotInitializedError):
service(settings, Mock())
def test_service_runs_worker():
worker = Mock()
db = MockDB(is_initialized=True)
settings = MockSettings(worker=worker, db=db)
service(settings, Mock())
worker.process.assert_called()