Explorar o código

HUE-8925 [fb] Fix AWS IDBroker expiration.

Jean-Francois Desjeans Gauthier %!s(int64=6) %!d(string=hai) anos
pai
achega
ba8d960b77
Modificáronse 2 ficheiros con 3 adicións e 2 borrados
  1. 2 1
      desktop/libs/aws/src/aws/client.py
  2. 1 1
      desktop/libs/aws/src/aws/tests.py

+ 2 - 1
desktop/libs/aws/src/aws/client.py

@@ -180,7 +180,8 @@ class Client(object):
       return cls(
         aws_access_key_id=credentials.get('AccessKeyId'),
         aws_secret_access_key=credentials.get('SecretAccessKey'),
-        aws_security_token=credentials.get('SessionToken')
+        aws_security_token=credentials.get('SessionToken'),
+        expiration=credentials.get('Expiration')
       )
 
   def get_s3_connection(self):

+ 1 - 1
desktop/libs/aws/src/aws/tests.py

@@ -45,7 +45,7 @@ class TestAWS(unittest.TestCase):
 
   def test_with_idbroker(self):
     try:
-      finish = conf.AWS_ACCOUNTS.set_for_testing({'default': {}})
+      finish = conf.AWS_ACCOUNTS.set_for_testing({}) # Set empty to test when no configs are set
       with patch('aws.client.conf_idbroker.get_conf') as get_conf:
         with patch('aws.client.Client.get_s3_connection'):
           with patch('aws.client.IDBroker.get_cab') as get_cab: