add update test and fix delete test

netbox_sys_plugin/tests/vm_assigned_virtual_machine_type/test_vm_assigned_virtual_machine_type_api.py

@@ -56,16 +56,16 @@ class VmAssignedVirtualMachineTypeApiTestCase(BaseAPITestCase):
             virtual_machine_type_desc='VMWare description'
         )
 
-        # # Assign VirtualMachineTypes to VirtualMachines
-        # VmAssignedVirtualMachineType.objects.create(
-        #     virtual_machine_type=cls.virtual_machine_type_aws,
-        #     assigned_object_id=cls.virtual_machine.id
-        # )
+        # Assign VirtualMachineTypes to VirtualMachines
+        VmAssignedVirtualMachineType.objects.create(
+            virtual_machine_type=cls.virtual_machine_type_aws,
+            assigned_object_id=cls.virtual_machine.id
+        )
 
-        # VmAssignedVirtualMachineType.objects.create(
-        #     virtual_machine_type=cls.virtual_machine_type_vmware,
-        #     assigned_object_id=cls.virtual_machine4.id
-        # )
+        VmAssignedVirtualMachineType.objects.create(
+            virtual_machine_type=cls.virtual_machine_type_vmware,
+            assigned_object_id=cls.virtual_machine4.id
+        )
 
         # Data for valid creation
         cls.valid_create_data = [
@@ -106,6 +106,28 @@ class VmAssignedVirtualMachineTypeApiTestCase(BaseAPITestCase):
         self.assertEqual(response.data["assigned_object_id"], form_data["assigned_object_id"])
         self.assertEqual(response.data["virtual_machine_type"], form_data["virtual_machine_type"])
 
+    def test_update_virtual_machine_type_assignment(self):
+        """Test updating an existing VmAssignedVirtualMachineType"""
+        vm_assignment = VmAssignedVirtualMachineType.objects.first()
+        obj_perm = ObjectPermission(
+            name="Update VmAssignedVirtualMachineType Permission",
+            actions=["change", "view"],
+        )
+        obj_perm.save()
+        obj_perm.users.add(self.user)
+        obj_perm.object_types.add(ContentType.objects.get_for_model(VmAssignedVirtualMachineType))
+
+        update_data = {
+            "virtual_machine_type": self.virtual_machine_type_vmware.id,
+            "assigned_object_id": self.virtual_machine4.id,
+            "assigned_object_type": self.vm_ct.pk
+        }
+
+        response = self.client.patch(self._get_detail_url(vm_assignment), update_data, format="json", **self.header)
+        self.assertHttpStatus(response, status.HTTP_200_OK)
+        self.assertEqual(response.data["virtual_machine_type"], update_data["virtual_machine_type"])
+        self.assertEqual(response.data["assigned_object_id"], update_data["assigned_object_id"])
+
     def test_create_invalid_virtual_machine_type_assignment(self):
         """Test creating invalid VmAssignedVirtualMachineType"""
         obj_perm = ObjectPermission(