simplify timeoutinterceptor (#840)

Co-authored-by: chenmusheng <chenmusheng@laoyuegou.com>
This commit is contained in:
masonchen2014
2021-07-24 21:51:46 +08:00
committed by GitHub
parent ff7443c6a7
commit cb8d9d413a
4 changed files with 12 additions and 42 deletions

View File

@@ -49,25 +49,6 @@ func TestTimeoutInterceptor_timeout(t *testing.T) {
assert.Nil(t, err)
}
func TestTimeoutInterceptor_timeoutExpire(t *testing.T) {
const timeout = time.Millisecond * 10
interceptor := TimeoutInterceptor(timeout)
ctx, cancel := context.WithTimeout(context.Background(), time.Millisecond)
defer cancel()
var wg sync.WaitGroup
wg.Add(1)
cc := new(grpc.ClientConn)
err := interceptor(ctx, "/foo", nil, nil, cc,
func(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn,
opts ...grpc.CallOption) error {
defer wg.Done()
time.Sleep(time.Millisecond * 50)
return nil
})
wg.Wait()
assert.Equal(t, context.DeadlineExceeded, err)
}
func TestTimeoutInterceptor_panic(t *testing.T) {
timeouts := []time.Duration{0, time.Millisecond * 10}
for _, timeout := range timeouts {