Add support for EXPIRE command's options (#2002)

* Add support for EXPIRE command's options

* Add requested changes

* Change method arguments

* add variables to the function header

Co-authored-by: dvora-h <67596500+dvora-h@users.noreply.github.com>
Co-authored-by: Chayim <chayim@users.noreply.github.com>
Co-authored-by: dvora-h <dvora.heller@redis.com>

1 files changed, 25 insertions, 0 deletions

diff --git a/tests/test_commands.py b/tests/test_commands.py
index a73541e..c8bcc9c 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -1058,6 +1058,31 @@ class TestRedisCommands:
         assert r.persist("a")
         assert r.ttl("a") == -1
 
+    @skip_if_server_version_lt("7.0.0")
+    def test_expire_option_nx(self, r):
+        r.set("key", "val")
+        assert r.expire("key", 100, nx=True) == 1
+        assert r.expire("key", 500, nx=True) == 0
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_expire_option_xx(self, r):
+        r.set("key", "val")
+        assert r.expire("key", 100, xx=True) == 0
+        assert r.expire("key", 100)
+        assert r.expire("key", 500, nx=True) == 1
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_expire_option_gt(self, r):
+        r.set("key", "val", 100)
+        assert r.expire("key", 50, gt=True) == 0
+        assert r.expire("key", 500, gt=True) == 1
+
+    @skip_if_server_version_lt("7.0.0")
+    def test_expire_option_lt(self, r):
+        r.set("key", "val", 100)
+        assert r.expire("key", 50, lt=True) == 1
+        assert r.expire("key", 150, lt=True) == 0
+
     def test_expireat_datetime(self, r):
         expire_at = redis_server_time(r) + datetime.timedelta(minutes=1)
         r["a"] = "foo"