fix test

nautobot · Jul 26, 2023 · 264d126 · 264d126
1 parent 03982d7
commit 264d126
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nautobot_device_onboarding/tests/test_nautobot_keeper.py b/nautobot_device_onboarding/tests/test_nautobot_keeper.py
@@ -542,7 +542,7 @@ def test_ensure_custom_fields(self):
         device = Device.objects.get(name="sw1")
 
         self.assertEqual(device.cf["cf_device"], False)
-        self.assertEqual(device.cf["cf_device_null"], None)
+        self.assertEqual(device.cf.get("cf_device_null"), None)
         self.assertEqual(device.platform.cf["cf_platform"], True)
         self.assertEqual(device.device_type.cf["cf_devicetype"], 5)
         self.assertEqual(device.device_role.cf["cf_devicerole"], 10)