def test_with_multiple_accounts_and_multiple_roles(self): client = Mock() client.get_account_and_role_list.return_value = \ {'ACCOUNT1': ['ROLE1', 'ROLE2'], 'ACCOUNT2': ['ROLE3', 'ROLE4']} self.assertEqual('ROLE1', get_first_role(client, 'ACCOUNT1'))
def test_with_single_account_and_single_role(self): client = Mock() client.get_account_and_role_list.return_value = \ {'ACCOUNT1': ['ROLE1']} self.assertEqual('ROLE1', get_first_role(client, 'ACCOUNT1'))