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 | |||||