tools/telemetry/third_party/gsutilz/gslib/tests/mock_logging_handler.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright 2014 Google Inc. All Rights Reserved.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at
   7 #
   8 #     http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 """Mock logging handler to check for expected logs."""
  16
  17 import logging
  18
  19
  20 class MockLoggingHandler(logging.Handler):
  21   """Mock logging handler to check for expected logs."""
  22
  23   def __init__(self, *args, **kwargs):
  24     self.reset()
  25     logging.Handler.__init__(self, *args, **kwargs)
  26
  27   def emit(self, record):
  28     self.messages[record.levelname.lower()].append(record.getMessage())
  29
  30   def reset(self):
  31     self.messages = {
  32         'debug': [],
  33         'info': [],
  34         'warning': [],
  35         'error': [],
  36         'critical': [],
  37     }