Release Vulpes OS 2.7-preview.22

tundra/tests/test_beatles_control_client.py

@@ -0,0 +1,60 @@
+# SPDX-License-Identifier: MPL-2.0
+import concurrent.futures
+from pathlib import Path
+import socket
+import subprocess
+import tempfile
+import unittest
+
+
+class ControlRelayTests(unittest.TestCase):
+    @classmethod
+    def setUpClass(cls):
+        cls.directory=tempfile.TemporaryDirectory()
+        cls.root=Path(cls.directory.name)
+        code=(Path(__file__).parents[1]/'probes/beatles-control-client.c').read_text()
+        for name in ('phone-control','wifi-control'):
+            code=code.replace('/work/'+name+'.sock',str(cls.root/(name+'.sock')))
+        source=cls.root/'client.c';source.write_text(code);cls.binary=cls.root/'client'
+        subprocess.run(['cc','-Wall','-Wextra','-Werror',str(source),'-o',str(cls.binary)],check=True)
+
+    @classmethod
+    def tearDownClass(cls):cls.directory.cleanup()
+
+    def exchange(self,mode,reply):
+        path=self.root/(('phone' if mode=='--radio' else 'wifi')+'-control.sock')
+        path.unlink(missing_ok=True)
+        with socket.socket(socket.AF_UNIX,socket.SOCK_STREAM) as server:
+            server.bind(str(path));server.listen(1);server.settimeout(3)
+            def broker():
+                with server.accept()[0] as conn:
+                    conn.settimeout(3);data=b''
+                    while b'\n' not in data:data+=conn.recv(4096)
+                    for part in reply:conn.sendall(part)
+                    return data
+            with concurrent.futures.ThreadPoolExecutor(max_workers=1) as pool:
+                future=pool.submit(broker)
+                result=subprocess.run([str(self.binary),mode],input=b'{"operation":"status"}',capture_output=True,timeout=4)
+                request=future.result()
+        return result,request
+
+    def test_radio_and_wifi_preserve_broker_envelopes(self):
+        for mode in ('--radio','--wifi'):
+            result,request=self.exchange(mode,[b'{"ok":true,',b'"result":[]}\n'])
+            self.assertEqual(result.returncode,0)
+            self.assertEqual(result.stdout,b'{"ok":true,"result":[]}\n')
+            self.assertEqual(request,b'{"operation":"status"}\n')
+
+    def test_disconnect_and_oversized_response_fail(self):
+        for reply in ([b'{"incomplete":'],[b'x'*65537]):
+            result,_=self.exchange('--wifi',reply)
+            self.assertNotEqual(result.returncode,0)
+            self.assertEqual(result.stdout,b'')
+
+    def test_unknown_target_and_multiple_requests_are_rejected(self):
+        for mode,data in [('--shell',b'{}'),('--radio',b'{}\n{}'),('--wifi',b'x'*8193)]:
+            result=subprocess.run([str(self.binary),mode],input=data,capture_output=True,timeout=3)
+            self.assertNotEqual(result.returncode,0)
+            self.assertEqual(result.stdout,b'')
+
+if __name__=='__main__':unittest.main()