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)
