Skip to content

Commit c97985f

Browse files
committed
Update test case for subv2 version
Signed-off-by: hwware <[email protected]>
1 parent 7bb578b commit c97985f

File tree

4 files changed

+30
-28
lines changed

4 files changed

+30
-28
lines changed

tests/integration/valkey-cli.tcl

+28-8
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,30 @@ start_server {tags {"cli"}} {
194194
assert_equal "bar" [r get key]
195195
}
196196

197+
test_interactive_cli "Subscribed mode -- deprecated" {
198+
if {$::force_resp3} {
199+
run_command $fd "hello 3"
200+
}
201+
202+
set reading "Reading messages... (press Ctrl-C to quit or any key to type command)\r"
203+
set erase "\033\[K"; # Erases the "Reading messages..." line.
204+
205+
# Subscribe to some channels.
206+
set sub1 "1) \"subscribe\"\n2) \"ch1\"\n3) (integer) 1\n"
207+
set sub2 "1) \"subscribe\"\n2) \"ch2\"\n3) (integer) 2\n"
208+
set sub3 "1) \"subscribe\"\n2) \"ch3\"\n3) (integer) 3\n"
209+
assert_equal $sub1$sub2$sub3$reading \
210+
[run_command $fd "subscribe ch1 ch2 ch3"]
211+
212+
# Unsubscribe all.
213+
set unsub1 "1) \"unsubscribe\"\n2) \"ch1\"\n3) (integer) 2\n"
214+
set unsub2 "1) \"unsubscribe\"\n2) \"ch2\"\n3) (integer) 1\n"
215+
set unsub3 "1) \"unsubscribe\"\n2) \"ch3\"\n3) (integer) 0\n"
216+
assert_equal $erase$unsub1$unsub2$unsub3$reading \
217+
[run_command $fd "unsubscribe ch1 ch2 ch3"]
218+
219+
}
220+
197221
test_interactive_cli "Subscribed mode" {
198222
if {$::force_resp3} {
199223
run_command $fd "hello 3"
@@ -202,15 +226,15 @@ start_server {tags {"cli"}} {
202226
set reading "Reading messages... (press Ctrl-C to quit or any key to type command)\r"
203227
set erase "\033\[K"; # Erases the "Reading messages..." line.
204228

229+
run_command $fd "client capa subv2"
230+
205231
# Subscribe to some channels.
206232
set sub1 "1) \"subscribe\"\n2) \"ch1\"\n3) (integer) 1\n"
207-
set sub2 "1) \"subscribe\"\n2) \"ch2\"\n3) (integer) 2\n"
208-
set sub3 "1) \"subscribe\"\n2) \"ch3\"\n3) (integer) 3\n"
233+
set sub2 "4) \"subscribe\"\n5) \"ch2\"\n6) (integer) 2\n"
234+
set sub3 "7) \"subscribe\"\n8) \"ch3\"\n9) (integer) 3\n"
209235
assert_equal $sub1$sub2$sub3$reading \
210236
[run_command $fd "subscribe ch1 ch2 ch3"]
211237

212-
# set sub2 "4) \"subscribe\"\n5) \"ch2\"\n6) (integer) 2\n"
213-
# set sub3 "7) \"subscribe\"\n8) \"ch3\"\n9) (integer) 3\n"
214238
# Receive pubsub message.
215239
r publish ch2 hello
216240
set message "1) \"message\"\n2) \"ch2\"\n3) \"hello\"\n"
@@ -243,10 +267,6 @@ start_server {tags {"cli"}} {
243267
[run_command $fd "subscribe ch1"]
244268
}
245269

246-
test_interactive_cli "Subscribed mode" {
247-
248-
}
249-
250270
test_interactive_nontty_cli "Subscribed mode" {
251271
# Raw output and no "Reading messages..." info message.
252272
# Use RESP3 in this test case.

tests/unit/cluster/pubsubshard.tcl

-3
Original file line numberDiff line numberDiff line change
@@ -59,9 +59,6 @@ test "sunsubscribe without specifying any channel would unsubscribe all shard ch
5959
set publishclient [valkey_client_by_addr $publishnode(host) $publishnode(port)]
6060
set subscribeclient [valkey_deferring_client_by_addr $publishnode(host) $publishnode(port)]
6161

62-
# assert_equal {1} [ssubscribe $subscribeclient {"\{channel.0\}1"}]
63-
# assert_equal {2} [ssubscribe $subscribeclient {"\{channel.0\}2"}]
64-
# assert_equal {3} [ssubscribe $subscribeclient {"\{channel.0\}3"}]
6562
set sub_res [ssubscribe $subscribeclient [list "\{channel.0\}1" "\{channel.0\}2" "\{channel.0\}3"]]
6663
assert_equal [list 1 2 3] $sub_res
6764

tests/unit/pubsub.tcl

+2-14
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,6 @@ start_server {tags {"pubsub network"}} {
4545
set rd1 [valkey_deferring_client]
4646

4747
# subscribe to two channels
48-
#assert_equal {1} [subscribe $rd1 {chan1}]
49-
#assert_equal {2} [subscribe $rd1 {chan2}]
5048
assert_equal {1 2} [subscribe $rd1 {chan1 chan2}]
5149
assert_equal 1 [r publish chan1 hello]
5250
assert_equal 1 [r publish chan2 world]
@@ -85,9 +83,6 @@ start_server {tags {"pubsub network"}} {
8583

8684
test "PUBLISH/SUBSCRIBE after UNSUBSCRIBE without arguments" {
8785
set rd1 [valkey_deferring_client]
88-
#assert_equal {1} [subscribe $rd1 {chan1}]
89-
#assert_equal {2} [subscribe $rd1 {chan2}]
90-
#assert_equal {3} [subscribe $rd1 {chan3}]
9186
assert_equal {1 2 3} [subscribe $rd1 {chan1 chan2 chan3}]
9287
unsubscribe $rd1
9388
# wait for the unsubscribe to take effect
@@ -106,9 +101,6 @@ start_server {tags {"pubsub network"}} {
106101

107102
test "SUBSCRIBE to one channel more than once" {
108103
set rd1 [valkey_deferring_client]
109-
#assert_equal {1} [subscribe $rd1 {chan1}]
110-
#assert_equal {2} [subscribe $rd1 {chan2}]
111-
#assert_equal {3} [subscribe $rd1 {chan3}]
112104
assert_equal {1 1 1} [subscribe $rd1 {chan1 chan1 chan1}]
113105
assert_equal 1 [r publish chan1 hello]
114106
assert_equal {message chan1 hello} [$rd1 read]
@@ -132,8 +124,6 @@ start_server {tags {"pubsub network"}} {
132124

133125
# subscribe to two patterns
134126
assert_equal {1 2} [psubscribe $rd1 {foo.* bar.*}]
135-
#assert_equal {1} [psubscribe $rd1 {foo.*}]
136-
#assert_equal {2} [psubscribe $rd1 {bar.*}]
137127
assert_equal 1 [r publish foo.1 hello]
138128
assert_equal 1 [r publish bar.1 hello]
139129
assert_equal 0 [r publish foo1 hello]
@@ -491,10 +481,8 @@ start_server {tags {"pubsub network"}} {
491481
# Note: SUBSCRIBE and UNSUBSCRIBE with multiple channels in the same command,
492482
# Only one response is returned
493483
# This update matches with Redis response: one command always returns one response
494-
#assert_equal "subscribe foo 1 subscribe bar 2 subscribe baz 3" [r subscribe foo bar baz]
495-
assert_equal "subscribe foo 1" [r subscribe foo bar baz]
496-
assert_equal "subscribe bar 2" [r read]
497-
assert_equal "subscribe baz 3" [r read]
484+
r client capa subv2
485+
assert_equal "subscribe foo 1 subscribe bar 2 subscribe baz 3" [r subscribe foo bar baz]
498486

499487
r multi
500488
r ping abc

tests/unit/pubsubshard.tcl

-3
Original file line numberDiff line numberDiff line change
@@ -64,9 +64,6 @@ start_server {tags {"pubsubshard external:skip"}} {
6464

6565
test "SSUBSCRIBE to one channel more than once" {
6666
set rd1 [valkey_deferring_client]
67-
#assert_equal {1} [ssubscribe $rd1 {chan1}]
68-
#assert_equal {1} [ssubscribe $rd1 {chan1}]
69-
#assert_equal {1} [ssubscribe $rd1 {chan1}]
7067
assert_equal {1 1 1} [ssubscribe $rd1 {chan1 chan1 chan1}]
7168
assert_equal 1 [r SPUBLISH chan1 hello]
7269
assert_equal {smessage chan1 hello} [$rd1 read]

0 commit comments

Comments
 (0)