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 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
106 def testUprevPackages(self): | 106 def testUprevPackages(self): |
107 """Test if we get actual revisions in revisions.pfq.""" | 107 """Test if we get actual revisions in revisions.pfq.""" |
108 self.mox.StubOutWithMock(__builtin__, 'open') | 108 self.mox.StubOutWithMock(__builtin__, 'open') |
109 | 109 |
110 # Mock out file interaction. | 110 # Mock out file interaction. |
111 m_file = self.mox.CreateMock(file) | 111 m_file = self.mox.CreateMock(file) |
112 __builtin__.open(self._revision_file).AndReturn(m_file) | 112 __builtin__.open(self._revision_file).AndReturn(m_file) |
113 m_file.read().AndReturn(self._test_string) | 113 m_file.read().AndReturn(self._test_string) |
114 m_file.close() | 114 m_file.close() |
115 | 115 |
116 cbuildbot.RunCommand(['./cros_mark_all_as_stable', | 116 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', |
117 '--tracking_branch="cros/master"'], | 117 '--tracking_branch="cros/master"', 'commit'], |
118 cwd='%s/src/scripts' % self._buildroot, | 118 cwd='%s/src/scripts' % self._buildroot, |
119 enter_chroot=True) | 119 enter_chroot=True) |
120 | 120 |
121 self.mox.ReplayAll() | 121 self.mox.ReplayAll() |
122 cbuildbot._UprevPackages(self._buildroot, self._revision_file, | 122 cbuildbot._UprevPackages(self._buildroot, self._revision_file, |
123 self._test_board) | 123 self._test_board) |
124 self.mox.VerifyAll() | 124 self.mox.VerifyAll() |
125 | 125 |
126 def testUprevAllPackages(self): | 126 def testUprevAllPackages(self): |
127 """Test if we get None in revisions.pfq indicating Full Builds.""" | 127 """Test if we get None in revisions.pfq indicating Full Builds.""" |
128 self.mox.StubOutWithMock(__builtin__, 'open') | 128 self.mox.StubOutWithMock(__builtin__, 'open') |
129 | 129 |
130 # Mock out file interaction. | 130 # Mock out file interaction. |
131 m_file = self.mox.CreateMock(file) | 131 m_file = self.mox.CreateMock(file) |
132 __builtin__.open(self._revision_file).AndReturn(m_file) | 132 __builtin__.open(self._revision_file).AndReturn(m_file) |
133 m_file.read().AndReturn('None') | 133 m_file.read().AndReturn('None') |
134 m_file.close() | 134 m_file.close() |
135 | 135 |
136 cbuildbot.RunCommand(['./cros_mark_all_as_stable', | 136 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', |
137 '--tracking_branch="cros/master"'], | 137 '--tracking_branch="cros/master"', 'commit'], |
138 cwd='%s/src/scripts' % self._buildroot, | 138 cwd='%s/src/scripts' % self._buildroot, |
139 enter_chroot=True) | 139 enter_chroot=True) |
140 | 140 |
141 self.mox.ReplayAll() | 141 self.mox.ReplayAll() |
142 cbuildbot._UprevPackages(self._buildroot, self._revision_file, | 142 cbuildbot._UprevPackages(self._buildroot, self._revision_file, |
143 self._test_board) | 143 self._test_board) |
144 self.mox.VerifyAll() | 144 self.mox.VerifyAll() |
145 | 145 |
146 | 146 |
147 if __name__ == '__main__': | 147 if __name__ == '__main__': |
148 unittest.main() | 148 unittest.main() |
OLD | NEW |