def test_good_init(self, mock): '''Test for a successful init''' token = '1234-1234-1234-1234' mock.register_uri('POST', '{0}/{1}/oauth2/token'.format( constants.OAUTH2_ENDPOINT, self.tenant_id), json={'access_token': token}, status_code=httplib.OK) connection.AzureConnection(logger=self.log)
def __init__(self, name, endpoint, api_version=None, logger=None, _ctx=ctx): # Set the active context self.ctx = _ctx # Configure logger self.log = utils.create_child_logger( 'resources.{0}'.format(name.replace(' ', '')), plogger=logger) # Set up labeling self.name = name # Build the partial endpoint self.endpoint = endpoint # Get a connection self.client = connection.AzureConnection( api_version=api_version, logger=self.log, _ctx=self.ctx)
def test_good_request(self, mock): '''Test for a successful request''' token = '1234-1234-1234-1234' mock.register_uri('POST', '{0}/{1}/oauth2/token'.format( constants.OAUTH2_ENDPOINT, self.tenant_id), json={'access_token': token}, status_code=httplib.OK) mock.register_uri('GET', '{0}/subscriptions/{1}{2}'.format( constants.CONN_API_ENDPOINT, self.subscription_id, '/some/resource'), json={'data': 'stuff'}, status_code=httplib.OK) conn = connection.AzureConnection(logger=self.log) res = conn.request(url='/some/resource', method='get') self.assertEqual(res.status_code, httplib.OK) self.assertEqual(res.json().get('data'), 'stuff')
def test_bad_credentials_request(self, mock): '''Test for a request using bad credentials''' token = '1234-1234-1234-1234' mock.register_uri('POST', '{0}/{1}/oauth2/token'.format( constants.OAUTH2_ENDPOINT, self.tenant_id), json={'access_token': token}, status_code=httplib.OK) mock.register_uri('GET', '{0}/subscriptions/{1}{2}'.format( constants.CONN_API_ENDPOINT, self.subscription_id, '/some/resource'), json={'error': 'invalid_client'}, status_code=httplib.UNAUTHORIZED) conn = connection.AzureConnection(logger=self.log) self.assertRaises(exceptions.InvalidCredentials, conn.request, url='/some/resource', method='get')
def test_unauthorized_request(self, mock): '''Test for an unauthorized request''' token = '1234-1234-1234-1234' mock.register_uri('POST', '{0}/{1}/oauth2/token'.format( constants.OAUTH2_ENDPOINT, self.tenant_id), json={'access_token': token}, status_code=httplib.OK) mock.register_uri('GET', '{0}/subscriptions/{1}{2}'.format( constants.CONN_API_ENDPOINT, self.subscription_id, '/some/resource'), json={'error': 'unknown_code'}, status_code=httplib.UNAUTHORIZED) conn = connection.AzureConnection(logger=self.log) self.assertRaises(exceptions.UnauthorizedRequest, conn.request, url='/some/resource', method='get')