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 |