@@ -21,27 +21,27 @@ private void assertDuration(long from, long to, long ms) {
2121 public void nullStatusesTest () {
2222 RetryConfig config = RetryConfig .retryForever ();
2323
24- Assert .assertNull (config .isThrowableRetryable (null ));
25- Assert .assertNull (config .isStatusRetryable (null ));
24+ Assert .assertNull (config .getThrowableRetryPolicy (null ));
25+ Assert .assertNull (config .getStatusCodeRetryPolicy (null ));
2626 }
2727
2828 @ Test
2929 public void throwableRetriesTest () {
3030 RetryConfig config = RetryConfig .retryUntilElapsed (1000 );
3131
32- Assert .assertNull (config .isThrowableRetryable (new RuntimeException ("test message" )));
33- Assert .assertNull (config .isThrowableRetryable (new Exception ("1" , new RuntimeException ("2" ))));
32+ Assert .assertNull (config .getThrowableRetryPolicy (new RuntimeException ("test message" )));
33+ Assert .assertNull (config .getThrowableRetryPolicy (new Exception ("1" , new RuntimeException ("2" ))));
3434
35- RetryPolicy immediatelly = config .isStatusRetryable (StatusCode .BAD_SESSION );
36- RetryPolicy fast = config .isStatusRetryable (StatusCode .NOT_FOUND );
35+ RetryPolicy immediatelly = config .getStatusCodeRetryPolicy (StatusCode .BAD_SESSION );
36+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .NOT_FOUND );
3737
38- Assert .assertEquals (immediatelly , config .isThrowableRetryable (
38+ Assert .assertEquals (immediatelly , config .getThrowableRetryPolicy (
3939 new UnexpectedResultException ("base" , Status .of (StatusCode .BAD_SESSION )))
4040 );
41- Assert .assertEquals (immediatelly , config .isThrowableRetryable (new Exception ("base" ,
41+ Assert .assertEquals (immediatelly , config .getThrowableRetryPolicy (new Exception ("base" ,
4242 new UnexpectedResultException ("cause" , Status .of (StatusCode .SESSION_BUSY )))
4343 ));
44- Assert .assertEquals (fast , config .isThrowableRetryable (new Exception ("base" ,
44+ Assert .assertEquals (fast , config .getThrowableRetryPolicy (new Exception ("base" ,
4545 new UnexpectedResultException ("cause" , Status .of (StatusCode .NOT_FOUND )))
4646 ));
4747 }
@@ -51,7 +51,7 @@ public void noRetryPolicyTest() {
5151 RetryConfig config = RetryConfig .noRetries ();
5252 // unretrayable
5353 for (StatusCode code : StatusCode .values ()) {
54- Assert .assertNull (config .isStatusRetryable (code ));
54+ Assert .assertNull (config .getStatusCodeRetryPolicy (code ));
5555 }
5656 }
5757
@@ -60,80 +60,80 @@ public void nonIdempotentRetryPolicyTest() {
6060 RetryConfig config = RetryConfig .retryForever ();
6161
6262 // unretrayable
63- Assert .assertNull (config .isStatusRetryable (StatusCode .SCHEME_ERROR ));
64- Assert .assertNull (config .isStatusRetryable (StatusCode .ALREADY_EXISTS ));
65- Assert .assertNull (config .isStatusRetryable (StatusCode .UNAUTHORIZED ));
66- Assert .assertNull (config .isStatusRetryable (StatusCode .UNAVAILABLE ));
67- Assert .assertNull (config .isStatusRetryable (StatusCode .TRANSPORT_UNAVAILABLE ));
68- Assert .assertNull (config .isStatusRetryable (StatusCode .CLIENT_CANCELLED ));
69- Assert .assertNull (config .isStatusRetryable (StatusCode .CLIENT_INTERNAL_ERROR ));
70- Assert .assertNull (config .isStatusRetryable (StatusCode .NOT_FOUND ));
71-
72- RetryPolicy immediatelly = config .isStatusRetryable (StatusCode .BAD_SESSION );
63+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .SCHEME_ERROR ));
64+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .ALREADY_EXISTS ));
65+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .UNAUTHORIZED ));
66+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .UNAVAILABLE ));
67+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .TRANSPORT_UNAVAILABLE ));
68+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .CLIENT_CANCELLED ));
69+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .CLIENT_INTERNAL_ERROR ));
70+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .NOT_FOUND ));
71+
72+ RetryPolicy immediatelly = config .getStatusCodeRetryPolicy (StatusCode .BAD_SESSION );
7373 Assert .assertNotNull (immediatelly );
74- Assert .assertEquals (immediatelly , config .isStatusRetryable (StatusCode .SESSION_BUSY ));
74+ Assert .assertEquals (immediatelly , config .getStatusCodeRetryPolicy (StatusCode .SESSION_BUSY ));
7575
76- RetryPolicy fast = config .isStatusRetryable (StatusCode .ABORTED );
76+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .ABORTED );
7777 Assert .assertNotNull (fast );
78- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .UNDETERMINED ));
78+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .UNDETERMINED ));
7979
80- RetryPolicy slow = config .isStatusRetryable (StatusCode .OVERLOADED );
80+ RetryPolicy slow = config .getStatusCodeRetryPolicy (StatusCode .OVERLOADED );
8181 Assert .assertNotNull (slow );
82- Assert .assertEquals (slow , config .isStatusRetryable (StatusCode .CLIENT_RESOURCE_EXHAUSTED ));
82+ Assert .assertEquals (slow , config .getStatusCodeRetryPolicy (StatusCode .CLIENT_RESOURCE_EXHAUSTED ));
8383 }
8484
8585 @ Test
8686 public void idempotentRetryPolicyTest () {
8787 RetryConfig config = RetryConfig .idempotentRetryForever ();
8888
8989 // unretrayable
90- Assert .assertNull (config .isStatusRetryable (StatusCode .SCHEME_ERROR ));
91- Assert .assertNull (config .isStatusRetryable (StatusCode .ALREADY_EXISTS ));
92- Assert .assertNull (config .isStatusRetryable (StatusCode .UNAUTHORIZED ));
93- Assert .assertNull (config .isStatusRetryable (StatusCode .NOT_FOUND ));
90+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .SCHEME_ERROR ));
91+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .ALREADY_EXISTS ));
92+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .UNAUTHORIZED ));
93+ Assert .assertNull (config .getStatusCodeRetryPolicy (StatusCode .NOT_FOUND ));
9494
95- RetryPolicy immediatelly = config .isStatusRetryable (StatusCode .BAD_SESSION );
95+ RetryPolicy immediatelly = config .getStatusCodeRetryPolicy (StatusCode .BAD_SESSION );
9696 Assert .assertNotNull (immediatelly );
97- Assert .assertEquals (immediatelly , config .isStatusRetryable (StatusCode .SESSION_BUSY ));
97+ Assert .assertEquals (immediatelly , config .getStatusCodeRetryPolicy (StatusCode .SESSION_BUSY ));
9898
99- RetryPolicy fast = config .isStatusRetryable (StatusCode .ABORTED );
99+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .ABORTED );
100100 Assert .assertNotNull (fast );
101- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .UNDETERMINED ));
102- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .UNAVAILABLE ));
103- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .TRANSPORT_UNAVAILABLE ));
104- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .CLIENT_CANCELLED ));
105- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .CLIENT_INTERNAL_ERROR ));
101+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .UNDETERMINED ));
102+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .UNAVAILABLE ));
103+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .TRANSPORT_UNAVAILABLE ));
104+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .CLIENT_CANCELLED ));
105+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .CLIENT_INTERNAL_ERROR ));
106106
107- RetryPolicy slow = config .isStatusRetryable (StatusCode .OVERLOADED );
107+ RetryPolicy slow = config .getStatusCodeRetryPolicy (StatusCode .OVERLOADED );
108108 Assert .assertNotNull (slow );
109- Assert .assertEquals (slow , config .isStatusRetryable (StatusCode .CLIENT_RESOURCE_EXHAUSTED ));
109+ Assert .assertEquals (slow , config .getStatusCodeRetryPolicy (StatusCode .CLIENT_RESOURCE_EXHAUSTED ));
110110 }
111111
112112 @ Test
113113 public void notFoundRetryPolicyTest () {
114114 RetryConfig config = RetryConfig .newConfig ().retryNotFound (true ).retryForever ();
115115
116- RetryPolicy fast = config .isStatusRetryable (StatusCode .ABORTED );
117- Assert .assertEquals (fast , config .isStatusRetryable (StatusCode .NOT_FOUND ));
116+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .ABORTED );
117+ Assert .assertEquals (fast , config .getStatusCodeRetryPolicy (StatusCode .NOT_FOUND ));
118118 }
119119
120120 @ Test
121121 public void foreverRetryTest () {
122122 RetryConfig config = RetryConfig .newConfig ().withSlowBackoff (100 , 5 ).withFastBackoff (10 , 10 ).retryForever ();
123123
124- RetryPolicy immediatelly = config .isStatusRetryable (StatusCode .BAD_SESSION );
124+ RetryPolicy immediatelly = config .getStatusCodeRetryPolicy (StatusCode .BAD_SESSION );
125125 Assert .assertEquals (0 , immediatelly .nextRetryMs (0 , 0 ));
126126 Assert .assertEquals (0 , immediatelly .nextRetryMs (0 , Integer .MAX_VALUE ));
127127 Assert .assertEquals (0 , immediatelly .nextRetryMs (Integer .MAX_VALUE , 0 ));
128128 Assert .assertEquals (0 , immediatelly .nextRetryMs (Integer .MAX_VALUE , Integer .MAX_VALUE ));
129129
130- RetryPolicy fast = config .isStatusRetryable (StatusCode .ABORTED );
130+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .ABORTED );
131131 assertDuration (10 , 20 , fast .nextRetryMs (0 , 0 ));
132132 assertDuration (10 , 20 , fast .nextRetryMs (0 , Integer .MAX_VALUE ));
133133 assertDuration (10240 , 20480 , fast .nextRetryMs (Integer .MAX_VALUE , 0 ));
134134 assertDuration (10240 , 20480 , fast .nextRetryMs (Integer .MAX_VALUE , Integer .MAX_VALUE ));
135135
136- RetryPolicy slow = config .isStatusRetryable (StatusCode .OVERLOADED );
136+ RetryPolicy slow = config .getStatusCodeRetryPolicy (StatusCode .OVERLOADED );
137137 assertDuration (100 , 200 , slow .nextRetryMs (0 , 0 ));
138138 assertDuration (100 , 200 , slow .nextRetryMs (0 , Integer .MAX_VALUE ));
139139 assertDuration (3200 , 6400 , slow .nextRetryMs (Integer .MAX_VALUE , 0 ));
@@ -144,22 +144,22 @@ public void foreverRetryTest() {
144144 public void untilElapsedRetryTest () {
145145 RetryConfig config = RetryConfig .idempotentRetryUntilElapsed (5000 );
146146
147- RetryPolicy immediatelly = config .isStatusRetryable (StatusCode .BAD_SESSION );
147+ RetryPolicy immediatelly = config .getStatusCodeRetryPolicy (StatusCode .BAD_SESSION );
148148 Assert .assertEquals (0 , immediatelly .nextRetryMs (0 , 0 ));
149149 Assert .assertEquals (0 , immediatelly .nextRetryMs (0 , 5000 ));
150150 Assert .assertEquals (0 , immediatelly .nextRetryMs (Integer .MAX_VALUE , 0 ));
151151 Assert .assertEquals (0 , immediatelly .nextRetryMs (Integer .MAX_VALUE , 5000 ));
152152 Assert .assertEquals (-1 , immediatelly .nextRetryMs (0 , 5001 ));
153153 Assert .assertEquals (-1 , immediatelly .nextRetryMs (Integer .MAX_VALUE , 5001 ));
154154
155- RetryPolicy fast = config .isStatusRetryable (StatusCode .ABORTED );
155+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .ABORTED );
156156 assertDuration (5 , 10 , fast .nextRetryMs (0 , 0 ));
157157 Assert .assertEquals (3 , fast .nextRetryMs (0 , 4997 ));
158158 Assert .assertEquals (5000 , fast .nextRetryMs (Integer .MAX_VALUE , 0 ));
159159 Assert .assertEquals (1 , fast .nextRetryMs (Integer .MAX_VALUE , 4999 ));
160160 Assert .assertEquals (-1 , fast .nextRetryMs (Integer .MAX_VALUE , 5000 ));
161161
162- RetryPolicy slow = config .isStatusRetryable (StatusCode .OVERLOADED );
162+ RetryPolicy slow = config .getStatusCodeRetryPolicy (StatusCode .OVERLOADED );
163163 assertDuration (500 , 1000 , slow .nextRetryMs (0 , 0 ));
164164 Assert .assertEquals (3 , slow .nextRetryMs (0 , 4997 ));
165165 Assert .assertEquals (5000 , slow .nextRetryMs (Integer .MAX_VALUE , 0 ));
@@ -171,23 +171,23 @@ public void untilElapsedRetryTest() {
171171 public void nTimesRetryTest () {
172172 RetryConfig config = RetryConfig .newConfig ().retryNTimes (8 );
173173
174- RetryPolicy immediatelly = config .isStatusRetryable (StatusCode .BAD_SESSION );
174+ RetryPolicy immediatelly = config .getStatusCodeRetryPolicy (StatusCode .BAD_SESSION );
175175 Assert .assertEquals (0 , immediatelly .nextRetryMs (0 , 0 ));
176176 Assert .assertEquals (0 , immediatelly .nextRetryMs (0 , Integer .MAX_VALUE ));
177177 Assert .assertEquals (0 , immediatelly .nextRetryMs (7 , 0 ));
178178 Assert .assertEquals (0 , immediatelly .nextRetryMs (7 , Integer .MAX_VALUE ));
179179 Assert .assertEquals (-1 , immediatelly .nextRetryMs (8 , 0 ));
180180 Assert .assertEquals (-1 , immediatelly .nextRetryMs (8 , Integer .MAX_VALUE ));
181181
182- RetryPolicy fast = config .isStatusRetryable (StatusCode .ABORTED );
182+ RetryPolicy fast = config .getStatusCodeRetryPolicy (StatusCode .ABORTED );
183183 assertDuration (5 , 10 , fast .nextRetryMs (0 , 0 ));
184184 assertDuration (5 , 10 , fast .nextRetryMs (0 , Integer .MAX_VALUE ));
185185 assertDuration (5 * 128 , 5 * 256 , fast .nextRetryMs (7 , 0 ));
186186 assertDuration (5 * 128 , 5 * 256 , fast .nextRetryMs (7 , Integer .MAX_VALUE ));
187187 Assert .assertEquals (-1 , fast .nextRetryMs (8 , 0 ));
188188 Assert .assertEquals (-1 , fast .nextRetryMs (8 , Integer .MAX_VALUE ));
189189
190- RetryPolicy slow = config .isStatusRetryable (StatusCode .OVERLOADED );
190+ RetryPolicy slow = config .getStatusCodeRetryPolicy (StatusCode .OVERLOADED );
191191 assertDuration (500 , 1000 , slow .nextRetryMs (0 , 0 ));
192192 assertDuration (500 , 1000 , slow .nextRetryMs (0 , Integer .MAX_VALUE ));
193193 assertDuration (500 * 64 , 500 * 128 , slow .nextRetryMs (7 , 0 ));
0 commit comments