def test_logout(self):
self.request_mock.get.return_value = Mock(ok=True, content=encode_response({}))
self.subject.logout('testToken')
self.request_mock.get.assert_called_once_with(params={'sessionToken': 'testToken', 'output': 'json'},
url='https://openapi.cdnetworks.com/api/rest/logout', verify=True)
def test_get_deploy_status_returns_pad_status(self):
get_mock = self.request_mock.get
get_mock.return_value = Mock(ok=True,
content=encode_response(STATUS_RESPONSE))
get_mock.return_value.raise_for_status.side_effect = RuntimeError(
'error')
padStatus = self.subject.get_deploy_status('testPadName')
self.assertEqual(padStatus, STATUS_RESPONSE)
def test_login_api_failure_raises_exception(self):
fake_response = encode_response(self.RESPONSE_LOGIN_FAILED)
self.request_mock.post.return_value = Mock(ok=False, content=fake_response)
self.request_mock.post.return_value.raise_for_status.side_effect = RuntimeError('TestError')
try:
self.subject.login()
self.fail('should throw')
except Exception as ve:
self.assertIsInstance(ve, RuntimeError)
self.assertEqual(str(ve), 'TestError')
def test_add_alias_to_pad_adds_alias_to_empty_pad(self):
self.request_mock.post.return_value = Mock(
ok=True, content=encode_response(EMPTY_PAD))
result = self.subject.add_alias_to_pad(EMPTY_PAD, 'testPadName',
'testDomain')
self.assertEqual(
'testDomain',
result['PadConfigResponse']['data']['data']['pad_aliases'])
def test_add_alias_to_pad_returns_modified_pad(self):
self.request_mock.post.return_value = Mock(
ok=True, content=encode_response(GET_PAD_RESPONSE))
result = self.subject.add_alias_to_pad(GET_PAD_RESPONSE, 'testPadName',
'testDomain')
aliases = result['PadConfigResponse']['data']['data'][
'pad_aliases'].split('\n')
self.assertIn('testDomain', aliases)
self.assertIn('alias1.url.com', aliases)
def test_clone_pad_raises_error_on_api_error(self):
post_mock = self.request_mock.post
post_mock.return_value = Mock(ok=False, content=encode_response({}))
post_mock.return_value.raise_for_status.side_effect = RuntimeError(
'error')
try:
self.subject.clone_pad('testContract', 'testSrcPad', 'testNewPad',
'testOrigin', 'testDescription')
except Exception as ex:
self.assertIsInstance(ex, RuntimeError)
self.assertEqual(str(ex), 'error')
def test_add_alias_to_pad_raises_error_on_bad_request(self):
post_mock = self.request_mock.post
post_mock.return_value = Mock(
ok=True, content=encode_response(ADD_PAD_ALIAS_FAILED_RESPONSE))
post_mock.return_value.raise_for_status.side_effect = RuntimeError(
'error')
try:
self.subject.add_alias_to_pad(GET_PAD_RESPONSE, 'testPadName',
'testDomain')
except Exception as ex:
self.assertIsInstance(ex, ValueError)
self.assertEqual(str(ex), 'error adding alias.')
def test_push_pad_configuration_pushes_to_staging_when_prod_is_false(self):
post_mock = self.request_mock.post
post_mock.return_value = Mock(ok=True,
content=encode_response(PUSH_RESPONSE))
self.subject.push_pad_configuration('testPadName', False)
post_mock.assert_called_once_with(
data={
'sessionToken': 'testSessionToken',
'apiKey': 'testApiKey',
'pad': 'testPadName',
'output': 'json'
},
url='https://openapi.cdnetworks.com/api/rest/pan/site/pushStaging')
def test_add_alias_to_pad_calls_proper_endpoint(self):
self.request_mock.post.return_value = Mock(
ok=True, content=encode_response(GET_PAD_RESPONSE))
self.subject.add_alias_to_pad(GET_PAD_RESPONSE, 'testPadName',
'testDomain')
self.request_mock.post.assert_called_once_with(
data={
'output': 'json',
'sessionToken': 'testSessionToken',
'apiKey': 'testApiKey',
'pad_aliases': 'alias1.url.com\ntestDomain',
'pad': 'testPadName'
},
url='https://openapi.cdnetworks.com/api/rest/pan/site/v2/edit')
def test_add_sam_to_pad_adds_sam(self):
post_mock = self.request_mock.post
post_mock.return_value = Mock(
ok=True, content=encode_response(GET_PAD_RESPONSE))
self.subject.add_sam_to_pad('testPadName', {'one': 'two'})
post_mock.assert_called_once_with(
data={
'sessionToken': 'testSessionToken',
'apiKey': 'testApiKey',
'sam_json': {
'one': 'two'
},
'output': 'json'
},
url=
'https://openapi.cdnetworks.com/api/rest/pan/sam/testPadName/add')
def test_clone_pad_calls_proper_endpoint(self):
expected_response = {}
self.request_mock.post.return_value = Mock(
ok=True, content=encode_response(expected_response))
result = self.subject.clone_pad('testContractNo', 'testSrcPad',
'testNewPad', 'testOrigin',
'testDescription')
assert isinstance(self.request_mock.post, Mock)
self.request_mock.post.assert_called_once_with(
data={
'output': 'json',
'sessionToken': 'testSessionToken',
'copy_settings_from': 'testSrcPad',
'apiKey': 'testApiKey',
'pad': 'testNewPad',
'origin': 'testOrigin',
'product': 'testContractNo',
'description': 'testDescription'
},
url='https://openapi.cdnetworks.com/api/rest/pan/site/v2/add')
self.assertEqual(result, expected_response)
def test_add_sam_raises_exception_on_api_error(self):
post_mock = self.request_mock.post
post_mock.return_value = Mock(
ok=False, content=encode_response(GET_PAD_RESPONSE))
post_mock.return_value.raise_for_status.side_effect = RuntimeError(
'error')
try:
self.subject.add_sam_to_pad('testPadName', {'one': 'two'})
except Exception as ex:
self.assertIsInstance(ex, RuntimeError)
self.assertEqual(str(ex), 'error')
post_mock.assert_called_once_with(
data={
'sessionToken': 'testSessionToken',
'apiKey': 'testApiKey',
'sam_json': {
'one': 'two'
},
'output': 'json'
},
url=
'https://openapi.cdnetworks.com/api/rest/pan/sam/testPadName/add')
def __setup_login(self, response_content):
fake_response = encode_response(response_content)
self.request_mock.post.return_value = Mock(ok=True, content=fake_response)
actual = self.subject.login()
return actual