94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
|
package webui
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"tea.karlcom.de/karl/bucketeer/internal/store"
|
||
|
|
)
|
||
|
|
|
||
|
|
type fakeStore struct {
|
||
|
|
lastFilter store.AssetFilter
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *fakeStore) ListAssets(ctx context.Context, fl store.AssetFilter) ([]store.Asset, error) {
|
||
|
|
f.lastFilter = fl
|
||
|
|
return []store.Asset{{ID: 1, Provider: "aws", Service: "s3", ProviderHost: "b.s3.amazonaws.com", Exposure: "public_read"}}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *fakeStore) Stats(ctx context.Context) (store.Stats, error) {
|
||
|
|
return store.Stats{Total: 1, Readable: 1}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestAPI(t *testing.T) {
|
||
|
|
fs := &fakeStore{}
|
||
|
|
srv := httptest.NewServer(New(fs, Config{}))
|
||
|
|
defer srv.Close()
|
||
|
|
|
||
|
|
resp, err := http.Get(srv.URL + "/api/assets?provider=aws&q=bucket&limit=5")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
if resp.StatusCode != http.StatusOK {
|
||
|
|
t.Fatalf("status = %d", resp.StatusCode)
|
||
|
|
}
|
||
|
|
var assets []store.Asset
|
||
|
|
if err := json.NewDecoder(resp.Body).Decode(&assets); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if len(assets) != 1 || fs.lastFilter.Provider != "aws" || fs.lastFilter.Query != "bucket" || fs.lastFilter.Limit != 5 {
|
||
|
|
t.Fatalf("unexpected assets/filter: %+v %+v", assets, fs.lastFilter)
|
||
|
|
}
|
||
|
|
|
||
|
|
statsResp, err := http.Get(srv.URL + "/api/stats")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer statsResp.Body.Close()
|
||
|
|
var st store.Stats
|
||
|
|
if err := json.NewDecoder(statsResp.Body).Decode(&st); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if st.Total != 1 {
|
||
|
|
t.Fatalf("stats = %+v", st)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBasicAuth(t *testing.T) {
|
||
|
|
srv := httptest.NewServer(New(&fakeStore{}, Config{User: "u", Pass: "p"}))
|
||
|
|
defer srv.Close()
|
||
|
|
|
||
|
|
resp, _ := http.Get(srv.URL + "/api/stats")
|
||
|
|
if resp.StatusCode != http.StatusUnauthorized {
|
||
|
|
t.Fatalf("status = %d, want 401", resp.StatusCode)
|
||
|
|
}
|
||
|
|
resp.Body.Close()
|
||
|
|
|
||
|
|
req, _ := http.NewRequest(http.MethodGet, srv.URL+"/api/stats", nil)
|
||
|
|
req.SetBasicAuth("u", "p")
|
||
|
|
resp, err := http.DefaultClient.Do(req)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
if resp.StatusCode != http.StatusOK {
|
||
|
|
t.Fatalf("status = %d, want 200", resp.StatusCode)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIndexServed(t *testing.T) {
|
||
|
|
srv := httptest.NewServer(New(&fakeStore{}, Config{}))
|
||
|
|
defer srv.Close()
|
||
|
|
resp, err := http.Get(srv.URL + "/")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer resp.Body.Close()
|
||
|
|
if ct := resp.Header.Get("Content-Type"); ct != "text/html; charset=utf-8" {
|
||
|
|
t.Fatalf("content-type = %q", ct)
|
||
|
|
}
|
||
|
|
}
|