chore(ci): fix some flaky test (#16980)

databendlabs · Dec 1, 2024 · b58d1eb · b58d1eb
1 parent 2bb93bf
commit b58d1eb
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 8 deletions.
diff --git a/tests/sqllogictests/suites/query/functions/02_0012_function_datetimes_tz.test b/tests/sqllogictests/suites/query/functions/02_0012_function_datetimes_tz.test
@@ -882,13 +882,22 @@ SELECT substr(DATE_ADD(month, 1, now())::String, 11)=substr(now()::String, 11);
 
 # format: 11
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 6,2)=substr(now()::String, 6, 2)+1;
+SELECT substr(DATE_ADD(month, 1, now())::String, 6, 2) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN '01'
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1
 
 # format: 2024
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 1,4)=substr(now()::String, 1,4);
+SELECT
+substr(DATE_ADD(month, 1, now())::String, 1,4) =
+CASE
+WHEN substr(now()::String, 6, 2) = '12' THEN (substr(now()::String, 1, 4)::int + 1)::String
+ELSE substr(now()::String, 1, 4)
+END;
 ----
 1
 
@@ -938,13 +947,21 @@ SELECT substr(DATE_ADD(month, 1, now())::String, 11)=substr(now()::String, 11);
 
 # format: 11
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 6,2)=substr(now()::String, 6, 2)+1;
+SELECT substr(DATE_ADD(month, 1, now())::String, 6, 2) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN '01'
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1
 
 # format: 2024
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 1,4)=substr(now()::String, 1,4);
+SELECT substr(DATE_ADD(month, 1, now())::String, 1, 4) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN (substr(now()::String, 1,4)::int+1)::String
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1
 

diff --git a/tests/sqllogictests/suites/query/functions/02_0075_function_datetimes_tz.test b/tests/sqllogictests/suites/query/functions/02_0075_function_datetimes_tz.test
@@ -756,13 +756,21 @@ SELECT substr(DATE_ADD(month, 1, now())::String, 11)=substr(now()::String, 11);
 
 # format: 11
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 6,2)=substr(now()::String, 6, 2)+1;
+SELECT substr(DATE_ADD(month, 1, now())::String, 6, 2) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN '01'
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1
 
 # format: 2024
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 1,4)=substr(now()::String, 1,4);
+SELECT substr(DATE_ADD(month, 1, now())::String, 1, 4) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN (substr(now()::String, 1,4)::int+1)::String
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1
 
@@ -802,13 +810,21 @@ SELECT substr(DATE_ADD(month, 1, now())::String, 11)=substr(now()::String, 11);
 
 # format: 11
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 6,2)=substr(now()::String, 6, 2)+1;
+SELECT substr(DATE_ADD(month, 1, now())::String, 6, 2) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN '01'
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1
 
 # format: 2024
 query T
-SELECT substr(DATE_ADD(month, 1, now())::String, 1,4)=substr(now()::String, 1,4);
+SELECT substr(DATE_ADD(month, 1, now())::String, 1, 4) =
+CASE
+   WHEN substr(now()::String, 6, 2) = '12' THEN (substr(now()::String, 1,4)::int+1)::String
+   ELSE (substr(now()::String, 6, 2)::int + 1)::String
+END;
 ----
 1