[raz] Skeleton of Python Raz Client tests

cloudera · Apr 21, 2021 · 9a56f9d · 9a56f9d
1 parent dbd9111
commit 9a56f9d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/desktop/core/src/desktop/lib/raz/clients.py b/desktop/core/src/desktop/lib/raz/clients.py
@@ -36,7 +36,7 @@ def __init__(self, **kwargs):
 
     self.ranger = RangerRazS3(RAZ.API_URL.get(), auth)
 
-  def get_url(self, bucket, path, perm='read'):
+  def get_url(self, bucket=None, path=None, perm='read'):
     # No GET/POST spec?
     # e.g. get_url('<storage_account?>', '<bucket>', '<relative_path>', 'read')
     return self.ranger.get_dsas_token(bucket, path, perm)

diff --git a/desktop/core/src/desktop/lib/raz/clients_test.py b/desktop/core/src/desktop/lib/raz/clients_test.py
@@ -34,6 +34,8 @@ def test_check_access_s3_list_buckets(self):
 
     url = S3RazClient().get_url()
 
+    assert_true('AWSAccessKeyId=' in url)
+    assert_true('Signature=' in url)
     assert_true('Expires=' in url)
 
 
@@ -54,4 +56,5 @@ def test_check_acccess_s3_list_file(self):
     assert_true('Signature=' in url)
     assert_true('Expires=' in url)
 
+
   def test_check_acccess_s3_list_file_no_access(self): pass