Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use host_has_agent method in autotest

Best Python code snippet using autotest_python

host_scheduler_unittests.py

Source:host_scheduler_unittests.py

...189        frontend_task = self.db_helper.create_special_task(host_id=host.id)190        self._dispatcher._schedule_special_tasks()191        # The frontend special task shouldn't get scheduled on the host till192        # the host is leased.193        self.assertFalse(self._dispatcher.host_has_agent(host))194        # Create a job for the same host and make the host scheduler lease the195        # host out to that job.196        job =  self.create_job(deps=set(['a']))197        self.host_scheduler._schedule_jobs()198        hqe = self.db_helper.get_hqes(job_id=job.id)[0]199        tasks = self.job_query_manager.get_prioritized_special_tasks(200                only_tasks_with_leased_hosts=True)201        # We should not find the frontend special task, even though its host is202        # now leased, because its leased by an active hqe.203        self.assertTrue(len(tasks) == 1 and tasks[0].queue_entry_id == hqe.id)204        self._dispatcher._schedule_special_tasks()205        self.assertTrue(self._dispatcher.host_has_agent(host))206        # Deactivate the hqe task and make sure the frontend task gets the host.207        task = tasks[0]208        self._dispatcher.remove_agent(self._check_agent_invariants(host, task))209        task.is_complete = 1210        task.is_active = 0211        task.save()212        self.db_helper.update_hqe(hqe.id, active=False)213        self._dispatcher._schedule_special_tasks()214        self.assertTrue(self._dispatcher.host_has_agent(host))215        self._check_agent_invariants(host, frontend_task)216        # Make sure we don't release the host being used by the incomplete task.217        self.host_scheduler._release_hosts()218        host = self.db_helper.get_host(hostname='h1')[0]219        self.assertTrue(host.leased == True)220class SuiteRecorderTest(rdb_testing_utils.AbstractBaseRDBTester,221                        unittest.TestCase):222    """Test the functionality of SuiteRecorder"""223    _config_section = 'AUTOTEST_WEB'224    def testGetSuiteHostAssignment(self):225        """Test the initialization of SuiteRecord."""226        hosts = []227        num = 4228        for i in range (0, num):...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.