def test_validate_service_instance_invalid(): mock_marathon_services = [("service1", "main"), ("service2", "main")] mock_chronos_services = [("service1", "worker"), ("service2", "tailer")] my_service = "bad_service" my_instance = "main" fake_cluster = "fake_cluster" fake_soa_dir = "fake_soa_dir" with contextlib.nested( mock.patch( "utils.get_services_for_cluster", autospec=True, side_effect=[mock_marathon_services, mock_chronos_services] ), mock.patch("sys.exit"), ) as (get_services_for_cluster_patch, sys_exit_patch): utils.validate_service_instance(my_service, my_instance, fake_cluster, fake_soa_dir) sys_exit_patch.assert_called_once_with(3)
def test_validate_service_instance_valid_chronos(): mock_marathon_services = [('service1', 'main'), ('service2', 'main')] mock_chronos_services = [('service1', 'worker'), ('service2', 'tailer')] my_service = 'service1' my_instance = 'worker' fake_cluster = 'fake_cluster' fake_soa_dir = 'fake_soa_dir' with contextlib.nested( mock.patch('utils.get_services_for_cluster', autospec=True, side_effect=[mock_marathon_services, mock_chronos_services]), ) as ( get_services_for_cluster_patch, ): assert utils.validate_service_instance( my_service, my_instance, fake_cluster, fake_soa_dir, ) == 'chronos' assert mock.call( cluster=fake_cluster, instance_type='chronos', soa_dir=fake_soa_dir, ) in get_services_for_cluster_patch.call_args_list
def test_validate_service_instance_invalid(): mock_marathon_services = [('service1', 'main'), ('service2', 'main')] mock_chronos_services = [('service1', 'worker'), ('service2', 'tailer')] my_service = 'bad_service' my_instance = 'main' fake_cluster = 'fake_cluster' fake_soa_dir = 'fake_soa_dir' with contextlib.nested( mock.patch('utils.get_services_for_cluster', autospec=True, side_effect=[mock_marathon_services, mock_chronos_services]), mock.patch('sys.exit'), ) as ( get_services_for_cluster_patch, sys_exit_patch, ): utils.validate_service_instance( my_service, my_instance, fake_cluster, fake_soa_dir, ) sys_exit_patch.assert_called_once_with(3)
def test_validate_service_instance_invalid(): mock_marathon_services = [('service1', 'main'), ('service2', 'main')] mock_chronos_services = [('service1', 'worker'), ('service2', 'tailer')] my_service = 'bad_service' my_instance = 'main' fake_cluster = 'fake_cluster' fake_soa_dir = 'fake_soa_dir' with contextlib.nested( mock.patch( 'utils.get_services_for_cluster', autospec=True, side_effect=[mock_marathon_services, mock_chronos_services]), mock.patch('sys.exit'), ) as ( get_services_for_cluster_patch, sys_exit_patch, ): utils.validate_service_instance( my_service, my_instance, fake_cluster, fake_soa_dir, ) sys_exit_patch.assert_called_once_with(3)
def test_validate_service_instance_valid_chronos(): mock_marathon_services = [("service1", "main"), ("service2", "main")] mock_chronos_services = [("service1", "worker"), ("service2", "tailer")] my_service = "service1" my_instance = "worker" fake_cluster = "fake_cluster" fake_soa_dir = "fake_soa_dir" with contextlib.nested( mock.patch( "utils.get_services_for_cluster", autospec=True, side_effect=[mock_marathon_services, mock_chronos_services] ) ) as (get_services_for_cluster_patch,): assert utils.validate_service_instance(my_service, my_instance, fake_cluster, fake_soa_dir) == "chronos" assert ( mock.call(cluster=fake_cluster, instance_type="chronos", soa_dir=fake_soa_dir) in get_services_for_cluster_patch.call_args_list )
def test_validate_service_instance_valid_chronos(): mock_marathon_services = [('service1', 'main'), ('service2', 'main')] mock_chronos_services = [('service1', 'worker'), ('service2', 'tailer')] my_service = 'service1' my_instance = 'worker' fake_cluster = 'fake_cluster' fake_soa_dir = 'fake_soa_dir' with contextlib.nested( mock.patch('utils.get_services_for_cluster', autospec=True, side_effect=[ mock_marathon_services, mock_chronos_services ]), ) as (get_services_for_cluster_patch, ): assert utils.validate_service_instance( my_service, my_instance, fake_cluster, fake_soa_dir, ) == 'chronos' assert mock.call( cluster=fake_cluster, instance_type='chronos', soa_dir=fake_soa_dir, ) in get_services_for_cluster_patch.call_args_list