diff --git a/swh/model/hypothesis_strategies.py b/swh/model/hypothesis_strategies.py --- a/swh/model/hypothesis_strategies.py +++ b/swh/model/hypothesis_strategies.py @@ -78,7 +78,9 @@ return builds( OriginVisit, visit=integers(0, 1000), - origin=origins()) + origin=origins(), + status=sampled_from(['ongoing', 'full', 'partial']), + type=pgsql_text()) @composite diff --git a/swh/model/model.py b/swh/model/model.py --- a/swh/model/model.py +++ b/swh/model/model.py @@ -104,8 +104,16 @@ SWH loader.""" origin = attr.ib(type=Origin) date = attr.ib(type=datetime.datetime) + status = attr.ib( + type=str, + validator=attr.validators.in_(['ongoing', 'full', 'partial'])) + type = attr.ib(type=str) + snapshot = attr.ib(type=Sha1Git) + metadata = attr.ib(type=Optional[Dict[str, object]], + default=None) + visit = attr.ib(type=Optional[int], - validator=attr.validators.optional([])) + default=None) """Should not be set before calling 'origin_visit_add()'.""" def to_dict(self): @@ -126,7 +134,7 @@ date=(date if isinstance(date, datetime.datetime) else dateutil.parser.parse(date)), - visit=d.get('visit')) + **d) class TargetType(Enum):