Changeset View
Changeset View
Standalone View
Standalone View
swh/provenance/tests/test_history_graph.py
| Show All 36 Lines | with open(get_datafile(filename)) as file: | ||||
| for expected in yaml.full_load(file): | for expected in yaml.full_load(file): | ||||
| entry = OriginEntry(expected["origin"], hash_to_bytes(expected["snapshot"])) | entry = OriginEntry(expected["origin"], hash_to_bytes(expected["snapshot"])) | ||||
| provenance.origin_add(entry) | provenance.origin_add(entry) | ||||
| for expected_graph_as_dict in expected["graphs"]: | for expected_graph_as_dict in expected["graphs"]: | ||||
| print("Expected graph:", expected_graph_as_dict) | print("Expected graph:", expected_graph_as_dict) | ||||
| computed_graph = HistoryGraph( | computed_graph = HistoryGraph( | ||||
| provenance, | |||||
| archive, | archive, | ||||
| RevisionEntry(hash_to_bytes(expected_graph_as_dict["head"]["rev"])), | RevisionEntry(hash_to_bytes(expected_graph_as_dict["head"])), | ||||
| ) | ) | ||||
| print("Computed graph:", computed_graph.as_dict()) | print("Computed graph:", computed_graph.as_dict()) | ||||
| assert computed_graph.as_dict() == expected_graph_as_dict | assert computed_graph.as_dict() == expected_graph_as_dict | ||||
| origin_add_revision(provenance, entry, computed_graph) | origin_add_revision(provenance, entry, computed_graph) | ||||
| if not batch: | if not batch: | ||||
| provenance.flush() | provenance.flush() | ||||