Add a unit test for input_api.change.tags.

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 os
 import StringIO
 import sys
@@ skipping 348 matching lines @@
                         description='\n'.join(description_lines),
                         files=files)
 
     output = StringIO.StringIO()
     input = StringIO.StringIO('y\n')
     # Always fail.
     DEFAULT_SCRIPT = """
 def CheckChangeOnUpload(input_api, output_api):
   print 'This is a test'
   return [output_api.PresubmitError("!!")]
+def CheckChangeOnCommit(input_api, output_api):
+  raise Exception("Test error")
 """
     def MockReadFile(dummy):
       return ''
     gcl.ReadFile = MockReadFile
     def MockIsFile(dummy):
       return False
     os.path.isfile = MockIsFile
     self.failIf(presubmit.DoPresubmitChecks(ci, False, True, output, input,
                                             DEFAULT_SCRIPT))
     self.assertEquals(output.getvalue().count('!!'), 1)
 
   def testDirectoryHandling(self):
     files = [
       ['A', 'isdir'],
       ['A', 'isdir\\blat.cc'],
     ]
     ci = gcl.ChangeInfo(name='mychange',
                         description='foo',
                         files=files)
     change = presubmit.GclChange(ci)
 
     affected_files = change.AffectedFiles(include_dirs=False)
     self.failUnless(len(affected_files) == 1)
     self.failUnless(affected_files[0].LocalPath().endswith('blat.cc'))
 
     affected_files_and_dirs = change.AffectedFiles(include_dirs=True)
     self.failUnless(len(affected_files_and_dirs) == 2)
 
+  def testTags(self):
+    DEFAULT_SCRIPT = """
+def CheckChangeOnUpload(input_api, output_api):
+ if input_api.change.tags['BUG'] != 'boo':
+   return [output_api.PresubmitError('Tag parsing failed. 1')]
+ if input_api.change.tags['STORY'] != 'http://tracker.com/42':
+   return [output_api.PresubmitError('Tag parsing failed. 2')]
+ if 'TEST' in input_api.change.tags:
+   return [output_api.PresubmitError('Tag parsing failed. 3')]
+ if input_api.change.DescriptionText() != 'Blah Blah':
+   return [output_api.PresubmitError('Tag parsing failed. 4 ' + 
+                                     input_api.change.DescriptionText())]
+ if (input_api.change.FullDescriptionText() !=
+     'Blah Blah\\n\\nSTORY=http://tracker.com/42\\nBUG=boo\\n'):
+   return [output_api.PresubmitError('Tag parsing failed. 5 ' +
+                                     input_api.change.FullDescriptionText())]
+ return [output_api.PresubmitNotifyResult(input_api.change.tags['STORY'])]
+def CheckChangeOnCommit(input_api, output_api):
+  raise Exception("Test error")
+"""
+    def MockReadFile(dummy):
+      return ''
+    gcl.ReadFile = MockReadFile
+    def MockIsFile(dummy):
+      return False
+    os.path.isfile = MockIsFile
+    change = gcl.ChangeInfo(
+        name='foo',
+        description="Blah Blah\n\nSTORY=http://tracker.com/42\nBUG=boo\n")
+    output = StringIO.StringIO()
+    input = StringIO.StringIO('y\n')
+    self.failUnless(presubmit.DoPresubmitChecks(change, False, True, output,
+                                                input, DEFAULT_SCRIPT))
+    self.assertEquals(output.getvalue(),
+                      ('Warning, no presubmit.py found.\n'
+                       'Running default presubmit script.\n\n'
+                       '** Presubmit Messages **\n\n'
+                       'http://tracker.com/42\n\n'))
+
 
 class InputApiUnittest(PresubmitTestsBase):
   """Tests presubmit.InputApi."""
   def testMembersChanged(self):
     members = [
       'AbsoluteLocalPaths', 'AffectedFiles', 'AffectedTextFiles',
       'DepotToLocalPath', 'FilterTextFiles', 'LocalPaths', 'LocalToDepotPath',
       'PresubmitLocalPath', 'RightHandSideLines', 'ServerPaths',
       'basename', 'cPickle', 'cStringIO', 'canned_checks', 'change',
       'marshal', 'os_path', 'pickle', 'platform',
@@ skipping 337 matching lines @@
     # TODO(maruel): Add real tests.
     self.failIf(presubmit_canned_checks.RunPythonUnitTests(
         self.MockInputApi(),
         presubmit.OutputApi, []))
     self.failUnless(presubmit_canned_checks.RunPythonUnitTests(
         self.MockInputApi(),
         presubmit.OutputApi, ['non_existent_module']))
 
 if __name__ == '__main__':
   unittest.main()