def test_api_add_pool(self, mock_url, mock_creds, mock_config, mock_post):
"""Test add_pool"""
_api = BatchAppsApi(mock_creds, mock_config)
mock_url.return_value = "https://test_endpoint.com/api/pools"
mock_post.return_value = {}
message = {"targetDedicated": "0", "maxTasksPerTVM": "1", "communication": False, "certificateReferences": []}
val = _api.add_pool()
mock_post.assert_called_with(mock_creds, "https://test_endpoint.com/api/pools", self.headers, message)
self.assertFalse(val.success)
mock_post.return_value = {"poolId": "1", "link": "2"}
val = _api.add_pool()
mock_post.assert_called_with(mock_creds, "https://test_endpoint.com/api/pools", self.headers, message)
self.assertTrue(val.success)
val = _api.add_pool(target_size="test")
self.assertFalse(val.success)
message["targetDedicated"] = "6"
val = _api.add_pool(target_size=6)
mock_post.assert_called_with(mock_creds, "https://test_endpoint.com/api/pools", self.headers, message)
self.assertTrue(val.success)
val = _api.add_pool(certs=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
message["certificateReferences"] = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
message["targetDedicated"] = "0"
mock_post.assert_called_with(mock_creds, "https://test_endpoint.com/api/pools", self.headers, message)
self.assertTrue(val.success)
mock_post.side_effect = RestCallException(None, "Boom~", None)
val = _api.add_pool()
self.assertFalse(val.success)
def test_api_add_pool(self, mock_url, mock_creds, mock_config, mock_post):
"""Test add_pool"""
_api = BatchAppsApi(mock_creds, mock_config)
mock_url.return_value = "https://test_endpoint.com/api/pools"
mock_post.return_value = {}
message = {
'targetDedicated': '0',
'maxTasksPerTVM': '1',
'communication': False,
'certificateReferences': []
}
val = _api.add_pool()
mock_post.assert_called_with(mock_creds,
"https://test_endpoint.com/api/pools",
self.headers, message)
self.assertFalse(val.success)
mock_post.return_value = {'poolId': '1', 'link': '2'}
val = _api.add_pool()
mock_post.assert_called_with(mock_creds,
"https://test_endpoint.com/api/pools",
self.headers, message)
self.assertTrue(val.success)
val = _api.add_pool(target_size="test")
self.assertFalse(val.success)
message['targetDedicated'] = '6'
val = _api.add_pool(target_size=6)
mock_post.assert_called_with(mock_creds,
"https://test_endpoint.com/api/pools",
self.headers, message)
self.assertTrue(val.success)
val = _api.add_pool(certs=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
message['certificateReferences'] = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
message['targetDedicated'] = '0'
mock_post.assert_called_with(mock_creds,
"https://test_endpoint.com/api/pools",
self.headers, message)
self.assertTrue(val.success)
mock_post.side_effect = RestCallException(None, "Boom~", None)
val = _api.add_pool()
self.assertFalse(val.success)
