Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/conftest.py
| Show All 11 Lines | |||||
| from _pytest.fixtures import SubRequest | from _pytest.fixtures import SubRequest | ||||
| from aiohttp.test_utils import TestClient, TestServer, loop_context | from aiohttp.test_utils import TestClient, TestServer, loop_context | ||||
| import msgpack | import msgpack | ||||
| import psycopg2.extensions | import psycopg2.extensions | ||||
| import pytest | import pytest | ||||
| from pytest_postgresql.factories import postgresql | from pytest_postgresql.factories import postgresql | ||||
| from swh.graph.http_server import make_app | from swh.graph.http_rpc_server import make_app | ||||
| from swh.journal.serializers import msgpack_ext_hook | from swh.journal.serializers import msgpack_ext_hook | ||||
| from swh.model.model import BaseModel, TimestampWithTimezone | from swh.model.model import BaseModel, TimestampWithTimezone | ||||
| from swh.provenance import get_provenance, get_provenance_storage | from swh.provenance import get_provenance, get_provenance_storage | ||||
| from swh.provenance.archive import ArchiveInterface | from swh.provenance.archive import ArchiveInterface | ||||
| from swh.provenance.interface import ProvenanceInterface, ProvenanceStorageInterface | from swh.provenance.interface import ProvenanceInterface, ProvenanceStorageInterface | ||||
| from swh.provenance.storage.archive import ArchiveStorage | from swh.provenance.storage.archive import ArchiveStorage | ||||
| from swh.storage.interface import StorageInterface | from swh.storage.interface import StorageInterface | ||||
| from swh.storage.replay import OBJECT_CONVERTERS, OBJECT_FIXERS, process_replay_objects | from swh.storage.replay import OBJECT_CONVERTERS, OBJECT_FIXERS, process_replay_objects | ||||
| ▲ Show 20 Lines • Show All 135 Lines • ▼ Show 20 Lines | |||||
| def ts2dt(ts: Dict[str, Any]) -> datetime: | def ts2dt(ts: Dict[str, Any]) -> datetime: | ||||
| return TimestampWithTimezone.from_dict(ts).to_datetime() | return TimestampWithTimezone.from_dict(ts).to_datetime() | ||||
| def run_grpc_server(queue, dataset_path): | def run_grpc_server(queue, dataset_path): | ||||
| try: | try: | ||||
| config = {"graph": {"path": dataset_path}} | config = { | ||||
| "graph": { | |||||
| "cls": "local", | |||||
| "grpc_server": {"path": dataset_path}, | |||||
| "http_rpc_server": {"debug": True}, | |||||
| } | |||||
| } | |||||
| with loop_context() as loop: | with loop_context() as loop: | ||||
| app = make_app(config=config, debug=True, spawn_rpc_port=None) | app = make_app(config=config) | ||||
| client = TestClient(TestServer(app), loop=loop) | client = TestClient(TestServer(app), loop=loop) | ||||
| loop.run_until_complete(client.start_server()) | loop.run_until_complete(client.start_server()) | ||||
| url = client.make_url("/graph/") | url = client.make_url("/graph/") | ||||
| queue.put((url, app["rpc_url"])) | queue.put((url, app["rpc_url"])) | ||||
| loop.run_forever() | loop.run_forever() | ||||
| except Exception as e: | except Exception as e: | ||||
| queue.put(e) | queue.put(e) | ||||
| Show All 19 Lines | |||||