Revert "Have the ability for the PFQ to both rev Chrome and ..."

cros_mark_as_stable_unittest.py

 #!/usr/bin/python
 
 # Copyright (c) 2010 The Chromium OS 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 cros_mark_as_stable.py."""
 
 import fileinput
 import mox
 import os
 import sys
 import unittest
 
 import cros_mark_as_stable
 
 class NonClassTests(mox.MoxTestBase):
   def setUp(self):
     mox.MoxTestBase.setUp(self)
     self.mox.StubOutWithMock(cros_mark_as_stable, '_SimpleRunCommand')
     self._branch = 'test_branch'
     self._tracking_branch = 'cros/test'
 
   def testPushChange(self):
     git_log = 'Marking test_one as stable\nMarking test_two as stable\n'
     fake_description = 'Marking set of ebuilds as stable\n\n%s' % git_log
-    self.mox.StubOutWithMock(cros_mark_as_stable, '_DoWeHaveLocalCommits')
+    self.mox.StubOutWithMock(cros_mark_as_stable, '_CheckOnStabilizingBranch')
     self.mox.StubOutWithMock(cros_mark_as_stable.GitBranch, 'CreateBranch')
     self.mox.StubOutWithMock(cros_mark_as_stable.GitBranch, 'Exists')
 
-    cros_mark_as_stable._DoWeHaveLocalCommits(
-        self._branch, self._tracking_branch).AndReturn(True)
+    cros_mark_as_stable._CheckOnStabilizingBranch(self._branch).AndReturn(True)
     cros_mark_as_stable.GitBranch.CreateBranch()
     cros_mark_as_stable.GitBranch.Exists().AndReturn(True)
     cros_mark_as_stable._SimpleRunCommand('git log --format=format:%s%n%n%b ' +
                           self._tracking_branch + '..').AndReturn(git_log)
     cros_mark_as_stable._SimpleRunCommand('repo sync .')
     cros_mark_as_stable._SimpleRunCommand('git merge --squash %s' %
                                           self._branch)
     cros_mark_as_stable._SimpleRunCommand('git commit -m "%s"' %
                                           fake_description)
     cros_mark_as_stable._SimpleRunCommand('git config push.default tracking')
     cros_mark_as_stable._SimpleRunCommand('git push')
     self.mox.ReplayAll()
     cros_mark_as_stable.PushChange(self._branch, self._tracking_branch)
     self.mox.VerifyAll()
 
 
 class GitBranchTest(mox.MoxTestBase):
 
   def setUp(self):
     mox.MoxTestBase.setUp(self)
     # Always stub RunCommmand out as we use it in every method.
     self.mox.StubOutWithMock(cros_mark_as_stable, '_SimpleRunCommand')
-    self._branch = self.mox.CreateMock(cros_mark_as_stable.GitBranch)
-    self._branch_name = 'test_branch'
-    self._branch.branch_name = self._branch_name
+    self._branch = 'test_branch'
     self._tracking_branch = 'cros/test'
-    self._branch.tracking_branch = self._tracking_branch
+
+  def testCreateBranchNoPrevious(self):
+    # Test init with no previous branch existing.
+    branch = cros_mark_as_stable.GitBranch(self._branch, self._tracking_branch)
+    self.mox.StubOutWithMock(branch, 'Exists')
+    self.mox.StubOutWithMock(branch, '_Checkout')
+    branch.Exists().AndReturn(False)
+    branch._Checkout(self._branch)
+    self.mox.ReplayAll()
+    branch.CreateBranch()
+    self.mox.VerifyAll()
+
+  def testCreateBranchWithPrevious(self):
+    # Test init with previous branch existing.
+    branch = cros_mark_as_stable.GitBranch(self._branch, self._tracking_branch)
+    self.mox.StubOutWithMock(branch, 'Exists')
+    self.mox.StubOutWithMock(branch, 'Delete')
+    self.mox.StubOutWithMock(branch, '_Checkout')
+    branch.Exists().AndReturn(True)
+    branch.Delete()
+    branch._Checkout(self._branch)
+    self.mox.ReplayAll()
+    branch.CreateBranch()
+    self.mox.VerifyAll()
 
   def testCheckoutCreate(self):
     # Test init with no previous branch existing.
-    self._branch.Exists().AndReturn(False)
     cros_mark_as_stable._SimpleRunCommand(
-        'git checkout -b %s %s' % (self._branch_name, self._tracking_branch))
+        'git checkout -b %s %s' % (self._branch, self._tracking_branch))
     self.mox.ReplayAll()
-    cros_mark_as_stable.GitBranch.Checkout(self._branch)
+    branch = cros_mark_as_stable.GitBranch(self._branch, self._tracking_branch)
+    branch._Checkout(self._branch)
     self.mox.VerifyAll()
 
   def testCheckoutNoCreate(self):
     # Test init with previous branch existing.
-    self._branch.Exists().AndReturn(True)
     cros_mark_as_stable._SimpleRunCommand('git checkout %s' % (
-        self._branch_name))
+        self._tracking_branch))
     self.mox.ReplayAll()
-    cros_mark_as_stable.GitBranch.Checkout(self._branch)
+    branch = cros_mark_as_stable.GitBranch(self._branch, self._tracking_branch)
+    branch._Checkout(self._tracking_branch, False)
     self.mox.VerifyAll()
 
   def testDelete(self):
-    self.mox.StubOutWithMock(cros_mark_as_stable.GitBranch, 'Checkout')
-    branch = cros_mark_as_stable.GitBranch(self._branch_name,
-                                           self._tracking_branch)
-    cros_mark_as_stable.GitBranch.Checkout(mox.IgnoreArg())
-    cros_mark_as_stable._SimpleRunCommand('git branch -D ' + self._branch_name)
+    branch = cros_mark_as_stable.GitBranch(self._branch, self._tracking_branch)
+    self.mox.StubOutWithMock(branch, '_Checkout')
+    branch._Checkout(self._tracking_branch, create=False)
+    cros_mark_as_stable._SimpleRunCommand('git branch -D ' + self._branch)
     self.mox.ReplayAll()
     branch.Delete()
     self.mox.VerifyAll()
 
   def testExists(self):
-    branch = cros_mark_as_stable.GitBranch(self._branch_name,
-                                           self._tracking_branch)
+    branch = cros_mark_as_stable.GitBranch(self._branch, self._tracking_branch)
+
     # Test if branch exists that is created
     cros_mark_as_stable._SimpleRunCommand('git branch').AndReturn(
-        '%s' % self._branch_name)
+        '%s %s' % (self._branch, self._tracking_branch))
     self.mox.ReplayAll()
     self.assertTrue(branch.Exists())
     self.mox.VerifyAll()
 
 
 class EBuildTest(mox.MoxTestBase):
 
   def setUp(self):
     mox.MoxTestBase.setUp(self)
 
@@ skipping 199 matching lines @@
     cros_mark_as_stable._FindUprevCandidates([]).AndReturn(package)
     self.mox.ReplayAll()
     cros_mark_as_stable._BuildEBuildDictionary(overlays, False, [])
     self.assertEquals(len(overlays), 1)
     self.assertEquals(overlays["/overlay"], [])
     self.mox.VerifyAll()
 
 
 if __name__ == '__main__':
   unittest.main()