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')
