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 Github

copy

Full Screen

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__':...

Full Screen

Full Screen

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.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run autotest automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful