Add post_process argument to Checkout.apply_patch().

tests/checkout_test.py

 #!/usr/bin/env python
 # Copyright (c) 2011 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 checkout.py."""
 
 from __future__ import with_statement
 import logging
 import os
@@ skipping 215 matching lines @@
     try:
       co.apply_patch([patch.FilePatchDiff('svn_utils_test.txt', BAD_PATCH, [])])
       self.fail()
     except checkout.PatchApplicationFailed, e:
       self.assertEquals(e.filename, 'svn_utils_test.txt')
       self.assertEquals(e.status, err_msg)
 
   def _log(self):
     raise NotImplementedError()
 
+  def _test_process(self, co):
+    """Makes sure the process lambda is called correctly."""
+    co.prepare()
+    ps = self.get_patches()
+    results = []
+    co.apply_patch(ps, [lambda *args: results.append(args)])
+    expected = [(co, p) for p in ps.patches]
+    self.assertEquals(expected, results)
+
 
 class SvnBaseTest(BaseTest):
   def setUp(self):
     super(SvnBaseTest, self).setUp()
     self.enabled = self.FAKE_REPOS.set_up_svn()
     self.assertTrue(self.enabled)
     self.svn_trunk = 'trunk'
     self.svn_url = self.svn_base + self.svn_trunk
     self.previous_log = self._log()
 
@@ skipping 124 matching lines @@
     co.apply_patch(patches)
     self.assertEquals(
         ['bin_file', 'extra', 'new_dir/subdir/new_file', 'svn_utils_test.txt'],
         sorted(patches.filenames))
     # *.txt = svn:eol-style=LF in subversion_config/config.
     out = subprocess2.check_output(
         ['svn', 'pget', 'svn:eol-style', 'svn_utils_test.txt'],
         cwd=co.project_path)
     self.assertEquals('LF\n', out)
 
+  def testProcess(self):
+    co = checkout.SvnCheckout(
+        self.root_dir, self.name,
+        None, None,
+        self.svn_url)
+    self._test_process(co)
+
 
 class GitSvnCheckout(SvnBaseTest):
   name = 'foo.git'
 
   def _get_co(self, read_only):
     co = checkout.GitSvnCheckout(
         self.root_dir, self.name[:-4],
         self.usr, self.pwd,
         self.svn_base, self.svn_trunk)
     if read_only:
@@ skipping 45 matching lines @@
       self.assertEquals(e.filename, 'svn_utils_test.txt')
       self.assertEquals(
           e.status,
           'Cannot apply svn property foo to file svn_utils_test.txt.')
     co.prepare()
     # svn:eol-style is ignored.
     svn_props = [('svn:eol-style', 'LF')]
     co.apply_patch(
         [patch.FilePatchDiff('svn_utils_test.txt', NAKED_PATCH, svn_props)])
 
+  def testProcess(self):
+    co = checkout.SvnCheckout(
+        self.root_dir, self.name,
+        None, None,
+        self.svn_url)
+    self._test_process(co)
+
 
 class RawCheckout(SvnBaseTest):
   def setUp(self):
     super(RawCheckout, self).setUp()
     # Use a svn checkout as the base.
     self.base_co = checkout.SvnCheckout(
         self.root_dir, self.name, None, None, self.svn_url)
     self.base_co.prepare()
 
   def _get_co(self, read_only):
@@ skipping 40 matching lines @@
     self._check(True)
 
   def testException(self):
     self._check_exception(
         self._get_co(True),
         'patching file svn_utils_test.txt\n'
         'Hunk #1 FAILED at 3.\n'
         '1 out of 1 hunk FAILED -- saving rejects to file '
         'svn_utils_test.txt.rej\n')
 
+  def testProcess(self):
+    co = checkout.SvnCheckout(
+        self.root_dir, self.name,
+        None, None,
+        self.svn_url)
+    self._test_process(co)
+
 
 if __name__ == '__main__':
   if '-v' in sys.argv:
     DEBUGGING = True
     logging.basicConfig(
         level=logging.DEBUG,
         format='%(levelname)5s %(filename)15s(%(lineno)3d): %(message)s')
   else:
     logging.basicConfig(
         level=logging.ERROR,
         format='%(levelname)5s %(filename)15s(%(lineno)3d): %(message)s')
   unittest.main()