From caa18746be7cdccb7562235e218570707eb55476 Mon Sep 17 00:00:00 2001
From: hwware <wen.hui.ware@gmail.com>
Date: Mon, 8 Jul 2024 19:04:08 +0000
Subject: [PATCH] update one test case for PUNSUBSCRIBE and UNSUBSCRIBE

Signed-off-by: hwware <wen.hui.ware@gmail.com>
---
 tests/unit/pubsub.tcl | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/tests/unit/pubsub.tcl b/tests/unit/pubsub.tcl
index 3214c5496f..cd1f74917e 100644
--- a/tests/unit/pubsub.tcl
+++ b/tests/unit/pubsub.tcl
@@ -235,15 +235,17 @@ start_server {tags {"pubsub network"}} {
         $rd1 close
     }
 
-    #test "PUNSUBSCRIBE and UNSUBSCRIBE should always reply" {
+    test "PUNSUBSCRIBE and UNSUBSCRIBE should always reply when they are executed in subscribe mode" {
         # Make sure we are not subscribed to any channel at all.
-    #    r punsubscribe
-    #    r unsubscribe
-        # Now check if the commands still reply correctly.
-    #    set reply1 [r punsubscribe]
-    #    set reply2 [r unsubscribe]
-    #    concat $reply1 $reply2
-    #} {punsubscribe {} 0 unsubscribe {} 0}
+        set rd1 [valkey_deferring_client]
+        assert_equal {1} [subscribe $rd1 {foo}]
+        assert_equal {2} [psubscribe $rd1 {foo.*}]
+        set reply1 [punsubscribe $rd1]
+        set reply2 [unsubscribe $rd1]
+        concat $reply1 $reply2
+        # clean up clients
+        $rd1 close
+    } {0}
 
     ### Keyspace events notification tests