make AgentTasksTest inherit from BaseSchedulerTest. it didn't used to, since it didn't have any DB dependencies, but the recent introduction of SpecialTasks has changed that, so we need AgentTasksTest to setup the DB now like everything else. It doesn't increase the unit test runtime too drastically.
Signed-off-by: Steve Howard <showard@google.com>
Autotest@test.kernel.org
http://test.kernel.org/cgi-bin/mailman/listinfo/autotest
git-svn-id: http://test.kernel.org/svn/autotest/trunk@3184 592f7852-d20e-0410-864c-8624ca9c26a4
diff --git a/scheduler/monitor_db_unittest.py b/scheduler/monitor_db_unittest.py
index 3bec0b3..5c754c0 100644
--- a/scheduler/monitor_db_unittest.py
+++ b/scheduler/monitor_db_unittest.py
@@ -1190,7 +1190,7 @@
self._test_agent_abort_before_started_helper(True)
-class AgentTasksTest(unittest.TestCase):
+class AgentTasksTest(BaseSchedulerTest):
TEMP_DIR = '/abspath/tempdir'
RESULTS_DIR = '/results/dir'
HOSTNAME = 'myhost'
@@ -1202,6 +1202,7 @@
JOB_AUTOSERV_PARAMS = set(['-u', JOB_OWNER, '-l', JOB_NAME])
def setUp(self):
+ super(AgentTasksTest, self).setUp()
self.god = mock.mock_god()
self.god.stub_with(drone_manager.DroneManager, 'get_temporary_path',
mock.mock_function('get_temporary_path',
@@ -1243,6 +1244,7 @@
def tearDown(self):
+ super(AgentTasksTest, self).tearDown()
self.god.unstub_all()