def test_update_instances_for_marathon_service():
with contextlib.nested(
mock.patch(
'paasta_tools.marathon_tools.load_marathon_service_config',
autospec=True),
mock.patch('paasta_tools.utils.KazooClient', autospec=True),
mock.patch('paasta_tools.utils.load_system_paasta_config',
autospec=True),
) as (
mock_load_marathon_service_config,
mock_zk_client,
_,
):
zk_client = mock.Mock(get=mock.Mock(side_effect=NoNodeError))
mock_zk_client.return_value = zk_client
mock_load_marathon_service_config.return_value = marathon_tools.MarathonServiceConfig(
service='service',
instance='instance',
cluster='cluster',
config_dict={
'min_instances': 5,
'max_instances': 10,
},
branch_dict={},
)
autoscaling_service_lib.set_instances_for_marathon_service(
'service', 'instance', instance_count=8)
zk_client.set.assert_called_once_with(
'/autoscaling/service/instance/instances', '8')
def zookeeper_scale_job(context, service, instance, number):
with contextlib.nested(
mock.patch.object(SystemPaastaConfig, 'get_zk_hosts', autospec=True, return_value=context.zk_hosts)
) as (
_,
):
set_instances_for_marathon_service(service, instance, number, soa_dir=context.soa_dir)
def test_update_instances_for_marathon_service():
with contextlib.nested(
mock.patch('paasta_tools.marathon_tools.load_marathon_service_config', autospec=True),
mock.patch('paasta_tools.utils.KazooClient', autospec=True),
mock.patch('paasta_tools.utils.load_system_paasta_config', autospec=True),
) as (
mock_load_marathon_service_config,
mock_zk_client,
_,
):
zk_client = mock.Mock(get=mock.Mock(side_effect=NoNodeError))
mock_zk_client.return_value = zk_client
mock_load_marathon_service_config.return_value = marathon_tools.MarathonServiceConfig(
service='service',
instance='instance',
cluster='cluster',
config_dict={
'min_instances': 5,
'max_instances': 10,
},
branch_dict={},
)
autoscaling_service_lib.set_instances_for_marathon_service('service', 'instance', instance_count=8)
zk_client.set.assert_called_once_with('/autoscaling/service/instance/instances', '8')
