Tests: Use DDL helper functions

This commit changes all DDL queries to use helper functions introduced in previous commit. Should fix #1126
scylladb · Nov 20, 2024 · d5b99ba · d5b99ba
1 parent 5787e16
commit d5b99ba
Show file tree

Hide file tree

Showing 18 changed files with 310 additions and 521 deletions.
diff --git a/scylla/src/transport/caching_session.rs b/scylla/src/transport/caching_session.rs
@@ -329,7 +329,9 @@ where
 mod tests {
     use crate::query::Query;
     use crate::statement::PagingState;
-    use crate::test_utils::{create_new_session_builder, scylla_supports_tablets, setup_tracing};
+    use crate::test_utils::{
+        create_new_session_builder, scylla_supports_tablets, setup_tracing, PerformDDL,
+    };
     use crate::transport::partitioner::PartitionerName;
     use crate::transport::session::Session;
     use crate::utils::test_utils::unique_keyspace_name;
@@ -358,18 +360,15 @@ mod tests {
         }
 
         session
-            .query_unpaged(create_ks, &[])
+            .ddl(create_ks)
             .await
             .expect("Could not create keyspace");
 
         session
-            .query_unpaged(
-                format!(
-                    "CREATE TABLE IF NOT EXISTS {}.test_table (a int primary key, b int)",
-                    ks
-                ),
-                &[],
-            )
+            .ddl(format!(
+                "CREATE TABLE IF NOT EXISTS {}.test_table (a int primary key, b int)",
+                ks
+            ))
             .await
             .expect("Could not create table");
 
@@ -566,10 +565,7 @@ mod tests {
         let session: CachingSession = create_caching_session().await;
 
         session
-            .execute_unpaged(
-                "CREATE TABLE IF NOT EXISTS test_batch_table (a int, b int, primary key (a, b))",
-                (),
-            )
+            .ddl("CREATE TABLE IF NOT EXISTS test_batch_table (a int, b int, primary key (a, b))")
             .await
             .unwrap();
 
@@ -689,7 +685,7 @@ mod tests {
         let session: CachingSession = CachingSession::from(new_for_test(true).await, 100);
 
         session
-            .execute_unpaged("CREATE TABLE tbl (a int PRIMARY KEY, b int)", ())
+            .ddl("CREATE TABLE tbl (a int PRIMARY KEY, b int)")
             .await
             .unwrap();
 
@@ -745,10 +741,7 @@ mod tests {
         let session: CachingSession = CachingSession::from(new_for_test(false).await, 100);
 
         session
-            .execute_unpaged(
-                "CREATE TABLE tbl (a int PRIMARY KEY) with cdc = {'enabled': true}",
-                &(),
-            )
+            .ddl("CREATE TABLE tbl (a int PRIMARY KEY) with cdc = {'enabled': true}")
             .await
             .unwrap();
 

diff --git a/scylla/src/transport/connection.rs b/scylla/src/transport/connection.rs
@@ -2397,7 +2397,7 @@ mod tests {
     use crate::transport::connection::open_connection;
     use crate::transport::node::ResolvedContactPoint;
     use crate::transport::topology::UntranslatedEndpoint;
-    use crate::utils::test_utils::unique_keyspace_name;
+    use crate::utils::test_utils::{unique_keyspace_name, PerformDDL};
     use crate::SessionBuilder;
     use futures::{StreamExt, TryStreamExt};
     use std::collections::HashMap;
@@ -2452,17 +2452,14 @@ mod tests {
                 .build()
                 .await
                 .unwrap();
-            session.query_unpaged(format!("CREATE KEYSPACE IF NOT EXISTS {} WITH REPLICATION = {{'class' : 'NetworkTopologyStrategy', 'replication_factor' : 1}}", ks.clone()), &[]).await.unwrap();
+            session.ddl(format!("CREATE KEYSPACE IF NOT EXISTS {} WITH REPLICATION = {{'class' : 'NetworkTopologyStrategy', 'replication_factor' : 1}}", ks.clone())).await.unwrap();
             session.use_keyspace(ks.clone(), false).await.unwrap();
             session
-                .query_unpaged("DROP TABLE IF EXISTS connection_query_iter_tab", &[])
+                .ddl("DROP TABLE IF EXISTS connection_query_iter_tab")
                 .await
                 .unwrap();
             session
-                .query_unpaged(
-                    "CREATE TABLE IF NOT EXISTS connection_query_iter_tab (p int primary key)",
-                    &[],
-                )
+                .ddl("CREATE TABLE IF NOT EXISTS connection_query_iter_tab (p int primary key)")
                 .await
                 .unwrap();
         }
@@ -2548,13 +2545,10 @@ mod tests {
                 .build()
                 .await
                 .unwrap();
-            session.query_unpaged(format!("CREATE KEYSPACE IF NOT EXISTS {} WITH REPLICATION = {{'class' : 'NetworkTopologyStrategy', 'replication_factor' : 1}}", ks.clone()), &[]).await.unwrap();
+            session.ddl(format!("CREATE KEYSPACE IF NOT EXISTS {} WITH REPLICATION = {{'class' : 'NetworkTopologyStrategy', 'replication_factor' : 1}}", ks.clone())).await.unwrap();
             session.use_keyspace(ks.clone(), false).await.unwrap();
             session
-                .query_unpaged(
-                    "CREATE TABLE IF NOT EXISTS t (p int primary key, v blob)",
-                    &[],
-                )
+                .ddl("CREATE TABLE IF NOT EXISTS t (p int primary key, v blob)")
                 .await
                 .unwrap();
         }
@@ -2580,7 +2574,7 @@ mod tests {
                 .await
                 .unwrap();
 
-            connection.query_unpaged("TRUNCATE t").await.unwrap();
+            connection.ddl("TRUNCATE t").await.unwrap();
 
             let mut futs = Vec::new();