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 a3230bd..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,6 +5,9 @@
 
 """Unittests for chrome stages."""
 
+from __future__ import print_function
+
+import mox
 import os
 import sys
 
@@ -14,6 +17,7 @@ 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
@@ -32,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):