示例#1
0
def test_ensure_pod_disruption_budget_noop_when_min_available_is_set():
    with mock.patch(
            "paasta_tools.kubernetes.application.controller_wrappers.pod_disruption_budget_for_service_instance",
            autospec=True,
    ) as mock_pdr_for_service_instance:
        mock_req_pdr = mock.Mock()
        mock_req_pdr.spec.max_unavailable = 10
        mock_pdr_for_service_instance.return_value = mock_req_pdr

        mock_client = mock.MagicMock()

        mock_pdr = mock.Mock()
        mock_pdr.spec.max_unavailable = 5
        mock_pdr.spec.min_available = 5

        mock_client.policy.read_namespaced_pod_disruption_budget.return_value = mock_pdr

        app = mock.MagicMock()
        app.soa_config.get_bounce_margin_factor.return_value = 0.1
        app.kube_deployment.service.return_value = "fake_service"
        app.kube_deployment.instance.return_value = "fake_instance"
        Application.ensure_pod_disruption_budget(self=app,
                                                 kube_client=mock_client)

        mock_client.policy.patch_namespaced_pod_disruption_budget.assert_not_called(
        )
def test_ensure_pod_disruption_budget_replaces_outdated(
        mock_pdr_for_service_instance, mock_load_system_paasta_config):
    mock_req_pdr = mock.Mock()
    mock_req_pdr.spec.max_unavailable = 10
    mock_pdr_for_service_instance.return_value = mock_req_pdr

    mock_client = mock.MagicMock()

    mock_pdr = mock.Mock()
    mock_pdr.spec.max_unavailable = 5
    mock_pdr.spec.min_available = None

    mock_client.policy.read_namespaced_pod_disruption_budget.return_value = mock_pdr

    app = mock.MagicMock()
    app.soa_config.get_bounce_margin_factor.return_value = 0.1
    app.kube_deployment.service.return_value = "fake_service"
    app.kube_deployment.instance.return_value = "fake_instance"
    Application.ensure_pod_disruption_budget(self=app, kube_client=mock_client)

    mock_client.policy.patch_namespaced_pod_disruption_budget.assert_called_once_with(
        name=mock_req_pdr.metadata.name,
        namespace=mock_req_pdr.metadata.namespace,
        body=mock_req_pdr,
    )
def test_ensure_pod_disruption_budget_create(
    bounce_margin_factor_set,
    mock_pdr_for_service_instance,
    mock_load_system_paasta_config,
):
    mock_load_system_paasta_config.return_value.get_pdb_max_unavailable.return_value = 3

    mock_req_pdr = mock.Mock()
    mock_req_pdr.spec.max_unavailable = 10 if bounce_margin_factor_set else 3
    mock_pdr_for_service_instance.return_value = mock_req_pdr

    mock_client = mock.MagicMock()

    mock_client.policy.read_namespaced_pod_disruption_budget.side_effect = ApiException(
        status=404)

    app = mock.MagicMock()
    if bounce_margin_factor_set:
        app.soa_config.config_dict = {"bounce_margin_factor": 0.1}
        app.soa_config.get_bounce_margin_factor.return_value = 0.1
    app.kube_deployment.service.return_value = "fake_service"
    app.kube_deployment.instance.return_value = "fake_instance"
    Application.ensure_pod_disruption_budget(self=app, kube_client=mock_client)
    mock_client.policy.create_namespaced_pod_disruption_budget.assert_called_once_with(
        namespace="paasta", body=mock_req_pdr)
示例#4
0
def test_ensure_pod_disruption_budget_create():
    with mock.patch(
            "paasta_tools.kubernetes.application.controller_wrappers.pod_disruption_budget_for_service_instance",
            autospec=True,
    ) as mock_pdr_for_service_instance:
        mock_req_pdr = mock.Mock()
        mock_req_pdr.spec.max_unavailable = 10
        mock_pdr_for_service_instance.return_value = mock_req_pdr

        mock_client = mock.MagicMock()

        mock_client.policy.read_namespaced_pod_disruption_budget.side_effect = ApiException(
            status=404)

        app = mock.MagicMock()
        app.soa_config.get_bounce_margin_factor.return_value = 0.1
        app.kube_deployment.service.return_value = "fake_service"
        app.kube_deployment.instance.return_value = "fake_instance"
        Application.ensure_pod_disruption_budget(self=app,
                                                 kube_client=mock_client)
        mock_client.policy.create_namespaced_pod_disruption_budget.assert_called_once_with(
            namespace="paasta", body=mock_req_pdr)