From: Martin Willi <martin@strongswan.org>
Date: Fri, 4 Nov 2022 13:46:53 +0000 (+0100)
Subject: vici: Add some initial session level unit-tests
X-Git-Tag: 5.9.9rc1~13^2~1
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=254d4075fed1cb3ed2aba70c03f6374ee13a61a8;p=thirdparty%2Fstrongswan.git

vici: Add some initial session level unit-tests
---

diff --git a/src/libcharon/plugins/vici/python/Makefile.am b/src/libcharon/plugins/vici/python/Makefile.am
index 6728e4eb7a..425623b08b 100644
--- a/src/libcharon/plugins/vici/python/Makefile.am
+++ b/src/libcharon/plugins/vici/python/Makefile.am
@@ -4,6 +4,7 @@ EXTRA_DIST = LICENSE README.rst MANIFEST.in \
 	tox.sh \
 	test/__init__.py \
 	test/test_protocol.py \
+	test/test_session.py \
 	vici/__init__.py \
 	vici/command_wrappers.py \
 	vici/exception.py \
diff --git a/src/libcharon/plugins/vici/python/test/test_session.py b/src/libcharon/plugins/vici/python/test/test_session.py
new file mode 100644
index 0000000000..eabacc0234
--- /dev/null
+++ b/src/libcharon/plugins/vici/python/test/test_session.py
@@ -0,0 +1,99 @@
+import pytest
+import socket
+import struct
+from collections import OrderedDict
+
+from vici.session import Session
+from vici.protocol import Transport, Packet, Message, FiniteStream
+from vici.exception import DeserializationException
+
+
+class MockedServer(object):
+
+    def __init__(self, sock):
+        self.transport = Transport(sock)
+
+    def send(self, kind, name=None, message=None):
+        if name is None:
+            payload = struct.pack("!B", kind)
+        else:
+            name = name.encode("UTF-8")
+            payload = struct.pack("!BB", kind, len(name)) + name
+        if message is not None:
+            payload += Message.serialize(message)
+        self.transport.send(payload)
+
+    def recv(self):
+        stream = FiniteStream(self.transport.receive())
+        kind, length = struct.unpack("!BB", stream.read(2))
+        name = stream.read(length)
+        data = stream.read()
+        if len(data):
+            return kind, name, Message.deserialize(data)
+        return kind, name
+
+
+class TestSession(object):
+
+    events = [
+        OrderedDict([('event', b'1')]),
+        OrderedDict([('event', b'2')]),
+        OrderedDict([('event', b'3')]),
+    ]
+
+    def interconnect(self):
+        c, s = socket.socketpair(socket.AF_UNIX)
+        return Session(c), MockedServer(s)
+
+    def test_request(self):
+        c, s = self.interconnect()
+
+        s.send(Packet.CMD_RESPONSE)
+        assert c.request("doit") == {}
+        assert s.recv() == (Packet.CMD_REQUEST, b"doit")
+
+        s.send(Packet.CMD_RESPONSE, message={"hey": b"hou"})
+        assert c.request("heyhou") == {"hey": b"hou"}
+        assert s.recv() == (Packet.CMD_REQUEST, b"heyhou")
+
+    def test_streamed(self):
+        c, s = self.interconnect()
+
+        s.send(Packet.EVENT_CONFIRM)
+        for e in self.events:
+            s.send(Packet.EVENT, name="stream", message=e)
+        s.send(Packet.CMD_RESPONSE)
+        s.send(Packet.EVENT_CONFIRM)
+
+        assert list(c.streamed_request("streamit", "stream")) == self.events
+        assert s.recv() == (Packet.EVENT_REGISTER, b"stream")
+        assert s.recv() == (Packet.CMD_REQUEST, b"streamit")
+        assert s.recv() == (Packet.EVENT_UNREGISTER, b"stream")
+
+    def test_timeout(self):
+        c, s = self.interconnect()
+
+        s.send(Packet.EVENT_CONFIRM)
+        s.send(Packet.EVENT_CONFIRM)
+        for e in self.events:
+            s.send(Packet.EVENT, name="event", message=e)
+
+        r = []
+        i = 0
+        for name, msg in c.listen(["xyz", "event"], timeout=0.1):
+            if name is None:
+                i += 1
+                if i > 2:
+                    s.send(Packet.EVENT_CONFIRM)
+                    s.send(Packet.EVENT_CONFIRM)
+                    break
+            else:
+                assert name == b"event"
+                r.append(msg)
+
+        assert s.recv() == (Packet.EVENT_REGISTER, b"xyz")
+        assert s.recv() == (Packet.EVENT_REGISTER, b"event")
+        assert s.recv() == (Packet.EVENT_UNREGISTER, b"xyz")
+        assert s.recv() == (Packet.EVENT_UNREGISTER, b"event")
+
+        assert r == self.events