Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 6 additions & 1 deletion api/segments/tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,12 @@

@register_task_handler()
def delete_segment(segment_id: int) -> None:
Segment.objects.get(pk=segment_id).delete()
try:
Segment.objects.get(pk=segment_id).delete()
except Segment.DoesNotExist:
# Already deleted, e.g. by a cascading delete of its parent segment
# or feature, or a previous run of this task.
return


@register_task_handler()
Expand Down
32 changes: 32 additions & 0 deletions api/tests/unit/segments/test_unit_segments_tasks.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
from projects.models import Project
from segments.models import Segment
from segments.tasks import delete_segment


def test_delete_segment__segment_exists__soft_deletes_segment(
project: Project,
) -> None:
# Given
segment = Segment.objects.create(name="Test Segment", project=project)

# When
delete_segment(segment_id=segment.id)

# Then
segment.refresh_from_db()
assert segment.deleted_at is not None


def test_delete_segment__segment_already_deleted__does_not_raise(
project: Project,
) -> None:
# Given
segment = Segment.objects.create(name="Test Segment", project=project)
segment_id = segment.id
segment.hard_delete()

# When
delete_segment(segment_id=segment_id) # should not raise

# Then
assert not Segment.objects.all_with_deleted().filter(id=segment_id).exists()
Loading