feat:add support for pub/sub

Signed-off-by: daz-3ux <daz-3ux@proton.me>
tair-opensource · Sep 1, 2023 · d344e8a · d344e8a
1 parent 98eae25
commit d344e8a
Show file tree

Hide file tree

Showing 2 changed files with 193 additions and 0 deletions.
diff --git a/cts.json b/cts.json
@@ -7314,5 +7314,196 @@
       0
     ],
     "since": "6.2.0"
+  },
+  {
+    "name": "psubscribe command",
+    "command": [
+      "psubscribe hello"
+    ],
+    "result": [
+      [
+        "psubscribe",
+        "hello",
+        1
+      ]
+    ],
+    "since": "2.0.0",
+    "is_pubsub": true
+  },
+  {
+    "name": "psubscribe with RESET",
+    "command": [
+      "psubscribe hello RESET"
+    ],
+    "result": [
+      [
+        "psubscribe",
+        "hello",
+        1
+      ]
+    ],
+    "since": "6.2.0"
+  },
+  {
+    "name": "publish command",
+    "command": [
+      "publish hello world"
+    ],
+    "result": [
+      0
+    ],
+    "since": "2.0.0"
+  },
+  {
+    "name": "pubsub channels command",
+    "command": [
+      "pubsub channels hello"
+    ],
+    "result": [
+      []
+    ],
+    "since": "2.8.0"
+  },
+  {
+    "name": "pubsub numpat command",
+    "command": [
+      "pubsub numpat"
+    ],
+    "result": [
+      0
+    ],
+    "since": "2.8.0"
+  },
+  {
+    "name": "pubsub numsub command",
+    "command": [
+      "pubsub numsub hello"
+    ],
+    "result": [
+      [
+        "hello",
+        0
+      ]
+    ],
+    "since": "2.8.0"
+  },
+  {
+    "name": "pubsub shardchannels command",
+    "command": [
+      "pubsub shardchannels hello"
+    ],
+    "result": [
+      []
+    ],
+    "since": "7.0.0"
+  },
+  {
+    "name": "pubsub shardnumsub command",
+    "command": [
+      "pubsub shardnumsub hello"
+    ],
+    "result": [
+      [
+        "hello",
+        0
+      ]
+    ],
+    "since": "7.0.0"
+  },
+  {
+    "name": "punsubscribe command",
+    "command": [
+      "punsubscribe hello"
+    ],
+    "result": [
+      [
+        "punsubscribe",
+        "hello",
+        0
+      ]
+    ],
+    "since": "2.0.0"
+  },
+  {
+    "name": "spublish command",
+    "command": [
+      "spublish hello world"
+    ],
+    "result": [
+      0
+    ],
+    "since": "7.0.0"
+  },
+  {
+    "name": "ssubscribe command",
+    "command": [
+      "ssubscribe hello"
+    ],
+    "result": [
+      [
+        "ssubscribe",
+        "hello",
+        1
+      ]
+    ],
+    "since": "7.0.0",
+    "is_pubsub": true
+  },
+  {
+    "name": "subscribe command",
+    "command": [
+      "subscribe hello"
+    ],
+    "result": [
+      [
+        "subscribe",
+        "hello",
+        1
+      ]
+    ],
+    "since": "2.0.0",
+    "is_pubsub": true
+  },
+  {
+    "name": "subscribe with RESET",
+    "command": [
+      "subscribe hello RESET"
+    ],
+    "result": [
+      [
+        "subscribe",
+        "hello",
+        1
+      ]
+    ],
+    "since": "6.2.0"
+  },
+  {
+    "name": "sunsubscribe command",
+    "command": [
+      "sunsubscribe hello"
+    ],
+    "result": [
+      [
+        "sunsubscribe",
+        "hello",
+        0
+      ]
+    ],
+    "since": "7.0.0"
+  },
+  {
+    "name": "unsubscribe command",
+    "command": [
+      "unsubscribe hello"
+    ],
+    "result": [
+      [
+        "unsubscribe",
+        "hello",
+        0
+      ]
+    ],
+    "since": "2.0.0"
   }
 ]
diff --git a/redis_compatibility_test.py b/redis_compatibility_test.py
@@ -195,6 +195,8 @@ def run_test(test):
             if 'sort_result' in test and isinstance(result[idx], list):
                 ret = sort_nested_list(ret)
                 result[idx] = sort_nested_list(result[idx])
+            if 'is_pubsub' in test:
+                r.quit()
             if result[idx] != ret:
                 test_failed(g_results[since], name, f"expected: {result[idx]}, result: {ret}")
                 return