From 6963e8f3e9d1357f9fbd4069cccd3d18d2a30326 Mon Sep 17 00:00:00 2001 From: Yuhuai Liu Date: Thu, 3 Sep 2026 14:46:58 -0400 Subject: [PATCH] fix tests --- osf_pigeon/pigeon.py | 5 +++-- tests/test_pigeon.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/osf_pigeon/pigeon.py b/osf_pigeon/pigeon.py index e9734cac40..483d65db64 100644 --- a/osf_pigeon/pigeon.py +++ b/osf_pigeon/pigeon.py @@ -94,7 +94,7 @@ async def get_metadata_for_ia_item(json_metadata): get_relationship_attribute( "creator", f'{settings.OSF_API_URL}v2/registrations/{json_metadata["data"]["id"]}/contributors/' - f"?filter[bibliographic]=true&", + f"?filter[bibliographic]=true", get_contributor_info, ), get_relationship_attribute( @@ -224,7 +224,8 @@ async def get_with_retry(url, retry_on=(), sleep_period=None, headers=None): async def get_pages(url, page, result=None, parse_json=None, semaphore=None): if result is None: result = {} - url = f"{url}?page={page}&page={page}" + separator = "&" if "?" in url else "?" + url = f"{url}{separator}page={page}" data = {} if semaphore is None: data = await get_with_retry(url, retry_on=(429,)) diff --git a/tests/test_pigeon.py b/tests/test_pigeon.py index 51612f6fd2..b3a0b1408b 100644 --- a/tests/test_pigeon.py +++ b/tests/test_pigeon.py @@ -120,7 +120,7 @@ async def test_stream_files_to_dir( body=page1, ) m.get( - f"{settings.OSF_API_URL}v2/registrations/{guid}/wikis/?page=2&page=2", + f"{settings.OSF_API_URL}v2/registrations/{guid}/wikis/?page=2", body=page2, ) with tempfile.TemporaryDirectory() as temp_dir: