def test_slog_with_scribe(self): mock = Mock() temp = douban.utils.slog.scribeclient douban.utils.slog.scribeclient = mock log('test', 'test-message') assert mock.send.called test_log = lambda x: log('test', x) test_log('tmsg') assert mock.send.called douban.utils.slog.scribeclient = temp
def test_slog_without_scribec(self): temp = douban.utils.slog.scribeclient douban.utils.slog.scribeclient = None with patch('sys.stderr') as mock_stderr: log('test', 'testmessage') assert mock_stderr.write.called test_log = lambda x: log('haha', x) test_log('tset_message') assert mock_stderr.write.called douban.utils.slog.scribeclient = temp