Harald Welte | 0b32725 | 2022-08-11 17:37:46 +0200 | [diff] [blame] | 1 | #!/usr/bin/env python3 |
| 2 | |
| 3 | import unittest |
| 4 | from pySim.utils import h2b, b2h |
| 5 | from pySim.sms import * |
| 6 | |
| 7 | class Test_SMS_UDH(unittest.TestCase): |
| 8 | def test_single_ie(self): |
| 9 | udh, tail = UserDataHeader.fromBytes('027100') |
| 10 | self.assertEqual(len(udh.ies), 1) |
| 11 | ie = udh.ies[0] |
| 12 | self.assertEqual(ie.iei, 0x71) |
| 13 | self.assertEqual(ie.length, 0) |
| 14 | self.assertEqual(ie.value, b'') |
| 15 | self.assertEqual(tail, b'') |
| 16 | |
| 17 | def test_single_ie_tail(self): |
| 18 | udh, tail = UserDataHeader.fromBytes('027100abcdef') |
| 19 | self.assertEqual(len(udh.ies), 1) |
| 20 | ie = udh.ies[0] |
| 21 | self.assertEqual(ie.iei, 0x71) |
| 22 | self.assertEqual(ie.length, 0) |
| 23 | self.assertEqual(ie.value, b'') |
| 24 | self.assertEqual(tail, b'\xab\xcd\xef') |
| 25 | |
| 26 | def test_single_ie_value(self): |
| 27 | udh, tail = UserDataHeader.fromBytes('03710110') |
| 28 | self.assertEqual(len(udh.ies), 1) |
| 29 | ie = udh.ies[0] |
| 30 | self.assertEqual(ie.iei, 0x71) |
| 31 | self.assertEqual(ie.length, 1) |
| 32 | self.assertEqual(ie.value, b'\x10') |
| 33 | self.assertEqual(tail, b'') |
| 34 | |
| 35 | def test_two_ie_data_tail(self): |
| 36 | udh, tail = UserDataHeader.fromBytes('0571007001ffabcd') |
| 37 | self.assertEqual(len(udh.ies), 2) |
| 38 | ie = udh.ies[0] |
| 39 | self.assertEqual(ie.iei, 0x71) |
| 40 | self.assertEqual(ie.length, 0) |
| 41 | self.assertEqual(ie.value, b'') |
| 42 | ie = udh.ies[1] |
| 43 | self.assertEqual(ie.iei, 0x70) |
| 44 | self.assertEqual(ie.length, 1) |
| 45 | self.assertEqual(ie.value, b'\xff') |
| 46 | self.assertEqual(tail, b'\xab\xcd') |
| 47 | |
| 48 | def test_toBytes(self): |
| 49 | indata = h2b('0571007001ff') |
| 50 | udh, tail = UserDataHeader.fromBytes(indata) |
| 51 | encoded = udh.toBytes() |
| 52 | self.assertEqual(encoded, indata) |
| 53 | |
| 54 | class Test_AddressField(unittest.TestCase): |
| 55 | def test_fromBytes(self): |
| 56 | encoded = h2b('0480214399') |
| 57 | af, trailer = AddressField.fromBytes(encoded) |
| 58 | self.assertEqual(trailer, b'\x99') |
| 59 | self.assertEqual(af.ton, 'unknown') |
| 60 | self.assertEqual(af.npi, 'unknown') |
| 61 | self.assertEqual(af.digits, '1234') |
| 62 | |
| 63 | def test_fromBytes_odd(self): |
| 64 | af, trailer = AddressField.fromBytes('038021f399') |
| 65 | self.assertEqual(trailer, b'\x99') |
| 66 | self.assertEqual(af.ton, 'unknown') |
| 67 | self.assertEqual(af.npi, 'unknown') |
| 68 | self.assertEqual(af.digits, '123') |
| 69 | |
| 70 | def test_toBytes(self): |
| 71 | encoded = h2b('04802143') |
| 72 | af, trailer = AddressField.fromBytes(encoded) |
| 73 | self.assertEqual(af.toBytes(), encoded) |
| 74 | |
| 75 | def test_toBytes_odd(self): |
| 76 | af = AddressField('12345', 'international', 'isdn_e164') |
| 77 | encoded = af.toBytes() |
| 78 | self.assertEqual(encoded, h2b('05912143f5')) |
| 79 | |
| 80 | |
| 81 | class Test_SUBMIT(unittest.TestCase): |
| 82 | def test_fromBytes(self): |
| 83 | s = SMS_SUBMIT.fromBytes('550d0b911614261771f000f5a78c0b050423f423f40003010201424547494e3a56434152440d0a56455253494f4e3a322e310d0a4e3a4d650d0a54454c3b505245463b43454c4c3b564f4943453a2b36313431363237313137300d0a54454c3b484f4d453b564f4943453a2b36313339353337303437310d0a54454c3b574f524b3b564f4943453a2b36313339363734373031350d0a454e443a') |
| 84 | self.assertEqual(s.tp_mti, 1) |
| 85 | self.assertEqual(s.tp_rd, True) |
| 86 | self.assertEqual(s.tp_vpf, 'relative') |
| 87 | self.assertEqual(s.tp_rp, False) |
| 88 | self.assertEqual(s.tp_udhi, True) |
| 89 | self.assertEqual(s.tp_srr, False) |
| 90 | self.assertEqual(s.tp_pid, 0) |
| 91 | self.assertEqual(s.tp_dcs, 0xf5) |
| 92 | self.assertEqual(s.tp_udl, 140) |
| 93 | |
| 94 | class Test_DELIVER(unittest.TestCase): |
| 95 | def test_fromBytes(self): |
| 96 | d = SMS_DELIVER.fromBytes('0408D0E5759A0E7FF6907090307513000824010101BB400101') |
| 97 | self.assertEqual(d.tp_mti, 0) |
| 98 | self.assertEqual(d.tp_mms, True) |
| 99 | self.assertEqual(d.tp_lp, False) |
| 100 | self.assertEqual(d.tp_rp, False) |
| 101 | self.assertEqual(d.tp_udhi, False) |
| 102 | self.assertEqual(d.tp_sri, False) |
| 103 | self.assertEqual(d.tp_pid, 0x7f) |
| 104 | self.assertEqual(d.tp_dcs, 0xf6) |
| 105 | self.assertEqual(d.tp_udl, 8) |