Skip to content
Snippets Groups Projects
Commit 0b9c5be2 authored by vlorentz's avatar vlorentz
Browse files

Make OriginVisit.origin a string instead of a dict.

parent b064a0bb
No related branches found
No related tags found
No related merge requests found
......@@ -79,7 +79,7 @@ def origin_visits():
return builds(
OriginVisit,
visit=integers(0, 1000),
origin=origins(),
origin=urls(),
status=sampled_from(['ongoing', 'full', 'partial']),
type=pgsql_text())
......
......@@ -121,7 +121,7 @@ class Origin(BaseModel):
class OriginVisit(BaseModel):
"""Represents a visit of an origin at a given point in time, by a
SWH loader."""
origin = attr.ib(type=Origin)
origin = attr.ib(type=str)
date = attr.ib(type=datetime.datetime)
status = attr.ib(
type=str,
......@@ -149,7 +149,6 @@ class OriginVisit(BaseModel):
d = d.copy()
date = d.pop('date')
return cls(
origin=Origin.from_dict(d.pop('origin')),
date=(date
if isinstance(date, datetime.datetime)
else dateutil.parser.parse(date)),
......
......@@ -3,7 +3,6 @@
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information
import attr
import copy
from hypothesis import given
......@@ -44,10 +43,7 @@ def test_todict_origins(origin):
def test_todict_origin_visits(origin_visit):
obj = origin_visit.to_dict()
assert 'type' not in obj['origin']
origin2 = attr.evolve(origin_visit.origin, type=None)
origin_visit2 = attr.evolve(origin_visit, origin=origin2)
assert origin_visit2 == type(origin_visit).from_dict(obj)
assert origin_visit == type(origin_visit).from_dict(obj)
def test_content_get_hash():
......
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