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