Merge "Add asserts for service providers"

This commit is contained in:
Jenkins 2016-01-22 00:11:58 +00:00 committed by Gerrit Code Review
commit 1147107441

View File

@ -1094,6 +1094,27 @@ class RestfulTestCase(unit.SQLDriverOverrides, rest.RestfulTestCase,
return entity return entity
# Service providers (federation)
def assertValidServiceProvider(self, entity, ref=None, *args, **kwargs):
attributes = frozenset(['auth_url', 'id', 'enabled', 'description',
'links', 'relay_state_prefix', 'sp_url'])
for attribute in attributes:
self.assertIsNotNone(entity.get(attribute))
def assertValidServiceProviderListResponse(self, resp, *args, **kwargs):
if kwargs.get('keys_to_check') is None:
kwargs['keys_to_check'] = ['auth_url', 'id', 'enabled',
'description', 'relay_state_prefix',
'sp_url']
return self.assertValidListResponse(
resp,
'service_providers',
self.assertValidServiceProvider,
*args,
**kwargs)
def build_external_auth_request(self, remote_user, def build_external_auth_request(self, remote_user,
remote_domain=None, auth_data=None, remote_domain=None, auth_data=None,
kerberos=False): kerberos=False):