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_lib.set_instances_for_marathon_service('service', 'instance', instance_count=8) zk_client.set.assert_called_once_with( '/autoscaling/service/instance/instances', '8')
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_lib.set_instances_for_marathon_service("service", "instance", instance_count=8) zk_client.set.assert_called_once_with("/autoscaling/service/instance/instances", "8")
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_lib.set_instances_for_marathon_service('service', 'instance', instance_count=8) zk_client.set.assert_called_once_with('/autoscaling/service/instance/instances', '8')