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