def setUp(self, inline_host_acquisition=True, setup_tables=True): """Common setup module for tests that need a jobs/host database. @param inline_host_acquisition: If True, the dispatcher tries to acquire hosts inline with the rest of the tick. """ self.db_helper = DBHelper() self._database = self.db_helper.database # Runs syncdb setting up initial database conditions self._frontend_common_setup(setup_tables=setup_tables) connection_manager = scheduler_lib.ConnectionManager(autocommit=False) self.god.stub_with(connection_manager, 'db_connection', self._database) self.god.stub_with(monitor_db, '_db_manager', connection_manager) self.god.stub_with(scheduler_models, '_db', self._database) self.god.stub_with(monitor_db, '_inline_host_acquisition', inline_host_acquisition) self._dispatcher = monitor_db.Dispatcher() self.host_scheduler = self._dispatcher._host_scheduler self.host_query_manager = query_managers.AFEHostQueryManager() self.job_query_manager = self._dispatcher._job_query_manager self._release_unused_hosts()
def _create_dispatcher(self): self.dispatcher = monitor_db.Dispatcher()
def setUp(self): self._frontend_common_setup() self._set_monitor_stubs() self._set_global_config_values() self._dispatcher = monitor_db.Dispatcher()
def setUp(self): self._frontend_common_setup() self._set_monitor_stubs() self._dispatcher = monitor_db.Dispatcher()