Changeset View
Changeset View
Standalone View
Standalone View
swh/model/tests/test_model.py
Show First 20 Lines • Show All 1,395 Lines • ▼ Show 20 Lines | _metadata_fetcher = MetadataFetcher( | ||||
name="test-fetcher", | name="test-fetcher", | ||||
version="0.0.1", | version="0.0.1", | ||||
) | ) | ||||
_content_swhid = ExtendedSWHID.from_string( | _content_swhid = ExtendedSWHID.from_string( | ||||
"swh:1:cnt:94a9ed024d3859793618152ea559a168bbcbb5e2" | "swh:1:cnt:94a9ed024d3859793618152ea559a168bbcbb5e2" | ||||
) | ) | ||||
_origin_url = "https://forge.softwareheritage.org/source/swh-model.git" | _origin_url = "https://forge.softwareheritage.org/source/swh-model.git" | ||||
_origin_swhid = ExtendedSWHID.from_string( | _origin_swhid = ExtendedSWHID.from_string( | ||||
"swh:1:ori:94a9ed024d3859793618152ea559a168bbcbb5e2" | "swh:1:ori:433b4f5612f0720ed51fa7aeaf43a3625870057b" | ||||
) | ) | ||||
_dummy_qualifiers = {"origin": "https://example.com", "lines": "42"} | _dummy_qualifiers = {"origin": "https://example.com", "lines": "42"} | ||||
_common_metadata_fields = dict( | _common_metadata_fields = dict( | ||||
discovery_date=datetime.datetime( | discovery_date=datetime.datetime( | ||||
2021, 1, 29, 13, 57, 9, tzinfo=datetime.timezone.utc | 2021, 1, 29, 13, 57, 9, tzinfo=datetime.timezone.utc | ||||
), | ), | ||||
authority=_metadata_authority, | authority=_metadata_authority, | ||||
fetcher=_metadata_fetcher, | fetcher=_metadata_fetcher, | ||||
Show All 10 Lines | def test_metadata_valid(): | ||||
# Object with an SWHID | # Object with an SWHID | ||||
RawExtrinsicMetadata( | RawExtrinsicMetadata( | ||||
target=_content_swhid, | target=_content_swhid, | ||||
**_common_metadata_fields, | **_common_metadata_fields, | ||||
) | ) | ||||
def test_metadata_from_old_dict(): | |||||
common_fields = { | |||||
"authority": {"type": "forge", "url": "https://forge.softwareheritage.org"}, | |||||
"fetcher": { | |||||
"name": "test-fetcher", | |||||
"version": "0.0.1", | |||||
}, | |||||
"discovery_date": _common_metadata_fields["discovery_date"], | |||||
"format": "json", | |||||
"metadata": b'{"origin": "https://example.com", "lines": "42"}', | |||||
} | |||||
m = RawExtrinsicMetadata( | |||||
target=_origin_swhid, | |||||
**_common_metadata_fields, | |||||
) | |||||
assert ( | |||||
RawExtrinsicMetadata.from_dict( | |||||
{"id": m.id, "target": _origin_url, "type": "origin", **common_fields} | |||||
) | |||||
== m | |||||
) | |||||
m = RawExtrinsicMetadata( | |||||
target=_content_swhid, | |||||
**_common_metadata_fields, | |||||
) | |||||
assert ( | |||||
RawExtrinsicMetadata.from_dict( | |||||
{"target": str(_content_swhid), "type": "content", **common_fields} | |||||
) | |||||
== m | |||||
) | |||||
def test_metadata_to_dict(): | def test_metadata_to_dict(): | ||||
"""Checks valid RawExtrinsicMetadata objects don't raise an error.""" | """Checks valid RawExtrinsicMetadata objects don't raise an error.""" | ||||
common_fields = { | common_fields = { | ||||
"authority": {"type": "forge", "url": "https://forge.softwareheritage.org"}, | "authority": {"type": "forge", "url": "https://forge.softwareheritage.org"}, | ||||
"fetcher": { | "fetcher": { | ||||
"name": "test-fetcher", | "name": "test-fetcher", | ||||
"version": "0.0.1", | "version": "0.0.1", | ||||
}, | }, | ||||
"discovery_date": _common_metadata_fields["discovery_date"], | "discovery_date": _common_metadata_fields["discovery_date"], | ||||
"format": "json", | "format": "json", | ||||
"metadata": b'{"origin": "https://example.com", "lines": "42"}', | "metadata": b'{"origin": "https://example.com", "lines": "42"}', | ||||
} | } | ||||
m = RawExtrinsicMetadata( | m = RawExtrinsicMetadata( | ||||
target=_origin_swhid, | target=_origin_swhid, | ||||
**_common_metadata_fields, | **_common_metadata_fields, | ||||
) | ) | ||||
assert m.to_dict() == { | assert m.to_dict() == { | ||||
"target": str(_origin_swhid), | "target": str(_origin_swhid), | ||||
"id": b"@j\xc9\x01\xbc\x1e#p*\xf3q9\xa7u\x97\x00\x14\x02xa", | "id": b"\xa3)q\x0f\xf7p\xc7\xb0\\O\xe8\x84\x83Z\xb0]\x81\xe9\x95\x13", | ||||
**common_fields, | **common_fields, | ||||
} | } | ||||
assert RawExtrinsicMetadata.from_dict(m.to_dict()) == m | assert RawExtrinsicMetadata.from_dict(m.to_dict()) == m | ||||
m = RawExtrinsicMetadata( | m = RawExtrinsicMetadata( | ||||
target=_content_swhid, | target=_content_swhid, | ||||
**_common_metadata_fields, | **_common_metadata_fields, | ||||
) | ) | ||||
▲ Show 20 Lines • Show All 327 Lines • Show Last 20 Lines |