Changeset View
Changeset View
Standalone View
Standalone View
swh/storage/tests/storage_tests.py
| Show First 20 Lines • Show All 1,742 Lines • ▼ Show 20 Lines | def test_origin_search_regexp_fullstring(self, swh_storage, sample_data): | ||||
| # offset=1 | # offset=1 | ||||
| actual_page = swh_storage.origin_search( | actual_page = swh_storage.origin_search( | ||||
| ".*/.*", page_token=next_page_token, limit=1, regexp=True | ".*/.*", page_token=next_page_token, limit=1, regexp=True | ||||
| ) | ) | ||||
| assert actual_page.next_page_token is None | assert actual_page.next_page_token is None | ||||
| assert actual_page.results == [origin2] | assert actual_page.results == [origin2] | ||||
| def test_origin_search_no_visit_types(self, swh_storage, sample_data): | |||||
| origin = sample_data.origins[0] | |||||
| swh_storage.origin_add([origin]) | |||||
| actual_page = swh_storage.origin_search(origin.url, visit_types=["git"]) | |||||
| assert actual_page.next_page_token is None | |||||
| assert actual_page.results == [] | |||||
| def test_origin_search_with_visit_types(self, swh_storage, sample_data): | |||||
| origin, origin2 = sample_data.origins[:2] | |||||
| swh_storage.origin_add([origin, origin2]) | |||||
| swh_storage.origin_visit_add( | |||||
| [ | |||||
| OriginVisit(origin=origin.url, date=now(), type="git"), | |||||
| OriginVisit(origin=origin2.url, date=now(), type="svn"), | |||||
| ] | |||||
| ) | |||||
| actual_page = swh_storage.origin_search(origin.url, visit_types=["git"]) | |||||
| assert actual_page.next_page_token is None | |||||
| assert actual_page.results == [origin] | |||||
| actual_page = swh_storage.origin_search(origin2.url, visit_types=["svn"]) | |||||
| assert actual_page.next_page_token is None | |||||
| assert actual_page.results == [origin2] | |||||
| def test_origin_search_multiple_visit_types(self, swh_storage, sample_data): | |||||
| origin = sample_data.origins[0] | |||||
| swh_storage.origin_add([origin]) | |||||
| def _add_visit_type(visit_type): | |||||
| swh_storage.origin_visit_add( | |||||
| [OriginVisit(origin=origin.url, date=now(), type=visit_type)] | |||||
| ) | |||||
| def _check_visit_types(visit_types): | |||||
| actual_page = swh_storage.origin_search(origin.url, visit_types=visit_types) | |||||
| assert actual_page.next_page_token is None | |||||
| assert actual_page.results == [origin] | |||||
| _add_visit_type("git") | |||||
| _check_visit_types(["git"]) | |||||
| _check_visit_types(["git", "hg"]) | |||||
| _add_visit_type("hg") | |||||
| _check_visit_types(["hg"]) | |||||
| _check_visit_types(["git", "hg"]) | |||||
| def test_origin_visit_add(self, swh_storage, sample_data): | def test_origin_visit_add(self, swh_storage, sample_data): | ||||
| origin1 = sample_data.origins[1] | origin1 = sample_data.origins[1] | ||||
| swh_storage.origin_add([origin1]) | swh_storage.origin_add([origin1]) | ||||
| date_visit = now() | date_visit = now() | ||||
| date_visit2 = date_visit + datetime.timedelta(minutes=1) | date_visit2 = date_visit + datetime.timedelta(minutes=1) | ||||
| date_visit = round_to_milliseconds(date_visit) | date_visit = round_to_milliseconds(date_visit) | ||||
| ▲ Show 20 Lines • Show All 2,200 Lines • Show Last 20 Lines | |||||