fix scrip_upgrade_master unit test

Signed-off-by: Pierre Tardy <pierre.tardy@intel.com>
buildbot · Jan 24, 2015 · c6b47e4 · c6b47e4
1 parent 5641396
commit c6b47e4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/master/buildbot/test/unit/test_scripts_upgrade_master.py b/master/buildbot/test/unit/test_scripts_upgrade_master.py
@@ -20,6 +20,7 @@
 
 from buildbot import config as config_module
 from buildbot.db import connector
+from buildbot.db import masters
 from buildbot.db import model
 from buildbot.scripts import upgrade_master
 from buildbot.test.util import compat
@@ -274,6 +275,9 @@ def test_upgradeDatabase(self):
         self.patch(connector.DBConnector, 'setup', setup)
         upgrade = mock.Mock(side_effect=lambda **kwargs: defer.succeed(None))
         self.patch(model.Model, 'upgrade', upgrade)
+        setAllMastersActiveLongTimeAgo = mock.Mock(side_effect=lambda **kwargs: defer.succeed(None))
+        self.patch(masters.MastersConnectorComponent,
+                   'setAllMastersActiveLongTimeAgo', setAllMastersActiveLongTimeAgo)
         yield upgrade_master.upgradeDatabase(
             mkconfig(basedir='test', quiet=True),
             config_module.MasterConfig())