chore: add cron test

Author: KKould

src/query/management/src/task/task_mgr.rs

@@ -274,12 +274,14 @@ impl TaskMgr {
             match schedule_options.schedule_type {
                 ScheduleType::IntervalType => (),
                 ScheduleType::CronType => {
-                    if let Err(e) = schedule_options.time_zone.as_ref().unwrap().parse::<Tz>() {
-                        return Ok(Err(TaskError::InvalidTimezone {
-                            tenant: self.tenant.tenant_name().to_string(),
-                            name: task.task_name.to_string(),
-                            reason: e.to_string(),
-                        }));
+                    if let Some(tz) = &schedule_options.time_zone {
+                        if let Err(e) = tz.parse::<Tz>() {
+                            return Ok(Err(TaskError::InvalidTimezone {
+                                tenant: self.tenant.tenant_name().to_string(),
+                                name: task.task_name.to_string(),
+                                reason: e.to_string(),
+                            }));
+                        }
                     }
                     if let Err(e) = Schedule::from_str(schedule_options.cron.as_ref().unwrap()) {
                         return Ok(Err(TaskError::InvalidCron {

src/query/service/src/task/service.rs

@@ -290,9 +290,9 @@ impl TaskService {
                                 let tz = schedule_options
                                     .time_zone
                                     .as_ref()
-                                    .unwrap()
-                                    .parse::<Tz>()
-                                    .unwrap();
+                                    .map(|tz| tz.parse::<Tz>())
+                                    .transpose()?
+                                    .unwrap_or_else(|| Tz::UCT);
                                 let schedule = Schedule::from_str(cron_expr).unwrap();
 
                                 runtime

tests/task/test-private-task.sh

@@ -228,3 +228,33 @@ if [ "$state14" = "Succeeded" ]; then
 else
   echo "✅ Passed"
 fi
+
+response15=$(curl -s -u root: -XPOST "http://localhost:8000/v1/query" -H 'Content-Type: application/json' -d "{\"sql\": \"CREATE TABLE t2 (c1 int)\"}")
+create_table_query_id_1=$(echo $response15 | jq -r '.id')
+echo "Create Table Query ID: $create_table_query_id_1"
+
+response16=$(curl -s -u root: -XPOST "http://localhost:8000/v1/query" -H 'Content-Type: application/json' -d "{\"sql\": \"CREATE TASK my_task_4 WAREHOUSE = 'mywh' SCHEDULE = USING CRON '*/5 * * * * ?' AS insert into t2 values(0)\"}")
+create_task_4_query_id=$(echo $response16 | jq -r '.id')
+echo "Create Task 4 Query ID: $create_task_4_query_id"
+
+sleep 1
+
+response17=$(curl -s -u root: -XPOST "http://localhost:8000/v1/query" -H 'Content-Type: application/json' -d "{\"sql\": \"ALTER TASK my_task_4 RESUME\"}")
+alter_task_4_query_id=$(echo $response17 | jq -r '.id')
+echo "Resume Task 4 ID: $alter_task_4_query_id"
+
+sleep 11
+
+response18=$(curl -s -u root: -XPOST "http://localhost:8000/v1/query" -H 'Content-Type: application/json' -d "{\"sql\": \"SELECT c1 FROM t2 ORDER BY c1\"}")
+
+actual=$(echo "$response18" | jq -c '.data')
+expected='[["0"],["0"]]'
+
+if [ "$actual" = "$expected" ]; then
+    echo "✅ Query result matches expected"
+else
+    echo "❌ Mismatch"
+    echo "Expected: $expected"
+    echo "Actual  : $actual"
+    exit 1
+fi