log_buffer_test.py 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. #!/usr/bin/env python
  2. # Licensed to Cloudera, Inc. under one
  3. # or more contributor license agreements. See the NOTICE file
  4. # distributed with this work for additional information
  5. # regarding copyright ownership. Cloudera, Inc. licenses this file
  6. # to you under the Apache License, Version 2.0 (the
  7. # "License"); you may not use this file except in compliance
  8. # with the License. You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing, software
  13. # distributed under the License is distributed on an "AS IS" BASIS,
  14. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. # See the License for the specific language governing permissions and
  16. # limitations under the License.
  17. #
  18. # A couple of test cases for the log buffer
  19. #
  20. import logging
  21. import unittest
  22. from django.test import TestCase
  23. from tools.ops.script_runner.lib.log import log_buffer
  24. class TestLogBuffer(TestCase):
  25. def test_logger(self):
  26. logger = logging.getLogger()
  27. handler = log_buffer.FixedBufferHandler()
  28. logger.addHandler(handler)
  29. msg = "My test logging message"
  30. logger.warn(msg)
  31. assert msg == str(handler.buf)
  32. def test_overflow(self):
  33. buffer = log_buffer.FixedBuffer(maxsize=10)
  34. buffer.insert("0123456789")
  35. buffer.insert("abcde")
  36. assert "56789\nabcde" == str(buffer)
  37. if __name__ == '__main__':
  38. unittest.main()