Ejemplos de update_autoscaler_count en Python

Ejemplo n.º 1

Archivo: test_autoscaler.py Proyecto: somic/paasta

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)

Ejemplo n.º 2

Archivo: test_autoscaler.py Proyecto: rirwin/paasta

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)

Ejemplo n.º 3

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"]

Ejemplo n.º 4

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']

Ejemplo n.º 5

Archivo: test_autoscaler.py Proyecto: zaitsevlab/paasta

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

Ejemplo n.º 6

Archivo: test_autoscaler.py Proyecto: zaitsevlab/paasta

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

Ejemplo n.º 7

Archivo: test_autoscaler.py Proyecto: somic/paasta

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']

Ejemplo n.º 8

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)

Ejemplo n.º 9

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
        )