| OLD | NEW |
| 1 #!/usr/bin/python | 1 #!/usr/bin/python |
| 2 | 2 |
| 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 3 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| 4 # Use of this source code is governed by a BSD-style license that can be | 4 # Use of this source code is governed by a BSD-style license that can be |
| 5 # found in the LICENSE file. | 5 # found in the LICENSE file. |
| 6 | 6 |
| 7 """Unittests for cbuildbot. Needs to be run inside of chroot for mox.""" | 7 """Unittests for cbuildbot. Needs to be run inside of chroot for mox.""" |
| 8 | 8 |
| 9 import __builtin__ | 9 import __builtin__ |
| 10 import mox | 10 import mox |
| (...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 159 def testUprevPackages(self): | 159 def testUprevPackages(self): |
| 160 """Test if we get actual revisions in revisions.pfq.""" | 160 """Test if we get actual revisions in revisions.pfq.""" |
| 161 self.mox.StubOutWithMock(__builtin__, 'open') | 161 self.mox.StubOutWithMock(__builtin__, 'open') |
| 162 | 162 |
| 163 # Mock out file interaction. | 163 # Mock out file interaction. |
| 164 m_file = self.mox.CreateMock(file) | 164 m_file = self.mox.CreateMock(file) |
| 165 __builtin__.open(self._revision_file).AndReturn(m_file) | 165 __builtin__.open(self._revision_file).AndReturn(m_file) |
| 166 m_file.read().AndReturn(self._test_string) | 166 m_file.read().AndReturn(self._test_string) |
| 167 m_file.close() | 167 m_file.close() |
| 168 | 168 |
| 169 overlays = [ '%s/src/third_party/chromiumos-overlay' % self._buildroot ] | |
| 170 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', | 169 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', |
| 171 '--board=%s' % self._test_board, | 170 '--board=%s' % self._test_board, |
| 172 '--overlays=%s' % ':'.join(overlays), | 171 '--tracking_branch="cros/master"', 'commit'], |
| 173 '--tracking_branch=cros/master', 'commit'], | |
| 174 cwd='%s/src/scripts' % self._buildroot, | 172 cwd='%s/src/scripts' % self._buildroot, |
| 175 enter_chroot=True) | 173 enter_chroot=True) |
| 176 | 174 |
| 177 self.mox.ReplayAll() | 175 self.mox.ReplayAll() |
| 178 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch, | 176 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch, |
| 179 self._revision_file, self._test_board, overlays) | 177 self._revision_file, self._test_board) |
| 180 self.mox.VerifyAll() | 178 self.mox.VerifyAll() |
| 181 | 179 |
| 182 def testUprevAllPackages(self): | 180 def testUprevAllPackages(self): |
| 183 """Test if we get None in revisions.pfq indicating Full Builds.""" | 181 """Test if we get None in revisions.pfq indicating Full Builds.""" |
| 184 self.mox.StubOutWithMock(__builtin__, 'open') | 182 self.mox.StubOutWithMock(__builtin__, 'open') |
| 185 | 183 |
| 186 # Mock out file interaction. | 184 # Mock out file interaction. |
| 187 m_file = self.mox.CreateMock(file) | 185 m_file = self.mox.CreateMock(file) |
| 188 __builtin__.open(self._revision_file).AndReturn(m_file) | 186 __builtin__.open(self._revision_file).AndReturn(m_file) |
| 189 m_file.read().AndReturn('None') | 187 m_file.read().AndReturn('None') |
| 190 m_file.close() | 188 m_file.close() |
| 191 | 189 |
| 192 overlays = [ '%s/src/third_party/chromiumos-overlay' % self._buildroot ] | |
| 193 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', | 190 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', |
| 194 '--board=%s' % self._test_board, | 191 '--board=%s' % self._test_board, |
| 195 '--overlays=%s' % ':'.join(overlays), | 192 '--tracking_branch="cros/master"', 'commit'], |
| 196 '--tracking_branch=cros/master', 'commit'], | |
| 197 cwd='%s/src/scripts' % self._buildroot, | 193 cwd='%s/src/scripts' % self._buildroot, |
| 198 enter_chroot=True) | 194 enter_chroot=True) |
| 199 | 195 |
| 200 self.mox.ReplayAll() | 196 self.mox.ReplayAll() |
| 201 overlays = [ '%s/src/third_party/chromiumos-overlay' % self._buildroot ] | |
| 202 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch, | 197 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch, |
| 203 self._revision_file, self._test_board, overlays) | 198 self._revision_file, self._test_board) |
| 204 self.mox.VerifyAll() | 199 self.mox.VerifyAll() |
| 205 | 200 |
| 206 | 201 |
| 207 if __name__ == '__main__': | 202 if __name__ == '__main__': |
| 208 unittest.main() | 203 unittest.main() |
| OLD | NEW |