61 lines
1.8 KiB
Go
61 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/tinode/chat/server/store"
|
|
"github.com/tinode/chat/server/store/types"
|
|
)
|
|
|
|
type defaultAvatarConfigStore map[string]string
|
|
|
|
func (s defaultAvatarConfigStore) Get(_ string, keys ...string) (map[string]string, error) {
|
|
out := make(map[string]string, len(keys))
|
|
for _, key := range keys {
|
|
out[key] = s[key]
|
|
}
|
|
return out, nil
|
|
}
|
|
|
|
func TestUserSummaryPayloadAppliesDefaultAvatarByGenderAndUserID(t *testing.T) {
|
|
oldConfig := store.ThirdPartyConfig
|
|
store.ThirdPartyConfig = defaultAvatarConfigStore{
|
|
imDefaultMaleAvatarURLsConfig: `["m0","m1","m2","m3","m4"]`,
|
|
imDefaultFemaleAvatarURLsConfig: `["f0","f1","f2","f3","f4"]`,
|
|
}
|
|
defer func() { store.ThirdPartyConfig = oldConfig }()
|
|
|
|
for i, gender := range []string{"male", "female"} {
|
|
uid := types.Uid(10 + i)
|
|
payload := userSummaryPayload(&types.User{
|
|
ObjHeader: types.ObjHeader{Id: uid.String()},
|
|
Public: map[string]any{"display_name": "Alice", "gender": gender},
|
|
}, "")
|
|
wantPrefix := "m"
|
|
if gender == "female" {
|
|
wantPrefix = "f"
|
|
}
|
|
want := fmt.Sprintf("%s%d", wantPrefix, uint64(uid)%5)
|
|
if payload["avatar_file_id"] != want {
|
|
t.Fatalf("gender=%s avatar=%v want %s", gender, payload["avatar_file_id"], want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestUserSummaryPayloadKeepsUploadedAvatar(t *testing.T) {
|
|
oldConfig := store.ThirdPartyConfig
|
|
store.ThirdPartyConfig = defaultAvatarConfigStore{
|
|
imDefaultMaleAvatarURLsConfig: `["m0","m1","m2","m3","m4"]`,
|
|
}
|
|
defer func() { store.ThirdPartyConfig = oldConfig }()
|
|
|
|
payload := userSummaryPayload(&types.User{
|
|
ObjHeader: types.ObjHeader{Id: types.Uid(10).String()},
|
|
Public: map[string]any{"display_name": "Alice", "gender": "male", "avatar_file_id": "uploaded.png"},
|
|
}, "")
|
|
if payload["avatar_file_id"] != "uploaded.png" {
|
|
t.Fatalf("avatar=%v", payload["avatar_file_id"])
|
|
}
|
|
}
|