tests: Add more scalar type tests, normal uuid and string type tests

This commit adds additional test to test various simple types such as
int and float, as well as one more uuid test and some simple string type
tests

Author: k0machi

scylla/tests/integration/serialization.rs

@@ -97,6 +97,60 @@ fn handle_incorrect_dates(error: &DeserializationError) -> bool {
     }
 }
 
+// Native type test, which expects value to have the display trait and be trivially copyable
+async fn run_native_serialize_test<T>(table_name: &str, type_name: &str, values: Vec<Option<T>>)
+where
+    T: SerializeValue
+        + DeserializeOwnedValue
+        + FromStr
+        + Debug
+        + Clone
+        + PartialEq
+        + Default
+        + Display
+        + Copy,
+{
+    let session = init_test(table_name, type_name).await;
+    tracing::debug!("Active keyspace: {}", session.get_keyspace().unwrap());
+
+    for original_value in values {
+        session
+            .query_unpaged(
+                format!("INSERT INTO {} (id, val) VALUES (1, ?)", table_name),
+                (&original_value,),
+            )
+            .await
+            .unwrap();
+
+        let selected_value: Vec<T> = session
+            .query_unpaged(format!("SELECT val FROM {}", table_name), &[])
+            .await
+            .unwrap()
+            .into_rows_result()
+            .unwrap()
+            .rows::<(T,)>()
+            .unwrap()
+            .map(|row| match row {
+                Ok((val,)) => val,
+                Err(e) => handle_deserialize_error(&e, || T::default()),
+            })
+            .collect::<Vec<_>>();
+
+        for value in selected_value {
+            tracing::debug!(
+                "Received: {} | Expected: {}",
+                value,
+                if original_value.is_none() {
+                    T::default()
+                } else {
+                    original_value.unwrap()
+                }
+            );
+            assert_eq!(value, original_value.unwrap_or(T::default()));
+        }
+    }
+}
+
 // Test for types that either don't have the Display trait or foreign types from other libraries
 async fn run_foreign_serialize_test<T>(table_name: &str, type_name: &str, values: Vec<Option<T>>)
 where
@@ -277,6 +331,146 @@ async fn run_literal_input_maybe_empty_output_test<X>(
     }
 }
 
+// Scalar types
+
+#[tokio::test]
+async fn test_serialize_deserialize_bigint() {
+    setup_tracing();
+    const TABLE_NAME: &str = "bigint_serialization_test";
+
+    let test_cases = vec![
+        Some(i64::MIN),
+        Some(i64::MAX),
+        Some(1),
+        Some(-1),
+        Some(0),
+        None,
+    ];
+
+    run_native_serialize_test(TABLE_NAME, "bigint", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_int() {
+    setup_tracing();
+    const TABLE_NAME: &str = "int_serialization_test";
+
+    let test_cases = vec![
+        Some(i32::MIN),
+        Some(i32::MAX),
+        Some(1),
+        Some(-1),
+        Some(0),
+        None,
+    ];
+
+    run_native_serialize_test(TABLE_NAME, "int", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_smallint() {
+    setup_tracing();
+    const TABLE_NAME: &str = "smallint_serialization_test";
+
+    let test_cases = vec![
+        Some(i16::MIN),
+        Some(i16::MAX),
+        Some(1),
+        Some(-1),
+        Some(0),
+        None,
+    ];
+
+    run_native_serialize_test(TABLE_NAME, "smallint", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_tinyint() {
+    setup_tracing();
+    const TABLE_NAME: &str = "tinyint_serialization_test";
+
+    let test_cases = vec![
+        Some(i8::MIN),
+        Some(i8::MAX),
+        Some(1),
+        Some(-1),
+        Some(0),
+        None,
+    ];
+
+    run_native_serialize_test(TABLE_NAME, "tinyint", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_float() {
+    setup_tracing();
+    const TABLE_NAME: &str = "float_serialization_test";
+
+    let test_cases = vec![
+        Some(f32::MIN),
+        Some(f32::MAX),
+        Some(1.0),
+        Some(-1.0),
+        Some(0.0),
+        None,
+    ];
+
+    run_native_serialize_test(TABLE_NAME, "float", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_bool() {
+    setup_tracing();
+    const TABLE_NAME: &str = "bool_serialization_test";
+
+    let test_cases = vec![Some(true), Some(false), None];
+
+    run_native_serialize_test(TABLE_NAME, "boolean", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_double() {
+    setup_tracing();
+    const TABLE_NAME: &str = "double_serialization_test";
+
+    let test_cases = vec![
+        Some(f64::MIN),
+        Some(f64::MAX),
+        Some(1.0),
+        Some(-1.0),
+        Some(0.0),
+        None,
+    ];
+
+    run_native_serialize_test(TABLE_NAME, "double", test_cases).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_double_literal() {
+    setup_tracing();
+    const TABLE_NAME: &str = "double_serialization_test";
+
+    let test_cases = [f64::MIN, f64::MAX, 1.0, -1.0, 0.0, 0.1]
+        .iter()
+        .map(|val| (val.to_string(), *val))
+        .collect();
+
+    run_literal_and_bound_value_test(TABLE_NAME, "double", test_cases, false).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_float_literal() {
+    setup_tracing();
+    const TABLE_NAME: &str = "float_serialization_test";
+
+    let test_cases = [f32::MIN, f32::MAX, 1.0, -1.0, 0.0, 0.1]
+        .iter()
+        .map(|val| (val.to_string(), *val))
+        .collect();
+
+    run_literal_and_bound_value_test(TABLE_NAME, "float", test_cases, false).await;
+}
+
 // Arbitrary precision types
 
 #[tokio::test]
@@ -1242,6 +1436,19 @@ async fn test_serialize_deserialize_offset_date_time_03_high_precision_rounding_
 
 //UUID Types
 
+#[tokio::test]
+async fn test_serialize_deserialize_uuid() {
+    setup_tracing();
+    const TABLE_NAME: &str = "uuid_serialization_test";
+
+    let tests = (0..100)
+        .map(|_| uuid::Uuid::new_v4())
+        .map(|new_uuid| (new_uuid.to_string(), new_uuid))
+        .collect();
+
+    run_literal_and_bound_value_test(TABLE_NAME, "uuid", tests, false).await;
+}
+
 #[tokio::test]
 async fn test_serialize_deserialize_timeuuid() {
     setup_tracing();
@@ -1350,6 +1557,34 @@ async fn test_serialize_deserialize_timeuuid_ordering() {
     }
 }
 
+#[tokio::test]
+async fn test_serialize_deserialize_strings_varchar() {
+    let table_name = "varchar_serialization_test";
+
+    let test_cases = vec![
+        ("Hello, World!", String::from("Hello, World!")),
+        ("Hello, Мир", String::from("Hello, Мир")), // multi-byte string
+        ("🦀A🦀B🦀C", String::from("🦀A🦀B🦀C")),
+        ("おはようございます", String::from("おはようございます")),
+    ];
+
+    run_literal_and_bound_value_test(table_name, "varchar", test_cases, true).await;
+}
+
+#[tokio::test]
+async fn test_serialize_deserialize_strings_ascii() {
+    let table_name = "ascii_serialization_test";
+
+    let test_cases = vec![
+        ("Hello, World!", String::from("Hello, World!")),
+        (
+            str::from_utf8(&[0x0, 0x7f]).unwrap(),
+            String::from_utf8(vec![0x0, 0x7f]).unwrap(),
+        ), // min/max ASCII values
+    ];
+
+    run_literal_and_bound_value_test(table_name, "varchar", test_cases, true).await;
+}
 
 // UDT Types