• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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
17import logging
18
19
20class 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    }
38