rules_parser_test.py

# Copyright 2015 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Unit tests for rules_parser.  Usage: ./rules_parser_test.py"""

import collections
import logging
from StringIO import StringIO
import unittest

import rules_parser


class RuleParserTest(unittest.TestCase):

  @classmethod
  def setUpClass(cls):
    if not logging.root.handlers:
      logging.basicConfig(level=logging.DEBUG,  # Enable log_url stdout.
                          format='%(asctime)s %(levelname)s %(message)s')

  def testCall(self):
    my_rules = rules_parser.Rules(StringIO(r'''
        [{"comment": "ignore me"},
         {"LogUrl": {"url": "example\\.com/ss.*"}},
         {"LogUrl": {"url": "example\\.com/blah$"}}]'''))
    log_url = my_rules.Find('log_url')
    self.assertEquals(True, log_url(FakeRequest(full_path='/ss'), None))
    self.assertEquals(True, log_url(FakeRequest(full_path='/ssxxxx'), None))
    self.assertEquals(True, log_url(FakeRequest(full_path='/blah'), None))
    self.assertEquals(None, log_url(FakeRequest(full_path='/blahxxx'), None))
    self.assertEquals(None, log_url(FakeRequest(full_path='/'), None))

  def testImport(self):
    my_rules = rules_parser.Rules(StringIO(r'''
        [{"rules.LogUrl": {"url": "example\\.com/ss.*"}}]'''))
    self.assertTrue(my_rules.Contains('log_url'))

  def testRaises(self):
    input_pairs = [
        'bad_json',
        '123',
        '{}',
        '[42]',
        '[{12:34}]',
        '[{"a":"b","c":"d"}]',
        '[{"bad+rule@name":{}}]',
        '["unallowed.Path":{}]',
        '["NoSuchRule":{}]',
        '["LogUrl":"bad"]',
        '["LogUrl":{}]',
        '["LogUrl":{"url":123}]',
        '["LogUrl":{"url":"", "bad_arg":123}]',
    ]
    for input_text in input_pairs:
      self.assertRaises(Exception, rules_parser.Rules, StringIO(input_text))


class FakeRequest(collections.namedtuple(
    'FakeRequest', ('command', 'host', 'full_path', 'request_body',
                    'headers', 'is_ssl'))):

  def __new__(cls, command='GET', host='example.com', full_path='/',
              request_body=None, headers=None, is_ssl=False):
    return super(FakeRequest, cls).__new__(
        cls, command, host, full_path, request_body, headers or {}, is_ssl)


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