Backport ccm main

scylla/Cargo.toml

@@ -84,12 +84,15 @@ num-bigint-04 = { package = "num-bigint", version = "0.4" }
 bigdecimal-04 = { package = "bigdecimal", version = "0.4" }
 scylla-proxy = { version = "0.0.3", path = "../scylla-proxy" }
 ntest = "0.9.3"
-criterion = "0.4"                                                      # Note: v0.5 needs at least rust 1.70.0
-tokio = { version = "1.34", features = ["test-util"] }
+criterion = "0.4"                                                       # Note: v0.5 needs at least rust 1.70.0
+tokio = { version = "1.34", features = ["test-util", "process", "fs"] }
 tracing-subscriber = { version = "0.3.14", features = ["env-filter"] }
 assert_matches = "1.5.0"
 rand_chacha = "0.9.0"
 time = "0.3"
+anyhow = "1"
+tempfile = "3"
+
 
 [[bench]]
 name = "benchmark"
@@ -103,4 +106,5 @@ unexpected_cfgs = { level = "warn", check-cfg = [
     'cfg(scylla_cloud_tests)',
     'cfg(cassandra_tests)',
     'cfg(cpp_rust_unstable)',
+    'cfg(ccm_tests)',
 ] }

scylla/tests/integration/ccm/example.rs

@@ -0,0 +1,65 @@
+use std::sync::Arc;
+
+use crate::ccm::lib::cluster::{Cluster, ClusterOptions};
+use crate::ccm::lib::{run_ccm_test, CLUSTER_VERSION};
+use crate::utils::setup_tracing;
+
+use tokio::sync::Mutex;
+use tracing::debug;
+
+fn cluster_1_node() -> ClusterOptions {
+    ClusterOptions {
+        name: "cluster_1_node".to_string(),
+        version: CLUSTER_VERSION.clone(),
+        nodes: vec![1],
+        ..ClusterOptions::default()
+    }
+}
+
+#[tokio::test]
+#[cfg_attr(not(ccm_tests), ignore)]
+async fn test_cluster_lifecycle1() {
+    setup_tracing();
+    async fn test(cluster: Arc<Mutex<Cluster>>) {
+        let cluster = cluster.lock().await;
+        let session = cluster.make_session_builder().await.build().await.unwrap();
+
+        let rows = session
+            .query_unpaged("select data_center from system.local", &[])
+            .await
+            .expect("failed to execute query")
+            .into_rows_result()
+            .expect("failed to get rows")
+            .rows::<(String,)>()
+            .expect("failed to deserialize rows")
+            .map(|res| res.map(|row| row.0))
+            .collect::<Result<Vec<_>, _>>()
+            .unwrap();
+        debug!("{:?}", rows);
+    }
+    run_ccm_test(cluster_1_node, test).await;
+}
+
+#[tokio::test]
+#[cfg_attr(not(ccm_tests), ignore)]
+async fn test_cluster_lifecycle2() {
+    setup_tracing();
+    async fn test(cluster: Arc<Mutex<Cluster>>) {
+        let cluster = cluster.lock().await;
+        let session = cluster.make_session_builder().await.build().await.unwrap();
+
+        let rows = session
+            .query_unpaged("select data_center from system.local", &[])
+            .await
+            .expect("failed to execute query")
+            .into_rows_result()
+            .expect("failed to get rows")
+            .rows::<(String,)>()
+            .expect("failed to deserialize rows")
+            .map(|res| res.map(|row| row.0))
+            .collect::<Result<Vec<_>, _>>()
+            .unwrap();
+        debug!("{:?}", rows);
+    }
+    run_ccm_test(cluster_1_node, test).await;
+}