def test_ensure_device_role_exist(self): """Verify ensure_device_role function when DeviceRole exist but is not assigned to the OT.""" nbk = NetboxKeeper(self.ndk1) nbk.ensure_device_role(create_device_role=True, default_device_role="firewall") self.assertEqual(nbk.netdev.ot.role, self.device_role1)
def test_ensure_device_role_assigned(self): """Verify ensure_device_role function when DeviceRole exist and is already assigned.""" nbk = NetboxKeeper(self.ndk2) nbk.ensure_device_role(create_device_role=True, default_device_role="firewall") self.assertEqual(nbk.netdev.ot.role, self.device_role1)
def test_ensure_device_type_present(self): """Verify ensure_device_type function when Manufacturer and DeviceType object are already present.""" nbk = NetboxKeeper(self.ndk2) nbk.ensure_device_type(create_manufacturer=False, create_device_type=False) self.assertEqual(nbk.manufacturer, self.manufacturer1) self.assertEqual(nbk.device_type, self.device_type1)
def test_ensure_device_instance_not_exist(self): """Verify ensure_device_instance function.""" nbk = NetboxKeeper(self.ndk2) nbk.device_type = self.device_type1 nbk.netdev.ot = self.onboarding_task3 nbk.ensure_device_instance() self.assertIsInstance(nbk.device, Device) self.assertEqual(nbk.device, nbk.netdev.ot.created_device) self.assertEqual(nbk.device.serial, "123456")
def test_ensure_primary_ip_not_exist(self): """Verify ensure_primary_ip function when the Ip address do not already exist.""" nbk = NetboxKeeper(self.ndk2) nbk.device_type = self.device_type1 nbk.netdev.ot = self.onboarding_task3 nbk.ensure_device_instance() nbk.ensure_interface() nbk.ensure_primary_ip() self.assertIsInstance(nbk.primary_ip, IPAddress) self.assertEqual(nbk.primary_ip.interface, nbk.interface)
def test_ensure_device_instance_exist(self): """Verify ensure_device_instance function.""" device = Device.objects.create( name=self.ndk2.hostname, site=self.site1, device_type=self.device_type1, device_role=self.device_role1, status="planned", serial="987654", ) nbk = NetboxKeeper(self.ndk2) nbk.netdev.ot = self.onboarding_task3 self.assertEqual(nbk.device, None) nbk.ensure_device_instance(default_status="active") self.assertIsInstance(nbk.device, Device) self.assertEqual(nbk.device.status, "planned") self.assertEqual(nbk.device.platform, self.platform1) self.assertEqual(nbk.device, device) self.assertEqual(nbk.device.serial, "123456")
def test_ensure_interface_not_exist(self): """Verify ensure_interface function when the interface do not exist.""" nbk = NetboxKeeper(self.ndk2) nbk.device_type = self.device_type1 nbk.netdev.ot = self.onboarding_task3 nbk.ensure_device_instance() nbk.ensure_interface() self.assertIsInstance(nbk.interface, Interface) self.assertEqual(nbk.interface.name, "ge-0/0/0")
def test_ensure_interface_exist(self): """Verify ensure_interface function when the interface already exist.""" nbk = NetboxKeeper(self.ndk2) nbk.device_type = self.device_type1 nbk.netdev.ot = self.onboarding_task3 nbk.ensure_device_instance() intf = Interface.objects.create(name=nbk.netdev.mgmt_ifname, device=nbk.device) nbk.ensure_interface() self.assertEqual(nbk.interface, intf)
def test_ensure_device_type_missing(self): """Verify ensure_device_type function when Manufacturer and DeviceType object are not present.""" nbk = NetboxKeeper(self.ndk1) with self.assertRaises(OnboardException) as exc_info: nbk.ensure_device_type(create_manufacturer=False, create_device_type=False) self.assertEqual(exc_info.exception.message, "ERROR manufacturer not found: cisco") self.assertEqual(exc_info.exception.reason, "fail-config") with self.assertRaises(OnboardException) as exc_info: nbk.ensure_device_type(create_manufacturer=True, create_device_type=False) self.assertEqual(exc_info.exception.message, "ERROR device type not found: csr1000v") self.assertEqual(exc_info.exception.reason, "fail-config") nbk.ensure_device_type(create_manufacturer=True, create_device_type=True) self.assertIsInstance(nbk.manufacturer, Manufacturer) self.assertIsInstance(nbk.device_type, DeviceType)
def test_ensure_device_role_not_exist(self): """Verify ensure_device_role function when DeviceRole do not already exist.""" nbk = NetboxKeeper(self.ndk1) with self.assertRaises(OnboardException) as exc_info: nbk.ensure_device_role(create_device_role=False, default_device_role="mytestrole") self.assertEqual(exc_info.exception.message, "ERROR device role not found: mytestrole") self.assertEqual(exc_info.exception.reason, "fail-config") nbk.ensure_device_role(create_device_role=True, default_device_role="mytestrole") self.assertIsInstance(nbk.netdev.ot.role, DeviceRole) self.assertEqual(nbk.netdev.ot.role.slug, "mytestrole")