mirror of
https://github.com/ramvignesh-b/pi-ku.git
synced 2026-05-04 08:56:52 +00:00
feat: add backend support for burning letters
This commit is contained in:
@@ -47,6 +47,12 @@ class LetterSerializer(serializers.ModelSerializer):
|
|||||||
fields["encrypted_dek"] = None
|
fields["encrypted_dek"] = None
|
||||||
except (ValueError, TypeError):
|
except (ValueError, TypeError):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
if fields["status"] == Letter.Status.BURNED:
|
||||||
|
fields["encrypted_content"] = None
|
||||||
|
fields["images"] = None
|
||||||
|
fields["encrypted_dek"] = None
|
||||||
|
|
||||||
return fields
|
return fields
|
||||||
|
|
||||||
def validate(self, data):
|
def validate(self, data):
|
||||||
|
|||||||
@@ -248,6 +248,38 @@ class LetterAPITest(APITestCase):
|
|||||||
self.assertEqual(response.data["encrypted_metadata"], "enc_meta==")
|
self.assertEqual(response.data["encrypted_metadata"], "enc_meta==")
|
||||||
self.assertEqual(response.data["encrypted_dek"], "enc_dek==")
|
self.assertEqual(response.data["encrypted_dek"], "enc_dek==")
|
||||||
|
|
||||||
|
def test_burn_letter(self):
|
||||||
|
"""
|
||||||
|
Test that a sealed letter can only be burned but not updated.
|
||||||
|
"""
|
||||||
|
letter = Letter.objects.create(
|
||||||
|
user=self.user,
|
||||||
|
type="KEPT",
|
||||||
|
status="SEALED",
|
||||||
|
public_id="4281edcc-5459-4ff2-bb5e-669fb44e0757",
|
||||||
|
encrypted_content="enc_content==",
|
||||||
|
encrypted_metadata="enc_meta==",
|
||||||
|
encrypted_dek="enc_dek==",
|
||||||
|
)
|
||||||
|
|
||||||
|
response_update_content = self.client.patch(
|
||||||
|
self.url + letter.public_id + "/",
|
||||||
|
{
|
||||||
|
"encrypted_content": "enc_content_new==",
|
||||||
|
"encrypted_metadata": "enc_meta_new==",
|
||||||
|
"encrypted_dek": "enc_dek_new==",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
response_burn = self.client.patch(self.url + letter.public_id + "/", {"status": "BURNED"})
|
||||||
|
|
||||||
|
self.assertEqual(response_update_content.status_code, 400)
|
||||||
|
self.assertEqual(response_update_content.data["error"], "Sealed letters can only be burned.")
|
||||||
|
self.assertEqual(Letter.objects.get().encrypted_content, "enc_content==")
|
||||||
|
|
||||||
|
self.assertEqual(response_burn.status_code, 200)
|
||||||
|
self.assertEqual(Letter.objects.count(), 1)
|
||||||
|
self.assertEqual(Letter.objects.get().status, "BURNED")
|
||||||
|
|
||||||
|
|
||||||
class LetterImageModelTest(TestCase):
|
class LetterImageModelTest(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
|
|||||||
@@ -62,3 +62,20 @@ class LetterDetailView(generics.RetrieveUpdateDestroyAPIView):
|
|||||||
|
|
||||||
response_serializer = self.get_serializer(letter)
|
response_serializer = self.get_serializer(letter)
|
||||||
return Response(response_serializer.data, status=201 if created else 200)
|
return Response(response_serializer.data, status=201 if created else 200)
|
||||||
|
|
||||||
|
def patch(self, request, public_id):
|
||||||
|
"""
|
||||||
|
Updates an existing letter.
|
||||||
|
"""
|
||||||
|
letter = Letter.objects.get(public_id=public_id, user=request.user)
|
||||||
|
|
||||||
|
if letter.status == Letter.Status.SEALED and (
|
||||||
|
len(request.data) > 1 or request.data.get("status") != Letter.Status.BURNED
|
||||||
|
):
|
||||||
|
return Response({"error": "Sealed letters can only be burned."}, status=400)
|
||||||
|
|
||||||
|
write_serializer = self.get_serializer(letter, data=request.data, partial=True)
|
||||||
|
write_serializer.is_valid(raise_exception=True)
|
||||||
|
write_serializer.save()
|
||||||
|
response_serializer = self.get_serializer(letter)
|
||||||
|
return Response(response_serializer.data, status=200)
|
||||||
|
|||||||
Reference in New Issue
Block a user