Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 1 | #!/usr/bin/python |
Mike Frysinger | 6cb624a | 2012-05-24 18:17:38 -0400 | [diff] [blame] | 2 | # Copyright (c) 2012 The Chromium OS Authors. All rights reserved. |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 3 | # Use of this source code is governed by a BSD-style license that can be |
| 4 | # found in the LICENSE file. |
| 5 | |
| 6 | """Unit tests for cros_mark_as_stable.py.""" |
| 7 | |
Mike Frysinger | 383367e | 2014-09-16 15:06:17 -0400 | [diff] [blame] | 8 | from __future__ import print_function |
| 9 | |
Mike Frysinger | 6cb624a | 2012-05-24 18:17:38 -0400 | [diff] [blame] | 10 | import os |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 11 | import sys |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 12 | |
Mike Frysinger | 6cb624a | 2012-05-24 18:17:38 -0400 | [diff] [blame] | 13 | sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(__file__)), |
| 14 | '..', '..')) |
J. Richard Barnette | d422f62 | 2011-11-17 09:39:46 -0800 | [diff] [blame] | 15 | from chromite.lib import cros_build_lib |
Mike Frysinger | de5ab0e | 2013-03-21 20:48:36 -0400 | [diff] [blame] | 16 | from chromite.lib import cros_build_lib_unittest |
Brian Harring | c92788f | 2012-09-21 18:07:15 -0700 | [diff] [blame] | 17 | from chromite.lib import cros_test_lib |
David James | 97d9587 | 2012-11-16 15:09:56 -0800 | [diff] [blame] | 18 | from chromite.lib import git |
David James | 59a0a2b | 2013-03-22 14:04:44 -0700 | [diff] [blame] | 19 | from chromite.lib import osutils |
Mike Frysinger | de5ab0e | 2013-03-21 20:48:36 -0400 | [diff] [blame] | 20 | from chromite.lib import parallel_unittest |
| 21 | from chromite.lib import partial_mock |
Mike Frysinger | 6cb624a | 2012-05-24 18:17:38 -0400 | [diff] [blame] | 22 | from chromite.scripts import cros_mark_as_stable |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 23 | |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 24 | # TODO(build): Finish test wrapper (http://crosbug.com/37517). |
| 25 | # Until then, this has to be after the chromite imports. |
| 26 | import mock |
Chris Sosa | 62ad852 | 2011-03-08 17:46:17 -0800 | [diff] [blame] | 27 | |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 28 | |
| 29 | # pylint: disable=W0212 |
| 30 | |
| 31 | |
Brian Harring | c92788f | 2012-09-21 18:07:15 -0700 | [diff] [blame] | 32 | class NonClassTests(cros_test_lib.MoxTestCase): |
Don Garrett | 25f309a | 2014-03-19 14:02:12 -0700 | [diff] [blame] | 33 | """Test the flow for pushing a change.""" |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 34 | def setUp(self): |
David James | 3a37309 | 2011-11-18 15:56:31 -0800 | [diff] [blame] | 35 | self.mox.StubOutWithMock(cros_build_lib, 'RunCommand') |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 36 | self._branch = 'test_branch' |
Brian Harring | eb23793 | 2012-05-07 02:08:06 -0700 | [diff] [blame] | 37 | self._target_manifest_branch = 'cros/master' |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 38 | |
Matt Tennant | cb52205 | 2013-11-25 14:23:43 -0800 | [diff] [blame] | 39 | def _TestPushChange(self, bad_cls): |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 40 | git_log = 'Marking test_one as stable\nMarking test_two as stable\n' |
| 41 | fake_description = 'Marking set of ebuilds as stable\n\n%s' % git_log |
| 42 | self.mox.StubOutWithMock(cros_mark_as_stable, '_DoWeHaveLocalCommits') |
| 43 | self.mox.StubOutWithMock(cros_mark_as_stable.GitBranch, 'CreateBranch') |
| 44 | self.mox.StubOutWithMock(cros_mark_as_stable.GitBranch, 'Exists') |
David James | 97d9587 | 2012-11-16 15:09:56 -0800 | [diff] [blame] | 45 | self.mox.StubOutWithMock(git, 'PushWithRetry') |
| 46 | self.mox.StubOutWithMock(git, 'GetTrackingBranch') |
| 47 | self.mox.StubOutWithMock(git, 'SyncPushBranch') |
| 48 | self.mox.StubOutWithMock(git, 'CreatePushBranch') |
| 49 | self.mox.StubOutWithMock(git, 'RunGit') |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 50 | |
Matt Tennant | cb52205 | 2013-11-25 14:23:43 -0800 | [diff] [blame] | 51 | # Run the flow. |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 52 | cros_mark_as_stable._DoWeHaveLocalCommits( |
Mike Frysinger | 2ebe373 | 2012-05-08 17:04:12 -0400 | [diff] [blame] | 53 | self._branch, self._target_manifest_branch, '.').AndReturn(True) |
David James | 97d9587 | 2012-11-16 15:09:56 -0800 | [diff] [blame] | 54 | git.GetTrackingBranch('.', for_push=True).AndReturn( |
Brian Harring | 609dc4e | 2012-05-07 02:17:44 -0700 | [diff] [blame] | 55 | ['gerrit', 'refs/remotes/gerrit/master']) |
David James | 97d9587 | 2012-11-16 15:09:56 -0800 | [diff] [blame] | 56 | git.SyncPushBranch('.', 'gerrit', 'refs/remotes/gerrit/master') |
David James | 6600946 | 2012-03-25 10:08:38 -0700 | [diff] [blame] | 57 | cros_mark_as_stable._DoWeHaveLocalCommits( |
Brian Harring | 609dc4e | 2012-05-07 02:17:44 -0700 | [diff] [blame] | 58 | self._branch, 'refs/remotes/gerrit/master', '.').AndReturn(True) |
Matt Tennant | cb52205 | 2013-11-25 14:23:43 -0800 | [diff] [blame] | 59 | |
| 60 | # Look for bad CLs. |
| 61 | cmd = ['log', '--format=short', '--perl-regexp', '--author', |
| 62 | '^(?!chrome-bot)', 'refs/remotes/gerrit/master..%s' % self._branch] |
| 63 | |
| 64 | if bad_cls: |
| 65 | result = cros_build_lib.CommandResult(output='Found bad stuff') |
| 66 | git.RunGit('.', cmd).AndReturn(result) |
| 67 | else: |
| 68 | result = cros_build_lib.CommandResult(output='\n') |
| 69 | git.RunGit('.', cmd).AndReturn(result) |
| 70 | result = cros_build_lib.CommandResult(output=git_log) |
| 71 | cmd = ['log', '--format=format:%s%n%n%b', |
| 72 | 'refs/remotes/gerrit/master..%s' % self._branch] |
| 73 | git.RunGit('.', cmd).AndReturn(result) |
| 74 | git.CreatePushBranch('merge_branch', '.') |
| 75 | git.RunGit('.', ['merge', '--squash', self._branch]) |
| 76 | git.RunGit('.', ['commit', '-m', fake_description]) |
| 77 | git.RunGit('.', ['config', 'push.default', 'tracking']) |
| 78 | git.PushWithRetry('merge_branch', '.', dryrun=False) |
| 79 | |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 80 | self.mox.ReplayAll() |
Mike Frysinger | 2ebe373 | 2012-05-08 17:04:12 -0400 | [diff] [blame] | 81 | cros_mark_as_stable.PushChange(self._branch, self._target_manifest_branch, |
| 82 | False, '.') |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 83 | self.mox.VerifyAll() |
| 84 | |
Matt Tennant | cb52205 | 2013-11-25 14:23:43 -0800 | [diff] [blame] | 85 | def testPushChange(self): |
| 86 | self._TestPushChange(bad_cls=False) |
| 87 | |
| 88 | def testPushChangeBadCls(self): |
| 89 | self.assertRaises(AssertionError, self._TestPushChange, bad_cls=True) |
| 90 | |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 91 | |
Mike Frysinger | de5ab0e | 2013-03-21 20:48:36 -0400 | [diff] [blame] | 92 | class CleanStalePackagesTest(cros_build_lib_unittest.RunCommandTestCase): |
Don Garrett | 25f309a | 2014-03-19 14:02:12 -0700 | [diff] [blame] | 93 | """Tests for cros_mark_as_stable.CleanStalePackages.""" |
Mike Frysinger | de5ab0e | 2013-03-21 20:48:36 -0400 | [diff] [blame] | 94 | |
David James | 59a0a2b | 2013-03-22 14:04:44 -0700 | [diff] [blame] | 95 | def setUp(self): |
| 96 | self.PatchObject(osutils, 'FindMissingBinaries', return_value=[]) |
| 97 | |
Mike Frysinger | de5ab0e | 2013-03-21 20:48:36 -0400 | [diff] [blame] | 98 | def testNormalClean(self): |
| 99 | """Clean up boards/packages with normal success""" |
| 100 | cros_mark_as_stable.CleanStalePackages(('board1', 'board2'), ['cow', 'car']) |
| 101 | |
| 102 | def testNothingToUnmerge(self): |
| 103 | """Clean up packages that don't exist (portage will exit 1)""" |
| 104 | self.rc.AddCmdResult(partial_mock.In('emerge'), returncode=1) |
| 105 | cros_mark_as_stable.CleanStalePackages((), ['no/pkg']) |
| 106 | |
| 107 | def testUnmergeError(self): |
| 108 | """Make sure random exit errors are not ignored""" |
| 109 | self.rc.AddCmdResult(partial_mock.In('emerge'), returncode=123) |
| 110 | with parallel_unittest.ParallelMock(): |
| 111 | self.assertRaises(cros_build_lib.RunCommandError, |
David James | 59a0a2b | 2013-03-22 14:04:44 -0700 | [diff] [blame] | 112 | cros_mark_as_stable.CleanStalePackages, |
| 113 | (), ['no/pkg']) |
Mike Frysinger | de5ab0e | 2013-03-21 20:48:36 -0400 | [diff] [blame] | 114 | |
| 115 | |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 116 | class GitBranchTest(cros_test_lib.MockTestCase): |
Don Garrett | 25f309a | 2014-03-19 14:02:12 -0700 | [diff] [blame] | 117 | """Tests for cros_mark_as_stable.GitBranch.""" |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 118 | |
| 119 | def setUp(self): |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 120 | # Always stub RunCommmand out as we use it in every method. |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 121 | self.rc_mock = self.PatchObject(cros_build_lib, 'RunCommand') |
| 122 | |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 123 | self._branch_name = 'test_branch' |
Brian Harring | eb23793 | 2012-05-07 02:08:06 -0700 | [diff] [blame] | 124 | self._target_manifest_branch = 'cros/test' |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 125 | self._branch = cros_mark_as_stable.GitBranch( |
| 126 | branch_name=self._branch_name, |
| 127 | tracking_branch=self._target_manifest_branch, |
| 128 | cwd='.') |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 129 | |
| 130 | def testCheckoutCreate(self): |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 131 | """Test init with no previous branch existing.""" |
| 132 | self.PatchObject(self._branch, 'Exists', return_value=False) |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 133 | cros_mark_as_stable.GitBranch.Checkout(self._branch) |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 134 | self.rc_mock.assert_call(mock.call( |
| 135 | ['repo', 'start', self._branch_name, '.'], |
| 136 | print_cmd=False, cwd='.', capture_output=True)) |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 137 | |
| 138 | def testCheckoutNoCreate(self): |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 139 | """Test init with previous branch existing.""" |
| 140 | self.PatchObject(self._branch, 'Exists', return_value=True) |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 141 | cros_mark_as_stable.GitBranch.Checkout(self._branch) |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 142 | self.rc_mock.assert_call(mock.call( |
| 143 | ['git', 'checkout', '-f', self._branch_name], |
| 144 | print_cmd=False, cwd='.', capture_output=True)) |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 145 | |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 146 | def testExists(self): |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 147 | """Test if branch exists that is created.""" |
Mike Frysinger | 7dafd0e | 2012-05-08 15:47:16 -0400 | [diff] [blame] | 148 | result = cros_build_lib.CommandResult(output=self._branch_name + '\n') |
Mike Frysinger | ef6ccb2 | 2014-11-07 14:59:48 -0500 | [diff] [blame^] | 149 | self.PatchObject(git, 'RunGit', return_value=result) |
| 150 | self.assertTrue(self._branch.Exists()) |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 151 | |
| 152 | |
Chris Sosa | dad0d32 | 2011-01-31 16:37:33 -0800 | [diff] [blame] | 153 | if __name__ == '__main__': |
Brian Harring | c92788f | 2012-09-21 18:07:15 -0700 | [diff] [blame] | 154 | cros_test_lib.main() |