acfunlive-remake/avatar_test.go

package main

import (
	"bytes"
	"encoding/base64"
	"image"
	"image/color"
	"image/gif"
	"image/jpeg"
	"image/png"
	"testing"
)

// 2x2 的静态 webp（纯红），用来确认 webp 解码器确实注册上了
const tinyWebPBase64 = "UklGRjwAAABXRUJQVlA4IDAAAADQAQCdASoCAAIAAgA0JaACdLoB+AADsAD+8Oj3/yC5YXXI1/8gP+QH/ID/+PIAAAA="

// 多帧 gif 解出来应该是第一帧（AcFun 上有动图头像，压成静态就用第一帧）
func TestDecodeAvatarGifTakesFirstFrame(t *testing.T) {
	pal := color.Palette{color.RGBA{R: 255, A: 255}, color.RGBA{B: 255, A: 255}}
	first := image.NewPaletted(image.Rect(0, 0, 8, 8), pal)
	second := image.NewPaletted(image.Rect(0, 0, 8, 8), pal)
	for i := range second.Pix {
		second.Pix[i] = 1
	}
	var buf bytes.Buffer
	if err := gif.EncodeAll(&buf, &gif.GIF{
		Image: []*image.Paletted{first, second},
		Delay: []int{10, 10},
	}); err != nil {
		t.Fatal(err)
	}

	src, err := decodeAvatar(buf.Bytes(), "image/gif")
	if err != nil {
		t.Fatalf("gif 解码失败：%v", err)
	}
	r, g, b, _ := src.At(4, 4).RGBA()
	if r>>8 != 255 || g>>8 != 0 || b>>8 != 0 {
		t.Errorf("应该取第一帧（红），实际拿到 r=%d g=%d b=%d", r>>8, g>>8, b>>8)
	}
}

// jpg / png / 静态 webp 都能解，解不出来的要给中文原因
func TestDecodeAvatarFormats(t *testing.T) {
	var pngBuf, jpgBuf bytes.Buffer
	img := image.NewRGBA(image.Rect(0, 0, 8, 8))
	if err := png.Encode(&pngBuf, img); err != nil {
		t.Fatal(err)
	}
	if err := jpeg.Encode(&jpgBuf, img, nil); err != nil {
		t.Fatal(err)
	}
	webpData, err := base64.StdEncoding.DecodeString(tinyWebPBase64)
	if err != nil {
		t.Fatal(err)
	}

	cases := []struct {
		name        string
		data        []byte
		contentType string
		wantErr     bool
	}{
		{"png", pngBuf.Bytes(), "image/png", false},
		{"jpeg", jpgBuf.Bytes(), "image/jpeg", false},
		{"webp", webpData, "image/webp", false},
		{"不是图片", []byte("这不是图片"), "text/html", true},
	}
	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			src, err := decodeAvatar(c.data, c.contentType)
			if c.wantErr {
				if err == nil {
					t.Fatal("应该报错")
				}
				if !bytes.Contains([]byte(err.Error()), []byte(c.contentType)) {
					t.Errorf("错误信息里应该带上 Content-Type 方便排查：%v", err)
				}
				return
			}
			if err != nil {
				t.Fatalf("解码失败：%v", err)
			}
			if src == nil {
				t.Fatal("解出来是空的")
			}
		})
	}
}

// 解出来的头像要能走通圆形裁切 + 缓存用的 PNG 编码（gif 是 *image.Paletted，和 jpg/png 的底层类型都不同）
func TestDecodeAvatarGifGoesThroughCircleAvatar(t *testing.T) {
	pal := color.Palette{color.RGBA{G: 255, A: 255}, color.RGBA{R: 255, B: 255, A: 255}}
	frame := image.NewPaletted(image.Rect(0, 0, 16, 16), pal)
	var buf bytes.Buffer
	if err := gif.EncodeAll(&buf, &gif.GIF{Image: []*image.Paletted{frame}, Delay: []int{10}}); err != nil {
		t.Fatal(err)
	}
	src, err := decodeAvatar(buf.Bytes(), "image/gif")
	if err != nil {
		t.Fatal(err)
	}

	var out bytes.Buffer
	if err := png.Encode(&out, circleAvatar(src, avatarCacheSize)); err != nil {
		t.Fatalf("圆形裁切后的 PNG 编码失败：%v", err)
	}
	if out.Len() == 0 {
		t.Fatal("编码出来是空的")
	}
}