apache_log_metrics_unittest.py

#!/usr/bin/python2

# Copyright 2016 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unit tests for apache_log_metrics.py"""

from __future__ import print_function

import mock
import unittest

import apache_log_metrics


STATIC_REQUEST_LINE = (
    '172.24.26.30 - - [30/Jun/2016:15:34:40 -0700] '
    '"GET /static/veyron_minnie-release/R52-8350.46.0/'
    'autotest_server_package.tar.bz2'
    ' HTTP/1.1" 200 13805917 "-" "Wget/1.15    (linux-gnu)'
)


class TestParsers(unittest.TestCase):
  """Tests the parsing functions in apache_log_metrics."""

  def testParseStaticResponse(self):
    match = apache_log_metrics.STATIC_GET_MATCHER.match(
        STATIC_REQUEST_LINE)
    self.assertTrue(match)

    ip = match.group('ip_addr')
    self.assertEqual(ip, '172.24.26.30')
    self.assertFalse(apache_log_metrics.InLab(ip))

    self.assertEqual(match.group('size'), '13805917')


class TestEmitters(unittest.TestCase):
  """Tests the emitter functions in apache_log_metrics."""

  def testEmitStaticResponse(self):
    match = apache_log_metrics.STATIC_GET_MATCHER.match(
        STATIC_REQUEST_LINE)
    # Calling the emitter should not raise any exceptions (for example, by
    # referencing regex match groups that don't exist.
    with mock.patch.object(apache_log_metrics, 'metrics'):
      apache_log_metrics.EmitStaticRequestMetric(match)


if __name__ == '__main__':
  unittest.main()