Add a git-drover.

tests/git_drover_test.py

+#!/usr/bin/env python
+# Copyright 2015 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.
+"""Tests for git_drover."""
+
+import os
+import shutil
+import subprocess
+import sys
+import tempfile
+import unittest
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+from testing_support import auto_stub
+import git_drover
+
+
+class GitDroverTest(auto_stub.TestCase):
+
+  REPO_CHECK_COMMANDS = [
+      ['git', '--help'],
+      ['git', 'status'],
+      ['git', 'fetch', 'origin'],
+      ['git', 'rev-parse', 'branch-heads/branch'],
+      ['git', 'rev-parse', 'cl'],
+      ['git', 'show', '-s', 'cl'],
+  ]
+  LOCAL_REPO_COMMANDS = [
+      ['git', 'config', 'core.sparsecheckout', 'true'],
+      ['git', 'checkout', '-b', 'drover_branch_123', 'branch-heads/branch'],
+      ['git', 'cherry-pick', '-x', 'cl'],
+      ['git', 'reset', '--hard'],
+  ]
+  UPLOAD_COMMAND = [['git', 'cl', 'upload']]
+  LAND_COMMAND = [['git', 'cl', 'land', '--bypass-hooks']]
+  ABORT_CHERRY_PICK_COMMAND = [['git', 'cherry-pick', '--abort']]
+  BRANCH_CLEANUP_COMMANDS = [
+      ['git', 'checkout', '--detach'],
+      ['git', 'branch', '-D', 'drover_branch_123'],
+  ]
+
+  def setUp(self):
+    super(GitDroverTest, self).setUp()
+    self._temp_directory = tempfile.mkdtemp()
+    self._parent_repo = os.path.join(self._temp_directory, 'parent_repo')
+    self._target_repo = os.path.join(self._temp_directory, 'drover_branch_123')
+    os.makedirs(os.path.join(self._parent_repo, '.git'))
+    with open(os.path.join(self._parent_repo, '.git', 'config'), 'w') as f:
+      f.write('config')
+    with open(os.path.join(self._parent_repo, '.git', 'HEAD'), 'w') as f:
+      f.write('HEAD')
+    self.mock(tempfile, 'mkdtemp', self._mkdtemp)
+    self.mock(__builtins__, 'raw_input', self._get_input)
+    self.mock(subprocess, 'check_call', self._check_call)
+    self._commands = []
+    self._input = []
+    self._fail_on_command = None
+    self._cwd = os.getcwd()
+    self._has_os_symlink = hasattr(os, 'symlink')
+    if not self._has_os_symlink:
+      os.symlink = lambda source, dest: None
+
+  def tearDown(self):
+    os.chdir(self._cwd)
+    shutil.rmtree(self._temp_directory)
+    if not self._has_os_symlink:
+      del os.symlink
+    super(GitDroverTest, self).tearDown()
+
+  def _mkdtemp(self, prefix='tmp'):
+    self.assertEquals('drover_branch_', prefix)
+    os.mkdir(self._target_repo)
+    return self._target_repo
+
+  def _get_input(self, message):
+    result = self._input.pop(0)
+    if result == 'EOF':
+      raise EOFError
+    return result
+
+  def _check_call(self, args, stderr=None, stdout=None, shell=''):
+    self.assertFalse(shell)
+    self._commands.append(args)
+    if (self._fail_on_command is not None and
+        self._fail_on_command == len(self._commands)):
+      raise subprocess.CalledProcessError(1, args[0])
+
+  def testSuccess(self):
+    self._input = ['y', 'y']
+    git_drover.merge_change('branch', 'cl', self._parent_repo, False)
+    self.assertEquals(
+        self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS +
+        self.UPLOAD_COMMAND + self.LAND_COMMAND + self.BRANCH_CLEANUP_COMMANDS,
+        self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testDryRun(self):
+    self._input = ['y']
+    git_drover.merge_change('branch', 'cl', self._parent_repo, True)
+    self.assertEquals(
+        self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS +
+        self.BRANCH_CLEANUP_COMMANDS, self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testCancelEarly(self):
+    self._input = ['n']
+    git_drover.merge_change('branch', 'cl', self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS, self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testEOFOnConfirm(self):
+    self._input = ['EOF']
+    git_drover.merge_change('branch', 'cl', self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS, self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testCancelLate(self):
+    self._input = ['y', 'n']
+    git_drover.merge_change('branch', 'cl', self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS +
+                      self.UPLOAD_COMMAND + self.BRANCH_CLEANUP_COMMANDS,
+                      self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testFailDuringCheck(self):
+    self._input = []
+    self._fail_on_command = 1
+    self.assertRaises(git_drover.Error, git_drover.merge_change, 'branch', 'cl',
+                      self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS[:1], self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testFailDuringBranchCreation(self):
+    self._input = ['y']
+    self._fail_on_command = 8
+    self.assertRaises(git_drover.Error, git_drover.merge_change, 'branch', 'cl',
+                      self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS[:2],
+                      self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testFailDuringCherryPick(self):
+    self._input = ['y']
+    self._fail_on_command = 9
+    self.assertRaises(git_drover.Error, git_drover.merge_change, 'branch', 'cl',
+                      self._parent_repo, False)
+    self.assertEquals(
+        self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS[:3] +
+        self.ABORT_CHERRY_PICK_COMMAND + self.BRANCH_CLEANUP_COMMANDS,
+        self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testFailAfterCherryPick(self):
+    self._input = ['y']
+    self._fail_on_command = 10
+    self.assertRaises(git_drover.Error, git_drover.merge_change, 'branch', 'cl',
+                      self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS +
+                      self.BRANCH_CLEANUP_COMMANDS, self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testFailOnUpload(self):
+    self._input = ['y']
+    self._fail_on_command = 11
+    self.assertRaises(git_drover.Error, git_drover.merge_change, 'branch', 'cl',
+                      self._parent_repo, False)
+    self.assertEquals(self.REPO_CHECK_COMMANDS + self.LOCAL_REPO_COMMANDS +
+                      self.UPLOAD_COMMAND + self.BRANCH_CLEANUP_COMMANDS,
+                      self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+  def testInvalidParentRepoDirectory(self):
+    self.assertRaises(
+        git_drover.Error, git_drover.merge_change, 'branch', 'cl',
+        os.path.join(self._parent_repo, 'fake'), False)
+    self.assertFalse(self._commands)
+    self.assertFalse(os.path.exists(self._target_repo))
+    self.assertFalse(self._input)
+    self.assertEquals(self._cwd, os.getcwd())
+
+
+if __name__ == '__main__':
+  unittest.main()