|
33 | 33 | import com.datastax.oss.driver.api.core.cql.ResultSet; |
34 | 34 | import com.datastax.oss.driver.api.core.cql.SimpleStatement; |
35 | 35 | import com.datastax.oss.driver.api.core.metadata.Node; |
36 | | -import com.datastax.oss.driver.api.core.servererrors.CoreWriteType; |
| 36 | +import com.datastax.oss.driver.api.core.servererrors.DefaultWriteType; |
37 | 37 | import com.datastax.oss.driver.api.core.servererrors.ReadTimeoutException; |
38 | 38 | import com.datastax.oss.driver.api.core.servererrors.ServerError; |
39 | 39 | import com.datastax.oss.driver.api.core.servererrors.UnavailableException; |
@@ -278,7 +278,7 @@ public void should_retry_on_write_timeout_if_write_type_batch_log() { |
278 | 278 | assertThat(wte.getConsistencyLevel()).isEqualTo(DefaultConsistencyLevel.LOCAL_QUORUM); |
279 | 279 | assertThat(wte.getReceived()).isEqualTo(1); |
280 | 280 | assertThat(wte.getBlockFor()).isEqualTo(3); |
281 | | - assertThat(wte.getWriteType()).isEqualTo(CoreWriteType.BATCH_LOG); |
| 281 | + assertThat(wte.getWriteType()).isEqualTo(DefaultWriteType.BATCH_LOG); |
282 | 282 | } |
283 | 283 |
|
284 | 284 | // there should have been a retry, and it should have been executed on the same host. |
@@ -342,7 +342,7 @@ public void should_not_retry_on_write_timeout_if_write_type_batch_log_but_non_id |
342 | 342 | assertThat(wte.getConsistencyLevel()).isEqualTo(DefaultConsistencyLevel.LOCAL_QUORUM); |
343 | 343 | assertThat(wte.getReceived()).isEqualTo(1); |
344 | 344 | assertThat(wte.getBlockFor()).isEqualTo(3); |
345 | | - assertThat(wte.getWriteType()).isEqualTo(CoreWriteType.BATCH_LOG); |
| 345 | + assertThat(wte.getWriteType()).isEqualTo(DefaultWriteType.BATCH_LOG); |
346 | 346 | } |
347 | 347 |
|
348 | 348 | // should not have been retried. |
|
0 commit comments