def test_publish_messages_raises(klio_job_config, monkeypatch, caplog):
monkeypatch.setattr(klio_job_config.job_config.events, "inputs", None)
with pytest.raises(SystemExit):
publish.publish_messages(klio_job_config, ["s0m3-ent1ty-1D"])
assert 1 == len(caplog.records)
def test_publish_messages(klio_job_config, expected_klio_message,
mock_publisher, caplog):
entity_id = "s0m3-ent1ty-1D"
msg_version = 1
expected_klio_message.metadata.force = False
expected_klio_message.metadata.ping = False
expected_klio_message.data.entity_id = entity_id
expected_klio_message.version = msg_version
publish.publish_messages(klio_job_config, [entity_id])
mock_publisher.return_value.publish.assert_called_once_with(
data=expected_klio_message.SerializeToString(), topic="an-input-topic")
assert 2 == len(caplog.records)
assert "INFO" == caplog.records[0].levelname
assert "INFO" == caplog.records[1].levelname