chore: supplement TaskMessage serialization test (#18426)

* test: supplement TaskMessage serialization test

* chore: codefmt

Author: KKould

src/meta/proto-conv/tests/it/v136_add_task.rs

@@ -73,6 +73,133 @@ fn test_decode_v136_add_task() -> anyhow::Result<()> {
     Ok(())
 }
 
+#[test]
+fn test_decode_v136_task_message() -> anyhow::Result<()> {
+    let want_task = || mt::Task {
+        task_id: 11,
+        task_name: "task_c".to_string(),
+        query_text: "SELECT * FROM t1".to_string(),
+        when_condition: Some("c1 > 1".to_string()),
+        after: vec!["task_a".to_string(), "task_b".to_string()],
+        comment: Some("comment".to_string()),
+        owner: "public".to_string(),
+        owner_user: "me".to_string(),
+        schedule_options: Some(ScheduleOptions {
+            interval: Some(11),
+            cron: Some("30 12 * * *".to_string()),
+            time_zone: Some("UTC".to_string()),
+            schedule_type: ScheduleType::IntervalType,
+            milliseconds_interval: Some(11),
+        }),
+        warehouse_options: Some(WarehouseOptions {
+            warehouse: Some("warehouse_a".to_string()),
+            using_warehouse_size: Some("10".to_string()),
+        }),
+        next_scheduled_at: Some(DateTime::from_timestamp(10, 0).unwrap()),
+        suspend_task_after_num_failures: Some(10),
+        error_integration: None,
+        status: mt::Status::Suspended,
+        created_at: DateTime::from_timestamp(11, 0).unwrap(),
+        updated_at: DateTime::from_timestamp(12, 0).unwrap(),
+        last_suspended_at: Some(DateTime::from_timestamp(13, 0).unwrap()),
+        session_params: btreemap! { s("a") => s("b") },
+    };
+
+    {
+        let task_message_execute_v136 = vec![
+            10, 239, 1, 8, 11, 18, 6, 116, 97, 115, 107, 95, 99, 34, 16, 83, 69, 76, 69, 67, 84,
+            32, 42, 32, 70, 82, 79, 77, 32, 116, 49, 42, 7, 99, 111, 109, 109, 101, 110, 116, 50,
+            6, 112, 117, 98, 108, 105, 99, 58, 22, 8, 11, 18, 11, 51, 48, 32, 49, 50, 32, 42, 32,
+            42, 32, 42, 26, 3, 85, 84, 67, 40, 11, 66, 17, 10, 11, 119, 97, 114, 101, 104, 111,
+            117, 115, 101, 95, 97, 18, 2, 49, 48, 74, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49,
+            32, 48, 48, 58, 48, 48, 58, 49, 48, 32, 85, 84, 67, 80, 10, 114, 23, 49, 57, 55, 48,
+            45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49, 49, 32, 85, 84, 67, 122, 23,
+            49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49, 50, 32, 85, 84,
+            67, 130, 1, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49,
+            51, 32, 85, 84, 67, 138, 1, 6, 116, 97, 115, 107, 95, 97, 138, 1, 6, 116, 97, 115, 107,
+            95, 98, 146, 1, 6, 99, 49, 32, 62, 32, 49, 154, 1, 6, 10, 1, 97, 18, 1, 98, 170, 1, 2,
+            109, 101, 160, 6, 136, 1, 168, 6, 24, 160, 6, 136, 1, 168, 6, 24,
+        ];
+        let want_execute = || mt::TaskMessage::ExecuteTask(want_task());
+
+        common::test_pb_from_to(func_name!(), want_execute())?;
+        common::test_load_old(
+            func_name!(),
+            task_message_execute_v136.as_slice(),
+            136,
+            want_execute(),
+        )?;
+    }
+    {
+        let task_message_schedule_v136 = vec![
+            18, 239, 1, 8, 11, 18, 6, 116, 97, 115, 107, 95, 99, 34, 16, 83, 69, 76, 69, 67, 84,
+            32, 42, 32, 70, 82, 79, 77, 32, 116, 49, 42, 7, 99, 111, 109, 109, 101, 110, 116, 50,
+            6, 112, 117, 98, 108, 105, 99, 58, 22, 8, 11, 18, 11, 51, 48, 32, 49, 50, 32, 42, 32,
+            42, 32, 42, 26, 3, 85, 84, 67, 40, 11, 66, 17, 10, 11, 119, 97, 114, 101, 104, 111,
+            117, 115, 101, 95, 97, 18, 2, 49, 48, 74, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49,
+            32, 48, 48, 58, 48, 48, 58, 49, 48, 32, 85, 84, 67, 80, 10, 114, 23, 49, 57, 55, 48,
+            45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49, 49, 32, 85, 84, 67, 122, 23,
+            49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49, 50, 32, 85, 84,
+            67, 130, 1, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49,
+            51, 32, 85, 84, 67, 138, 1, 6, 116, 97, 115, 107, 95, 97, 138, 1, 6, 116, 97, 115, 107,
+            95, 98, 146, 1, 6, 99, 49, 32, 62, 32, 49, 154, 1, 6, 10, 1, 97, 18, 1, 98, 170, 1, 2,
+            109, 101, 160, 6, 136, 1, 168, 6, 24, 160, 6, 136, 1, 168, 6, 24,
+        ];
+        let want_schedule = || mt::TaskMessage::ScheduleTask(want_task());
+
+        common::test_pb_from_to(func_name!(), want_schedule())?;
+        common::test_load_old(
+            func_name!(),
+            task_message_schedule_v136.as_slice(),
+            136,
+            want_schedule(),
+        )?;
+    }
+    {
+        let task_message_after_v136 = vec![
+            34, 239, 1, 8, 11, 18, 6, 116, 97, 115, 107, 95, 99, 34, 16, 83, 69, 76, 69, 67, 84,
+            32, 42, 32, 70, 82, 79, 77, 32, 116, 49, 42, 7, 99, 111, 109, 109, 101, 110, 116, 50,
+            6, 112, 117, 98, 108, 105, 99, 58, 22, 8, 11, 18, 11, 51, 48, 32, 49, 50, 32, 42, 32,
+            42, 32, 42, 26, 3, 85, 84, 67, 40, 11, 66, 17, 10, 11, 119, 97, 114, 101, 104, 111,
+            117, 115, 101, 95, 97, 18, 2, 49, 48, 74, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49,
+            32, 48, 48, 58, 48, 48, 58, 49, 48, 32, 85, 84, 67, 80, 10, 114, 23, 49, 57, 55, 48,
+            45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49, 49, 32, 85, 84, 67, 122, 23,
+            49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49, 50, 32, 85, 84,
+            67, 130, 1, 23, 49, 57, 55, 48, 45, 48, 49, 45, 48, 49, 32, 48, 48, 58, 48, 48, 58, 49,
+            51, 32, 85, 84, 67, 138, 1, 6, 116, 97, 115, 107, 95, 97, 138, 1, 6, 116, 97, 115, 107,
+            95, 98, 146, 1, 6, 99, 49, 32, 62, 32, 49, 154, 1, 6, 10, 1, 97, 18, 1, 98, 170, 1, 2,
+            109, 101, 160, 6, 136, 1, 168, 6, 24, 160, 6, 136, 1, 168, 6, 24,
+        ];
+        let want_after = || mt::TaskMessage::AfterTask(want_task());
+
+        common::test_pb_from_to(func_name!(), want_after())?;
+        common::test_load_old(
+            func_name!(),
+            task_message_after_v136.as_slice(),
+            136,
+            want_after(),
+        )?;
+    }
+    {
+        let task_message_delete_v136 =
+            vec![26, 6, 116, 97, 115, 107, 95, 99, 160, 6, 136, 1, 168, 6, 24];
+        let want_delete = || {
+            let task = want_task();
+            mt::TaskMessage::DeleteTask(task.task_name)
+        };
+
+        common::test_pb_from_to(func_name!(), want_delete())?;
+        common::test_load_old(
+            func_name!(),
+            task_message_delete_v136.as_slice(),
+            136,
+            want_delete(),
+        )?;
+    }
+
+    Ok(())
+}
+
 fn s(ss: impl ToString) -> String {
     ss.to_string()
 }