def test_return_scale_v2(self):
bundles_endpoint_reply = """
[{
"bundleId": "a101449418187d92c789d1adc240b6d6",
"bundleExecutions": [
{
"host": "127.0.0.1",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": true
},
{
"host": "127.0.0.2",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": false
}
]
}]
"""
http_method = self.respond_with(text=bundles_endpoint_reply)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'dcos_mode': False,
'scheme': 'http',
'ip': '127.0.0.1',
'port': '9005',
'base_path': '/',
'api_version': '2',
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
}
input_args = MagicMock(**args)
with patch('requests.get', http_method):
result = bundle_scale.get_scale(bundle_id, True, input_args)
self.assertEqual(1, result)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles',
auth=self.conductr_auth,
verify=self.server_verification_file,
headers={'Host': '127.0.0.1'})
def test_return_zero_v1(self):
bundles_endpoint_reply = '[]'
http_method = self.respond_with(text=bundles_endpoint_reply)
with patch('requests.get', http_method):
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'ip': '127.0.0.1',
'port': '9005',
'api_version': '1'
}
result = bundle_scale.get_scale(bundle_id, MagicMock(**args))
self.assertEqual(0, result)
http_method.assert_called_with('http://127.0.0.1:9005/bundles')
def test_return_scale_v2(self):
bundles_endpoint_reply = """
[{
"bundleId": "a101449418187d92c789d1adc240b6d6",
"bundleExecutions": [
{
"host": "127.0.0.1",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": true
},
{
"host": "127.0.0.2",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": false
}
]
}]
"""
http_method = self.respond_with(text=bundles_endpoint_reply)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'dcos_mode': False,
'scheme': 'http',
'ip': '127.0.0.1',
'port': '9005',
'base_path': '/',
'api_version': '2',
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
}
input_args = MagicMock(**args)
with patch('requests.get', http_method):
result = bundle_scale.get_scale(bundle_id, input_args)
self.assertEqual(1, result)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles', auth=self.conductr_auth,
verify=self.server_verification_file, headers={'Host': '127.0.0.1'})
def test_return_scale_v2(self):
request_headers_mock = MagicMock(return_value=self.mock_headers)
bundles_endpoint_reply = """
[{
"bundleId": "a101449418187d92c789d1adc240b6d6",
"bundleExecutions": [
{
"host": "127.0.0.1",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": true
},
{
"host": "127.0.0.2",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": false
}
]
}]
"""
http_method = self.respond_with(text=bundles_endpoint_reply)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'ip': '127.0.0.1',
'port': '9005',
'api_version': '2'
}
input_args = MagicMock(**args)
with patch('requests.get', http_method), \
patch('conductr_cli.conduct_url.request_headers', request_headers_mock):
result = bundle_scale.get_scale(bundle_id, input_args)
self.assertEqual(1, result)
request_headers_mock.assert_called_with(input_args)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles', headers=self.mock_headers)
def test_return_zero_v2(self):
request_headers_mock = MagicMock(return_value=self.mock_headers)
bundles_endpoint_reply = '[]'
http_method = self.respond_with(text=bundles_endpoint_reply)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'ip': '127.0.0.1',
'port': '9005',
'api_version': '2'
}
input_args = MagicMock(**args)
with patch('requests.get', http_method), \
patch('conductr_cli.conduct_url.request_headers', request_headers_mock):
result = bundle_scale.get_scale(bundle_id, input_args)
self.assertEqual(0, result)
request_headers_mock.assert_called_with(input_args)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles', headers=self.mock_headers)
def test_return_scale_v2(self):
bundles_endpoint_reply = """
[{
"bundleId": "a101449418187d92c789d1adc240b6d6",
"bundleExecutions": [
{
"host": "127.0.0.1",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": true
},
{
"host": "127.0.0.2",
"endpoints": {
"web": {
"bindPort": 10822,
"hostPort": 10822
}
},
"isStarted": false
}
]
}]
"""
http_method = self.respond_with(text=bundles_endpoint_reply)
with patch('requests.get', http_method):
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'ip': '127.0.0.1',
'port': '9005',
'api_version': '2'
}
result = bundle_scale.get_scale(bundle_id, MagicMock(**args))
self.assertEqual(1, result)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles')
def test_return_zero_v2(self):
bundles_endpoint_reply = '[]'
http_method = self.respond_with(text=bundles_endpoint_reply)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'dcos_mode': False,
'scheme': 'http',
'host': '127.0.0.1',
'port': '9005',
'base_path': '/',
'api_version': '2',
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
}
input_args = MagicMock(**args)
with patch('requests.get', http_method):
result = bundle_scale.get_scale(bundle_id, input_args)
self.assertEqual(0, result)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles', auth=self.conductr_auth,
verify=self.server_verification_file, headers={'Host': '127.0.0.1'})
def test_return_zero_v2(self):
bundles_endpoint_reply = '[]'
http_method = self.respond_with(text=bundles_endpoint_reply)
bundle_id = 'a101449418187d92c789d1adc240b6d6'
args = {
'dcos_mode': False,
'scheme': 'http',
'host': '127.0.0.1',
'port': '9005',
'base_path': '/',
'api_version': '2',
'conductr_auth': self.conductr_auth,
'server_verification_file': self.server_verification_file
}
input_args = MagicMock(**args)
with patch('requests.get', http_method):
result = bundle_scale.get_scale(bundle_id, True, input_args)
self.assertEqual(0, result)
http_method.assert_called_with('http://127.0.0.1:9005/v2/bundles',
auth=self.conductr_auth,
verify=self.server_verification_file,
headers={'Host': '127.0.0.1'})
