OSDN Git Service

modify ci
[bytom/vapor.git] / vendor / github.com / go-kit / kit / ratelimit / token_bucket_test.go
diff --git a/vendor/github.com/go-kit/kit/ratelimit/token_bucket_test.go b/vendor/github.com/go-kit/kit/ratelimit/token_bucket_test.go
deleted file mode 100644 (file)
index 3845c9e..0000000
+++ /dev/null
@@ -1,46 +0,0 @@
-package ratelimit_test
-
-import (
-       "context"
-       "strings"
-       "testing"
-       "time"
-
-       "golang.org/x/time/rate"
-
-       "github.com/go-kit/kit/endpoint"
-       "github.com/go-kit/kit/ratelimit"
-)
-
-var nopEndpoint = func(context.Context, interface{}) (interface{}, error) { return struct{}{}, nil }
-
-func TestXRateErroring(t *testing.T) {
-       limit := rate.NewLimiter(rate.Every(time.Minute), 1)
-       testSuccessThenFailure(
-               t,
-               ratelimit.NewErroringLimiter(limit)(nopEndpoint),
-               ratelimit.ErrLimited.Error())
-}
-
-func TestXRateDelaying(t *testing.T) {
-       limit := rate.NewLimiter(rate.Every(time.Minute), 1)
-       testSuccessThenFailure(
-               t,
-               ratelimit.NewDelayingLimiter(limit)(nopEndpoint),
-               "exceed context deadline")
-}
-
-func testSuccessThenFailure(t *testing.T, e endpoint.Endpoint, failContains string) {
-       ctx, cxl := context.WithTimeout(context.Background(), 500*time.Millisecond)
-       defer cxl()
-
-       // First request should succeed.
-       if _, err := e(ctx, struct{}{}); err != nil {
-               t.Errorf("unexpected: %v\n", err)
-       }
-
-       // Next request should fail.
-       if _, err := e(ctx, struct{}{}); !strings.Contains(err.Error(), failContains) {
-               t.Errorf("expected `%s`: %v\n", failContains, err)
-       }
-}