Page MenuHomeSoftware Heritage
Files F9123227

D7617.id27602.diff
No OneTemporary
Actions

D7617.id27602.diff
View Options

diff --git a/swh/scheduler/tests/test_utils.py b/swh/scheduler/tests/test_utils.py
--- 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)

File Metadata

Mime Type
text/plain
Expires
Fri, Jun 20, 5:11 PM (1 w, 6 d ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3214229

Event Timeline

About · Developer information · Legal