Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/conftest.py
Show First 20 Lines • Show All 76 Lines • ▼ Show 20 Lines | elif request.param == "rabbitmq": | ||||
rabbitmq_params: Dict[str, Any] = { | rabbitmq_params: Dict[str, Any] = { | ||||
"url": f"amqp://guest:guest@{host}:{port}/%2f", | "url": f"amqp://guest:guest@{host}:{port}/%2f", | ||||
"storage_config": { | "storage_config": { | ||||
"cls": "postgresql", # TODO: also test with underlying mongodb storage | "cls": "postgresql", # TODO: also test with underlying mongodb storage | ||||
"db": provenance_postgresqldb, | "db": provenance_postgresqldb, | ||||
"raise_on_commit": True, | "raise_on_commit": True, | ||||
}, | }, | ||||
} | } | ||||
server = ProvenanceStorageRabbitMQServer( | server = ProvenanceStorageRabbitMQServer( | ||||
url=rabbitmq_params["url"], storage_config=rabbitmq_params["storage_config"] | url=rabbitmq_params["url"], storage_config=rabbitmq_params["storage_config"] | ||||
) | ) | ||||
server.start() | server.start() | ||||
with get_provenance_storage(cls=request.param, **rabbitmq_params) as storage: | with get_provenance_storage(cls=request.param, **rabbitmq_params) as storage: | ||||
yield storage | yield storage | ||||
server.stop() | server.stop(terminate_after=60) | ||||
else: | else: | ||||
# in test sessions, we DO want to raise any exception occurring at commit time | # in test sessions, we DO want to raise any exception occurring at commit time | ||||
with get_provenance_storage( | with get_provenance_storage( | ||||
cls=request.param, db=provenance_postgresqldb, raise_on_commit=True | cls=request.param, db=provenance_postgresqldb, raise_on_commit=True | ||||
) as storage: | ) as storage: | ||||
yield storage | yield storage | ||||
▲ Show 20 Lines • Show All 77 Lines • Show Last 20 Lines |