How to use _user_or_global_default_drone_set method in autotest

Best Python code snippet using autotest_python

monitor_db_unittest.py

Source:monitor_db_unittest.py

...714        task._user_or_global_default_drone_set.expect_call(715                task.task, MockSpecialTask.requested_by).and_return(result)716        self.assertEqual(result, task.get_drone_hostnames_allowed())717        self.god.check_playback()718    def _setup_test_user_or_global_default_drone_set(self):719        result = object()720        class MockDroneSet(object):721            def get_drone_hostnames(self):722                return result723        self.god.stub_function(models.DroneSet, 'get_default')724        models.DroneSet.get_default.expect_call().and_return(MockDroneSet())725        return result726    def test_user_or_global_default_drone_set(self):727        expected = object()728        class MockDroneSet(object):729            def get_drone_hostnames(self):730                return expected731        class MockUser(object):732            drone_set = MockDroneSet()733        self._setup_test_user_or_global_default_drone_set()734        actual = agent_task.AgentTask()._user_or_global_default_drone_set(735                None, MockUser())736        self.assertEqual(expected, actual)737        self.god.check_playback()738    def test_user_or_global_default_drone_set_no_user(self):739        expected = self._setup_test_user_or_global_default_drone_set()740        actual = agent_task.AgentTask()._user_or_global_default_drone_set(741                None, None)742        self.assertEqual(expected, actual)743        self.god.check_playback()744    def test_user_or_global_default_drone_set_no_user_drone_set(self):745        class MockUser(object):746            drone_set = None747            login = None748        expected = self._setup_test_user_or_global_default_drone_set()749        actual = agent_task.AgentTask()._user_or_global_default_drone_set(750                None, MockUser())751        self.assertEqual(expected, actual)752        self.god.check_playback()753    def test_abort_HostlessQueueTask(self):754        hqe = self.god.create_mock_class(scheduler_models.HostQueueEntry,755                                         'HostQueueEntry')756        # If hqe is still in STARTING status, aborting the task should finish757        # without changing hqe's status.758        hqe.status = models.HostQueueEntry.Status.STARTING759        hqe.job = None760        hqe.id = 0761        task = monitor_db.HostlessQueueTask(hqe)762        task.abort()763        # If hqe is in RUNNING status, aborting the task should change hqe's...

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.