CBuildbot - Adds ability to only packages given by the sourcestamp.

bin/cbuildbot_unittest.py

Index: bin/cbuildbot_unittest.py
diff --git a/bin/cbuildbot_unittest.py b/bin/cbuildbot_unittest.py
new file mode 100755
index 0000000000000000000000000000000000000000..eb5313cb827454db7ac4e142d00b6875bff35511
--- /dev/null
+++ b/bin/cbuildbot_unittest.py
@@ -0,0 +1,100 @@
+#!/usr/bin/python
+
+# Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import __builtin__
+import mox
+import unittest
+
+import cbuildbot
+
+class CBuildBotTest(mox.MoxTestBase):
+
+  def setUp(self):
+    mox.MoxTestBase.setUp(self)
+    # Always stub RunCommmand out as we use it in every method.
+    self.mox.StubOutWithMock(cbuildbot, 'RunCommand')
+    self._test_repos = [['kernel', 'third_party/kernel/files'],
+                        ['login_manager', 'platform/login_manager']
+                       ]
+    self._test_cros_workon_packages = \
+      'chromeos-base/kernel\nchromeos-base/chromeos-login\n'
+    self._test_board = 'test-board'
+    self._buildroot = '.'
+    self._test_dict = {'kernel' : ['chromos-base/kernel', 'dev-util/perf'],
+                       'cros' : ['chromos-base/libcros']
+                      }
+    self._test_string = "kernel.git@12345test cros.git@12333test"
+    self._revision_file = 'test-revisions.pfq'
+    self._test_parsed_string_array = [
+                                      ['chromeos-base/kernel', '12345test'],
+                                      ['dev-util/perf', '12345test'],
+                                      ['chromos-base/libcros', '12345test']
+                                     ]
+
+  def testParseRevisionString(self):
+    """Test whether _ParseRevisionString parses string correctly."""
+    return_array = cbuildbot._ParseRevisionString(self._test_string,
+                                                  self._test_dict)
+    self.assertEqual(len(return_array), 3)
+    self.assertTrue(
+      'chromeos-base/kernel', '12345test' in return_array)
+    self.assertTrue(
+      'dev-util/perf', '12345test' in return_array)
+    self.assertTrue(
+      'chromos-base/libcros', '12345test' in return_array)
+
+  def testCreateDictionary(self):
+    self.mox.StubOutWithMock(cbuildbot, '_GetAllGitRepos')
+    self.mox.StubOutWithMock(cbuildbot, '_GetCrosWorkOnSrcPath')
+    cbuildbot._GetAllGitRepos(mox.IgnoreArg()).AndReturn(self._test_repos)
+    cbuildbot.RunCommand(mox.IgnoreArg(),
+                         cwd='%s/src/scripts' % self._buildroot,
+                         redirect_stdout=True,
+                         redirect_stderr=True,
+                         enter_chroot=True,
+                         print_cmd=False).AndReturn(
+                             self._test_cros_workon_packages)
+    cbuildbot._GetCrosWorkOnSrcPath(self._buildroot, self._test_board,
+                                    'chromeos-base/kernel').AndReturn(
+        '/home/test/third_party/kernel/files')
+    cbuildbot._GetCrosWorkOnSrcPath(self._buildroot, self._test_board,
+                                    'chromeos-base/chromeos-login').AndReturn(
+        '/home/test/platform/login_manager')
+    self.mox.ReplayAll()
+    repo_dict = cbuildbot._CreateRepoDictionary(self._buildroot,
+                                                self._test_board)
+    self.assertEqual(repo_dict['kernel'], ['chromeos-base/kernel'])
+    self.assertEqual(repo_dict['login_manager'],
+                     ['chromeos-base/chromeos-login'])
+    self.mox.VerifyAll()
+
+  def testUprevPackages(self):
+    self.mox.StubOutWithMock(cbuildbot, '_CreateRepoDictionary')
+    self.mox.StubOutWithMock(cbuildbot, '_ParseRevisionString')
+    self.mox.StubOutWithMock(cbuildbot, '_UprevFromRevisionList')
+    self.mox.StubOutWithMock(__builtin__, 'open')
+
+    # Mock out file interaction.
+    m_file = self.mox.CreateMock(file)
+    __builtin__.open(self._revision_file).AndReturn(m_file)
+    m_file.read().AndReturn(self._test_string)
+    m_file.close()
+
+    cbuildbot._CreateRepoDictionary(self._buildroot,
+                                    self._test_board).AndReturn(self._test_dict)
+    cbuildbot._ParseRevisionString(self._test_string,
+                                   self._test_dict).AndReturn(
+                                       self._test_parsed_string_array)
+    cbuildbot._UprevFromRevisionList(self._buildroot,
+                                     self._test_parsed_string_array)
+    self.mox.ReplayAll()
+    cbuildbot._UprevPackages(self._buildroot, self._revision_file,
+                             self._test_board)
+    self.mox.VerifyAll()
+
+
+if __name__ == '__main__':
+  unittest.main()