Part of http://patchwork.test.kernel.org/patch/1850/ didn't make it into the
repository. Trying again.
git-svn-id: http://test.kernel.org/svn/autotest/trunk@4278 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/scheduler/metahost_scheduler_unittest.py b/scheduler/metahost_scheduler_unittest.py
index 5933769..c2ec728 100644
--- a/scheduler/metahost_scheduler_unittest.py
+++ b/scheduler/metahost_scheduler_unittest.py
@@ -3,7 +3,9 @@
import common
import unittest
from autotest_lib.client.common_lib.test_utils import mock
-from autotest_lib.scheduler import metahost_scheduler, monitor_db
+from autotest_lib.frontend import setup_django_environment
+from autotest_lib.frontend import setup_test_environment
+from autotest_lib.scheduler import metahost_scheduler, scheduler_models
class LabelMetahostSchedulerTest(unittest.TestCase):
def setUp(self):
@@ -13,8 +15,13 @@
self.metahost_scheduler = metahost_scheduler.LabelMetahostScheduler()
+ def tearDown(self):
+ self.god.unstub_all()
+
+
def entry(self):
- return self.god.create_mock_class(monitor_db.HostQueueEntry, 'entry')
+ return self.god.create_mock_class(scheduler_models.HostQueueEntry,
+ 'entry')
def test_can_schedule_metahost(self):