diff --git a/netbox/utilities/tests/test_counters.py b/netbox/utilities/tests/test_counters.py index e9561c91b..0c61c0890 100644 --- a/netbox/utilities/tests/test_counters.py +++ b/netbox/utilities/tests/test_counters.py @@ -29,13 +29,17 @@ class CountersTest(TestCase): self.assertEqual(device1.interface_count, 2) self.assertEqual(device2.interface_count, 2) - Interface.objects.create(device=device1, name='Interface 5') + interface1 = Interface.objects.create(device=device1, name='Interface 5') Interface.objects.create(device=device2, name='Interface 6') device1.refresh_from_db() device2.refresh_from_db() self.assertEqual(device1.interface_count, 3) self.assertEqual(device2.interface_count, 3) + interface1.save() + device1.refresh_from_db() + self.assertEqual(device1.interface_count, 3) + def test_interface_count_deletion(self): """ When a tracked object (Interface) is deleted the tracking counter should be updated.