Revert "Include role path in get_execution_role() result (#268)"

This reverts commit 7c2073c.

Author: nadiaya

CHANGELOG.rst

@@ -2,11 +2,6 @@
 CHANGELOG
 =========
 
-1.5.3dev
-========
-
-* bug-fix: Session: include role path in ``get_execution_role()`` result
-
 1.5.2
 =====
 

src/sagemaker/session.py

@@ -697,11 +697,6 @@ def get_caller_identity_arn(self):
             return role
 
         role = re.sub(r'^(.+)sts::(\d+):assumed-role/(.+?)/.*$', r'\1iam::\2:role/\3', assumed_role)
-
-        # Call IAM to get the role's path
-        role_name = role[role.rfind('/') + 1:]
-        role = self.boto_session.client('iam').get_role(RoleName=role_name)['Role']['Arn']
-
         return role
 
     def logs_for_job(self, job_name, wait=False, poll=10):  # noqa: C901 - suppress complexity warning for this method

tests/unit/test_session.py

@@ -41,7 +41,7 @@ def test_get_execution_role():
     assert actual == 'arn:aws:iam::369233609183:role/SageMakerRole'
 
 
-def test_get_execution_role_works_with_service_role():
+def test_get_execution_role_works_with_servie_role():
     session = Mock()
     session.get_caller_identity_arn.return_value = \
         'arn:aws:iam::369233609183:role/service-role/AmazonSageMaker-ExecutionRole-20171129T072388'
@@ -61,9 +61,7 @@ def test_get_execution_role_throws_exception_if_arn_is_not_role():
 
 def test_get_caller_identity_arn_from_an_user(boto_session):
     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.return_value = {'Role': {'Arn': arn}}
+    sess.boto_session.client('sts').get_caller_identity.return_value = {'Arn': 'arn:aws:iam::369233609183:user/mia'}
 
     actual = sess.get_caller_identity_arn()
     assert actual == 'arn:aws:iam::369233609183:user/mia'
@@ -74,37 +72,19 @@ def test_get_caller_identity_arn_from_a_role(boto_session):
     arn = 'arn:aws:sts::369233609183:assumed-role/SageMakerRole/6d009ef3-5306-49d5-8efc-78db644d8122'
     sess.boto_session.client('sts').get_caller_identity.return_value = {'Arn': arn}
 
-    expected_role = 'arn:aws:iam::369233609183:role/SageMakerRole'
-    sess.boto_session.client('iam').get_role.return_value = {'Role': {'Arn': expected_role}}
-
     actual = sess.get_caller_identity_arn()
-    assert actual == expected_role
+    assert actual == 'arn:aws:iam::369233609183:role/SageMakerRole'
 
 
 def test_get_caller_identity_arn_from_a_execution_role(boto_session):
     sess = Session(boto_session)
     arn = 'arn:aws:sts::369233609183:assumed-role/AmazonSageMaker-ExecutionRole-20171129T072388/SageMaker'
     sess.boto_session.client('sts').get_caller_identity.return_value = {'Arn': arn}
-    sess.boto_session.client('iam').get_role.return_value = {'Role': {'Arn': arn}}
 
     actual = sess.get_caller_identity_arn()
     assert actual == 'arn:aws:iam::369233609183:role/service-role/AmazonSageMaker-ExecutionRole-20171129T072388'
 
 
-def test_get_caller_identity_arn_from_role_with_path(boto_session):
-    sess = Session(boto_session)
-    arn_prefix = 'arn:aws:iam::369233609183:role'
-    role_name = 'name'
-    sess.boto_session.client('sts').get_caller_identity.return_value = {'Arn': '/'.join([arn_prefix, role_name])}
-
-    role_path = 'path'
-    role_with_path = '/'.join([arn_prefix, role_path, role_name])
-    sess.boto_session.client('iam').get_role.return_value = {'Role': {'Arn': role_with_path}}
-
-    actual = sess.get_caller_identity_arn()
-    assert actual == role_with_path
-
-
 def test_delete_endpoint(boto_session):
     sess = Session(boto_session)
     sess.delete_endpoint('my_endpoint')
@@ -115,15 +95,15 @@ def test_delete_endpoint(boto_session):
 def test_s3_input_all_defaults():
     prefix = 'pre'
     actual = s3_input(s3_data=prefix)
-    expected = {
-        'DataSource': {
+    expected = \
+        {'DataSource': {
             'S3DataSource': {
                 'S3DataDistributionType': 'FullyReplicated',
                 'S3DataType': 'S3Prefix',
                 'S3Uri': prefix
             }
         }
-    }
+        }
     assert actual.config == expected