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

141 lines
3.9 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_MixedURLs tests embed with external URL for thumbnail, attachment:// for image
func TestEmbedAttachmentURL_MixedURLs(t *testing.T) {
client := newTestClient(t)
user := createTestAccount(t, client)
ensureSessionStarted(t, client, user.Token)
guild := createGuild(t, client, user.Token, "Mixed URLs 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 mixed URL types
var body bytes.Buffer
writer := multipart.NewWriter(&body)
payload := map[string]any{
"content": "Mixed URL types",
"attachments": []map[string]any{
{"id": 0, "filename": "yeah.png"},
},
"embeds": []map[string]any{
{
"title": "Mixed URLs",
"description": "External thumbnail, attached image",
"thumbnail": map[string]string{
"url": "https://example.com/thumbnail.png",
},
"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 {
Embeds []struct {
Image struct {
URL string `json:"url"`
} `json:"image"`
Thumbnail struct {
URL string `json:"url"`
} `json:"thumbnail"`
} `json:"embeds"`
}
decodeJSONResponse(t, resp, &msgResp)
if len(msgResp.Embeds) != 1 {
t.Fatalf("expected 1 embed, got %d", len(msgResp.Embeds))
}
embed := msgResp.Embeds[0]
if embed.Thumbnail.URL != "https://example.com/thumbnail.png" {
t.Errorf("expected thumbnail URL 'https://example.com/thumbnail.png', got '%s'", embed.Thumbnail.URL)
}
if embed.Image.URL == "" {
t.Error("expected image URL to be populated")
}
if strings.Contains(embed.Image.URL, "attachment://") {
t.Errorf("expected image attachment:// URL to be resolved, got '%s'", embed.Image.URL)
}
if !strings.Contains(embed.Image.URL, "yeah.png") {
t.Errorf("expected image URL to contain 'yeah.png', got '%s'", embed.Image.URL)
}
}