Python cluster_to_scribe_envの例

コード例 #1

ファイル: test_cmds_logs.py プロジェクト: ashwinaj/paasta

def test_cluster_to_scribe_env_bad():
    mock_system_paasta_config = mock.Mock(autospec='paasta_tools.utils.SystemPaastaConfig')
    mock_system_paasta_config.get_scribe_map.return_value = {}
    with mock.patch('paasta_tools.paasta_cli.cmds.logs.load_system_paasta_config', autospec=True) as (
        mock_load_system_paasta_config
    ):
        mock_load_system_paasta_config.return_value = mock_system_paasta_config
        with raises(SystemExit) as sys_exit:
            logs.cluster_to_scribe_env('dne')
            assert sys_exit.value.code == 1

コード例 #2

ファイル: test_cmds_logs.py プロジェクト: georgekola/paasta

def test_cluster_to_scribe_env_bad():
    mock_system_paasta_config = mock.Mock(
        autospec='paasta_tools.utils.SystemPaastaConfig')
    mock_system_paasta_config.get_scribe_map.return_value = {}
    with mock.patch(
            'paasta_tools.paasta_cli.cmds.logs.load_system_paasta_config',
            autospec=True) as (mock_load_system_paasta_config):
        mock_load_system_paasta_config.return_value = mock_system_paasta_config
        with raises(SystemExit) as sys_exit:
            logs.cluster_to_scribe_env('dne')
            assert sys_exit.value.code == 1

コード例 #3

ファイル: test_cmds_logs.py プロジェクト: ashwinaj/paasta

def test_cluster_to_scribe_env_good():
    mock_system_paasta_config = mock.Mock(autospec='paasta_tools.utils.SystemPaastaConfig')
    mock_system_paasta_config.get_scribe_map.return_value = {'mesosstage': 'env1'}
    with mock.patch('paasta_tools.paasta_cli.cmds.logs.load_system_paasta_config', autospec=True) as (
        mock_load_system_paasta_config
    ):
        mock_load_system_paasta_config.return_value = mock_system_paasta_config
        actual = logs.cluster_to_scribe_env('mesosstage')
        assert actual == 'env1'

コード例 #4

ファイル: test_cmds_logs.py プロジェクト: georgekola/paasta

def test_cluster_to_scribe_env_good():
    mock_system_paasta_config = mock.Mock(
        autospec='paasta_tools.utils.SystemPaastaConfig')
    mock_system_paasta_config.get_scribe_map.return_value = {
        'mesosstage': 'env1'
    }
    with mock.patch(
            'paasta_tools.paasta_cli.cmds.logs.load_system_paasta_config',
            autospec=True) as (mock_load_system_paasta_config):
        mock_load_system_paasta_config.return_value = mock_system_paasta_config
        actual = logs.cluster_to_scribe_env('mesosstage')
        assert actual == 'env1'