mocks

type BurstLimiterMock struct {
	rate.Limiter

	OnBurst func() uint64
	OnTakeN func(n uint64) rate.Token
}

type CookiesJar struct {
	OnSetCookies func(u *url.URL, cookies []*http.Cookie)
	OnCookies    func(u *url.URL) []*http.Cookie
}

type LimiterMock struct {
	OnLimit  func() uint64
	OnTokens func() uint64
	OnTake   func() rate.Token
}

type LimiterStoreMock struct {
	OnLimit func(key string) rate.Limiter
}

type ResponseWriter struct {
	OnHeader      func() http.Header
	OnWrite       func([]byte) (int, error)
	OnWriteHeader func(int)
	OnResponse    func(int, interface{}) error
}

type RoundTripper struct {
	OnRoundTrip func(*http.Request) (*http.Response, error)
}

type Stringer struct {
	Value string
}

type TokenMock struct {
	OnAllow    func() bool
	OnResetsAt func() time.Time
	OnUse      func() error
	OnCancel   func()
}

type Writer struct {
	// contains filtered or unexported fields
}