PRESUBMIT_unittest.py

   1 #!/usr/bin/python
   2 # Copyright (c) 2009 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Unit tests for top-level Chromium presubmit script.
   7 """
   8
   9
  10 import os
  11 import PRESUBMIT
  12 import re
  13 import unittest
  14
  15
  16 class MockInputApi(object):
  17   def __init__(self):
  18     self.affected_files = []
  19     self.re = re
  20     self.os_path = os.path
  21
  22   def AffectedFiles(self):
  23     return self.affected_files
  24
  25   def AffectedTextFiles(self, include_deletes=True):
  26     return self.affected_files
  27
  28
  29 class MockAffectedFile(object):
  30   def __init__(self, path):
  31     self.path = path
  32
  33   def LocalPath(self):
  34     return self.path
  35
  36
  37 class MockOutputApi(object):
  38   class PresubmitError(object):
  39     def __init__(self, msg, items=[], long_text=''):
  40       self.msg = msg
  41       self.items = items
  42
  43
  44 class PresubmitUnittest(unittest.TestCase):
  45   def setUp(self):
  46     self.file_contents = ''
  47     def MockReadFile(path):
  48       self.failIf(path.endswith('notsource'))
  49       return self.file_contents
  50     self._ReadFile = PRESUBMIT.ReadFile
  51     PRESUBMIT.ReadFile = MockReadFile
  52
  53   def tearDown(self):
  54     PRESUBMIT.ReadFile = self._ReadFile
  55
  56   def testLocalChecks(self):
  57     api = MockInputApi()
  58     api.affected_files = [
  59       MockAffectedFile('foo/blat/yoo.notsource'),
  60       MockAffectedFile('third_party/blat/source.cc'),
  61       MockAffectedFile('foo/blat/source.h'),
  62       MockAffectedFile('foo/blat/source.mm'),
  63       MockAffectedFile('foo/blat/source.py'),
  64     ]
  65     self.file_contents = 'file with \n\terror\nhere\r\nyes there'
  66     # 3 source files, 2 errors by file + 1 global CR error.
  67     self.failUnless(len(PRESUBMIT.LocalChecks(api, MockOutputApi)) == 7)
  68
  69     self.file_contents = 'file\twith\ttabs'
  70     # 3 source files, 1 error by file.
  71     self.failUnless(len(PRESUBMIT.LocalChecks(api, MockOutputApi)) == 3)
  72
  73     self.file_contents = 'file\rusing\rCRs'
  74     # One global CR error.
  75     self.failUnless(len(PRESUBMIT.LocalChecks(api, MockOutputApi)) == 1)
  76     self.failUnless(
  77       len(PRESUBMIT.LocalChecks(api, MockOutputApi)[0].items) == 3)
  78
  79     self.file_contents = 'both\ttabs and\r\nCRLF'
  80     # 3 source files, 1 error by file + 1 global CR error.
  81     self.failUnless(len(PRESUBMIT.LocalChecks(api, MockOutputApi)) == 4)
  82
  83     self.file_contents = 'file with\nzero \\t errors \\r\\n'
  84     self.failIf(PRESUBMIT.LocalChecks(api, MockOutputApi))
  85
  86
  87 if __name__ == '__main__':
  88   unittest.main()