#!/usr/bin/env python3
+# © Copyright 2021-2022, Scott Gasch
+
+"""logging_utils unittest."""
+
import os
import sys
import tempfile
class TestLoggingUtils(unittest.TestCase):
-
def test_output_context(self):
unique_suffix = sutils.generate_uuid(True)
filename = f'/tmp/logging_utils_test.{unique_suffix}'
with tempfile.SpooledTemporaryFile(mode='r+') as tmpfile1:
with uu.RecordStdout() as record:
with lutils.OutputMultiplexerContext(
- lutils.OutputMultiplexer.Destination.FILENAMES |
- lutils.OutputMultiplexer.Destination.FILEHANDLES |
- lutils.OutputMultiplexer.Destination.LOG_INFO,
- filenames = [filename, '/dev/null'],
- handles = [tmpfile1, sys.stdout],
+ lutils.OutputMultiplexer.Destination.FILENAMES
+ | lutils.OutputMultiplexer.Destination.FILEHANDLES
+ | lutils.OutputMultiplexer.Destination.LOG_INFO,
+ filenames=[filename, '/dev/null'],
+ handles=[tmpfile1, sys.stdout],
) as mplex:
mplex.print(secret_message, end='')
with uu.RecordMultipleStreams(sys.stderr, sys.stdout) as record:
print("This is a test!")
print("This is one too.", file=sys.stderr)
- self.assertEqual(record().readlines(),
- ["This is a test!\n", "This is one too.\n"])
+ self.assertEqual(record().readlines(), ["This is a test!\n", "This is one too.\n"])
if __name__ == '__main__':