Upstream version 10.39.225.0

[platform/framework/web/crosswalk.git] / src / third_party / chromite / cbuildbot / stages / chrome_stages_unittest.py

diff --git a/src/third_party/chromite/cbuildbot/stages/chrome_stages_unittest.py b/src/third_party/chromite/cbuildbot/stages/chrome_stages_unittest.py
index c2502fa..424fd22 100755 (executable)
--- a/src/third_party/chromite/cbuildbot/stages/chrome_stages_unittest.py
+++ b/src/third_party/chromite/cbuildbot/stages/chrome_stages_unittest.py
@@ -5,22 +5,25 @@
 
 """Unittests for chrome stages."""
 
+from __future__ import print_function
+
+import mox
 import os
 import sys
 
 sys.path.insert(0, os.path.abspath('%s/../../..' % os.path.dirname(__file__)))
 from chromite.cbuildbot import commands
 from chromite.cbuildbot import constants
+from chromite.cbuildbot.cbuildbot_unittest import BuilderRunMock
 from chromite.cbuildbot.stages import chrome_stages
 from chromite.cbuildbot.stages import generic_stages_unittest
+from chromite.lib import cidb
 from chromite.lib import cros_build_lib
 from chromite.lib import cros_build_lib_unittest
 from chromite.lib import cros_test_lib
 from chromite.lib import osutils
 from chromite.lib import parallel_unittest
 
-from chromite.cbuildbot.stages.generic_stages_unittest import BuilderRunMock
-
 
 # pylint: disable=R0901,W0212
 class ChromeSDKStageTest(generic_stages_unittest.AbstractStageTest,
@@ -33,6 +36,12 @@ class ChromeSDKStageTest(generic_stages_unittest.AbstractStageTest,
     self.StartPatcher(BuilderRunMock())
     self.StartPatcher(parallel_unittest.ParallelMock())
 
+    # Set up a general purpose cidb mock. Tests with more specific
+    # mock requirements can replace this with a separate call to
+    # SetupMockCidb
+    mock_cidb = mox.MockObject(cidb.CIDBConnection)
+    cidb.CIDBConnectionFactory.SetupMockCidb(mock_cidb)
+
     self._Prepare()
 
   def _Prepare(self, bot_id=None, **kwargs):