Host scheduler refactoring. Move HostScheduler out of monitor_db.

scheduler/monitor_db_unittest.py

 #!/usr/bin/python
 
 import gc, logging, time
 import common
 from autotest_lib.frontend import setup_django_environment
 from autotest_lib.frontend.afe import frontend_test_utils
 from autotest_lib.client.common_lib.test_utils import mock
 from autotest_lib.client.common_lib.test_utils import unittest
 from autotest_lib.database import database_connection
 from autotest_lib.frontend.afe import models
 from autotest_lib.scheduler import monitor_db, drone_manager, email_manager
-from autotest_lib.scheduler import scheduler_config, gc_stats
+from autotest_lib.scheduler import scheduler_config, gc_stats, host_scheduler
 from autotest_lib.scheduler import monitor_db_functional_test
 from autotest_lib.scheduler import scheduler_models
 
 _DEBUG = False
 
 
 class DummyAgentTask(object):
     num_processes = 1
     owner_username = 'my_user'
 
@@ skipping 1268 matching lines @@
         self.assert_(queue_entry.execution_subdir)
         self.god.check_playback()
 
         class dummy_test_agent(object):
             task = 'dummy_test_agent'
         self._dispatcher._register_agent_for_ids(
                 self._dispatcher._host_agents, [queue_entry.host.id],
                 dummy_test_agent)
 
         # Attempted to schedule on a host that already has an agent.
-        self.assertRaises(monitor_db.SchedulerError,
+        self.assertRaises(host_scheduler.SchedulerError,
                           self._dispatcher._schedule_running_host_queue_entries)
 
 
     def test_schedule_hostless_job(self):
         job = self._create_job(hostless=True)
         self.assertEqual(1, job.hostqueueentry_set.count())
         hqe_query = scheduler_models.HostQueueEntry.fetch(
                 'id = %s' % job.hostqueueentry_set.all()[0].id)
         self.assertEqual(1, len(hqe_query))
         hqe = hqe_query[0]
@@ skipping 198 matching lines @@
         expected = self._setup_test_user_or_global_default_drone_set()
         actual = monitor_db.AgentTask()._user_or_global_default_drone_set(
                 None, MockUser())
 
         self.assertEqual(expected, actual)
         self.god.check_playback()
 
 
 if __name__ == '__main__':
     unittest.main()