format

Author: abhi-airspace-intelligence

etl-destinations/tests/integration/delta_test.rs

@@ -13,7 +13,7 @@ use chrono::{DateTime, NaiveDate, NaiveDateTime, Utc};
 
 use deltalake::DeltaTableError;
 use deltalake::arrow::array::RecordBatch;
-use deltalake::kernel::{DataType as DeltaDataType};
+use deltalake::kernel::DataType as DeltaDataType;
 use deltalake::operations::collect_sendable_stream;
 
 use crate::common::delta::{MinioDeltaLakeDatabase, setup_delta_connection};
@@ -793,16 +793,16 @@ async fn comprehensive_data_type_mapping() {
 
     let columns = vec![
         ("id", "bigint primary key"), // Manually define id column without sequence
-        ("name", "text"),              // TEXT -> STRING
-        ("age", "int4"),               // INT4 -> INTEGER
-        ("height", "float8"),          // FLOAT8 -> DOUBLE
-        ("active", "bool"),            // BOOL -> BOOLEAN
-        ("birth_date", "date"),        // DATE -> DATE
-        ("created_at", "timestamp"),   // TIMESTAMP -> TIMESTAMP_NTZ (no timezone)
+        ("name", "text"),             // TEXT -> STRING
+        ("age", "int4"),              // INT4 -> INTEGER
+        ("height", "float8"),         // FLOAT8 -> DOUBLE
+        ("active", "bool"),           // BOOL -> BOOLEAN
+        ("birth_date", "date"),       // DATE -> DATE
+        ("created_at", "timestamp"),  // TIMESTAMP -> TIMESTAMP_NTZ (no timezone)
         ("updated_at", "timestamptz"), // TIMESTAMPTZ -> TIMESTAMP (with timezone)
-        ("profile_data", "bytea"),     // BYTEA -> BINARY
-        //("salary", "numeric(10,2)"), // NUMERIC -> DECIMAL
-        // TODO(abhi): Decimal type is currently causing hangs
+        ("profile_data", "bytea"),    // BYTEA -> BINARY
+                                      //("salary", "numeric(10,2)"), // NUMERIC -> DECIMAL
+                                      // TODO(abhi): Decimal type is currently causing hangs
     ];
 
     let table_id = database
@@ -846,15 +846,38 @@ async fn comprehensive_data_type_mapping() {
         .await;
 
     let birth_date = NaiveDate::from_ymd_opt(1993, 1, 15).unwrap();
-    let created_at = NaiveDateTime::parse_from_str("2023-01-01 12:00:00", "%Y-%m-%d %H:%M:%S").unwrap();
-    let updated_at = DateTime::parse_from_rfc3339("2023-01-01T12:00:00+00:00").unwrap().with_timezone(&Utc);
+    let created_at =
+        NaiveDateTime::parse_from_str("2023-01-01 12:00:00", "%Y-%m-%d %H:%M:%S").unwrap();
+    let updated_at = DateTime::parse_from_rfc3339("2023-01-01T12:00:00+00:00")
+        .unwrap()
+        .with_timezone(&Utc);
     let profile_data = b"Hello".to_vec();
 
     database
         .insert_values(
             table_name.clone(),
-            &["id", "name", "age", "height", "active", "birth_date", "created_at", "updated_at", "profile_data"],
-            &[&1i64, &"John Doe", &30i32, &5.9f64, &true, &birth_date, &created_at, &updated_at, &profile_data],
+            &[
+                "id",
+                "name",
+                "age",
+                "height",
+                "active",
+                "birth_date",
+                "created_at",
+                "updated_at",
+                "profile_data",
+            ],
+            &[
+                &1i64,
+                &"John Doe",
+                &30i32,
+                &5.9f64,
+                &true,
+                &birth_date,
+                &created_at,
+                &updated_at,
+                &profile_data,
+            ],
         )
         .await
         .unwrap();
@@ -909,7 +932,8 @@ async fn comprehensive_data_type_mapping() {
         assert_eq!(
             batch.num_columns(),
             columns.len(),
-            "Should have {} columns for comprehensive data types", columns.len()
+            "Should have {} columns for comprehensive data types",
+            columns.len()
         );
 
         // Verify all expected columns are present