From ca809d8cc047e0b2085efbd3e77d6eff6ebf7d5b Mon Sep 17 00:00:00 2001
From: kim <grufwub@gmail.com>
Date: Fri, 29 Nov 2024 16:59:42 +0000
Subject: [PATCH] add test for deleting status edits

---
 internal/db/bundb/statusedit_test.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/internal/db/bundb/statusedit_test.go b/internal/db/bundb/statusedit_test.go
index dc75fd089..b6a15e825 100644
--- a/internal/db/bundb/statusedit_test.go
+++ b/internal/db/bundb/statusedit_test.go
@@ -114,7 +114,20 @@ func (suite *StatusEditTestSuite) TestGetStatusEditsByIDs() {
 }
 
 func (suite *StatusEditTestSuite) TestDeleteStatusEdits() {
+	// Create a new context for this test.
+	ctx, cncl := context.WithCancel(context.Background())
+	defer cncl()
 
+	for _, status := range suite.testStatuses {
+		// Delete all edits for status with given IDs from database.
+		err := suite.state.DB.DeleteStatusEdits(ctx, status.EditIDs)
+		suite.NoError(err)
+
+		// Now attempt to fetch these edits from database, should be empty.
+		edits, err := suite.state.DB.GetStatusEditsByIDs(ctx, status.EditIDs)
+		suite.NoError(err)
+		suite.Empty(edits)
+	}
 }
 
 func TestStatusEditTestSuite(t *testing.T) {