Alessandro Cabbia
accf15410c
* fix: offload prometheus logic to the monitoring package * fix: offload ui logic to the asset package * refactor: use handlers coming from assets and monitoring package * ci: add unit test run and vet command * fix: make vet happy and adjust code
124 lines
3.1 KiB
Go
124 lines
3.1 KiB
Go
package assets
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/sha256"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
// TestAPI is a small test to make sure that behavior didn't change for the
|
|
// worse when switching over to the go1.16+ "embed" package from go-bindata.
|
|
func TestAPIAssets(t *testing.T) {
|
|
tt := []struct {
|
|
Name string
|
|
Path string
|
|
Check func(testing.TB, *http.Response)
|
|
}{
|
|
{
|
|
Name: "Root",
|
|
Path: "/",
|
|
Check: func(t testing.TB, res *http.Response) {
|
|
if res.StatusCode != http.StatusOK {
|
|
t.Fatalf("unexpected response: %v", res.Status)
|
|
return
|
|
}
|
|
h := sha256.New()
|
|
if _, err := io.Copy(h, res.Body); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if got, want := h.Sum(nil), mkDigest(t, "index.html"); !bytes.Equal(got, want) {
|
|
t.Fatalf("bad digest: got: %x, want %x", got, want)
|
|
}
|
|
},
|
|
},
|
|
{
|
|
Name: "NotFound",
|
|
Path: "/glauth.js",
|
|
Check: func(t testing.TB, res *http.Response) {
|
|
if res.StatusCode != http.StatusNotFound {
|
|
t.Fatalf("unexpected response: %v", res.Status)
|
|
}
|
|
},
|
|
},
|
|
{
|
|
Name: "Content-Type",
|
|
Path: "/assets/js/glauth.js",
|
|
Check: func(t testing.TB, res *http.Response) {
|
|
if res.StatusCode != http.StatusOK {
|
|
t.Fatalf("unexpected response: %v", res.Status)
|
|
}
|
|
if got, want := res.Header.Get("Content-Type"), `text/javascript; charset=utf-8`; got != want {
|
|
t.Fatalf("bad content-type: got: %q, want %q", got, want)
|
|
}
|
|
h := sha256.New()
|
|
if _, err := io.Copy(h, res.Body); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if got, want := h.Sum(nil), mkDigest(t, "js/glauth.js"); !bytes.Equal(got, want) {
|
|
t.Fatalf("bad digest: got: %x, want %x", got, want)
|
|
}
|
|
},
|
|
},
|
|
{
|
|
Name: "Content-Type",
|
|
Path: "/assets/css/glauth.css",
|
|
Check: func(t testing.TB, res *http.Response) {
|
|
if res.StatusCode != http.StatusOK {
|
|
t.Fatalf("unexpected response: %v", res.Status)
|
|
}
|
|
// This isn't exactly the same, it adds a charset argument.
|
|
if got, want := res.Header.Get("Content-Type"), `text/css; charset=utf-8`; got != want {
|
|
t.Fatalf("bad content-type: got: %q, want %q", got, want)
|
|
}
|
|
h := sha256.New()
|
|
if _, err := io.Copy(h, res.Body); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if got, want := h.Sum(nil), mkDigest(t, "css/glauth.css"); !bytes.Equal(got, want) {
|
|
t.Fatalf("bad digest: got: %x, want %x", got, want)
|
|
}
|
|
},
|
|
},
|
|
}
|
|
|
|
mux := http.NewServeMux()
|
|
log := zerolog.New(os.Stdout).Level(zerolog.InfoLevel)
|
|
NewAPI(log).RegisterEndpoints(mux)
|
|
srv := httptest.NewServer(mux)
|
|
defer srv.Close()
|
|
for _, tc := range tt {
|
|
t.Run(tc.Name, func(t *testing.T) {
|
|
req, err := http.NewRequest(http.MethodGet, srv.URL+tc.Path, nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
res, err := srv.Client().Do(req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer res.Body.Close()
|
|
tc.Check(t, res)
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func mkDigest(t testing.TB, path string) []byte {
|
|
f, err := Content.Open(path)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer f.Close()
|
|
h := sha256.New()
|
|
if _, err := io.Copy(h, f); err != nil {
|
|
t.Error(err)
|
|
}
|
|
return h.Sum(nil)
|
|
}
|