81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"path"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestRun(t *testing.T) {
|
|
targetCalls := 0
|
|
target := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
targetCalls += 1
|
|
}))
|
|
defer target.Close()
|
|
|
|
p := func() int {
|
|
s := httptest.NewServer(http.HandlerFunc(http.NotFound))
|
|
s.Close()
|
|
u := s.URL
|
|
ps := strings.Split(u, ":")[2]
|
|
n, err := strconv.Atoi(ps)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return n
|
|
}()
|
|
tmpl := `{{ . }}`
|
|
x := fmt.Sprintf(`http://localhost:%d`, p)
|
|
y := target.URL
|
|
db := path.Join(t.TempDir(), "db")
|
|
|
|
do := func(method, body string) bool {
|
|
req, _ := http.NewRequest(method, x, strings.NewReader(body))
|
|
resp, err := httpc.Do(req)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
resp.Body.Close()
|
|
return true
|
|
}
|
|
|
|
go func() {
|
|
if err := run(p, tmpl, y, db); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
for !do(http.MethodGet, "") {
|
|
time.Sleep(time.Millisecond * 100)
|
|
}
|
|
|
|
if targetCalls != 1 {
|
|
t.Error("empty req body no called target")
|
|
}
|
|
|
|
if !do(http.MethodGet, "") {
|
|
t.Error("couldnt get a second time with no body")
|
|
} else if targetCalls != 1 {
|
|
t.Error("no dedupe no body")
|
|
}
|
|
|
|
if !do(http.MethodPost, "1") {
|
|
t.Error("couldnt get a third time with new body")
|
|
} else if targetCalls != 2 {
|
|
t.Error("deduped new body")
|
|
} else if !do(http.MethodPost, "1") {
|
|
t.Error("couldnt get a fourth time with new body again")
|
|
} else if targetCalls != 2 {
|
|
t.Error("no deduped new body again")
|
|
} else if !do(http.MethodPost, "2") {
|
|
t.Error("couldnt get a fifth time with new new body")
|
|
} else if targetCalls != 3 {
|
|
t.Error("deduped new new body")
|
|
}
|
|
}
|