def test_get_selected_roles_all(self):
creds = GimmeAWSCreds()
selections = creds._get_selected_roles('all', self.APP_INFO)
self.assertEqual(selections, {'test1', 'test2'})
def test_get_selected_roles_from_config_1(self):
creds = GimmeAWSCreds()
selections = creds._get_selected_roles('test2', self.APP_INFO)
self.assertEqual(selections, {'test2'})
def test_get_selected_roles_multiple_list(self):
creds = GimmeAWSCreds()
selections = creds._get_selected_roles(['test1', 'test2'], self.APP_INFO)
self.assertEqual(selections, {'test1', 'test2'})
def test_missing_role_from_config(self, mock):
creds = GimmeAWSCreds()
selections = creds._get_selected_roles('test3', self.APP_INFO)
self.assertEqual(selections, {'test1'})
