pr updates

Author: Andre Moeller

src/sagemaker/session.py

@@ -751,7 +751,7 @@ def expand_role(self, role):
     def get_caller_identity_arn(self):
         """Returns the ARN user or role whose credentials are used to call the API.
         Returns:
-            (str): The ARN uer or role
+            (str): The ARN user or role
         """
         assumed_role = self.boto_session.client('sts').get_caller_identity()['Arn']
 
@@ -766,7 +766,8 @@ def get_caller_identity_arn(self):
         try:
             role = self.boto_session.client('iam').get_role(RoleName=role_name)['Role']['Arn']
         except ClientError:
-            LOGGER.warning("Couldn't call 'get_role' to get Role ARN from role name {}.".format(role_name))
+            LOGGER.warning("Couldn't call 'get_role' to get Role ARN from role name {} to get Role path."
+                           .format(role_name))
 
         return role
 

tests/unit/test_session.py

@@ -69,14 +69,17 @@ def test_get_caller_identity_arn_from_an_user(boto_session):
     assert actual == 'arn:aws:iam::369233609183:user/mia'
 
 
-def test_get_caller_identity_arn_from_an_user_without_permissions(boto_session):
+@patch('logging.Logger.warning')
+def test_get_caller_identity_arn_from_an_user_without_permissions(boto_session, mock_logger):
     sess = Session(boto_session)
     arn = 'arn:aws:iam::369233609183:user/mia'
     sess.boto_session.client('sts').get_caller_identity.return_value = {'Arn': arn}
     sess.boto_session.client('iam').get_role.side_effect = ClientError('Bad permissions!', {})
 
+
     actual = sess.get_caller_identity_arn()
     assert actual == 'arn:aws:iam::369233609183:user/mia'
+    mock_logger.assert_called_once()
 
 
 def test_get_caller_identity_arn_from_a_role(boto_session):