Join the 1-day Testing & QA Summit featuring 15+ Expert Speakers.Register for FREE! Join TestMu Conference

How to use drone_sets_enabled method in autotest

Best Python code snippet using autotest_python

agent_task_unittest.py

Source:agent_task_unittest.py

1#!/usr/bin/python2import unittest3import common4from autotest_lib.frontend import setup_django_environment5from autotest_lib.frontend.afe import frontend_test_utils6from autotest_lib.frontend.afe import models7from autotest_lib.scheduler import agent_task8from autotest_lib.server import system_utils9class RestrictedSubnetTest(unittest.TestCase,10                           frontend_test_utils.FrontendTestMixin):11    """Test server election based on restricted subnet setting.12    """13    DRONE_IN_RESTRICTED_SUBNET = '192.168.0.9'14    DRONE_NOT_IN_RESTRICTED_SUBNET = '127.0.0.9'15    HOST_IN_RESTRICTED_SUBNET = '192.168.0.3'16    HOST_NOT_IN_RESTRICTED_SUBNET = '127.0.0.3'17    RESTRICTED_SUBNETS = [('192.168.0.1', 16)]18    def setUp(self):19        self._drones = [self.DRONE_IN_RESTRICTED_SUBNET,20                        self.DRONE_NOT_IN_RESTRICTED_SUBNET]21        system_utils.DroneCache.unrestricted_drones = None22        system_utils.DroneCache.drone_ip_map = None23        self._frontend_common_setup()24    def tearDown(self):25        self._frontend_common_teardown()26    def test_get_drone_hostnames_allowed_with_restricted_subnet(self):27        """Test method get_drone_hostnames_allowed work as expected when28        restricted subnet is set, and host is inside restricted subnet.29        """30        self.god.stub_function(system_utils, 'get_drones')31        system_utils.get_drones.expect_call().and_return(self._drones)32        self.god.stub_function(models.DroneSet, 'drone_sets_enabled')33        models.DroneSet.drone_sets_enabled.expect_call().and_return(False)34        task = agent_task.AgentTask()35        task.hostnames = {1: self.HOST_IN_RESTRICTED_SUBNET}36        self.assertEqual(37                set([self.DRONE_IN_RESTRICTED_SUBNET]),38                task.get_drone_hostnames_allowed(self.RESTRICTED_SUBNETS, True))39        self.god.check_playback()40    def test_get_drone_hostnames_allowed_not_in_restricted_subnet(self):41        """Test method get_drone_hostnames_allowed work as expected when42        restricted subnet is set, and host is not in restricted subnet.43        """44        self.god.stub_function(system_utils, 'get_drones')45        system_utils.get_drones.expect_call().and_return(self._drones)46        self.god.stub_function(models.DroneSet, 'drone_sets_enabled')47        models.DroneSet.drone_sets_enabled.expect_call().and_return(False)48        task = agent_task.AgentTask()49        task.hostnames = {1: self.HOST_NOT_IN_RESTRICTED_SUBNET}50        self.assertEqual(51                set([self.DRONE_NOT_IN_RESTRICTED_SUBNET]),52                task.get_drone_hostnames_allowed(self.RESTRICTED_SUBNETS, True))53        self.god.check_playback()54    def test_get_drone_hostnames_allowed_in_mixed_subnet(self):55        """Test method get_drone_hostnames_allowed work as expected when56        restricted subnet is set, and hosts are distributed across restricted57        subnet and unrestricted subnet.58        """59        task = agent_task.AgentTask()60        task.hostnames = {1: self.HOST_NOT_IN_RESTRICTED_SUBNET,61                          2: self.HOST_IN_RESTRICTED_SUBNET}62        self.assertEqual(63                set(),64                task.get_drone_hostnames_allowed(self.RESTRICTED_SUBNETS, True))65        self.god.check_playback()66if __name__ == '__main__':...

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.