gclient: kill git fetch operation that hangs.

tests/gclient_utils_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 import os
 import StringIO
 import sys
+import threading
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from testing_support.super_mox import SuperMoxTestBase
 from testing_support import trial_dir
 
 import gclient_utils
 import subprocess2
 
 
@@ skipping 55 matching lines @@
 
   def testNoLF(self):
     # Exactly as testCheckCallAndFilterAndHeader without trailing \n
     args = ['boo', 'foo', 'bar']
     test_string = 'ahah\naccb\nallo\naddb'
     self._inner(args, test_string)
     self.checkstdout('\n________ running \'boo foo bar\' in \'bleh\'\n'
         'ahah\naccb\nallo\naddb\n'
         '________ running \'boo foo bar\' in \'bleh\'\nahah\naccb\nallo\naddb')
 
+  def _checkKillTimeout(self, output_block_for=0, kill_raises=False):
+    cv = threading.Condition()
+    order = []
+
+    output = list(reversed('output'))
+    def kid_stdout_read(_):
+      if output:
+        return output.pop()
+      else:
+        with cv:
+          order.append('block')
+          cv.wait(timeout=output_block_for)
+          order.append('unblock')
+        return ''
+    def kid_kill():
+      with cv:
+        order.append('killed')
+        cv.notify()  # unblock reader.
+      if kill_raises:
+        raise OSError('somethign went wrong')
+
+    kid = self.ProcessIdMock('')
+    kid.stdout.read = kid_stdout_read
+    kid.kill = kid_kill
+    cwd = 'bleh'
+    args = ['ar', 'gs']
+    gclient_utils.sys.stdout.write(
+        '\n________ running \'ar gs\' in \'bleh\'\noutput')
+    os.getcwd()
+    subprocess2.Popen(
+        args, cwd=cwd,
+        stdout=subprocess2.PIPE,
+        stderr=subprocess2.STDOUT,
+        bufsize=0).AndReturn(kid)
+    self.mox.ReplayAll()
+
+    # This test relies on the testing machine's ability to process 1 char
+    # of output in <0.01 seconds set in kill_timeout.
+    gclient_utils.CheckCallAndFilterAndHeader(
+        args, cwd=cwd, always=True, kill_timeout=0.01)
+    self.checkstdout(
+        '\n________ running \'ar gs\' in \'bleh\'\noutput\n'
+        '________ running \'ar gs\' in \'bleh\'\noutput')
+    return order
+
+  def testKillTimeout(self):
+    order = self._checkKillTimeout(output_block_for=1.0)
+    self.assertEquals(order, ['block', 'killed', 'unblock'])
+
+  def testKillRaise(self):
+    order = self._checkKillTimeout(output_block_for=1.0, kill_raises=True)
+    self.assertEquals(order, ['block', 'killed', 'unblock'])
+
+  def testNoKill(self):
+    order = self._checkKillTimeout(output_block_for=0.0)
+    self.assertEquals(order, ['block', 'unblock'])
+
 
 class SplitUrlRevisionTestCase(GclientUtilBase):
   def testSSHUrl(self):
     url = "ssh://test@example.com/test.git"
     rev = "ac345e52dc"
     out_url, out_rev = gclient_utils.SplitUrlRevision(url)
     self.assertEquals(out_rev, None)
     self.assertEquals(out_url, url)
     out_url, out_rev = gclient_utils.SplitUrlRevision("%s@%s" % (url, rev))
     self.assertEquals(out_rev, rev)
@@ skipping 102 matching lines @@
         ],
         ['VIEW_VC:http://r/s', {'VIEW_VC': 'https://r/s'}],
     ]
     for content, expected in values:
       self.assertEquals(
           expected, gclient_utils.ParseCodereviewSettingsContent(content))
 
 
 if __name__ == '__main__':
   import unittest
+  import logging
+  level = logging.DEBUG if '-v' in sys.argv else logging.FATAL
+  logging.basicConfig(
+      level=level,
+      format='%(asctime).19s %(levelname)s %(filename)s:'
+             '%(lineno)s %(message)s')
   unittest.main()
 
 # vim: ts=2:sw=2:tw=80:et: