initial commit

tests/integration/attachment_decay_referenced_message_test.go

@@ -0,0 +1,128 @@
+/*
+ * Copyright (C) 2026 Fluxer Contributors
+ *
+ * This file is part of Fluxer.
+ *
+ * Fluxer is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Fluxer is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with Fluxer. If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package integration
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+	"time"
+)
+
+func TestAttachmentDecayPopulatesReferencedMessage(t *testing.T) {
+	client := newTestClient(t)
+	author := createTestAccount(t, client)
+	recipient := createTestAccount(t, client)
+	ensureSessionStarted(t, client, author.Token)
+	ensureSessionStarted(t, client, recipient.Token)
+
+	createFriendship(t, client, author, recipient)
+
+	channel := createDmChannel(t, client, author.Token, parseSnowflake(t, recipient.UserID))
+
+	originalMessage, originalAttachmentID := sendChannelMessageWithAttachment(
+		t,
+		client,
+		author.Token,
+		parseSnowflake(t, channel.ID),
+		"Original message for reference",
+		"document.pdf",
+	)
+
+	replyPayload := map[string]any{
+		"content": "Reply referencing expired attachment",
+		"message_reference": map[string]any{
+			"message_id": originalMessage.ID,
+			"channel_id": channel.ID,
+			"type":       0,
+		},
+	}
+
+	resp, err := client.postJSONWithAuth(
+		fmt.Sprintf("/channels/%d/messages", parseSnowflake(t, channel.ID)),
+		replyPayload,
+		author.Token,
+	)
+	if err != nil {
+		t.Fatalf("failed to post reply: %v", err)
+	}
+	assertStatus(t, resp, http.StatusOK)
+	var reply messageResponse
+	decodeJSONResponse(t, resp, &reply)
+
+	rows := []map[string]any{
+		{
+			"attachment_id":    fmt.Sprintf("%d", originalAttachmentID),
+			"channel_id":       channel.ID,
+			"message_id":       originalMessage.ID,
+			"expires_at":       time.Now().Add(-1 * time.Hour).Format(time.RFC3339Nano),
+			"uploaded_at":      time.Now().Add(-2 * time.Hour).Format(time.RFC3339Nano),
+			"last_accessed_at": time.Now().Add(-1 * time.Hour).Format(time.RFC3339Nano),
+			"filename":         "expired-reference.bin",
+			"size_bytes":       2048,
+			"cost":             2,
+			"lifetime_days":    1,
+		},
+	}
+
+	resp, err = client.postJSONWithAuth("/test/attachment-decay/rows", map[string]any{"rows": rows}, author.Token)
+	if err != nil {
+		t.Fatalf("failed to seed attachment decay rows: %v", err)
+	}
+	assertStatus(t, resp, http.StatusOK)
+
+	resp, err = client.getWithAuth(
+		fmt.Sprintf("/test/messages/%d/%d/with-reference", parseSnowflake(t, channel.ID), parseSnowflake(t, reply.ID)),
+		author.Token,
+	)
+	if err != nil {
+		t.Fatalf("failed to fetch message with reference: %v", err)
+	}
+	assertStatus(t, resp, http.StatusOK)
+
+	var fetched struct {
+		ID                string `json:"id"`
+		ReferencedMessage struct {
+			Attachments []struct {
+				ID        string  `json:"id"`
+				URL       *string `json:"url"`
+				Expired   *bool   `json:"expired"`
+				ExpiresAt *string `json:"expires_at"`
+			} `json:"attachments"`
+		} `json:"referenced_message"`
+	}
+	decodeJSONResponse(t, resp, &fetched)
+
+	if len(fetched.ReferencedMessage.Attachments) == 0 {
+		t.Fatalf("expected referenced message to include attachments")
+	}
+
+	for _, attachment := range fetched.ReferencedMessage.Attachments {
+		if attachment.URL != nil {
+			t.Fatalf("expected referenced message attachment to hide url")
+		}
+		if attachment.Expired == nil || !*attachment.Expired {
+			t.Fatalf("expected referenced message attachment to be marked expired")
+		}
+		if attachment.ExpiresAt == nil {
+			t.Fatalf("expected referenced message attachment to include expires_at")
+		}
+	}
+}