SHUTDOWN - add support for the new NOW, FORCE and ABORT modifiers (#2150)

* add support for NOW, FORCE and ABORT modifiers

* linters

* test

* linters

* test params

* fix tests

Co-authored-by: Chayim <chayim@users.noreply.github.com>

1 files changed, 14 insertions, 0 deletions

diff --git a/tests/test_commands.py b/tests/test_commands.py
index 1ae6219..fcd2ed1 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -3,6 +3,7 @@ import datetime
 import re
 import time
 from string import ascii_letters
+from unittest import mock
 
 import pytest
 
@@ -4631,6 +4632,19 @@ class TestRedisCommands:
             assert r.replicaof("NO ONE")
         assert r.replicaof("NO", "ONE")
 
+    def test_shutdown(self, r: redis.Redis):
+        r.execute_command = mock.MagicMock()
+        r.execute_command("SHUTDOWN", "NOSAVE")
+        r.execute_command.assert_called_once_with("SHUTDOWN", "NOSAVE")
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_shutdown_with_params(self, r: redis.Redis):
+        r.execute_command = mock.MagicMock()
+        r.execute_command("SHUTDOWN", "SAVE", "NOW", "FORCE")
+        r.execute_command.assert_called_once_with("SHUTDOWN", "SAVE", "NOW", "FORCE")
+        r.execute_command("SHUTDOWN", "ABORT")
+        r.execute_command.assert_called_with("SHUTDOWN", "ABORT")
+
     @pytest.mark.replica
     @skip_if_server_version_lt("2.8.0")
     @skip_if_redis_enterprise()