Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
Show First 20 Lines • Show All 1,377 Lines • ▼ Show 20 Lines | def test_origin_search_regexp_fullstring(self, swh_storage): | ||||
# check both origins were returned | # check both origins were returned | ||||
assert found_origins0 != found_origins1 | assert found_origins0 != found_origins1 | ||||
def test_origin_visit_add(self, swh_storage): | def test_origin_visit_add(self, swh_storage): | ||||
# given | # given | ||||
swh_storage.origin_add_one(data.origin2) | swh_storage.origin_add_one(data.origin2) | ||||
origin_url = data.origin2['url'] | origin_url = data.origin2['url'] | ||||
date_visit = datetime.datetime.now(datetime.timezone.utc) | |||||
# Round to milliseconds before insertion, so equality doesn't fail | |||||
# after a round-trip through a DB (eg. Cassandra) | |||||
date_visit = date_visit.replace( | |||||
microsecond=round(date_visit.microsecond, -3)) | |||||
# when | # when | ||||
date_visit = datetime.datetime.now(datetime.timezone.utc) | |||||
origin_visit1 = swh_storage.origin_visit_add( | origin_visit1 = swh_storage.origin_visit_add( | ||||
origin_url, | origin_url, | ||||
type=data.type_visit1, | type=data.type_visit1, | ||||
date=date_visit) | date=date_visit) | ||||
actual_origin_visits = list(swh_storage.origin_visit_get( | actual_origin_visits = list(swh_storage.origin_visit_get( | ||||
origin_url)) | origin_url)) | ||||
assert { | assert { | ||||
Show All 25 Lines | |||||
def test_origin_visit_add_default_type(self, swh_storage): | def test_origin_visit_add_default_type(self, swh_storage): | ||||
# given | # given | ||||
swh_storage.origin_add_one(data.origin2) | swh_storage.origin_add_one(data.origin2) | ||||
origin_url = data.origin2['url'] | origin_url = data.origin2['url'] | ||||
# when | # when | ||||
date_visit = datetime.datetime.now(datetime.timezone.utc) | date_visit = datetime.datetime.now(datetime.timezone.utc) | ||||
date_visit2 = date_visit + datetime.timedelta(minutes=1) | date_visit2 = date_visit + datetime.timedelta(minutes=1) | ||||
# Round to milliseconds before insertion, so equality doesn't fail | |||||
# after a round-trip through a DB (eg. Cassandra) | |||||
date_visit = date_visit.replace( | |||||
microsecond=round(date_visit.microsecond, -3)) | |||||
date_visit2 = date_visit2.replace( | |||||
microsecond=round(date_visit2.microsecond, -3)) | |||||
origin_visit1 = swh_storage.origin_visit_add( | origin_visit1 = swh_storage.origin_visit_add( | ||||
origin_url, | origin_url, | ||||
date=date_visit, | date=date_visit, | ||||
type=data.type_visit1, | type=data.type_visit1, | ||||
) | ) | ||||
origin_visit2 = swh_storage.origin_visit_add( | origin_visit2 = swh_storage.origin_visit_add( | ||||
origin_url, | origin_url, | ||||
date=date_visit2, | date=date_visit2, | ||||
▲ Show 20 Lines • Show All 46 Lines • ▼ Show 20 Lines | def test_origin_visit_add_validation(self, swh_storage): | ||||
== psycopg2.errorcodes.UNDEFINED_FUNCTION | == psycopg2.errorcodes.UNDEFINED_FUNCTION | ||||
def test_origin_visit_update(self, swh_storage): | def test_origin_visit_update(self, swh_storage): | ||||
# given | # given | ||||
swh_storage.origin_add_one(data.origin) | swh_storage.origin_add_one(data.origin) | ||||
origin_url = data.origin['url'] | origin_url = data.origin['url'] | ||||
date_visit = datetime.datetime.now(datetime.timezone.utc) | date_visit = datetime.datetime.now(datetime.timezone.utc) | ||||
date_visit2 = date_visit + datetime.timedelta(minutes=1) | |||||
# Round to milliseconds before insertion, so equality doesn't fail | |||||
# after a round-trip through a DB (eg. Cassandra) | |||||
date_visit = date_visit.replace( | |||||
microsecond=round(date_visit.microsecond, -3)) | |||||
date_visit2 = date_visit2.replace( | |||||
microsecond=round(date_visit2.microsecond, -3)) | |||||
origin_visit1 = swh_storage.origin_visit_add( | origin_visit1 = swh_storage.origin_visit_add( | ||||
origin_url, | origin_url, | ||||
date=date_visit, | date=date_visit, | ||||
type=data.type_visit1, | type=data.type_visit1, | ||||
) | ) | ||||
date_visit2 = date_visit + datetime.timedelta(minutes=1) | |||||
origin_visit2 = swh_storage.origin_visit_add( | origin_visit2 = swh_storage.origin_visit_add( | ||||
origin_url, | origin_url, | ||||
date=date_visit2, | date=date_visit2, | ||||
type=data.type_visit2 | type=data.type_visit2 | ||||
) | ) | ||||
swh_storage.origin_add_one(data.origin2) | swh_storage.origin_add_one(data.origin2) | ||||
origin_url2 = data.origin2['url'] | origin_url2 = data.origin2['url'] | ||||
▲ Show 20 Lines • Show All 2,274 Lines • Show Last 20 Lines |