def test_flush(self): logger = mock.MagicMock() logger.log = mock.MagicMock() log = StreamLogWriter(logger, 1) msg = "test_message" log.write(msg) self.assertEqual(log._buffer, msg) log.flush() logger.log.assert_called_once_with(1, msg) self.assertEqual(log._buffer, "")
def test_flush(self): logger = mock.MagicMock() logger.log = mock.MagicMock() log = StreamLogWriter(logger, 1) msg = "test_message" log.write(msg) assert log._buffer == msg log.flush() logger.log.assert_called_once_with(1, msg) assert log._buffer == ""
def test_write(self): logger = mock.MagicMock() logger.log = mock.MagicMock() log = StreamLogWriter(logger, 1) msg = "test_message" log.write(msg) self.assertEqual(log._buffer, msg) log.write(" \n") logger.log.assert_called_once_with(1, msg) self.assertEqual(log._buffer, "")