Changeset View
Changeset View
Standalone View
Standalone View
swh/web/tests/conftest.py
| Show First 20 Lines • Show All 222 Lines • ▼ Show 20 Lines | def snapshot_get_latest(self, origin_url): | ||||
| snp = snapshot_get_latest(self.storage, origin_url) | snp = snapshot_get_latest(self.storage, origin_url) | ||||
| return converters.from_snapshot(snp.to_dict()) | return converters.from_snapshot(snp.to_dict()) | ||||
| def origin_get(self, origin_urls): | def origin_get(self, origin_urls): | ||||
| origins = self.storage.origin_get(origin_urls) | origins = self.storage.origin_get(origin_urls) | ||||
| return [converters.from_origin(o.to_dict()) for o in origins] | return [converters.from_origin(o.to_dict()) for o in origins] | ||||
| def origin_visit_get(self, origin_url): | def origin_visit_get(self, origin_url): | ||||
| visits = list(self.storage.origin_visit_get(origin_url)) | next_page_token = None | ||||
| for i in range(len(visits)): | visits = [] | ||||
| while True: | |||||
| visit_page = self.storage.origin_visit_get( | |||||
| origin_url, page_token=next_page_token | |||||
| ) | |||||
| next_page_token = visit_page.next_page_token | |||||
| for visit in visit_page.results: | |||||
| visit_status = self.storage.origin_visit_status_get_latest( | visit_status = self.storage.origin_visit_status_get_latest( | ||||
| origin_url, visits[i]["visit"] | origin_url, visit.visit | ||||
| ) | |||||
| visits.append( | |||||
| converters.from_origin_visit( | |||||
| {**visit_status.to_dict(), "type": visit.type} | |||||
| ) | ) | ||||
| visits[i] = converters.from_origin_visit( | |||||
| {**visits[i], **visit_status.to_dict()} | |||||
| ) | ) | ||||
| if not next_page_token: | |||||
| break | |||||
| return visits | return visits | ||||
| def origin_visit_get_by(self, origin_url: str, visit_id: int) -> OriginVisitInfo: | def origin_visit_get_by(self, origin_url: str, visit_id: int) -> OriginVisitInfo: | ||||
| visit = self.storage.origin_visit_get_by(origin_url, visit_id) | visit = self.storage.origin_visit_get_by(origin_url, visit_id) | ||||
| assert visit is not None | assert visit is not None | ||||
| visit_status = self.storage.origin_visit_status_get_latest(origin_url, visit_id) | visit_status = self.storage.origin_visit_status_get_latest(origin_url, visit_id) | ||||
| assert visit_status is not None | assert visit_status is not None | ||||
| return converters.from_origin_visit( | return converters.from_origin_visit( | ||||
| ▲ Show 20 Lines • Show All 100 Lines • Show Last 20 Lines | |||||