Fix Tests

This commit is contained in:
Daniel Sheppard 2025-01-20 23:44:36 -06:00
parent 22e320084a
commit bec97df242

View File

@ -1534,15 +1534,16 @@ class MACAddress(PrimaryModel):
def clean(self, *args, **kwargs): def clean(self, *args, **kwargs):
super().clean() super().clean()
if self._original_assigned_object_id and self._original_assigned_object_type_id: if self._original_assigned_object_id and self._original_assigned_object_type_id:
assigned_object = getattr(self.assigned_object, 'parent_object', None) assigned_object = self.assigned_object
ct = ObjectType.objects.get_for_id(self._original_assigned_object_type_id) ct = ObjectType.objects.get_for_id(self._original_assigned_object_type_id)
original_assigned_object = ct.get_object_for_this_type(pk=self._original_assigned_object_id) original_assigned_object = ct.get_object_for_this_type(pk=self._original_assigned_object_id)
if original_assigned_object and not assigned_object: if original_assigned_object.primary_mac_address:
raise ValidationError( if not assigned_object:
_("Cannot unassign MAC Address while it is designated as the primary MAC for an object") raise ValidationError(
) _("Cannot unassign MAC Address while it is designated as the primary MAC for an object")
elif original_assigned_object and original_assigned_object != assigned_object: )
raise ValidationError( elif original_assigned_object != assigned_object:
_("Cannot reassign MAC Address while it is designated as the primary MAC for an object") raise ValidationError(
) _("Cannot reassign MAC Address while it is designated as the primary MAC for an object")
)