Update cbuildbot.py to upload prebuilts from preflight buildbot.

bin/cbuildbot_unittest.py

Index: bin/cbuildbot_unittest.py
diff --git a/bin/cbuildbot_unittest.py b/bin/cbuildbot_unittest.py
index 979cda55d40d2a258ce75a1c1f7ccd2706b0fe10..56796ae955b4ddb6e99ea887ce211b1ff5326521 100755
--- a/bin/cbuildbot_unittest.py
+++ b/bin/cbuildbot_unittest.py
@@ -171,10 +171,13 @@ class CBuildBotTest(mox.MoxTestBase):
     m_file.read().AndReturn(self._test_string)
     m_file.close()
 
+    drop_file = '%s/src/scripts/cbuildbot_package.list' % self._buildroot
     cbuildbot.RunCommand(['./cros_mark_as_stable', '--all',
                      '--board=%s' % self._test_board,
                      '--overlays=%s' % ':'.join(self._chroot_overlays),
-                     '--tracking_branch=cros/master', 'commit'],
+                     '--tracking_branch=cros/master',
+                     '--drop_file=%s' % ReinterpretPathForChroot(drop_file),
+                     'commit'],
                      cwd='%s/src/scripts' % self._buildroot,
                      enter_chroot=True)
 
@@ -194,10 +197,13 @@ class CBuildBotTest(mox.MoxTestBase):
     m_file.read().AndReturn('None')
     m_file.close()
 
+    drop_file = '%s/src/scripts/cbuildbot_package.list' % self._buildroot
     cbuildbot.RunCommand(['./cros_mark_as_stable', '--all',
                          '--board=%s' % self._test_board,
                          '--overlays=%s' % ':'.join(self._chroot_overlays),
-                         '--tracking_branch=cros/master', 'commit'],
+                         '--tracking_branch=cros/master',
+                         '--drop_file=%s' % ReinterpretPathForChroot(drop_file),
+                         'commit'],
                          cwd='%s/src/scripts' % self._buildroot,
                          enter_chroot=True)
 
@@ -207,6 +213,40 @@ class CBuildBotTest(mox.MoxTestBase):
                              self._overlays)
     self.mox.VerifyAll()
 
+  def testGetPortageEnvVar(self):
+    envvar = 'EXAMPLE'
+    cbuildbot.RunCommand(mox.And(mox.IsA(list), mox.In(envvar)),
+                         cwd='%s/src/scripts' % self._buildroot,
+                         redirect_stdout=True, enter_chroot=True,
+                         error_ok=True).AndReturn('RESULT\n')
+    self.mox.ReplayAll()
+    result = cbuildbot._GetPortageEnvVar(self._buildroot, self._test_board,
+                                         envvar)
+    self.mox.VerifyAll()
+    self.assertEqual(result, 'RESULT')
+
+  def testUploadPublicPrebuilts(self):
+    binhost = 'http://www.example.com'
+    binhosts = [binhost, None]
+    check = mox.And(mox.IsA(list), mox.In(binhost), mox.Not(mox.In(None)),
+                    mox.In('gs://chromeos-prebuilt'))
+    cbuildbot.RunCommand(check)
+    self.mox.ReplayAll()
+    cbuildbot._UploadPrebuilts(self._buildroot, self._test_board, 'public',
+                               binhosts)
+    self.mox.VerifyAll()
+
+  def testUploadPrivatePrebuilts(self):
+    binhost = 'http://www.example.com'
+    binhosts = [binhost, None]
+    check = mox.And(mox.IsA(list), mox.In(binhost), mox.Not(mox.In(None)),
+                    mox.In('chromeos-images:/var/www/prebuilt/'))
+    cbuildbot.RunCommand(check)
+    self.mox.ReplayAll()
+    cbuildbot._UploadPrebuilts(self._buildroot, self._test_board, 'private',
+                               binhosts)
+    self.mox.VerifyAll()
+
 
 if __name__ == '__main__':
   unittest.main()