From b8f16f1a14dcd57b3a3884ae6bfa9f7b2eae1d11 Mon Sep 17 00:00:00 2001
From: YuanYuYuan <az6980522@gmail.com>
Date: Mon, 11 Sep 2023 22:55:33 +0800
Subject: [PATCH] Add get/queryable into session unit test (#109)

* Add get/queryable into session unit test

* No error is allowed

* Remove unused imports
---
 .github/workflows/ci.yml | 16 ++++++--
 tests/test_session.py    | 88 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 101 insertions(+), 3 deletions(-)
 create mode 100644 tests/test_session.py

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 0a27440f..90adedb0 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -18,7 +18,11 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.7", "3.8", "3.9", "3.10"]
+        python-version:
+          - "3.7"
+          - "3.8"
+          - "3.9"
+          - "3.10"
 
     steps:
       - uses: actions/checkout@v2
@@ -49,9 +53,15 @@ jobs:
         with:
           command: fmt
           args: -- --check
-      
+
       - name: Install zenoh-python
         run: pip3 install ./target/wheels/*.whl
-      
+
       - name: Run examples check
         run: python3 tests/examples_check.py
+
+      - name: Install pytest
+        run: pip3 install pytest pytest-xdist
+
+      - name: Run pytest
+        run: pytest -n auto --import-mode=append
diff --git a/tests/test_session.py b/tests/test_session.py
new file mode 100644
index 00000000..3890a737
--- /dev/null
+++ b/tests/test_session.py
@@ -0,0 +1,88 @@
+import zenoh
+import json
+from zenoh import Session, Query, Sample
+from typing import List, Tuple
+import time
+
+SLEEP = 1
+MSG_COUNT = 1_000;
+MSG_SIZE = [1_024, 131_072];
+
+
+def open_session(endpoints: List[str]) -> Tuple[Session, Session]:
+    # listen peer
+    conf = zenoh.Config()
+    conf.insert_json5("listen/endpoints", json.dumps(endpoints))
+    conf.insert_json5("scouting/multicast/enabled", "false")
+    print("[  ][01a] Opening peer01 session");
+    peer01 = zenoh.open(conf)
+
+    # connect peer
+    conf = zenoh.Config()
+    conf.insert_json5("connect/endpoints", json.dumps(endpoints))
+    conf.insert_json5("scouting/multicast/enabled", "false")
+    print("[  ][02a] Opening peer02 session");
+    peer02 = zenoh.open(conf)
+
+    return (peer01, peer02)
+
+
+def close_session(peer01: Session, peer02: Session):
+    print("[  ][01d] Closing peer01 session");
+    peer01.close()
+    print("[  ][02d] Closing peer02 session");
+    peer02.close()
+
+
+def run_session_qryrep(peer01: Session, peer02: Session):
+    keyexpr = "test/session"
+
+    for size in MSG_SIZE:
+        num_requests = 0
+        num_replies = 0
+        num_errors = 0
+
+        def queryable_callback(query: Query):
+            nonlocal num_requests
+            query.reply(Sample(keyexpr, bytes(size)))
+            num_requests += 1
+
+        print("[QR][01c] Queryable on peer01 session");
+        queryable = peer01.declare_queryable(
+            keyexpr,
+            queryable_callback,
+            complete=False
+        )
+
+        time.sleep(SLEEP)
+
+        print(f"[QR][02c] Getting on peer02 session. {MSG_COUNT} msgs.");
+        for _ in range(MSG_COUNT):
+            replies = peer02.get(keyexpr, zenoh.Queue())
+            for reply in replies.receiver:
+                try:
+                    unwraped_reply = reply.ok
+                except:
+                    unwraped_reply = None
+
+                if unwraped_reply:
+                    assert len(unwraped_reply.payload) == size
+                    num_replies += 1
+                else:
+                    num_errors += 1
+
+        time.sleep(SLEEP)
+        print(f"[QR][02c] Got on peer02 session. {num_replies}/{MSG_COUNT} msgs.");
+        assert num_replies == MSG_COUNT
+        assert num_requests == MSG_COUNT
+        assert num_errors == 0
+
+        print("[QR][03c] Unqueryable on peer01 session");
+        queryable.undeclare()
+
+
+def test_session():
+    zenoh.init_logger()
+    (peer01, peer02) = open_session(["tcp/127.0.0.1:17447"])
+    run_session_qryrep(peer01, peer02)
+    close_session(peer01, peer02)