Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/test_journal_client.py
Show All 22 Lines | writer_config = { | ||||
"client_id": "kafka_writer", | "client_id": "kafka_writer", | ||||
"prefix": kafka_prefix, | "prefix": kafka_prefix, | ||||
"anonymize": False, | "anonymize": False, | ||||
} | } | ||||
yield {**swh_storage_backend_config, "journal_writer": writer_config} | yield {**swh_storage_backend_config, "journal_writer": writer_config} | ||||
@pytest.mark.origin_layer | @pytest.mark.origin_layer | ||||
@pytest.mark.kafka | |||||
def test_cli_origin_from_journal_client( | def test_cli_origin_from_journal_client( | ||||
swh_storage: StorageInterface, | swh_storage: StorageInterface, | ||||
swh_storage_backend_config: Dict, | swh_storage_backend_config: Dict, | ||||
kafka_prefix: str, | kafka_prefix: str, | ||||
kafka_server: str, | kafka_server: str, | ||||
consumer: Consumer, | consumer: Consumer, | ||||
tmp_path: str, | tmp_path: str, | ||||
provenance, | provenance, | ||||
Show All 38 Lines | ) -> None: | ||||
origin_sha1 = MultiHash.from_data( | origin_sha1 = MultiHash.from_data( | ||||
origin_url.encode(), hash_names=["sha1"] | origin_url.encode(), hash_names=["sha1"] | ||||
).digest()["sha1"] | ).digest()["sha1"] | ||||
actual_result = provenance.storage.origin_get([origin_sha1]) | actual_result = provenance.storage.origin_get([origin_sha1]) | ||||
assert actual_result == {origin_sha1: origin_url} | assert actual_result == {origin_sha1: origin_url} | ||||
@pytest.mark.kafka | |||||
def test_cli_revision_from_journal_client( | def test_cli_revision_from_journal_client( | ||||
swh_storage: StorageInterface, | swh_storage: StorageInterface, | ||||
swh_storage_backend_config: Dict, | swh_storage_backend_config: Dict, | ||||
kafka_prefix: str, | kafka_prefix: str, | ||||
kafka_server: str, | kafka_server: str, | ||||
consumer: Consumer, | consumer: Consumer, | ||||
tmp_path: str, | tmp_path: str, | ||||
provenance, | provenance, | ||||
▲ Show 20 Lines • Show All 44 Lines • Show Last 20 Lines |