Skip to content

Commit

Permalink
Merge acc7916 into 0eefbed
Browse files Browse the repository at this point in the history
  • Loading branch information
np5 committed May 25, 2024
2 parents 0eefbed + acc7916 commit 6406a61
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 3 deletions.
3 changes: 3 additions & 0 deletions tests/monolith/test_monolith_enrollment_package_api_views.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
from accounts.models import APIToken, User
from zentral.contrib.inventory.models import Tag
from zentral.contrib.monolith.models import ManifestEnrollmentPackage
from zentral.contrib.munki.models import Enrollment as MunkiEnrollment
from tests.munki.utils import force_enrollment as force_munki_enrollment
from .utils import force_manifest, force_manifest_enrollment_package

Expand Down Expand Up @@ -420,6 +421,7 @@ def test_delete_manifest_enrollment_package_not_found(self):

def test_delete_manifest_enrollment_package(self):
mep = force_manifest_enrollment_package()
enrollment = mep.get_enrollment()
manifest = mep.manifest
manifest.refresh_from_db()
self.assertEqual(manifest.version, 2)
Expand All @@ -428,3 +430,4 @@ def test_delete_manifest_enrollment_package(self):
self.assertEqual(response.status_code, 204)
manifest.refresh_from_db()
self.assertEqual(manifest.version, 3)
self.assertTrue(MunkiEnrollment.objects.filter(pk=enrollment.pk).exists())
3 changes: 1 addition & 2 deletions zentral/contrib/monolith/api_views.py
Original file line number Diff line number Diff line change
Expand Up @@ -289,9 +289,8 @@ class ManifestEnrollmentPackageDetail(generics.RetrieveUpdateDestroyAPIView):

def perform_destroy(self, instance):
manifest = instance.manifest
response = super().perform_destroy(instance)
instance.delete(delete_enrollment=False)
manifest.bump_version()
return response


# manifest sub manifests
Expand Down
3 changes: 2 additions & 1 deletion zentral/contrib/monolith/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -1140,10 +1140,11 @@ class ManifestEnrollmentPackage(models.Model):
updated_at = models.DateTimeField(auto_now=True)

def delete(self, *args, **kwargs):
delete_enrollment = kwargs.pop("delete_enrollment", True)
self.file.delete(save=False)
enrollment = self.get_enrollment()
super().delete(*args, **kwargs)
if enrollment:
if delete_enrollment and enrollment:
enrollment.delete()

def get_installer_item_filename(self):
Expand Down

0 comments on commit 6406a61

Please sign in to comment.