diff --git a/swh/scheduler/tests/test_utils.py b/swh/scheduler/tests/test_utils.py index cba3036..4f3a8c9 100644 --- a/swh/scheduler/tests/test_utils.py +++ b/swh/scheduler/tests/test_utils.py @@ -1,82 +1,82 @@ -# Copyright (C) 2017-2018 The Software Heritage developers +# Copyright (C) 2017-2022 The Software Heritage developers # See the AUTHORS file at the top-level directory of this distribution # License: GNU General Public License version 3, or any later version # See top-level LICENSE file for more information from datetime import timezone -import unittest from unittest.mock import patch from swh.scheduler import utils -class UtilsTest(unittest.TestCase): - @patch("swh.scheduler.utils.datetime") - def test_create_oneshot_task_dict_simple(self, mock_datetime): - mock_datetime.now.return_value = "some-date" - - actual_task = utils.create_oneshot_task_dict("some-task-type") - - expected_task = { - "policy": "oneshot", - "type": "some-task-type", - "next_run": "some-date", - "arguments": { - "args": [], - "kwargs": {}, - }, - } - - self.assertEqual(actual_task, expected_task) - mock_datetime.now.assert_called_once_with(tz=timezone.utc) - - @patch("swh.scheduler.utils.datetime") - def test_create_oneshot_task_dict_other_call(self, mock_datetime): - mock_datetime.now.return_value = "some-other-date" - - actual_task = utils.create_oneshot_task_dict( - "some-task-type", "arg0", "arg1", priority="high", other_stuff="normal" - ) - - expected_task = { - "policy": "oneshot", - "type": "some-task-type", - "next_run": "some-other-date", - "arguments": { - "args": ("arg0", "arg1"), - "kwargs": {"other_stuff": "normal"}, - }, - "priority": "high", - } - - self.assertEqual(actual_task, expected_task) - mock_datetime.now.assert_called_once_with(tz=timezone.utc) - - @patch("swh.scheduler.utils.datetime") - def test_create_task_dict(self, mock_datetime): - mock_datetime.now.return_value = "date" - - actual_task = utils.create_task_dict( - "task-type", - "recurring", - "arg0", - "arg1", - priority="low", - other_stuff="normal", - retries_left=3, - ) - - expected_task = { - "policy": "recurring", - "type": "task-type", - "next_run": "date", - "arguments": { - "args": ("arg0", "arg1"), - "kwargs": {"other_stuff": "normal"}, - }, - "priority": "low", - "retries_left": 3, - } - - self.assertEqual(actual_task, expected_task) - mock_datetime.now.assert_called_once_with(tz=timezone.utc) +@patch("swh.scheduler.utils.datetime") +def test_create_oneshot_task_dict_simple(mock_datetime): + mock_datetime.now.return_value = "some-date" + + actual_task = utils.create_oneshot_task_dict("some-task-type") + + expected_task = { + "policy": "oneshot", + "type": "some-task-type", + "next_run": "some-date", + "arguments": { + "args": [], + "kwargs": {}, + }, + } + + assert actual_task == expected_task + mock_datetime.now.assert_called_once_with(tz=timezone.utc) + + +@patch("swh.scheduler.utils.datetime") +def test_create_oneshot_task_dict_other_call(mock_datetime): + mock_datetime.now.return_value = "some-other-date" + + actual_task = utils.create_oneshot_task_dict( + "some-task-type", "arg0", "arg1", priority="high", other_stuff="normal" + ) + + expected_task = { + "policy": "oneshot", + "type": "some-task-type", + "next_run": "some-other-date", + "arguments": { + "args": ("arg0", "arg1"), + "kwargs": {"other_stuff": "normal"}, + }, + "priority": "high", + } + + assert actual_task == expected_task + mock_datetime.now.assert_called_once_with(tz=timezone.utc) + + +@patch("swh.scheduler.utils.datetime") +def test_create_task_dict(mock_datetime): + mock_datetime.now.return_value = "date" + + actual_task = utils.create_task_dict( + "task-type", + "recurring", + "arg0", + "arg1", + priority="low", + other_stuff="normal", + retries_left=3, + ) + + expected_task = { + "policy": "recurring", + "type": "task-type", + "next_run": "date", + "arguments": { + "args": ("arg0", "arg1"), + "kwargs": {"other_stuff": "normal"}, + }, + "priority": "low", + "retries_left": 3, + } + + assert actual_task == expected_task + mock_datetime.now.assert_called_once_with(tz=timezone.utc)