def test_pid_decision_policy():
current_time = datetime.now()
zookeeper_get_payload = {
'pid_iterm': 0,
'pid_last_error': 0,
'pid_last_time': (current_time - timedelta(seconds=600)).strftime('%s'),
}
with contextlib.nested(
mock.patch('paasta_tools.utils.KazooClient', autospec=True,
return_value=mock.Mock(get=mock.Mock(
side_effect=lambda x: (zookeeper_get_payload[x.split('/')[-1]], None)))),
mock.patch('paasta_tools.autoscaling_lib.datetime', autospec=True),
mock.patch('paasta_tools.utils.load_system_paasta_config', autospec=True,
return_value=mock.Mock(get_zk_hosts=mock.Mock())),
) as (
mock_zk_client,
mock_datetime,
_,
):
mock_datetime.now.return_value = current_time
assert autoscaling_lib.pid_decision_policy('/autoscaling/fake-service/fake-instance', 10, 1, 100, 0.0) == 0
assert autoscaling_lib.pid_decision_policy('/autoscaling/fake-service/fake-instance', 10, 1, 100, 0.2) == 1
assert autoscaling_lib.pid_decision_policy('/autoscaling/fake-service/fake-instance', 10, 1, 100, -0.2) == -1
mock_zk_client.return_value.set.assert_has_calls([
mock.call('/autoscaling/fake-service/fake-instance/pid_iterm', '0.0'),
mock.call('/autoscaling/fake-service/fake-instance/pid_last_error', '0.0'),
mock.call('/autoscaling/fake-service/fake-instance/pid_last_time',
'%s' % current_time.strftime('%s')),
], any_order=True)
def test_pid_decision_policy():
current_time = datetime.now()
zookeeper_get_payload = {
'pid_iterm': 0,
'pid_last_error': 0,
'pid_last_time': (current_time - timedelta(seconds=600)).strftime('%s'),
}
with contextlib.nested(
mock.patch('paasta_tools.utils.KazooClient', autospec=True,
return_value=mock.Mock(get=mock.Mock(
side_effect=lambda x: (zookeeper_get_payload[x.split('/')[-1]], None)))),
mock.patch('paasta_tools.autoscaling_lib.datetime', autospec=True),
mock.patch('paasta_tools.utils.load_system_paasta_config', autospec=True,
return_value=mock.Mock(get_zk_hosts=mock.Mock())),
) as (
mock_zk_client,
mock_datetime,
_,
):
mock_datetime.now.return_value = current_time
assert autoscaling_lib.pid_decision_policy('/autoscaling/fake-service/fake-instance', 10, 1, 100, 0.0) == 0
assert autoscaling_lib.pid_decision_policy('/autoscaling/fake-service/fake-instance', 10, 1, 100, 0.2) == 1
assert autoscaling_lib.pid_decision_policy('/autoscaling/fake-service/fake-instance', 10, 1, 100, -0.2) == -1
mock_zk_client.return_value.set.assert_has_calls([
mock.call('/autoscaling/fake-service/fake-instance/pid_iterm', '0.0'),
mock.call('/autoscaling/fake-service/fake-instance/pid_last_error', '0.0'),
mock.call('/autoscaling/fake-service/fake-instance/pid_last_time',
'%s' % current_time.strftime('%s')),
], any_order=True)
def test_pid_decision_policy():
fake_marathon_service_config = marathon_tools.MarathonServiceConfig(
service='fake-service',
instance='fake-instance',
cluster='fake-cluster',
config_dict={},
branch_dict={},
)
current_time = datetime.now()
zookeeper_get_payload = {
'pid_iterm': 0,
'pid_last_error': 0,
'pid_last_time': (current_time - timedelta(seconds=600)).strftime('%s'),
}
with contextlib.nested(
mock.patch('paasta_tools.utils.KazooClient', autospec=True,
return_value=mock.Mock(get=mock.Mock(
side_effect=lambda x: (zookeeper_get_payload[x.split('/')[-1]], None)))),
mock.patch('paasta_tools.autoscaling_lib.datetime', autospec=True),
mock.patch('paasta_tools.utils.load_system_paasta_config', autospec=True,
return_value=mock.Mock(get_zk_hosts=mock.Mock())),
) as (
mock_zk_client,
mock_datetime,
_,
):
mock_datetime.now.return_value = current_time
mock_metrics_provider = mock.Mock(return_value=0.8)
assert autoscaling_lib.pid_decision_policy(fake_marathon_service_config, mock_metrics_provider,
mock.Mock(), mock.Mock()) == 0
mock_zk_client.return_value.set.assert_has_calls([
mock.call('/autoscaling/fake-service/fake-instance/pid_iterm', '0.0'),
mock.call('/autoscaling/fake-service/fake-instance/pid_last_error', '0.0'),
mock.call('/autoscaling/fake-service/fake-instance/pid_last_time',
'%s' % current_time.strftime('%s')),
], any_order=True)