From 1be36c97c696eeab31ae8c2d12b7c3e3978be072 Mon Sep 17 00:00:00 2001 From: Simon Sawicki Date: Mon, 12 Feb 2024 21:48:50 +0100 Subject: [PATCH] Add tests --- tests/utils/test_logging_mixin.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tests/utils/test_logging_mixin.py b/tests/utils/test_logging_mixin.py index bcc7a408856e0b..4289336d95d2b7 100644 --- a/tests/utils/test_logging_mixin.py +++ b/tests/utils/test_logging_mixin.py @@ -131,11 +131,13 @@ def test_write(self): log = StreamLogWriter(logger, 1) msg = "test_message" - log.write(msg) + length = log.write(msg) + assert length == len(msg) assert log._buffer == msg - log.write(" \n") + length = log.write(" \n") + assert length == 0 logger.log.assert_called_once_with(1, msg) assert log._buffer == "" @@ -168,7 +170,7 @@ def test_encoding(self): logger.log = mock.MagicMock() log = StreamLogWriter(logger, 1) - assert log.encoding is None + assert log.encoding == "None" def test_iobase_compatibility(self): log = StreamLogWriter(None, 1) @@ -177,6 +179,13 @@ def test_iobase_compatibility(self): # has no specific effect log.close() + def test_textio_compatibility(self): + log = StreamLogWriter(None, 1) + + # Make sure some common operations don't fail + assert "b" not in log.mode + assert log.name.partition(".")[0] == "" + @pytest.mark.parametrize(["maintain_propagate"], [[SetContextPropagate.MAINTAIN_PROPAGATE], [None]]) def test_set_context_propagation(parent_child_handlers, child_logger, maintain_propagate):