|
6 | 6 | import unittest |
7 | 7 |
|
8 | 8 | from fair_research_login.client import NativeClient |
9 | | -from globus_sdk import DeleteData, TransferClient |
10 | | -from globus_sdk.exc import TransferAPIError |
| 9 | +from globus_sdk import DeleteData, TransferAPIError, TransferClient |
11 | 10 |
|
12 | 11 | from tests.base import TOP_LEVEL, ZSTASH_PATH, TestZstash, print_starred, run_cmd |
13 | 12 |
|
@@ -63,7 +62,7 @@ def preactivate_globus(self): |
63 | 62 | transfer_authorizer = native_client.get_authorizers().get( |
64 | 63 | "transfer.api.globus.org" |
65 | 64 | ) |
66 | | - self.transfer_client = TransferClient(transfer_authorizer) |
| 65 | + self.transfer_client = TransferClient(authorizer=transfer_authorizer) |
67 | 66 |
|
68 | 67 | for ep_id in [hpss_globus_endpoint, local_endpoint]: |
69 | 68 | r = self.transfer_client.endpoint_autoactivate(ep_id, if_expires_in=600) |
@@ -96,7 +95,9 @@ def delete_files_globus(self): |
96 | 95 | with 5 second timeout limit. If the task is ACTIVE after time runs |
97 | 96 | out 'task_wait' returns False, and True otherwise. |
98 | 97 | """ |
99 | | - while not self.transfer_client.task_wait(task_id, 5, 5): |
| 98 | + while not self.transfer_client.task_wait( |
| 99 | + task_id, timeout=5, polling_interval=5 |
| 100 | + ): |
100 | 101 | task = self.transfer_client.get_task(task_id) |
101 | 102 | if task.get("is_paused"): |
102 | 103 | break |
|
0 commit comments