def test_get_username_principalColon(self): evt = { 'detail': { 'userIdentity': { 'type': 'foo', 'principalId': 'bar:baz' } } } self.assertEqual(utils.get_aws_username_from_event(Mock(), evt), 'baz')
def test_get_username_iam(self): evt = { 'detail': { 'userIdentity': { 'type': 'IAMUser', 'userName': '******' } } } self.assertEqual(utils.get_aws_username_from_event(Mock(), evt), 'bar')
def test_get_username_assumed_role_lambda(self): evt = { 'detail': { 'userIdentity': { 'type': 'AssumedRole', 'arn': 'foo/awslambda' } } } self.assertEqual(utils.get_aws_username_from_event(Mock(), evt), None)
def test_get_username_assumed_role_colons(self): evt = { 'detail': { 'userIdentity': { 'type': 'AssumedRole', 'arn': 'foo/bar:baz:blam' } } } self.assertEqual(utils.get_aws_username_from_event(Mock(), evt), 'baz:blam')
def test_get_username_root(self): evt = { 'detail': { 'userIdentity': { 'type': 'Root' } } } self.assertEqual( utils.get_aws_username_from_event(Mock(), evt), None )
def test_get_username_identity_none(self): evt = {'detail': {}} self.assertEqual( utils.get_aws_username_from_event(Mock(), evt), None )
def test_get_username_none(self): self.assertEqual( utils.get_aws_username_from_event(Mock(), None), None )
def test_get(self): username = utils.get_aws_username_from_event( Mock(), self.CLOUDTRAIL_EVENT ) self.assertEqual(username, 'michael_bolton')