Release Vulpes OS 2.7-preview.22

tundra/tests/test_beatles_radio.py

@@ -0,0 +1,65 @@
+# SPDX-License-Identifier: MPL-2.0
+import importlib.util
+from pathlib import Path
+import socket
+import struct
+import tempfile
+import unittest
+from unittest.mock import patch, Mock
+
+spec=importlib.util.spec_from_file_location('beatles_radio',Path(__file__).resolve().parents[1]/'devices/beatles/radio_status.py')
+radio=importlib.util.module_from_spec(spec);spec.loader.exec_module(radio)
+def integers(*values):return struct.pack('<'+'i'*len(values),*values)
+def string(value):
+    if value is None:return integers(-1)
+    data=integers(len(value))+value.encode('utf-16-le')+b'\0\0'
+    return data+b'\0'*((-len(data))%4)
+
+class RadioTest(unittest.TestCase):
+    def test_registration_does_not_publish_cell_location(self):
+        payload=integers(4)+b''.join(map(string,['5','ffff','123456','14']))
+        result=radio.decode(20,radio.Parcel(payload))['voice']
+        self.assertEqual(result['type'],'lte');self.assertTrue(result['connected']);self.assertTrue(result['roaming'])
+        self.assertNotIn('123456',str(result));self.assertNotIn('ffff',str(result))
+    def test_lte_signal_and_registration_without_data_bearer(self):
+        values=[99,0,0,0,0,0,0,99,95,0,0,0,0]
+        self.assertEqual(radio.decode(19,radio.Parcel(integers(*values)))['signal'],47)
+        values[8]=2147483647
+        self.assertIsNone(radio.decode(19,radio.Parcel(integers(*values)))['signal'])
+        payload=integers(4)+b''.join(map(string,['1','','','14']))
+        state=radio.decode(21,radio.Parcel(payload))['data']
+        self.assertEqual(state['state'],'registered');self.assertFalse(state['connected'])
+    def test_sim_absent_and_locked(self):
+        self.assertEqual(radio.decode(1,radio.Parcel(integers(0,0,-1,-1,-1,0))),{'sim':{'present':False,'cardState':'absent'}})
+        payload=integers(1,0,0,-1,-1,1,2,2,0)+string('PRIVATE_AID')+string('SIM')+integers(0,1,0)
+        result=radio.decode(1,radio.Parcel(payload))
+        self.assertEqual(result,{'sim':{'present':True,'cardState':'pinRequired'}})
+    def test_bounds_and_unknown_signal(self):
+        self.assertIsNone(radio.decode(19,radio.Parcel(integers(99)))['signal'])
+        for data in [b'',integers(999999),integers(2)+b'a\0']:
+            with self.assertRaises(ValueError):radio.Parcel(data).string()
+        with self.assertRaises(ValueError):radio.decode(25,radio.Parcel(b''))
+    def test_partial_frame_and_disconnect_drop_state(self):
+        client,server=socket.socketpair()
+        r=radio.Radio('/unused',0);r.sock=client;r.pending={3:(1,0)}
+        body=integers(0,3,0,0,0,-1,-1,-1,0);frame=struct.pack('>I',len(body))+body
+        server.sendall(frame[:7]);r.receive();self.assertFalse(r.state['available'])
+        server.sendall(frame[7:]);r.receive();self.assertFalse(r.state['sim']['present'])
+        r.close();server.close();self.assertEqual(r.state,{'slot':0,'available':False,'radioState':'unknown'})
+    def test_handover_grace_and_bounded_reconnect_backoff(self):
+        r=radio.Radio('/unused',0);r.sock=Mock();r.serial=1;r.last_poll=100;r.pending={1:(9,90)}
+        with patch.object(radio.time,'monotonic',return_value=110):
+            r.poll();self.assertIsNotNone(r.sock)
+        with patch.object(radio.time,'monotonic',return_value=140):
+            r.poll();self.assertIsNone(r.sock);self.assertEqual(r.retry_at,145)
+            r.close();self.assertEqual(r.retry_at,150)
+            for _ in range(8):r.close()
+            self.assertEqual(r.retry_at,200)
+
+    def test_battery_unknown_is_not_empty(self):
+        with tempfile.TemporaryDirectory() as root:
+            p=Path(root);self.assertIsNone(radio.battery(p))
+            (p/'capacity').write_text('73');(p/'status').write_text('Discharging')
+            self.assertEqual(radio.battery(p),{'level':.73,'charging':False})
+
+if __name__=='__main__':unittest.main()