mirror of
https://github.com/fumiama/deepinfra.git
synced 2026-06-05 00:32:46 +08:00
34 lines
805 B
Go
34 lines
805 B
Go
package model
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
const contentTextSmallImage = "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAQAAAC1HAwCAAAAC0lEQVR42mNk+A8AAQUBAScY42YAAAAASUVORK5CYII="
|
|
|
|
func TestContentImageDataBytes(t *testing.T) {
|
|
// 从 smallimg 中提取 base64 数据
|
|
parts := strings.Split(contentTextSmallImage, ",")
|
|
if len(parts) != 2 {
|
|
t.Fatal("Invalid data URL format")
|
|
}
|
|
|
|
// 解码 base64 数据
|
|
data, err := base64.StdEncoding.DecodeString(parts[1])
|
|
if err != nil {
|
|
t.Fatalf("Failed to decode base64: %v", err)
|
|
}
|
|
|
|
s, err := NewContentImageDataBase64URL(data)
|
|
if err != nil {
|
|
t.Fatalf("NewContentImageDataBytes failed: %v", err)
|
|
}
|
|
|
|
// 比较结果
|
|
if s != contentTextSmallImage {
|
|
t.Errorf("Expected %s, got %s", contentTextSmallImage, s)
|
|
}
|
|
}
|