Add presubmit_canned_checks.CheckLicense()

tests/presubmit_unittest.py

 #!/usr/bin/python
 # Copyright (c) 2006-2009 The Chromium 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 presubmit_support.py and presubmit_canned_checks.py."""
 
 import StringIO
 
 import presubmit_support as presubmit
@@ skipping 1051 matching lines @@
     self.mox.ReplayAll()
     members = [
       'CheckChangeHasBugField', 'CheckChangeHasDescription',
       'CheckChangeHasNoStrayWhitespace',
       'CheckChangeHasOnlyOneEol', 'CheckChangeHasNoCR',
       'CheckChangeHasNoCrAndHasOnlyOneEol', 'CheckChangeHasNoTabs',
       'CheckChangeHasQaField', 'CheckChangeHasTestedField',
       'CheckChangeHasTestField',
       'CheckChangeLintsClean',
       'CheckChangeSvnEolStyle',
+      'CheckLicense',
       'CheckSvnModifiedDirectories',
       'CheckSvnForCommonMimeTypes', 'CheckSvnProperty',
       'CheckDoNotSubmit',
       'CheckDoNotSubmitInDescription', 'CheckDoNotSubmitInFiles',
       'CheckLongLines', 'CheckTreeIsOpen', 'RunPythonUnitTests',
     ]
     # If this test fails, you should add the relevant test.
     self.compareMembers(presubmit_canned_checks, members)
 
   def DescriptionTest(self, check, description1, description2, error_type,
@@ skipping 193 matching lines @@
     # Test CheckSvnProperty at the same time.
     self.SvnPropertyTest(presubmit_canned_checks.CheckChangeSvnEolStyle,
                          'svn:eol-style', 'LF', '', True,
                          presubmit.OutputApi.PresubmitError, True)
 
   def testCheckChangeSvnEolStyleUpload(self):
     self.SvnPropertyTest(presubmit_canned_checks.CheckChangeSvnEolStyle,
                          'svn:eol-style', 'LF', '', False,
                          presubmit.OutputApi.PresubmitNotifyResult, True)
 
+  def _LicenseCheck(self, text, license, committing, expected_result):
+    change = self.mox.CreateMock(presubmit.SvnChange)
+    change.scm = 'svn'
+    input_api = self.MockInputApi(change, committing)
+    affected_file = self.mox.CreateMock(presubmit.SvnAffectedFile)
+    input_api.AffectedSourceFiles(42).AndReturn([affected_file])
+    input_api.ReadFile(affected_file, 'rb').AndReturn(text)
+    if expected_result:
+      affected_file.LocalPath().AndReturn('bleh')
+
+    self.mox.ReplayAll()
+    result = presubmit_canned_checks.CheckLicense(
+                 input_api, presubmit.OutputApi, license, 42)
+    if expected_result:
+      self.assertEqual(len(result), 1)
+      self.assertEqual(result[0].__class__, expected_result)
+    else:
+      self.assertEqual(result, [])
+
+  def testCheckLicenseSuccess(self):
+    text = (
+        "#!/bin/python\n"
+        "# Copyright (c) 2037 Nobody.\n"
+        "# All Rights Reserved.\n"
+        "print 'foo'\n"
+    )
+    license = (
+        r".*? Copyright \(c\) 2037 Nobody." "\n"
+        r".*? All Rights Reserved\." "\n"
+    )
+    self._LicenseCheck(text, license, True, None)
+
+  def testCheckLicenseFailCommit(self):
+    text = (
+        "#!/bin/python\n"
+        "# Copyright (c) 2037 Nobody.\n"
+        "# All Rights Reserved.\n"
+        "print 'foo'\n"
+    )
+    license = (
+        r".*? Copyright \(c\) 0007 Nobody." "\n"
+        r".*? All Rights Reserved\." "\n"
+    )
+    self._LicenseCheck(text, license, True,
+                       presubmit.OutputApi.PresubmitPromptWarning)
+
+  def testCheckLicenseFailUpload(self):
+    text = (
+        "#!/bin/python\n"
+        "# Copyright (c) 2037 Nobody.\n"
+        "# All Rights Reserved.\n"
+        "print 'foo'\n"
+    )
+    license = (
+        r".*? Copyright \(c\) 0007 Nobody." "\n"
+        r".*? All Rights Reserved\." "\n"
+    )
+    self._LicenseCheck(text, license, False,
+                       presubmit.OutputApi.PresubmitNotifyResult)
+
   def testCannedCheckSvnAccidentalSubmission(self):
     modified_dir_file = 'foo/'
     accidental_submssion_file = 'foo/bar.cc'
 
     change = self.mox.CreateMock(presubmit.SvnChange)
     change.scm = 'svn'
     change.GetModifiedFiles().AndReturn([modified_dir_file])
     change.GetAllModifiedFiles().AndReturn([modified_dir_file,
                                             accidental_submssion_file])
     input_api = self.MockInputApi(change, True)
@@ skipping 166 matching lines @@
     self.mox.ReplayAll()
 
     results = presubmit_canned_checks.RunPythonUnitTests(
         input_api, presubmit.OutputApi, ['test_module'])
     self.assertEquals(len(results), 0)
 
 
 if __name__ == '__main__':
   import unittest
   unittest.main()