Skip to content
Snippets Groups Projects
Commit 4e26f7dc authored by vlorentz's avatar vlorentz
Browse files

Add missing fields status/type/snapshot/metadata to OriginVisit.

parent 54957d2f
No related branches found
No related tags found
No related merge requests found
...@@ -78,7 +78,9 @@ def origin_visits(): ...@@ -78,7 +78,9 @@ def origin_visits():
return builds( return builds(
OriginVisit, OriginVisit,
visit=integers(0, 1000), visit=integers(0, 1000),
origin=origins()) origin=origins(),
status=sampled_from(['ongoing', 'full', 'partial']),
type=pgsql_text())
@composite @composite
......
...@@ -104,8 +104,16 @@ class OriginVisit(BaseModel): ...@@ -104,8 +104,16 @@ class OriginVisit(BaseModel):
SWH loader.""" SWH loader."""
origin = attr.ib(type=Origin) origin = attr.ib(type=Origin)
date = attr.ib(type=datetime.datetime) 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], visit = attr.ib(type=Optional[int],
validator=attr.validators.optional([])) default=None)
"""Should not be set before calling 'origin_visit_add()'.""" """Should not be set before calling 'origin_visit_add()'."""
def to_dict(self): def to_dict(self):
...@@ -126,7 +134,7 @@ class OriginVisit(BaseModel): ...@@ -126,7 +134,7 @@ class OriginVisit(BaseModel):
date=(date date=(date
if isinstance(date, datetime.datetime) if isinstance(date, datetime.datetime)
else dateutil.parser.parse(date)), else dateutil.parser.parse(date)),
visit=d.get('visit')) **d)
class TargetType(Enum): class TargetType(Enum):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment