Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(384)

Side by Side Diff: bin/cbuildbot_unittest.py

Issue 4969003: Update cbuildbot.py to upload prebuilts from preflight buildbot. (Closed) Base URL: ssh://git@gitrw.chromium.org:9222/crosutils.git@master
Patch Set: Address comments by dianders Created 10 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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 153 matching lines...) Expand 10 before | Expand all | Expand 10 after
164 def testUprevPackages(self): 164 def testUprevPackages(self):
165 """Test if we get actual revisions in revisions.pfq.""" 165 """Test if we get actual revisions in revisions.pfq."""
166 self.mox.StubOutWithMock(__builtin__, 'open') 166 self.mox.StubOutWithMock(__builtin__, 'open')
167 167
168 # Mock out file interaction. 168 # Mock out file interaction.
169 m_file = self.mox.CreateMock(file) 169 m_file = self.mox.CreateMock(file)
170 __builtin__.open(self._revision_file).AndReturn(m_file) 170 __builtin__.open(self._revision_file).AndReturn(m_file)
171 m_file.read().AndReturn(self._test_string) 171 m_file.read().AndReturn(self._test_string)
172 m_file.close() 172 m_file.close()
173 173
174 drop_file = '%s/src/scripts/cbuildbot_package.list' % self._buildroot
sosa 2010/11/23 23:56:21 Can you reference the var in cbuildbot instead?
174 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', 175 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all',
175 '--board=%s' % self._test_board, 176 '--board=%s' % self._test_board,
176 '--overlays=%s' % ':'.join(self._chroot_overlays), 177 '--overlays=%s' % ':'.join(self._chroot_overlays),
177 '--tracking_branch=cros/master', 'commit'], 178 '--tracking_branch=cros/master',
179 '--drop_file=%s' % ReinterpretPathForChroot(drop_file),
180 'commit'],
178 cwd='%s/src/scripts' % self._buildroot, 181 cwd='%s/src/scripts' % self._buildroot,
179 enter_chroot=True) 182 enter_chroot=True)
180 183
181 self.mox.ReplayAll() 184 self.mox.ReplayAll()
182 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch, 185 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch,
183 self._revision_file, self._test_board, 186 self._revision_file, self._test_board,
184 self._overlays) 187 self._overlays)
185 self.mox.VerifyAll() 188 self.mox.VerifyAll()
186 189
187 def testUprevAllPackages(self): 190 def testUprevAllPackages(self):
188 """Test if we get None in revisions.pfq indicating Full Builds.""" 191 """Test if we get None in revisions.pfq indicating Full Builds."""
189 self.mox.StubOutWithMock(__builtin__, 'open') 192 self.mox.StubOutWithMock(__builtin__, 'open')
190 193
191 # Mock out file interaction. 194 # Mock out file interaction.
192 m_file = self.mox.CreateMock(file) 195 m_file = self.mox.CreateMock(file)
193 __builtin__.open(self._revision_file).AndReturn(m_file) 196 __builtin__.open(self._revision_file).AndReturn(m_file)
194 m_file.read().AndReturn('None') 197 m_file.read().AndReturn('None')
195 m_file.close() 198 m_file.close()
196 199
200 drop_file = '%s/src/scripts/cbuildbot_package.list' % self._buildroot
sosa 2010/11/23 23:56:21 Same here
197 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all', 201 cbuildbot.RunCommand(['./cros_mark_as_stable', '--all',
198 '--board=%s' % self._test_board, 202 '--board=%s' % self._test_board,
199 '--overlays=%s' % ':'.join(self._chroot_overlays), 203 '--overlays=%s' % ':'.join(self._chroot_overlays),
200 '--tracking_branch=cros/master', 'commit'], 204 '--tracking_branch=cros/master',
205 '--drop_file=%s' % ReinterpretPathForChroot(drop_file),
206 'commit'],
201 cwd='%s/src/scripts' % self._buildroot, 207 cwd='%s/src/scripts' % self._buildroot,
202 enter_chroot=True) 208 enter_chroot=True)
203 209
204 self.mox.ReplayAll() 210 self.mox.ReplayAll()
205 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch, 211 cbuildbot._UprevPackages(self._buildroot, self.tracking_branch,
206 self._revision_file, self._test_board, 212 self._revision_file, self._test_board,
207 self._overlays) 213 self._overlays)
208 self.mox.VerifyAll() 214 self.mox.VerifyAll()
209 215
216 def testGetPortageEnvVar(self):
sosa 2010/11/23 23:56:21 Doc string line...useful when test fails
davidjames 2010/11/24 00:29:23 Done.
217 envvar = 'EXAMPLE'
218 cbuildbot.RunCommand(mox.And(mox.IsA(list), mox.In(envvar)),
219 cwd='%s/src/scripts' % self._buildroot,
220 redirect_stdout=True, enter_chroot=True,
221 error_ok=True).AndReturn('RESULT\n')
222 self.mox.ReplayAll()
223 result = cbuildbot._GetPortageEnvVar(self._buildroot, self._test_board,
224 envvar)
225 self.mox.VerifyAll()
226 self.assertEqual(result, 'RESULT')
227
228 def testUploadPublicPrebuilts(self):
sosa 2010/11/23 23:56:21 Doc string line...useful when test fails
davidjames 2010/11/24 00:29:23 Done.
229 check = mox.And(mox.IsA(list), mox.In('gs://chromeos-prebuilt'))
230 cbuildbot.RunCommand(check)
231 self.mox.ReplayAll()
232 cbuildbot._UploadPrebuilts(self._buildroot, self._test_board, 'public')
233 self.mox.VerifyAll()
234
235 def testUploadPrivatePrebuilts(self):
sosa 2010/11/23 23:56:21 Doc string line...useful when test fails
davidjames 2010/11/24 00:29:23 Done.
236 check = mox.And(mox.IsA(list), mox.In('chromeos-images:/var/www/prebuilt/'))
237 cbuildbot.RunCommand(check)
238 self.mox.ReplayAll()
239 cbuildbot._UploadPrebuilts(self._buildroot, self._test_board, 'private')
240 self.mox.VerifyAll()
241
210 242
211 if __name__ == '__main__': 243 if __name__ == '__main__':
212 unittest.main() 244 unittest.main()
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698