fluxer/tests/integration/embed_attachment_url_basic_functionality_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_BasicFunctionality tests embed with image using attachment:// references uploaded file
func TestEmbedAttachmentURL_BasicFunctionality(t *testing.T) {
client := newTestClient(t)
user := createTestAccount(t, client)
ensureSessionStarted(t, client, user.Token)
guild := createGuild(t, client, user.Token, "Embed Attachment 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 embed referencing attachment
var body bytes.Buffer
writer := multipart.NewWriter(&body)
payload := map[string]any{
"content": "Test message with embed",
"attachments": []map[string]any{
{"id": 0, "filename": "yeah.png"},
},
"embeds": []map[string]any{
{
"title": "Test Embed",
"description": "This embed uses an attached image",
"image": map[string]string{
"url": "attachment://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]", "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 {
Content string `json:"content"`
Embeds []struct {
Title string `json:"title"`
Description string `json:"description"`
Image struct {
URL string `json:"url"`
} `json:"image"`
} `json:"embeds"`
}
decodeJSONResponse(t, resp, &msgResp)
if msgResp.Content != "Test message with embed" {
t.Errorf("expected content 'Test message with embed', got '%s'", msgResp.Content)
}
if len(msgResp.Embeds) != 1 {
t.Fatalf("expected 1 embed, got %d", len(msgResp.Embeds))
}
embed := msgResp.Embeds[0]
if embed.Title != "Test Embed" {
t.Errorf("expected embed title 'Test Embed', got '%s'", embed.Title)
}
if embed.Image.URL == "" {
t.Error("expected image URL to be populated")
}
if strings.Contains(embed.Image.URL, "attachment://") {
t.Errorf("expected attachment:// URL to be resolved, got '%s'", embed.Image.URL)
}
}