def test_scribe_tail_handles_StreamTailerSetupError():
env = "fake_env"
stream_name = "fake_stream"
service = "fake_service"
levels = ["fake_level1"]
components = ["build"]
clusters = ["fake_cluster1"]
queue = Queue()
filter_fn = mock.Mock(return_value=True)
with mock.patch(
"paasta_tools.cli.cmds.logs.scribereader", autospec=True
) as mock_scribereader, mock.patch(
"paasta_tools.cli.cmds.logs.log", autospec=True
) as mock_log:
mock_scribereader.get_stream_tailer.side_effect = StreamTailerSetupError(
"bla", "unused1", "unused2"
)
with raises(StreamTailerSetupError):
logs.scribe_tail(
env,
stream_name,
service,
levels,
components,
clusters,
queue,
filter_fn,
)
mock_log.error.assert_any_call(
"Failed to setup stream tailing for %s in fake_env" % stream_name
)
def test_scribe_tail_handles_StreamTailerSetupError():
env = 'fake_env'
stream_name = 'fake_stream'
service = 'fake_service'
levels = ['fake_level1']
components = ['build']
clusters = ['fake_cluster1']
queue = Queue()
filter_fn = mock.Mock(return_value=True)
with contextlib.nested(
mock.patch('paasta_tools.cli.cmds.logs.scribereader',
autospec=True),
mock.patch('paasta_tools.cli.cmds.logs.log', autospec=True),
) as (
mock_scribereader,
mock_log,
):
mock_scribereader.get_stream_tailer.side_effect = StreamTailerSetupError(
'bla', 'unused1', 'unused2')
with raises(StreamTailerSetupError):
logs.scribe_tail(
env,
stream_name,
service,
levels,
components,
clusters,
queue,
filter_fn,
)
mock_log.error.assert_any_call(
'Failed to setup stream tailing for %s in fake_env' % stream_name)
