Use message.requeue instead of message.reject

Message.requeue does exactly the same as message.reject with requeue=True,
so why not to use it directly.

Change-Id: I42462172c2eb5d5c534e3c55ce7dfcc1a7169e31

1 files changed, 5 insertions, 5 deletions

diff --git a/taskflow/tests/unit/worker_based/test_server.py b/taskflow/tests/unit/worker_based/test_server.py
index 34af585..30e2746 100644
--- a/taskflow/tests/unit/worker_based/test_server.py
+++ b/taskflow/tests/unit/worker_based/test_server.py
@@ -139,26 +139,26 @@ class TestServer(test.MockTestCase):
         ]
         self.assertEqual(self.master_mock.mock_calls, master_mock_calls)
 
-    def test_on_message_proxy_not_running_reject_success(self):
+    def test_on_message_proxy_not_running_requeue_success(self):
         self.proxy_inst_mock.is_running = False
         s = self.server(reset_master_mock=True)
         s._on_message({}, self.message_mock)
 
         # check calls
         master_mock_calls = [
-            mock.call.message.reject(requeue=True)
+            mock.call.message.requeue()
         ]
         self.assertEqual(self.master_mock.mock_calls, master_mock_calls)
 
-    def test_on_message_proxy_not_running_reject_failure(self):
-        self.message_mock.reject.side_effect = exc.MessageStateError('Woot!')
+    def test_on_message_proxy_not_running_requeue_failure(self):
+        self.message_mock.requeue.side_effect = exc.MessageStateError('Woot!')
         self.proxy_inst_mock.is_running = False
         s = self.server(reset_master_mock=True)
         s._on_message({}, self.message_mock)
 
         # check calls
         master_mock_calls = [
-            mock.call.message.reject(requeue=True)
+            mock.call.message.requeue()
         ]
         self.assertEqual(self.master_mock.mock_calls, master_mock_calls)