mirror of
https://github.com/pomerium/pomerium.git
synced 2025-05-01 03:16:31 +02:00
37 lines
936 B
Go
37 lines
936 B
Go
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)
|
|
}
|