mirror of
https://github.com/FunnyWolf/agentic-soc-platform.git
synced 2026-08-22 13:12:56 +02:00
funnywolfandCopilot
cef99c4b11
Cache relation parents before delete so audit relation events do not query parents already removed by cascade deletes. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
41 lines
1.4 KiB
Python
41 lines
1.4 KiB
Python
from django.contrib.contenttypes.models import ContentType
|
|
from django.test import TestCase
|
|
|
|
from apps.artifacts.models import Artifact, ArtifactName, ArtifactRole, ArtifactType
|
|
from apps.enrichments.models import Enrichment, EnrichmentProvider, EnrichmentType
|
|
from .models import AuditLog
|
|
|
|
|
|
class AuditDeleteCascadeTests(TestCase):
|
|
def test_deleting_parent_with_fk_relation_audit_child_does_not_query_deleted_parent(self):
|
|
artifact = Artifact.objects.create(
|
|
name=ArtifactName.HOSTNAME,
|
|
type=ArtifactType.HOSTNAME,
|
|
role=ArtifactRole.RELATED,
|
|
value="audit-delete-repro.example",
|
|
)
|
|
enrichment = Enrichment.objects.create(
|
|
artifact=artifact,
|
|
name="audit delete repro enrichment",
|
|
type=EnrichmentType.OBSERVATION,
|
|
provider=EnrichmentProvider.INTERNAL,
|
|
uid="audit-delete-repro",
|
|
value="audit-delete-repro.example",
|
|
)
|
|
|
|
artifact_id = artifact.id
|
|
enrichment_id = enrichment.id
|
|
|
|
artifact.delete()
|
|
|
|
artifact_content_type = ContentType.objects.get_for_model(Artifact)
|
|
self.assertTrue(
|
|
AuditLog.objects.filter(
|
|
content_type=artifact_content_type,
|
|
object_id=str(artifact_id),
|
|
action="deleted",
|
|
metadata__relation="enrichments",
|
|
metadata__related_id=str(enrichment_id),
|
|
).exists()
|
|
)
|