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

How to use _setup_test_drone_restrictions method in autotest

Best Python code snippet using autotest_python

drone_manager_unittest.py

Source:drone_manager_unittest.py

...123                                                              None))124        drone = self.manager._choose_drone_for_execution(125            1, username=self._USERNAME, drone_hostnames_allowed=None)126        self.assertEquals(drone.name, 2)127    def _setup_test_drone_restrictions(self, active_processes=0):128        self.manager._enqueue_drone(MockDrone(129            1, active_processes=active_processes, max_processes=10))130        self.manager._enqueue_drone(MockDrone(131            2, active_processes=active_processes, max_processes=5))132        self.manager._enqueue_drone(MockDrone(133            3, active_processes=active_processes, max_processes=2))134    def test_drone_restrictions_allow_any(self):135        self._setup_test_drone_restrictions()136        self.assertEquals(10,137                          self.manager.max_runnable_processes(self._USERNAME,138                                                              None))139        drone = self.manager._choose_drone_for_execution(140            1, username=self._USERNAME, drone_hostnames_allowed=None)141        self.assertEqual(drone.name, 1)142    def test_drone_restrictions_under_capacity(self):143        self._setup_test_drone_restrictions()144        drone_hostnames_allowed = (2, 3)145        self.assertEquals(146            5, self.manager.max_runnable_processes(self._USERNAME,147                                                   drone_hostnames_allowed))148        drone = self.manager._choose_drone_for_execution(149            1, username=self._USERNAME,150            drone_hostnames_allowed=drone_hostnames_allowed)151        self.assertEqual(drone.name, 2)152    def test_drone_restrictions_over_capacity(self):153        self._setup_test_drone_restrictions(active_processes=6)154        drone_hostnames_allowed = (2, 3)155        self.assertEquals(156            0, self.manager.max_runnable_processes(self._USERNAME,157                                                   drone_hostnames_allowed))158        drone = self.manager._choose_drone_for_execution(159            7, username=self._USERNAME,160            drone_hostnames_allowed=drone_hostnames_allowed)161        self.assertEqual(drone.name, 2)162    def test_drone_restrictions_allow_none(self):163        self._setup_test_drone_restrictions()164        drone_hostnames_allowed = ()165        self.assertEquals(166            0, self.manager.max_runnable_processes(self._USERNAME,167                                                   drone_hostnames_allowed))168        drone = self.manager._choose_drone_for_execution(169            1, username=self._USERNAME,170            drone_hostnames_allowed=drone_hostnames_allowed)171        self.assertEqual(drone, None)172    def test_initialize(self):173        results_hostname = 'results_repo'174        results_install_dir = '/results/install'175        settings.override_value(scheduler_config.CONFIG_SECTION,176                                'results_host_installation_directory',177                                results_install_dir)...

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.