1- # pylint:disable=unused-variable
2- # pylint:disable=unused-argument
3- # pylint:disable=redefined-outer-name
1+ # pylint: disable=redefined-outer-name
2+ # pylint: disable=unused-argument
3+ # pylint: disable=unused-variable
44
5- from typing import Dict
5+ from typing import Iterator
6+ from unittest .mock import Mock
67
78import httpx
89import pytest
1415
1516
1617@pytest .fixture
17- async def pennsieve_files_mock (pennsieve_subsystem_mock , pennsieve_file_id : str ):
18+ async def pennsieve_files_mock (
19+ pennsieve_subsystem_mock : Mock , pennsieve_file_id : str
20+ ) -> Iterator [Mock ]:
1821 mock = pennsieve_subsystem_mock
1922 if mock :
2023 FAKE_FILE_ID = "123434"
@@ -39,9 +42,9 @@ async def pennsieve_files_mock(pennsieve_subsystem_mock, pennsieve_file_id: str)
3942
4043async def test_download_file_entrypoint (
4144 async_client : httpx .AsyncClient ,
42- pennsieve_subsystem_mock ,
43- pennsieve_files_mock ,
44- pennsieve_api_headers : Dict [str , str ],
45+ pennsieve_subsystem_mock : Mock ,
46+ pennsieve_files_mock : Mock ,
47+ pennsieve_api_headers : dict [str , str ],
4548 pennsieve_file_id : str ,
4649):
4750 file_id = pennsieve_file_id
@@ -57,9 +60,9 @@ async def test_download_file_entrypoint(
5760
5861async def test_delete_file_entrypoint (
5962 async_client : httpx .AsyncClient ,
60- pennsieve_subsystem_mock ,
61- pennsieve_files_mock ,
62- pennsieve_api_headers : Dict [str , str ],
63+ pennsieve_subsystem_mock : Mock ,
64+ pennsieve_files_mock : Mock ,
65+ pennsieve_api_headers : dict [str , str ],
6366 pennsieve_file_id : str ,
6467):
6568 file_id = pennsieve_file_id
@@ -68,5 +71,4 @@ async def test_delete_file_entrypoint(
6871 headers = pennsieve_api_headers ,
6972 )
7073 assert response .status_code == status .HTTP_204_NO_CONTENT
71- data = response .json ()
72- assert not data
74+ assert response .num_bytes_downloaded == 0
0 commit comments