Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/test_storage.py
| Show First 20 Lines • Show All 1,379 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 | |||||