Change to set_bearer_token

OCHA-DAP · Mar 13, 2024 · aac66ed · aac66ed
1 parent c5e0360
commit aac66ed
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/src/hdx/utilities/downloader.py b/src/hdx/utilities/downloader.py
@@ -306,8 +306,8 @@ def normal_setup(
             ) from e
         return self.response
 
-    def update_bearer_token(self, bearer_token: str) -> None:
-        """Update bearer token
+    def set_bearer_token(self, bearer_token: str) -> None:
+        """Set bearer token
 
         Args:
             bearer_token (str): Bearer token

diff --git a/src/hdx/utilities/retriever.py b/src/hdx/utilities/retriever.py
@@ -166,16 +166,16 @@ def get_filename(
         filename = slugify(f"{filename}{extension}")
         return f"{prefix}{filename}{first_ext}", kwargs
 
-    def update_bearer_token(self, bearer_token: str) -> None:
-        """Update bearer token in downloader
+    def set_bearer_token(self, bearer_token: str) -> None:
+        """Set bearer token in downloader
 
         Args:
             bearer_token (str): Bearer token
 
         Returns:
             None
         """
-        self.downloader.update_bearer_token(bearer_token)
+        self.downloader.set_bearer_token(bearer_token)
 
     def download_file(
         self,

diff --git a/tests/hdx/utilities/test_downloader.py b/tests/hdx/utilities/test_downloader.py
@@ -266,7 +266,7 @@ def test_init(self, monkeypatch, downloaderfolder):
                 == f"Bearer {bearertoken}"
             )
             bearertoken = "FGHIJ"
-            downloader.update_bearer_token(bearertoken)
+            downloader.set_bearer_token(bearertoken)
             assert (
                 downloader.session.headers["Authorization"]
                 == f"Bearer {bearertoken}"

diff --git a/tests/hdx/utilities/test_retriever.py b/tests/hdx/utilities/test_retriever.py
@@ -156,7 +156,7 @@ def test_error(self, dirs, retrieverfolder, fallback_dir):
                     use_saved=True,
                 )
 
-    def test_update_bearer_token(self, dirs, retrieverfolder, fallback_dir):
+    def test_set_bearer_token(self, dirs, retrieverfolder, fallback_dir):
         saved_dir, temp_dir = dirs
         bearertoken = "12345"
         with Download(bearer_token=bearertoken) as downloader:
@@ -173,7 +173,7 @@ def test_update_bearer_token(self, dirs, retrieverfolder, fallback_dir):
                 use_saved=False,
             ) as retriever:
                 bearertoken = "67890"
-                retriever.update_bearer_token(bearertoken)
+                retriever.set_bearer_token(bearertoken)
                 assert (
                     downloader.session.headers["Authorization"]
                     == f"Bearer {bearertoken}"