Python main 예제들

예제 #1

파일: test_check_marathon_services_replication.py 프로젝트: iomedhealth/paasta

def test_main():
    soa_dir = 'anw'
    crit = 1
    services = [('a', 'main'), ('b', 'main'), ('c', 'main')]
    args = mock.Mock(soa_dir=soa_dir, crit=crit, verbose=False)
    with contextlib.nested(
        mock.patch('paasta_tools.check_marathon_services_replication.parse_args',
                   return_value=args, autospec=True),
        mock.patch('paasta_tools.check_marathon_services_replication.get_services_for_cluster',
                   return_value=services, autospec=True),
        mock.patch('paasta_tools.check_marathon_services_replication.check_service_replication',
                   autospec=True),
        mock.patch('paasta_tools.check_marathon_services_replication.load_system_paasta_config',
                   autospec=True),
        mock.patch('paasta_tools.check_marathon_services_replication.marathon_tools.load_marathon_config')
    ) as (
        mock_parse_args,
        mock_get_services_for_cluster,
        mock_check_service_replication,
        mock_load_system_paasta_config,
        mock_load_marathon_config,
    ):
        mock_config = mock.Mock()
        mock_load_marathon_config.return_value = mock_config
        mock_load_system_paasta_config.return_value.get_cluster = mock.Mock(return_value='fake_cluster')
        check_marathon_services_replication.main()
        mock_parse_args.assert_called_once_with()
        mock_get_services_for_cluster.assert_called_once_with(
            cluster='fake_cluster', instance_type='marathon', soa_dir=soa_dir)

예제 #2

파일: test_check_marathon_services_replication.py 프로젝트: snafis/paasta

def test_main():
    soa_dir = 'anw'
    crit = 1
    services = [('a', 'main'), ('b', 'main'), ('c', 'main')]
    args = mock.Mock(soa_dir=soa_dir, crit=crit, verbose=False)
    with mock.patch(
            'paasta_tools.check_marathon_services_replication.parse_args',
            return_value=args,
            autospec=True,
    ) as mock_parse_args, mock.patch(
            'paasta_tools.check_marathon_services_replication.get_services_for_cluster',
            return_value=services,
            autospec=True,
    ) as mock_get_services_for_cluster, mock.patch(
            'paasta_tools.check_marathon_services_replication.check_service_replication',
            autospec=True,
    ), mock.patch(
            'paasta_tools.check_marathon_services_replication.load_system_paasta_config',
            autospec=True,
    ) as mock_load_system_paasta_config, mock.patch(
            'paasta_tools.check_marathon_services_replication.marathon_tools.load_marathon_config',
            autospec=True,
    ) as mock_load_marathon_config:
        mock_config = mock.Mock()
        mock_load_marathon_config.return_value = mock_config
        mock_load_system_paasta_config.return_value.get_cluster = mock.Mock(
            return_value='fake_cluster')
        check_marathon_services_replication.main()
        mock_parse_args.assert_called_once_with()
        mock_get_services_for_cluster.assert_called_once_with(
            cluster='fake_cluster', instance_type='marathon', soa_dir=soa_dir)

예제 #3

파일: test_check_marathon_services_replication.py 프로젝트: somic/paasta

def test_main():
    soa_dir = "anw"
    crit = 1
    services = [("a", "main"), ("b", "main"), ("c", "main")]
    args = mock.Mock(soa_dir=soa_dir, crit=crit, verbose=False)
    with contextlib.nested(
        mock.patch("paasta_tools.check_marathon_services_replication.parse_args", return_value=args, autospec=True),
        mock.patch(
            "paasta_tools.check_marathon_services_replication.get_services_for_cluster",
            return_value=services,
            autospec=True,
        ),
        mock.patch("paasta_tools.check_marathon_services_replication.check_service_replication", autospec=True),
        mock.patch("paasta_tools.check_marathon_services_replication.load_system_paasta_config", autospec=True),
        mock.patch(
            "paasta_tools.check_marathon_services_replication.marathon_tools.load_marathon_config", autospec=True
        ),
    ) as (
        mock_parse_args,
        mock_get_services_for_cluster,
        mock_check_service_replication,
        mock_load_system_paasta_config,
        mock_load_marathon_config,
    ):
        mock_config = mock.Mock()
        mock_load_marathon_config.return_value = mock_config
        mock_load_system_paasta_config.return_value.get_cluster = mock.Mock(return_value="fake_cluster")
        check_marathon_services_replication.main()
        mock_parse_args.assert_called_once_with()
        mock_get_services_for_cluster.assert_called_once_with(
            cluster="fake_cluster", instance_type="marathon", soa_dir=soa_dir
        )

예제 #4

파일: test_check_marathon_services_replication.py 프로젝트: xcorail/paasta

def test_main(instance_config):
    soa_dir = 'anw'
    crit = 1
    args = mock.Mock(soa_dir=soa_dir, crit=crit, verbose=False)
    instance_config.get_docker_image.return_value = True
    with mock.patch(
            'paasta_tools.check_marathon_services_replication.parse_args',
            return_value=args,
            autospec=True,
    ) as mock_parse_args, mock.patch(
            'paasta_tools.check_marathon_services_replication.list_services',
            autospec=True,
            return_value=['a'],
    ), mock.patch(
            'paasta_tools.check_marathon_services_replication.check_service_replication',
            autospec=True,
    ) as mock_check_service_replication, mock.patch(
            'paasta_tools.check_marathon_services_replication.load_system_paasta_config',
            autospec=True,
    ) as mock_load_system_paasta_config, mock.patch(
            'paasta_tools.check_marathon_services_replication.marathon_tools.get_marathon_clients',
            autospec=True,
    ) as mock_get_marathon_clients, mock.patch(
            'paasta_tools.check_marathon_services_replication.get_slaves',
            autospec=True,
    ), mock.patch(
            'paasta_tools.check_marathon_services_replication.PaastaServiceConfigLoader',
            autospec=True,
    ) as mock_paasta_service_config_loader:
        mock_paasta_service_config_loader.return_value.instance_configs.return_value = [
            instance_config
        ]
        mock_client = mock.Mock()
        mock_client.list_tasks.return_value = []
        mock_get_marathon_clients.return_value = mock.Mock(
            get_all_clients=mock.Mock(return_value=[mock_client]))
        mock_load_system_paasta_config.return_value.get_cluster = mock.Mock(
            return_value='fake_cluster')
        check_marathon_services_replication.main()
        mock_parse_args.assert_called_once_with()
        mock_paasta_service_config_loader.assert_called_once_with(
            service='a',
            soa_dir=soa_dir,
        )
        instance_config.get_docker_image.assert_called_once_with()
        assert mock_check_service_replication.called