diff --git a/test/test_task_on_kart.py b/test/test_task_on_kart.py index e3946b49..86a8a682 100644 --- a/test/test_task_on_kart.py +++ b/test/test_task_on_kart.py @@ -1,5 +1,6 @@ import os import pathlib +import pickle import unittest from datetime import datetime from typing import Any, Dict, List, cast @@ -657,5 +658,18 @@ def test_run_when_complete_check_at_run_is_true_and_task_is_completed(self): task.dump.assert_not_called() +class TestPickleTaskOnKart: + def test_pickle_and_unpickle(self): + task = _DummyTask(redis_host='0.0.0.0', redis_port=12345, redis_timeout=180, should_lock_run=True) + pickled = pickle.dumps(task) + unpickled = pickle.loads(pickled) + assert task.to_str_params() == unpickled.to_str_params() + + task = _DummyTask(should_lock_run=False) + pickled = pickle.dumps(task) + unpickled = pickle.loads(pickled) + assert task.to_str_params() == unpickled.to_str_params() + + if __name__ == '__main__': unittest.main()