3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-08-04 01:09:36 +02:00
Code Issues Projects Releases Packages Wiki Activity

proxy: add logo discovery (#5448)

Browse source 
* proxy: add logo discovery

* use a static url for testing
This commit is contained in:
Caleb Doxsey 2025-01-29 12:57:26 -07:00 • committed by GitHub
parent 936bd28ae4
commit 3c5c7fbd31
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 295 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

37
proxy/portal/logo_provider_test.go Normal file
View file

@ -0,0 +1,37 @@
package portal_test
import (
"io"
"net/http"
"net/http/httptest"
"testing"
"time"
"github.com/stretchr/testify/assert"
"github.com/pomerium/pomerium/internal/testutil"
"github.com/pomerium/pomerium/proxy/portal"
)
func TestLogoProvider(t *testing.T) {
t.Parallel()
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
switch r.URL.Path {
case "/icon":
w.Header().Set("Content-Type", "image/vnd.microsoft.icon")
io.WriteString(w, "NOT ACTUALLY AN ICON")
case "/":
io.WriteString(w, `<!doctype html><html><head><link rel="icon" href="/icon" /></head><body></body></html>`)
default:
http.NotFound(w, r)
}
}))
t.Cleanup(srv.Close)
ctx := testutil.GetContext(t, time.Minute)
p := portal.NewLogoProvider()
u, err := p.GetLogoURL(ctx, "", srv.URL)
assert.NoError(t, err)
assert.Equal(t, "data:image/vnd.microsoft.icon;base64,Tk9UIEFDVFVBTExZIEFOIElDT04=", u)
}