fluxer/tests/integration/embed_attachment_url_spoiler_with_attachment_test.go
2026-01-01 21:05:54 +00:00

138 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 (
"bytes"
"encoding/json"
"fmt"
"mime/multipart"
"net/http"
"strings"
"testing"
)
// TestEmbedAttachmentURL_SpoilerWithAttachment tests combine spoiler flag with attachment:// URL
func TestEmbedAttachmentURL_SpoilerWithAttachment(t *testing.T) {
client := newTestClient(t)
user := createTestAccount(t, client)
ensureSessionStarted(t, client, user.Token)
guild := createGuild(t, client, user.Token, "Spoiler Test Guild")
channelID := parseSnowflake(t, guild.SystemChannel)
fileData, err := fixturesFS.ReadFile("fixtures/yeah.png")
if err != nil {
t.Fatalf("failed to read fixture yeah.png: %v", err)
}
// Create multipart form with spoiler attachment
var body bytes.Buffer
writer := multipart.NewWriter(&body)
payload := map[string]any{
"content": "Spoiler image in embed",
"embeds": []map[string]any{
{
"title": "Spoiler Embed",
"description": "This uses a spoiler attachment",
"image": map[string]string{
"url": "attachment://SPOILER_yeah.png",
},
},
},
"attachments": []map[string]any{
{
"id": 0,
"filename": "SPOILER_yeah.png",
},
},
}
payloadJSON, err := json.Marshal(payload)
if err != nil {
t.Fatalf("failed to encode payload JSON: %v", err)
}
if err := writer.WriteField("payload_json", string(payloadJSON)); err != nil {
t.Fatalf("failed to write payload_json field: %v", err)
}
fileWriter, err := writer.CreateFormFile("files[0]", "SPOILER_yeah.png")
if err != nil {
t.Fatalf("failed to create files[0] field: %v", err)
}
if _, err := fileWriter.Write(fileData); err != nil {
t.Fatalf("failed to write file data: %v", err)
}
if err := writer.Close(); err != nil {
t.Fatalf("failed to close multipart writer: %v", err)
}
req, err := http.NewRequest(
http.MethodPost,
fmt.Sprintf("%s/channels/%d/messages", client.baseURL, channelID),
&body,
)
if err != nil {
t.Fatalf("failed to create request: %v", err)
}
req.Header.Set("Content-Type", writer.FormDataContentType())
client.applyCommonHeaders(req)
req.Header.Set("Authorization", user.Token)
resp, err := client.httpClient.Do(req)
if err != nil {
t.Fatalf("failed to send request: %v", err)
}
defer resp.Body.Close()
assertStatus(t, resp, http.StatusOK)
var msgResp struct {
Attachments []struct {
Filename string `json:"filename"`
} `json:"attachments"`
Embeds []struct {
Image struct {
URL string `json:"url"`
} `json:"image"`
} `json:"embeds"`
}
decodeJSONResponse(t, resp, &msgResp)
if len(msgResp.Attachments) != 1 {
t.Fatalf("expected 1 attachment, got %d", len(msgResp.Attachments))
}
if msgResp.Attachments[0].Filename != "SPOILER_yeah.png" {
t.Errorf("expected filename 'SPOILER_yeah.png', got '%s'", msgResp.Attachments[0].Filename)
}
if len(msgResp.Embeds) != 1 {
t.Fatalf("expected 1 embed, got %d", len(msgResp.Embeds))
}
if msgResp.Embeds[0].Image.URL == "" {
t.Error("expected image URL to be populated")
}
if strings.Contains(msgResp.Embeds[0].Image.URL, "attachment://") {
t.Errorf("expected attachment:// URL to be resolved, got '%s'", msgResp.Embeds[0].Image.URL)
}
}