60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package broker
|
|
|
|
import (
|
|
"fmt"
|
|
"local/storage"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/time/rate"
|
|
)
|
|
|
|
func TestDoCookies(t *testing.T) {
|
|
limiter = rate.NewLimiter(rate.Limit(20.0), 1)
|
|
calls := 0
|
|
db := storage.NewMap()
|
|
s := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
calls += 1
|
|
if calls == 1 {
|
|
http.SetCookie(w, &http.Cookie{
|
|
Name: "name",
|
|
Value: "value",
|
|
Expires: time.Now().Add(time.Hour),
|
|
})
|
|
} else {
|
|
if !strings.Contains(fmt.Sprint(r.Header["Cookie"]), "name=value") {
|
|
w.WriteHeader(http.StatusBadRequest)
|
|
t.Error("cookie not set on calls after first")
|
|
}
|
|
}
|
|
}))
|
|
defer s.Close()
|
|
req, _ := http.NewRequest(http.MethodGet, s.URL, nil)
|
|
|
|
if cookies, err := getCookies(db, "cookies_"+req.URL.Host); err == nil {
|
|
t.Fatal(err)
|
|
} else if len(cookies) != 0 {
|
|
t.Fatal(cookies)
|
|
}
|
|
|
|
for i := 0; i < 3; i++ {
|
|
resp, err := _do(db, req)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if resp.StatusCode != http.StatusOK {
|
|
t.Fatal(resp.StatusCode)
|
|
}
|
|
resp.Body.Close()
|
|
|
|
if cookies, err := getCookies(db, "cookies_"+req.URL.Host); err != nil {
|
|
t.Fatal(err)
|
|
} else if len(cookies) == 0 {
|
|
t.Fatal(cookies)
|
|
}
|
|
}
|
|
}
|