def test_update_autoscaler_count(mock_load_marathon_service_config): request = testing.DummyRequest() request.swagger_data = { 'service': 'fake_service', 'instance': 'fake_instance', 'json_body': {'desired_instances': 123}, } mock_load_marathon_service_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=100), get_max_instances=mock.MagicMock(return_value=200) ) with mock.patch('paasta_tools.api.views.autoscaler.set_instances_for_marathon_service', autospec=True) as mock_set_instances: response = autoscaler.update_autoscaler_count(request) assert response.json_body['desired_instances'] == 123 mock_set_instances.assert_called_once_with(service='fake_service', instance='fake_instance', instance_count=123)
def test_update_autoscaler_count(): request = testing.DummyRequest() request.swagger_data = { 'service': 'fake_service', 'instance': 'fake_instance', 'json_body': { 'desired_instances': 123 }, } with mock.patch( 'paasta_tools.api.views.autoscaler.set_instances_for_marathon_service', autospec=True) as mock_set_instances: response = autoscaler.update_autoscaler_count(request) assert response.json_body['desired_instances'] == 123 mock_set_instances.assert_called_once_with(service='fake_service', instance='fake_instance', instance_count=123)
def test_update_autoscaler_count_warning( mock_set_instances_for_marathon_service, mock_load_marathon_service_config ): request = testing.DummyRequest() request.swagger_data = { "service": "fake_service", "instance": "fake_instance", "json_body": {"desired_instances": 123}, } mock_load_marathon_service_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=10), get_max_instances=mock.MagicMock(return_value=100), ) response = autoscaler.update_autoscaler_count(request) assert response.json_body["desired_instances"] == 100 assert "WARNING" in response.json_body["status"]
def test_update_autoscaler_count_warning( mock_set_instances_for_marathon_service, mock_load_marathon_service_config): request = testing.DummyRequest() request.swagger_data = { 'service': 'fake_service', 'instance': 'fake_instance', 'json_body': { 'desired_instances': 123 }, } mock_load_marathon_service_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=10), get_max_instances=mock.MagicMock(return_value=100)) response = autoscaler.update_autoscaler_count(request) assert response.json_body['desired_instances'] == 100 assert 'WARNING' in response.json_body['status']
def test_update_autoscaler_count_kubernetes(mock_get_instance_config): request = testing.DummyRequest() request.swagger_data = { "service": "fake_kubernetes_service", "instance": "fake__kubernetes_instance", "json_body": { "desired_instances": 155 }, } mock_get_instance_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=100), get_max_instances=mock.MagicMock(return_value=200), spec=KubernetesDeploymentConfig, ) response = autoscaler.update_autoscaler_count(request) assert response.json_body["desired_instances"] == 155 assert response.status_code == 202
def test_update_autoscaler_count_marathon(mock_get_instance_config): request = testing.DummyRequest() request.swagger_data = { "service": "fake_marathon_service", "instance": "fake_marathon_instance", "json_body": { "desired_instances": 123 }, } mock_get_instance_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=100), get_max_instances=mock.MagicMock(return_value=200), spec=MarathonServiceConfig, ) response = autoscaler.update_autoscaler_count(request) assert response.json_body["desired_instances"] == 123 assert response.status_code == 202
def test_update_autoscaler_count_warning( mock_set_instances_for_marathon_service, mock_load_marathon_service_config ): request = testing.DummyRequest() request.swagger_data = { 'service': 'fake_service', 'instance': 'fake_instance', 'json_body': {'desired_instances': 123}, } mock_load_marathon_service_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=10), get_max_instances=mock.MagicMock(return_value=100) ) response = autoscaler.update_autoscaler_count(request) assert response.json_body['desired_instances'] == 100 assert 'WARNING' in response.json_body['status']
def test_update_autoscaler_count(mock_load_marathon_service_config): request = testing.DummyRequest() request.swagger_data = { 'service': 'fake_service', 'instance': 'fake_instance', 'json_body': { 'desired_instances': 123 }, } mock_load_marathon_service_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=100), get_max_instances=mock.MagicMock(return_value=200)) with mock.patch( 'paasta_tools.api.views.autoscaler.set_instances_for_marathon_service', autospec=True) as mock_set_instances: response = autoscaler.update_autoscaler_count(request) assert response.json_body['desired_instances'] == 123 mock_set_instances.assert_called_once_with(service='fake_service', instance='fake_instance', instance_count=123)
def test_update_autoscaler_count(mock_load_marathon_service_config): request = testing.DummyRequest() request.swagger_data = { "service": "fake_service", "instance": "fake_instance", "json_body": {"desired_instances": 123}, } mock_load_marathon_service_config.return_value = mock.MagicMock( get_min_instances=mock.MagicMock(return_value=100), get_max_instances=mock.MagicMock(return_value=200), ) with mock.patch( "paasta_tools.api.views.autoscaler.set_instances_for_marathon_service", autospec=True, ) as mock_set_instances: response = autoscaler.update_autoscaler_count(request) assert response.json_body["desired_instances"] == 123 mock_set_instances.assert_called_once_with( service="fake_service", instance="fake_instance", instance_count=123 )