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