1package tinyclient23import (4 "context"5 "encoding/json"6 "net/http"7 "net/http/httptest"8 "strings"9 "testing"1011 "github.com/FelineStateMachine/tinyrelay/internal/policy"12)1314type pictureRoomsBackend struct{ *roomsBackend }1516func (b *pictureRoomsBackend) Query(ctx context.Context, method string, params []json.RawMessage, actor string) (any, error) {17 result, err := b.roomsBackend.Query(ctx, method, params, actor)18 if err != nil {19 return result, err20 }21 page, ok := result.(map[string]any)22 if !ok {23 return result, nil24 }25 if method == "browserooms" {26 for _, item := range page["items"].([]any) {27 item.(map[string]any)["picture"] = "https://cdn.example/general.png"28 }29 } else if room, ok := page["room"].(map[string]any); ok {30 room["picture"] = "https://cdn.example/general.png"31 }32 return page, nil33}3435func TestRoomSurfacesRenderStableSeedmarkFallbackAvatars(t *testing.T) {36 app, _ := roomsApp(t, roomOwner)37 for _, path := range []string{"/chat", "/rooms/general", "/rooms/general/thread/" + roomThread} {38 body := roomsPage(t, app, path)39 if !strings.Contains(body, `fallback="/avatars/v1/`) {40 t.Fatalf("%s has no Seedmark room fallback: %s", path, body)41 }42 if strings.Count(body, "<room-avatar") == 0 {43 t.Fatalf("%s has no room avatar element", path)44 }45 }46}4748func TestRoomAvatarSeedAndTenantScoping(t *testing.T) {49 first := roomAvatarURL("https://relay.example", "general")50 if first == "" || first != roomAvatarURL("https://relay.example", "general") {51 t.Fatalf("room avatar URL is not stable: %q", first)52 }53 if first == roomAvatarURL("https://relay.example", "build") {54 t.Fatal("different room IDs share an avatar URL")55 }56 if scoped := injectBase(`<room-avatar fallback="`+first+`"><img src="`+first+`"></room-avatar>`, "/r/alice"); !strings.Contains(scoped, `fallback="/r/alice`+first+`"`) || !strings.Contains(scoped, `src="/r/alice`+first+`"`) {57 t.Fatalf("tenant prefix missed room avatar URL: %s", scoped)58 }59}6061func TestRoomAvatarUsesValidatedPictureForSSR(t *testing.T) {62 base := &roomsBackend{fakeBackend: fakeBackend{policy: policy.Defaults(roomOwner)}}63 app, err := New(&pictureRoomsBackend{roomsBackend: base}, Options{Actor: func(*http.Request) (string, error) { return roomOwner, nil }})64 if err != nil {65 t.Fatal(err)66 }67 recorder := httptest.NewRecorder()68 app.ServeHTTP(recorder, httptest.NewRequest(http.MethodGet, "/rooms/general", nil))69 if recorder.Code != http.StatusOK {70 t.Fatalf("status = %d", recorder.Code)71 }72 body := recorder.Body.String()73 if !strings.Contains(body, `picture="https://cdn.example/general.png"`) || !strings.Contains(body, `<img src="https://cdn.example/general.png"`) {74 t.Fatalf("validated room picture was not preferred in SSR: %s", body)75 }76}