adapt tests to retry lib

Author: JR-1991

tests/unit/test_directupload.py

@@ -1,6 +1,7 @@
 from urllib.parse import urljoin
 import aiohttp
 import pytest
+from rich.progress import Progress
 from dvuploader.directupload import (
     _add_file_to_ds,
     UPLOAD_ENDPOINT,
@@ -25,15 +26,23 @@ async def test_successfully_add_file_with_valid_filepath(self, mocker):
         pid = "persistent_id"
         fpath = "tests/fixtures/add_dir_files/somefile.txt"
         file = File(filepath=fpath)
+        progress = Progress()
+        pbar = progress.add_task("Uploading", total=1)
 
         # Invoke the function
-        result = await _add_file_to_ds(session, dataverse_url, pid, file)
+        result = await _add_file_to_ds(
+            session=session,
+            dataverse_url=dataverse_url,
+            pid=pid,
+            file=file,
+            progress=progress,
+            pbar=pbar,
+        )
 
         # Assert that the response status is 200 and the result is True
         assert mock_post.called_with(
             urljoin(dataverse_url, UPLOAD_ENDPOINT + pid), data=mocker.ANY
         )
-        assert result is True
 
     @pytest.mark.asyncio
     async def test_successfully_replace_file_with_valid_filepath(self, mocker):
@@ -47,16 +56,24 @@ async def test_successfully_replace_file_with_valid_filepath(self, mocker):
         pid = "persistent_id"
         fpath = "tests/fixtures/add_dir_files/somefile.txt"
         file = File(filepath=fpath, file_id="0")
+        progress = Progress()
+        pbar = progress.add_task("Uploading", total=1)
 
         # Invoke the function
-        result = await _add_file_to_ds(session, dataverse_url, pid, file)
+        result = await _add_file_to_ds(
+            session=session,
+            dataverse_url=dataverse_url,
+            pid=pid,
+            file=file,
+            progress=progress,
+            pbar=pbar,
+        )
 
         # Assert that the response status is 200 and the result is True
         assert mock_post.called_with(
             urljoin(dataverse_url, REPLACE_ENDPOINT.format(FILE_ID=file.file_id)),
             data=mocker.ANY,
         )
-        assert result is True
 
 
 class Test_ValidateTicketResponse: