2727import static org .assertj .core .api .Assertions .fail ;
2828
2929import com .datastax .oss .driver .api .core .AllNodesFailedException ;
30- import com .datastax .oss .driver .api .core .CoreConsistencyLevel ;
3130import com .datastax .oss .driver .api .core .CqlSession ;
31+ import com .datastax .oss .driver .api .core .DefaultConsistencyLevel ;
3232import com .datastax .oss .driver .api .core .connection .ClosedConnectionException ;
3333import com .datastax .oss .driver .api .core .cql .ResultSet ;
3434import com .datastax .oss .driver .api .core .cql .SimpleStatement ;
@@ -114,7 +114,7 @@ public void should_not_retry_on_read_timeout_when_data_present() {
114114 fail ("Expected a ReadTimeoutException" );
115115 } catch (ReadTimeoutException rte ) {
116116 // then a read timeout exception is thrown
117- assertThat (rte .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
117+ assertThat (rte .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
118118 assertThat (rte .getReceived ()).isEqualTo (1 );
119119 assertThat (rte .getBlockFor ()).isEqualTo (3 );
120120 assertThat (rte .wasDataPresent ()).isTrue ();
@@ -137,7 +137,7 @@ public void should_not_retry_on_read_timeout_when_less_than_blockFor_received()
137137 fail ("Expected a ReadTimeoutException" );
138138 } catch (ReadTimeoutException rte ) {
139139 // then a read timeout exception is thrown
140- assertThat (rte .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
140+ assertThat (rte .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
141141 assertThat (rte .getReceived ()).isEqualTo (2 );
142142 assertThat (rte .getBlockFor ()).isEqualTo (3 );
143143 assertThat (rte .wasDataPresent ()).isFalse ();
@@ -160,7 +160,7 @@ public void should_retry_on_read_timeout_when_enough_responses_and_data_not_pres
160160 fail ("Expected a ReadTimeoutException" );
161161 } catch (ReadTimeoutException rte ) {
162162 // then a read timeout exception is thrown.
163- assertThat (rte .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
163+ assertThat (rte .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
164164 assertThat (rte .getReceived ()).isEqualTo (3 );
165165 assertThat (rte .getBlockFor ()).isEqualTo (3 );
166166 assertThat (rte .wasDataPresent ()).isFalse ();
@@ -275,7 +275,7 @@ public void should_retry_on_write_timeout_if_write_type_batch_log() {
275275 fail ("WriteTimeoutException expected" );
276276 } catch (WriteTimeoutException wte ) {
277277 // then a write timeout exception is thrown
278- assertThat (wte .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
278+ assertThat (wte .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
279279 assertThat (wte .getReceived ()).isEqualTo (1 );
280280 assertThat (wte .getBlockFor ()).isEqualTo (3 );
281281 assertThat (wte .getWriteType ()).isEqualTo (CoreWriteType .BATCH_LOG );
@@ -314,7 +314,7 @@ public void should_not_retry_on_write_timeout_if_write_type_non_batch_log(
314314 fail ("WriteTimeoutException expected" );
315315 } catch (WriteTimeoutException wte ) {
316316 // then a write timeout exception is thrown
317- assertThat (wte .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
317+ assertThat (wte .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
318318 assertThat (wte .getReceived ()).isEqualTo (1 );
319319 assertThat (wte .getBlockFor ()).isEqualTo (3 );
320320 }
@@ -339,7 +339,7 @@ public void should_not_retry_on_write_timeout_if_write_type_batch_log_but_non_id
339339 fail ("WriteTimeoutException expected" );
340340 } catch (WriteTimeoutException wte ) {
341341 // then a write timeout exception is thrown
342- assertThat (wte .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
342+ assertThat (wte .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
343343 assertThat (wte .getReceived ()).isEqualTo (1 );
344344 assertThat (wte .getBlockFor ()).isEqualTo (3 );
345345 assertThat (wte .getWriteType ()).isEqualTo (CoreWriteType .BATCH_LOG );
@@ -389,7 +389,7 @@ public void should_only_retry_once_on_unavailable() {
389389 // tried).
390390 assertThat (ue .getCoordinator ().getConnectAddress ())
391391 .isEqualTo (simulacron .cluster ().node (1 ).inetSocketAddress ());
392- assertThat (ue .getConsistencyLevel ()).isEqualTo (CoreConsistencyLevel .LOCAL_QUORUM );
392+ assertThat (ue .getConsistencyLevel ()).isEqualTo (DefaultConsistencyLevel .LOCAL_QUORUM );
393393 assertThat (ue .getRequired ()).isEqualTo (3 );
394394 assertThat (ue .getAlive ()).isEqualTo (0 );
395395 }
0 commit comments