From 0fff740ba6920e6c48dfeeca63c824ee9ecf62a4 Mon Sep 17 00:00:00 2001 From: Noel Kwan Date: Fri, 30 Dec 2022 22:32:12 +0800 Subject: [PATCH] gen timestamp with time zone --- src/tests/sqlsmith/src/sql_gen/scalar.rs | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/tests/sqlsmith/src/sql_gen/scalar.rs b/src/tests/sqlsmith/src/sql_gen/scalar.rs index 8581463491322..4366de7fa3d29 100644 --- a/src/tests/sqlsmith/src/sql_gen/scalar.rs +++ b/src/tests/sqlsmith/src/sql_gen/scalar.rs @@ -126,7 +126,12 @@ impl<'a, R: Rng> SqlGenerator<'a, R> { let tm = DateTime::::from(SystemTime::now() - Duration::from_secs(*secs)); match typ { T::Date => tm.format("%F").to_string(), - T::Timestamp | T::Timestamptz => tm.format("%Y-%m-%d %H:%M:%S").to_string(), + T::Timestamp => tm.format("%Y-%m-%d %H:%M:%S").to_string(), + T::Timestamptz => { + let timestamp = tm.format("%Y-%m-%d %H:%M:%S"); + let timezone = self.rng.gen_range(0..=15); + format!("{}+{}", timestamp, timezone) + } T::Time => tm.format("%T").to_string(), T::Interval => secs.to_string(), _ => unreachable!(),