diff --git a/swh/model/hypothesis_strategies.py b/swh/model/hypothesis_strategies.py index df90cc714209aa9f68a7359a3f9303f2b70fa51a..af145a14eff59a5f02ee2a7738ab6494e604d55a 100644 --- a/swh/model/hypothesis_strategies.py +++ b/swh/model/hypothesis_strategies.py @@ -38,13 +38,13 @@ from .model import ( DirectoryEntry, MetadataAuthority, MetadataFetcher, - ObjectType, Origin, OriginVisit, OriginVisitStatus, Person, RawExtrinsicMetadata, Release, + ReleaseTargetType, Revision, RevisionType, SkippedContent, @@ -224,7 +224,7 @@ def origin_visit_statuses(**kwargs): @composite def releases_d(draw, **kwargs): defaults = dict( - target_type=sampled_from([x.value for x in ObjectType]), + target_type=sampled_from([x.value for x in ReleaseTargetType]), name=binary(), message=optional(binary()), synthetic=booleans(),