fixes and improvements for new server settings menu

internal/mtprotoedge/server_info_http_test.go

@@ -0,0 +1,105 @@
+package mtprotoedge
+
+import (
+	"crypto/rand"
+	"crypto/rsa"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"telesrv/internal/identity"
+)
+
+func TestServeServerInfoIncludesIdentity(t *testing.T) {
+	key, err := rsa.GenerateKey(rand.Reader, 2048)
+	if err != nil {
+		t.Fatal(err)
+	}
+	pem := rsaPublicKeyPEM(key)
+
+	store := identity.NewStore(t.TempDir())
+	if err := store.SetText("OwpenGram", "A self-hosted server."); err != nil {
+		t.Fatal(err)
+	}
+
+	fallback := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { http.NotFound(w, r) })
+	h := serverInfoHTTPHandler(fallback, 2, pem, store)
+
+	req := httptest.NewRequest(http.MethodGet, ServerInfoPath, nil)
+	rec := httptest.NewRecorder()
+	h.ServeHTTP(rec, req)
+	if rec.Code != http.StatusOK {
+		t.Fatalf("status = %d", rec.Code)
+	}
+	var resp ServerInfoResponse
+	if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
+		t.Fatal(err)
+	}
+	if resp.Name != "OwpenGram" || resp.Description != "A self-hosted server." {
+		t.Fatalf("got %+v", resp)
+	}
+	if resp.HasIcon {
+		t.Fatal("has_icon should be false before any upload")
+	}
+}
+
+func TestServeServerInfoWithoutIdentityStore(t *testing.T) {
+	key, err := rsa.GenerateKey(rand.Reader, 2048)
+	if err != nil {
+		t.Fatal(err)
+	}
+	pem := rsaPublicKeyPEM(key)
+
+	fallback := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { http.NotFound(w, r) })
+	h := serverInfoHTTPHandler(fallback, 2, pem, nil)
+
+	req := httptest.NewRequest(http.MethodGet, ServerInfoPath, nil)
+	rec := httptest.NewRecorder()
+	h.ServeHTTP(rec, req)
+	if rec.Code != http.StatusOK {
+		t.Fatalf("status = %d", rec.Code)
+	}
+	var resp ServerInfoResponse
+	if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
+		t.Fatal(err)
+	}
+	if resp.Name != "" || resp.Description != "" || resp.HasIcon {
+		t.Fatalf("expected empty identity fields when store is nil, got %+v", resp)
+	}
+	if resp.RSAPublicKeyPEM == "" {
+		t.Fatal("RSA key should still serve when identity store is nil")
+	}
+}
+
+func TestServeServerIcon(t *testing.T) {
+	store := identity.NewStore(t.TempDir())
+	fallback := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { http.NotFound(w, r) })
+	h := serverInfoHTTPHandler(fallback, 2, []byte("pem"), store)
+
+	// No icon configured yet -> 404.
+	req := httptest.NewRequest(http.MethodGet, ServerIconPath, nil)
+	rec := httptest.NewRecorder()
+	h.ServeHTTP(rec, req)
+	if rec.Code != http.StatusNotFound {
+		t.Fatalf("status = %d, want 404 before upload", rec.Code)
+	}
+
+	png := []byte{0x89, 'P', 'N', 'G', 1, 2, 3, 4}
+	if err := store.SetIcon(png, ".png"); err != nil {
+		t.Fatal(err)
+	}
+
+	req = httptest.NewRequest(http.MethodGet, ServerIconPath, nil)
+	rec = httptest.NewRecorder()
+	h.ServeHTTP(rec, req)
+	if rec.Code != http.StatusOK {
+		t.Fatalf("status = %d", rec.Code)
+	}
+	if rec.Body.String() != string(png) {
+		t.Fatal("icon body mismatch")
+	}
+	if ct := rec.Header().Get("Content-Type"); ct != "image/png" {
+		t.Fatalf("Content-Type = %q", ct)
+	}
+}