Move test_prepared_config to statement.rs and add additional cases to it

Author: sylwiaszunejko

scylla/tests/integration/session.rs

@@ -1013,21 +1013,6 @@ async fn test_request_timeout() {
     }
 }
 
-#[tokio::test]
-async fn test_prepared_config() {
-    setup_tracing();
-    let session = create_new_session_builder().build().await.unwrap();
-
-    let mut query = Query::new("SELECT * FROM system_schema.tables");
-    query.set_is_idempotent(true);
-    query.set_page_size(42);
-
-    let prepared_statement = session.prepare(query).await.unwrap();
-
-    assert!(prepared_statement.get_is_idempotent());
-    assert_eq!(prepared_statement.get_page_size(), 42);
-}
-
 fn udt_type_a_def(ks: &str) -> Arc<UserDefinedType<'_>> {
     Arc::new(UserDefinedType {
         name: "type_a".into(),

scylla/tests/integration/statement.rs

@@ -3,10 +3,45 @@ use scylla::{
     batch::{Batch, BatchType},
     query::Query,
     response::PagingState,
+    statement::SerialConsistency,
 };
+use scylla_cql::Consistency;
 use std::collections::HashSet;
 use std::sync::{Arc, Mutex};
 
+#[tokio::test]
+async fn test_prepared_config() {
+    use std::time::Duration;
+
+    setup_tracing();
+    let session = create_new_session_builder().build().await.unwrap();
+
+    let mut query = Query::new("SELECT * FROM system_schema.tables");
+    query.set_is_idempotent(true);
+    query.set_page_size(42);
+    query.set_consistency(Consistency::One);
+    query.set_serial_consistency(Some(SerialConsistency::LocalSerial));
+    query.set_tracing(true);
+    query.set_request_timeout(Some(Duration::from_millis(1)));
+    query.set_timestamp(Some(42));
+
+    let prepared_statement = session.prepare(query).await.unwrap();
+
+    assert!(prepared_statement.get_is_idempotent());
+    assert_eq!(prepared_statement.get_page_size(), 42);
+    assert_eq!(prepared_statement.get_consistency(), Some(Consistency::One));
+    assert_eq!(
+        prepared_statement.get_serial_consistency(),
+        Some(SerialConsistency::LocalSerial)
+    );
+    assert!(prepared_statement.get_tracing());
+    assert_eq!(
+        prepared_statement.get_request_timeout(),
+        Some(Duration::from_millis(1))
+    );
+    assert_eq!(prepared_statement.get_timestamp(), Some(42));
+}
+
 #[tokio::test]
 async fn test_timestamp() {
     setup_tracing();