Add "archive" command to git_cl.py.

tests/git_cl_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.
 
 """Unit tests for git_cl.py."""
 
 import os
 import StringIO
 import stat
@@ skipping 1378 matching lines @@
         ((['git', 'symbolic-ref', 'HEAD'],), 'feature'),
         ((['git', 'config', 'branch.feature.rietveldissue'],), ''),
         ((['git', 'config', 'branch.feature.gerritissue'],), '123'),
         ((['git', 'config', 'branch.feature.gerritserver'],),
          'https://chromium-review.googlesource.com'),
         ((['SetReview', 'chromium-review.googlesource.com', 123,
            {'Commit-Queue': 1}],), ''),
     ]
     # TODO(tandrii): consider testing just set-commit and set-commit --clear,
     # but without copy-pasting tons of expectations, as modifying them later is
-    # super tedious.
+#super tedious.
     self.assertEqual(0, git_cl.main(['set-commit', '-d']))
 
   def test_description_display(self):
     out = StringIO.StringIO()
     self.mock(git_cl.sys, 'stdout', out)
 
     self.mock(git_cl, 'Changelist', ChangelistMock)
     ChangelistMock.desc = 'foo\n'
 
     self.assertEqual(0, git_cl.main(['description', '-d']))
@@ skipping 37 matching lines @@
   def test_description_set_stdin(self):
     out = StringIO.StringIO()
     self.mock(git_cl.sys, 'stdout', out)
 
     self.mock(git_cl, 'Changelist', ChangelistMock)
     self.mock(git_cl.sys, 'stdin', StringIO.StringIO('hi \r\n\t there\n\nman'))
 
     self.assertEqual(0, git_cl.main(['description', '-n', '-']))
     self.assertEqual('hi\n\t there\n\nman', ChangelistMock.desc)
 
+  def test_cleanup(self):
+    self.calls = \
+        [((['git', 'for-each-ref', '--format=%(refname)', 'refs/heads'],),
+          'refs/heads/master\nrefs/heads/foo\nrefs/heads/bar'),
+         ((['git', 'config', 'branch.master.rietveldissue'],), '1'),
+         ((['git', 'config', 'rietveld.autoupdate'],), ''),
+         ((['git', 'config', 'rietveld.server'],), ''),
+         ((['git', 'config', 'rietveld.server'],), ''),
+         ((['git', 'config', 'branch.foo.rietveldissue'],), '456'),
+         ((['git', 'config', 'rietveld.server'],), ''),
+         ((['git', 'config', 'rietveld.server'],), ''),
+         ((['git', 'config', 'branch.bar.rietveldissue'],), ''),
+         ((['git', 'config', 'branch.bar.gerritissue'],), '789'),
+         ((['git', 'symbolic-ref', 'HEAD', '--short'],), 'master'),
+         ((['git', 'tag', 'git-cl-closed-foo-456', 'foo'],), ''),
+         ((['git', 'branch', '-D', 'foo'],), '')]
+
+    class MockChangelist():
+      def __init__(self, branch, issue):
+        self.branch = branch
+        self.issue = issue
+      def GetBranch(self):
+        return self.branch
+      def GetIssue(self):
+        return self.issue
+
+    self.mock(git_cl, 'get_cl_statuses',
+              lambda branches, fine_grained, max_processes:
+              [(MockChangelist('master', 1), 'open'),
+               (MockChangelist('foo', 456), 'closed'),
+               (MockChangelist('bar', 789), 'open')])
+
+    self.assertEqual(0, git_cl.main(['cleanup', '-f']))
+
+  def test_cleanup_current_branch_fails(self):
+    self.calls = \
+        [((['git', 'for-each-ref', '--format=%(refname)', 'refs/heads'],),
+          'refs/heads/master'),
+         ((['git', 'config', 'branch.master.rietveldissue'],), '1'),
+         ((['git', 'config', 'rietveld.autoupdate'],), ''),
+         ((['git', 'config', 'rietveld.server'],), ''),
+         ((['git', 'config', 'rietveld.server'],), ''),
+         ((['git', 'symbolic-ref', 'HEAD', '--short'],), 'master')]
+
+    class MockChangelist():
+      def __init__(self, branch, issue):
+        self.branch = branch
+        self.issue = issue
+      def GetBranch(self):
+        return self.branch
+      def GetIssue(self):
+        return self.issue
+
+    self.mock(git_cl, 'get_cl_statuses',
+              lambda branches, fine_grained, max_processes:
+              [(MockChangelist('master', 1), 'closed')])
+
+    self.assertEqual(1, git_cl.main(['cleanup', '-f']))
 
 if __name__ == '__main__':
   git_cl.logging.basicConfig(
       level=git_cl.logging.DEBUG if '-v' in sys.argv else git_cl.logging.ERROR)
   unittest.main()