def test_get_source_profile_no_target_profile_name():
    profiles = {
        'myuser': {
            'aws_access_key_id': 'AKIA...',
            'aws_secret_access_key': 'SECRET',
        },
        'myrole': {
            'role_arn': 'arn:aws:iam:XXX:role/role_name',
            'source_profile': 'myuser',
        },
    }
    assert profile.get_source_profile(profiles, 'admin') == None
def test_get_source_profile_return_default():
    profiles = {
        'default': {
            'aws_access_key_id': 'AKIA...',
            'aws_secret_access_key': 'SECRET',
        },
        'myuser': {
            'aws_access_key_id': 'AKIA...',
            'aws_secret_access_key': 'SECRET',
        },
        'myrole': {
            'role_arn': 'arn:aws:iam:XXX:role/role_name',
        },
    }
    assert profile.get_source_profile(profiles,
                                      'myrole') == profiles['default']