1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# TRX Toolkit
# Unit tests for DATA capture management
#
# (C) 2019 by Vadim Yanitskiy <axilirator@gmail.com>
#
# All Rights Reserved
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
import unittest
import tempfile
import random
from gsm_shared import *
from data_dump import *
class DATADump_Test(unittest.TestCase):
def setUp(self):
# Create a temporary file
self._tf = tempfile.TemporaryFile(mode = 'w+b')
# Create an instance of DATA dump manager
self._ddf = DATADumpFile(self._tf)
# Compare message a with message b
def _compare_msg(self, a, b):
# Make sure we're comparing messages of the same type
self.assertEqual(a.__class__, b.__class__)
# Compare common header fields
self.assertEqual(a.ver, b.ver)
self.assertEqual(a.fn, b.fn)
self.assertEqual(a.tn, b.tn)
# Burst bits (if present)
self.assertEqual(a.burst, b.burst)
# TxMsg specific fields
if isinstance(a, TxMsg):
self.assertEqual(a.pwr, b.pwr)
# RxMsg specific fields
if isinstance(a, RxMsg):
# Version independent fields
self.assertEqual(a.toa256, b.toa256)
self.assertEqual(a.rssi, b.rssi)
# Version specific fields
if a.ver >= 1:
self.assertEqual(a.nope_ind, b.nope_ind)
self.assertEqual(a.mod_type, b.mod_type)
self.assertEqual(a.tsc_set, b.tsc_set)
self.assertEqual(a.tsc, b.tsc)
self.assertEqual(a.ci, b.ci)
# Generate a random message of a given type / version
def _gen_rand_message(self, cls, ver = 1):
msg = cls(ver = ver)
msg.rand_hdr()
msg.rand_burst()
return msg
# Generate a list of random messages
def _gen_rand_messages(self, cls, count, ver = 1):
msg_list = []
for _ in range(count):
msg = self._gen_rand_message(cls, ver)
msg_list.append(msg)
return msg_list
# Generate a mixed list of random messages
def _gen_rand_message_mix(self, count, ver = 1):
msg_list = []
msg_list += self._gen_rand_messages(RxMsg, count)
msg_list += self._gen_rand_messages(TxMsg, count)
random.shuffle(msg_list)
return msg_list
def _test_store_and_parse(self, cls):
msg_ref = self._gen_rand_message(cls)
self._ddf.append_msg(msg_ref)
msg = self._ddf.parse_msg(0)
self._compare_msg(msg, msg_ref)
# Store one Rx message in a file, read it back and compare
def test_store_and_parse_rx_msg(self):
self._test_store_and_parse(RxMsg)
# Store one Tx message in a file, read it back and compare
def test_store_and_parse_tx_msg(self):
self._test_store_and_parse(TxMsg)
# Store multiple Rx/Tx messages in a file, read them back and compare
def test_store_and_parse_all(self):
# Store a mixed list of random messages (19 + 19)
msg_list_ref = self._gen_rand_message_mix(19)
self._ddf.append_all(msg_list_ref)
# Retrieve and compare stored messages
msg_list = self._ddf.parse_all()
for i in range(len(msg_list_ref)):
self._compare_msg(msg_list[i], msg_list_ref[i])
# Verify random access to stored messages
def test_parse_msg_idx(self):
# Store a mixed list of random messages (19 + 19)
msg_list_ref = self._gen_rand_message_mix(19)
self._ddf.append_all(msg_list_ref)
# Random access
for _ in range(100):
idx = random.randrange(len(msg_list_ref))
msg = self._ddf.parse_msg(idx)
self._compare_msg(msg, msg_list_ref[idx])
def test_parse_empty(self):
with self.assertLogs(level = 'ERROR'):
for idx in range(100):
msg = self._ddf.parse_msg(idx)
self.assertEqual(msg, None)
def test_parse_all_empty(self):
msg_list = self._ddf.parse_all()
self.assertEqual(msg_list, [])
def test_parse_len_overflow(self):
# Write a malformed message directly
self._tf.write(DATADump.TAG_TxMsg)
self._tf.write(b'\x00\x63') # 99
self._tf.write(b'\xff' * 90)
with self.assertLogs(level = 'ERROR'):
msg = self._ddf.parse_msg(0)
self.assertEqual(msg, None)
def test_parse_unknown_tag(self):
# Write a malformed message directly
self._tf.write(b'\x33')
self._tf.write(b'\x00\x63') # 99
self._tf.write(b'\xff' * 90)
with self.assertLogs(level = 'ERROR'):
msg = self._ddf.parse_msg(0)
self.assertEqual(msg, None)
if __name__ == '__main__':
unittest.main()
|