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_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")