Changeset View
Changeset View
Standalone View
Standalone View
swh/model/tests/test_model.py
| Show All 12 Lines | |||||
| from hypothesis import given | from hypothesis import given | ||||
| from hypothesis.strategies import binary | from hypothesis.strategies import binary | ||||
| import pytest | import pytest | ||||
| from swh.model.collections import ImmutableDict | from swh.model.collections import ImmutableDict | ||||
| from swh.model.from_disk import DentryPerms | from swh.model.from_disk import DentryPerms | ||||
| from swh.model.hashutil import MultiHash, hash_to_bytes | from swh.model.hashutil import MultiHash, hash_to_bytes | ||||
| import swh.model.hypothesis_strategies as strategies | import swh.model.hypothesis_strategies as strategies | ||||
| import swh.model.model | |||||
| from swh.model.model import ( | from swh.model.model import ( | ||||
| BaseModel, | BaseModel, | ||||
| Content, | Content, | ||||
| Directory, | Directory, | ||||
| MetadataAuthority, | MetadataAuthority, | ||||
| MetadataAuthorityType, | MetadataAuthorityType, | ||||
| MetadataFetcher, | MetadataFetcher, | ||||
| MissingData, | MissingData, | ||||
| Origin, | Origin, | ||||
| OriginVisit, | OriginVisit, | ||||
| OriginVisitStatus, | OriginVisitStatus, | ||||
| Person, | Person, | ||||
| RawExtrinsicMetadata, | RawExtrinsicMetadata, | ||||
| Release, | Release, | ||||
| Revision, | Revision, | ||||
| SkippedContent, | SkippedContent, | ||||
| Snapshot, | Snapshot, | ||||
| TargetType, | TargetType, | ||||
| Timestamp, | Timestamp, | ||||
| TimestampWithTimezone, | TimestampWithTimezone, | ||||
| type_validator, | type_validator, | ||||
| ) | ) | ||||
| import swh.model.swhids | |||||
| from swh.model.swhids import CoreSWHID, ExtendedSWHID, ObjectType | from swh.model.swhids import CoreSWHID, ExtendedSWHID, ObjectType | ||||
| from swh.model.tests.swh_model_data import TEST_OBJECTS | from swh.model.tests.swh_model_data import TEST_OBJECTS | ||||
| from swh.model.tests.test_identifiers import ( | from swh.model.tests.test_identifiers import ( | ||||
| TS_DATETIMES, | TS_DATETIMES, | ||||
| TS_TIMEZONES, | TS_TIMEZONES, | ||||
| directory_example, | directory_example, | ||||
| metadata_example, | metadata_example, | ||||
| release_example, | release_example, | ||||
| Show All 19 Lines | def test_todict_inverse_fromdict(objtype_and_obj): | ||||
| # Check from_dict() does not change the input dict | # Check from_dict() does not change the input dict | ||||
| assert obj_as_dict == obj_as_dict_copy | assert obj_as_dict == obj_as_dict_copy | ||||
| # Check the composition of from_dict and to_dict is the identity | # Check the composition of from_dict and to_dict is the identity | ||||
| assert obj_as_dict == type(obj).from_dict(obj_as_dict).to_dict() | assert obj_as_dict == type(obj).from_dict(obj_as_dict).to_dict() | ||||
| @given(strategies.objects()) | |||||
| def test_repr(objtype_and_obj): | |||||
| """Checks every model object has a working repr(), and that it can be eval()uated | |||||
| (so that printed objects can be copy-pasted to write test cases.)""" | |||||
| (obj_type, obj) = objtype_and_obj | |||||
| if obj_type == "content" and obj.ctime is not None: | |||||
| # eval() does not always work with non-UTC timezones | |||||
| obj = attr.evolve(obj, ctime=obj.ctime.replace(tzinfo=datetime.timezone.utc)) | |||||
| r = repr(obj) | |||||
| env = { | |||||
| "tzutc": lambda: datetime.timezone.utc, | |||||
| "hash_to_bytes": hash_to_bytes, | |||||
| **swh.model.swhids.__dict__, | |||||
| **swh.model.model.__dict__, | |||||
| } | |||||
| try: | |||||
| obj2 = eval(r, env) | |||||
| except Exception: | |||||
| raise | |||||
| assert obj2 == obj | |||||
| @attr.s | @attr.s | ||||
| class Cls1: | class Cls1: | ||||
| pass | pass | ||||
| @attr.s | @attr.s | ||||
| class Cls2(Cls1): | class Cls2(Cls1): | ||||
| pass | pass | ||||
| ▲ Show 20 Lines • Show All 1,187 Lines • Show Last 20 Lines | |||||