Cassandra protocol v5 (#1104)

shotover · Apr 3, 2023 · 9128d66 · 9128d66
1 parent 6c7388a
commit 9128d66
Show file tree

Hide file tree

Showing 4 changed files with 455 additions and 110 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/shotover-proxy/tests/cassandra_int_tests/mod.rs b/shotover-proxy/tests/cassandra_int_tests/mod.rs
@@ -867,7 +867,7 @@ async fn test_protocol_v4(#[case] driver: CassandraDriver) {
 #[case::cdrs(CdrsTokio)]
 #[tokio::test(flavor = "multi_thread")]
 #[serial]
-async fn test_protocol_v5(#[case] _driver: CassandraDriver) {
+async fn test_protocol_v5_single(#[case] driver: CassandraDriver) {
     let _docker_compose =
         DockerCompose::new("example-configs/cassandra-passthrough/docker-compose.yaml");
 
@@ -877,13 +877,13 @@ async fn test_protocol_v5(#[case] _driver: CassandraDriver) {
     .start()
     .await;
 
-    // let connection = || {
-    //     CassandraConnectionBuilder::new("127.0.0.1", 9042, driver)
-    //         .with_protocol_version(ProtocolVersion::V5)
-    //         .build()
-    // };
+    let connection = || {
+        CassandraConnectionBuilder::new("127.0.0.1", 9042, driver)
+            .with_protocol_version(ProtocolVersion::V5)
+            .build()
+    };
 
-    // standard_test_suite(&connection, driver).await;
+    standard_test_suite(&connection, driver).await;
 
     shotover.shutdown_and_then_consume_events(&[]).await;
 }
diff --git a/shotover/Cargo.toml b/shotover/Cargo.toml
@@ -13,6 +13,7 @@ license = "Apache-2.0"
 alpha-transforms = []
 
 [dependencies]
+atomic_enum = "0.2.0"
 pretty-hex = "0.3.0"
 tokio-stream = "0.1.2"
 bytes-utils = "0.1.1"