120 lines
3.0 KiB
Go
120 lines
3.0 KiB
Go
package with_test
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"slices"
|
|
"strings"
|
|
"testing"
|
|
|
|
"gitea.bel.blue/bel/with"
|
|
)
|
|
|
|
func TestCachedHTTP(t *testing.T) {
|
|
called := false
|
|
code := http.StatusAccepted
|
|
headers := http.Header{
|
|
"K": []string{"v"},
|
|
"K2": []string{"v2a", "v2b"},
|
|
}
|
|
body := "response body"
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
called = true
|
|
for k, v := range headers {
|
|
for _, subv := range v {
|
|
w.Header().Add(k, subv)
|
|
}
|
|
}
|
|
w.WriteHeader(code)
|
|
w.Write([]byte(body))
|
|
}))
|
|
defer server.Close()
|
|
|
|
testResp := func(resp *http.Response) {
|
|
if resp.StatusCode != code {
|
|
t.Errorf("initial request wrong status code: %d", resp.StatusCode)
|
|
} else if !slices.Equal(resp.Header["K"], headers["K"]) {
|
|
t.Errorf("initial request wrong headers[k]: %+v in %+v", resp.Header["K"], resp.Header)
|
|
} else if !slices.Equal(resp.Header["K2"], headers["K2"]) {
|
|
t.Errorf("initial request wrong headers[k2]: %+v in %+v", resp.Header["K2"], resp.Header)
|
|
} else if b, _ := io.ReadAll(resp.Body); string(b) != body {
|
|
t.Errorf("initial request wrong resp body: %q", b)
|
|
}
|
|
}
|
|
|
|
if err := with.CachedHTTP(context.Background(), func(c *http.Client) error {
|
|
req := func() *http.Request {
|
|
req, _ := http.NewRequest(http.MethodGet, server.URL+"/my/path", strings.NewReader("my body"))
|
|
return req
|
|
}
|
|
|
|
if resp, err := c.Do(req()); err != nil {
|
|
t.Fatalf("failed initial request: %v", err)
|
|
} else if !called {
|
|
t.Errorf("initial request didnt hit server")
|
|
} else {
|
|
testResp(resp)
|
|
}
|
|
|
|
called = false
|
|
if resp, err := c.Do(req()); err != nil {
|
|
t.Fatalf("failed second request: %v", err)
|
|
} else if called {
|
|
t.Errorf("second request didnt hit cache")
|
|
} else {
|
|
testResp(resp)
|
|
}
|
|
|
|
called = false
|
|
reqDiffURLPath := req()
|
|
reqDiffURLPath.URL.Path += "/teehee"
|
|
if resp, err := c.Do(reqDiffURLPath); err != nil {
|
|
t.Fatalf("failed diff url path request: %v", err)
|
|
} else if !called {
|
|
t.Errorf("new initial hit cache")
|
|
} else {
|
|
testResp(resp)
|
|
}
|
|
|
|
called = false
|
|
reqDiffURLQuery := req()
|
|
reqDiffURLQuery.URL.RawQuery += "hello=world"
|
|
if resp, err := c.Do(reqDiffURLQuery); err != nil {
|
|
t.Fatalf("failed diff url query request: %v", err)
|
|
} else if !called {
|
|
t.Errorf("new initial hit cache")
|
|
} else {
|
|
testResp(resp)
|
|
}
|
|
|
|
called = false
|
|
reqDiffHeader := req()
|
|
reqDiffHeader.Header.Set("Misc", "Misc")
|
|
if resp, err := c.Do(reqDiffHeader); err != nil {
|
|
t.Fatalf("failed diff header request: %v", err)
|
|
} else if !called {
|
|
t.Errorf("new initial hit cache")
|
|
} else {
|
|
testResp(resp)
|
|
}
|
|
|
|
called = false
|
|
reqDiffBody := req()
|
|
reqDiffBody.Body = io.NopCloser(strings.NewReader("diff"))
|
|
reqDiffBody.ContentLength = int64(len("diff"))
|
|
if resp, err := c.Do(reqDiffBody); err != nil {
|
|
t.Fatalf("failed diff body request: %v", err)
|
|
} else if !called {
|
|
t.Errorf("new initial hit cache")
|
|
} else {
|
|
testResp(resp)
|
|
}
|
|
|
|
return nil
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|