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'})