129 lines
3.8 KiB
Go
129 lines
3.8 KiB
Go
/*
|
|
* 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")
|
|
}
|
|
}
|
|
}
|