소스 검색

rename retry_test package

v2ray 9 년 전
부모
커밋
7e481fe943
2개의 변경된 파일5개의 추가작업 그리고 4개의 파일을 삭제
  1. 2 2
      common/retry/retry.go
  2. 3 2
      common/retry/retry_test.go

+ 2 - 2
common/retry/retry.go

@@ -6,7 +6,7 @@ import (
 )
 
 var (
-	errorRetryFailed = errors.New("All retry attempts failed.")
+	ErrorRetryFailed = errors.New("All retry attempts failed.")
 )
 
 // Strategy is a way to retry on a specific function.
@@ -29,7 +29,7 @@ func (r *retryer) On(method func() error) error {
 		}
 		delay := r.NextDelay(attempt)
 		if delay < 0 {
-			return errorRetryFailed
+			return ErrorRetryFailed
 		}
 		<-time.After(time.Duration(delay) * time.Millisecond)
 		attempt++

+ 3 - 2
common/retry/retry_test.go

@@ -1,10 +1,11 @@
-package retry
+package retry_test
 
 import (
 	"errors"
 	"testing"
 	"time"
 
+	. "github.com/v2ray/v2ray-core/common/retry"
 	v2testing "github.com/v2ray/v2ray-core/testing"
 	"github.com/v2ray/v2ray-core/testing/assert"
 )
@@ -76,6 +77,6 @@ func TestRetryExhausted(t *testing.T) {
 	})
 	duration := time.Since(startTime)
 
-	assert.Error(err).Equals(errorRetryFailed)
+	assert.Error(err).Equals(ErrorRetryFailed)
 	assert.Int64(int64(duration / time.Millisecond)).AtLeast(1900)
 }