3131import com .datastax .oss .driver .api .core .cql .ExecutionInfo ;
3232import com .datastax .oss .driver .api .core .cql .PreparedStatement ;
3333import com .datastax .oss .driver .api .core .cql .SimpleStatement ;
34+ import com .datastax .oss .driver .api .core .cql .Statement ;
3435import com .datastax .oss .driver .api .testinfra .ccm .CcmRule ;
3536import com .datastax .oss .driver .api .testinfra .ccm .SchemaChangeSynchronizer ;
3637import com .datastax .oss .driver .api .testinfra .session .SessionRule ;
37- import com .datastax .oss .driver .api .testinfra .session .SessionUtils ;
3838import com .datastax .oss .driver .categories .ParallelizableTests ;
3939import com .datastax .oss .driver .internal .core .cql .EmptyColumnDefinitions ;
4040import com .tngtech .java .junit .dataprovider .DataProvider ;
4141import com .tngtech .java .junit .dataprovider .DataProviderRunner ;
4242import edu .umd .cs .findbugs .annotations .NonNull ;
4343import io .reactivex .Flowable ;
44- import java .time .Duration ;
4544import java .util .LinkedHashSet ;
4645import java .util .List ;
4746import java .util .Set ;
@@ -60,14 +59,7 @@ public class DefaultReactiveResultSetIT {
6059
6160 private static CcmRule ccmRule = CcmRule .getInstance ();
6261
63- private static SessionRule <CqlSession > sessionRule =
64- SessionRule .builder (ccmRule )
65- .withConfigLoader (
66- SessionUtils .configLoaderBuilder ()
67- .withDuration (
68- DefaultDriverOption .METADATA_SCHEMA_REQUEST_TIMEOUT , Duration .ofSeconds (20 ))
69- .build ())
70- .build ();
62+ private static SessionRule <CqlSession > sessionRule = SessionRule .builder (ccmRule ).build ();
7163
7264 @ ClassRule public static TestRule chain = RuleChain .outerRule (ccmRule ).around (sessionRule );
7365
@@ -76,19 +68,15 @@ public static void initialize() {
7668 CqlSession session = sessionRule .session ();
7769 SchemaChangeSynchronizer .withLock (
7870 () -> {
79- session .execute ("DROP TABLE IF EXISTS test_reactive_read" );
80- session .execute ("DROP TABLE IF EXISTS test_reactive_write" );
71+ session .execute (createSlowStatement ( "DROP TABLE IF EXISTS test_reactive_read" ) );
72+ session .execute (createSlowStatement ( "DROP TABLE IF EXISTS test_reactive_write" ) );
8173 session .checkSchemaAgreement ();
8274 session .execute (
83- SimpleStatement .builder (
84- "CREATE TABLE test_reactive_read (pk int, cc int, v int, PRIMARY KEY ((pk), cc))" )
85- .setExecutionProfile (sessionRule .slowProfile ())
86- .build ());
75+ createSlowStatement (
76+ "CREATE TABLE test_reactive_read (pk int, cc int, v int, PRIMARY KEY ((pk), cc))" ));
8777 session .execute (
88- SimpleStatement .builder (
89- "CREATE TABLE test_reactive_write (pk int, cc int, v int, PRIMARY KEY ((pk), cc))" )
90- .setExecutionProfile (sessionRule .slowProfile ())
91- .build ());
78+ createSlowStatement (
79+ "CREATE TABLE test_reactive_write (pk int, cc int, v int, PRIMARY KEY ((pk), cc))" ));
9280 session .checkSchemaAgreement ();
9381 });
9482 for (int i = 0 ; i < 1000 ; i ++) {
@@ -101,6 +89,12 @@ public static void initialize() {
10189 }
10290 }
10391
92+ static Statement <?> createSlowStatement (String statement ) {
93+ return SimpleStatement .builder (statement )
94+ .setExecutionProfile (sessionRule .slowProfile ())
95+ .build ();
96+ }
97+
10498 @ Before
10599 public void truncateTables () throws Exception {
106100 CqlSession session = sessionRule .session ();
0 commit comments