Esempio n. 1
0
def test_pause_autoscaler_long():
    args = mock.Mock(
        cluster='cluster1',
        duration=MAX_PAUSE_DURATION + 10,
        force=False,
        resume=False,
        info=False,
    )

    with mock.patch(
            'paasta_tools.cli.cmds.pause_service_autoscaler.update_service_autoscale_pause_time',
            autospec=True,
    ):
        return_code = paasta_pause_service_autoscaler(args)
        assert return_code == 3
Esempio n. 2
0
def test_pause_autoscaler_long(mock_log_audit):
    args = mock.Mock(
        cluster="cluster1",
        duration=MAX_PAUSE_DURATION + 10,
        force=False,
        resume=False,
        info=False,
    )

    with mock.patch(
            "paasta_tools.cli.cmds.pause_service_autoscaler.update_service_autoscale_pause_time",
            autospec=True,
    ):
        return_code = paasta_pause_service_autoscaler(args)
        assert return_code == 3
        assert not mock_log_audit.called
Esempio n. 3
0
def test_pause_autoscaler_defaults(mock_log_audit):
    args = mock.Mock(cluster="cluster1", duration=30, resume=False, info=False)

    with mock.patch(
            "paasta_tools.cli.cmds.pause_service_autoscaler.update_service_autoscale_pause_time",
            autospec=True,
    ) as mock_exc:
        mock_exc.return_value = 0
        return_code = paasta_pause_service_autoscaler(args)
        mock_exc.assert_called_once_with("cluster1", 30)
        assert return_code == 0
        mock_log_audit.assert_called_once_with(
            action="pause-service-autoscaler",
            action_details={"duration": 30},
            cluster="cluster1",
        )
Esempio n. 4
0
def test_pause_autoscaler_info(mock_log_audit):
    args = mock.Mock(cluster="cluster1",
                     duration=30,
                     force=False,
                     resume=False,
                     info=True)

    with mock.patch(
            "paasta_tools.cli.cmds.pause_service_autoscaler.get_service_autoscale_pause_time",
            autospec=True,
    ) as mock_exc:
        mock_exc.return_value = 0
        return_code = paasta_pause_service_autoscaler(args)
        mock_exc.assert_called_once_with("cluster1")
        assert return_code == 0
        assert not mock_log_audit.called
Esempio n. 5
0
def test_pause_autoscaler_defaults():
    args = mock.Mock(
        cluster='cluster1',
        duration=30,
        resume=False,
        info=False,
    )

    with mock.patch(
            'paasta_tools.cli.cmds.pause_service_autoscaler.update_service_autoscale_pause_time',
            autospec=True,
    ) as mock_exc:
        mock_exc.return_value = 0
        return_code = paasta_pause_service_autoscaler(args)
        mock_exc.assert_called_once_with('cluster1', 30)
        assert return_code == 0
Esempio n. 6
0
def test_pause_autoscaler_resume(mock_log_audit):
    args = mock.Mock(cluster="cluster1",
                     duration=120,
                     force=False,
                     resume=True,
                     info=False)

    with mock.patch(
            "paasta_tools.cli.cmds.pause_service_autoscaler.delete_service_autoscale_pause_time",
            autospec=True,
    ) as mock_exc:
        mock_exc.return_value = 0
        return_code = paasta_pause_service_autoscaler(args)
        mock_exc.assert_called_once_with("cluster1")
        assert return_code == 0
        mock_log_audit.assert_called_once_with(
            action="resume-service-autoscaler", cluster="cluster1")
Esempio n. 7
0
def test_pause_autoscaler_force():
    args = mock.Mock(
        cluster='cluster1',
        duration=MAX_PAUSE_DURATION + 10,
        force=True,
        resume=False,
        info=False,
    )

    with mock.patch(
            'paasta_tools.cli.cmds.pause_service_autoscaler.update_service_autoscale_pause_time',
            autospec=True,
    ) as mock_exc:
        mock_exc.return_value = 0
        return_code = paasta_pause_service_autoscaler(args)
        assert return_code == 0
        mock_exc.assert_called_once_with('cluster1', 330)