Changeset View
Changeset View
Standalone View
Standalone View
swh/web/tests/common/test_origin_save.py
| Show First 20 Lines • Show All 318 Lines • ▼ Show 20 Lines | |||||
| @pytest.mark.parametrize("visit_date", [None, "some-date"]) | @pytest.mark.parametrize("visit_date", [None, "some-date"]) | ||||
| def test_from_save_origin_request_to_save_request_info_dict(visit_date): | def test_from_save_origin_request_to_save_request_info_dict(visit_date): | ||||
| """Ensure save request to json serializable dict is fine | """Ensure save request to json serializable dict is fine | ||||
| """ | """ | ||||
| request_date = datetime.now(tz=timezone.utc) | request_date = datetime.now(tz=timezone.utc) | ||||
| _visit_date = request_date + timedelta(minutes=5) if visit_date else None | _visit_date = request_date + timedelta(minutes=5) if visit_date else None | ||||
| request_date = datetime.now(tz=timezone.utc) | request_date = datetime.now(tz=timezone.utc) | ||||
| note = "request succeeded" | |||||
| sor = SaveOriginRequest( | sor = SaveOriginRequest( | ||||
| request_date=request_date, | request_date=request_date, | ||||
| visit_type=_visit_type, | visit_type=_visit_type, | ||||
| visit_status=VISIT_STATUS_FULL, | visit_status=VISIT_STATUS_FULL, | ||||
| origin_url=_origin_url, | origin_url=_origin_url, | ||||
| status=SAVE_REQUEST_ACCEPTED, | status=SAVE_REQUEST_ACCEPTED, | ||||
| loading_task_status=None, | loading_task_status=None, | ||||
| visit_date=_visit_date, | visit_date=_visit_date, | ||||
| loading_task_id=1, | loading_task_id=1, | ||||
| note=note, | |||||
| ) | ) | ||||
| assert sor.to_dict() == SaveOriginRequestInfo( | assert sor.to_dict() == SaveOriginRequestInfo( | ||||
| id=sor.id, | id=sor.id, | ||||
| origin_url=sor.origin_url, | origin_url=sor.origin_url, | ||||
| visit_type=sor.visit_type, | visit_type=sor.visit_type, | ||||
| save_request_date=sor.request_date.isoformat(), | save_request_date=sor.request_date.isoformat(), | ||||
| save_request_status=sor.status, | save_request_status=sor.status, | ||||
| save_task_status=sor.loading_task_status, | save_task_status=sor.loading_task_status, | ||||
| visit_status=sor.visit_status, | visit_status=sor.visit_status, | ||||
| visit_date=_visit_date.isoformat() if _visit_date else None, | visit_date=_visit_date.isoformat() if _visit_date else None, | ||||
| loading_task_id=sor.loading_task_id, | loading_task_id=sor.loading_task_id, | ||||
| note=note, | |||||
| ) | ) | ||||
| def test__check_origin_exists_404(requests_mock): | def test__check_origin_exists_404(requests_mock): | ||||
| url_ko = "https://example.org/some-inexistant-url" | url_ko = "https://example.org/some-inexistant-url" | ||||
| requests_mock.head(url_ko, status_code=404) | requests_mock.head(url_ko, status_code=404) | ||||
| with pytest.raises(BadInputExc, match="not exist"): | with pytest.raises(BadInputExc, match="not exist"): | ||||
| ▲ Show 20 Lines • Show All 407 Lines • Show Last 20 Lines | |||||