From a59c2bee986718be155ae71e94ef00b251b5b070 Mon Sep 17 00:00:00 2001
From: conorbros <conorbros@protonmail.com>
Date: Thu, 8 Feb 2024 18:06:32 +1100
Subject: [PATCH] fix batch size error

---
 shotover-proxy/tests/cassandra_int_tests/batch_statements.rs | 2 +-
 shotover-proxy/tests/cassandra_int_tests/mod.rs              | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/shotover-proxy/tests/cassandra_int_tests/batch_statements.rs b/shotover-proxy/tests/cassandra_int_tests/batch_statements.rs
index e8668548e..422efec48 100644
--- a/shotover-proxy/tests/cassandra_int_tests/batch_statements.rs
+++ b/shotover-proxy/tests/cassandra_int_tests/batch_statements.rs
@@ -11,7 +11,7 @@ async fn large_batch(connection: &CassandraConnection) {
     let random_bytes = rand::thread_rng().gen::<[u8; 32]>();
 
     let mut queries = Vec::new();
-    for id in 0..1000 {
+    for id in 0..500 {
         let statement = format!(
             "INSERT INTO batch_keyspace.my_table (id, data) VALUES ({}, 0x{})",
             id,
diff --git a/shotover-proxy/tests/cassandra_int_tests/mod.rs b/shotover-proxy/tests/cassandra_int_tests/mod.rs
index b09a90492..33dc79d7b 100644
--- a/shotover-proxy/tests/cassandra_int_tests/mod.rs
+++ b/shotover-proxy/tests/cassandra_int_tests/mod.rs
@@ -56,7 +56,7 @@ where
     // functions::test(&connection).await;
     prepared_statements_simple::test(&connection, connection_creator, 1).await;
     prepared_statements_all::test(&connection, 1).await;
-    // batch_statements::test(&connection).await;
+    batch_statements::test(&connection).await;
     timestamp::test(&connection).await;
 }