12345678910111213141516171819202122232425262728293031323334353637383940 |
- #!/usr/bin/env python3
- #
- # Copyright (C) 2016 Intel Corporation
- #
- # SPDX-License-Identifier: MIT
- #
- import unittest
- import logging
- import tempfile
- from common import setup_sys_path, TestBase
- setup_sys_path()
- from oeqa.core.runner import OEStreamLogger
- class TestRunner(TestBase):
- def test_stream_logger(self):
- fp = tempfile.TemporaryFile(mode='w+')
- logging.basicConfig(format='%(message)s', stream=fp)
- logger = logging.getLogger()
- logger.setLevel(logging.INFO)
- oeSL = OEStreamLogger(logger)
- lines = ['init', 'bigline_' * 65535, 'morebigline_' * 65535 * 4, 'end']
- for line in lines:
- oeSL.write(line)
- fp.seek(0)
- fp_lines = fp.readlines()
- for i, fp_line in enumerate(fp_lines):
- fp_line = fp_line.strip()
- self.assertEqual(lines[i], fp_line)
- fp.close()
- if __name__ == '__main__':
- unittest.main()
|