def test_enum(self):
enum_schema_data = bytes([17, 4, 1, 0])
actual_type = reader.Type.from_binary(io.BytesIO(enum_schema_data))
self.assertTrue(isinstance(actual_type, reader.EnumType))
actual_value = actual_type.read(reader.Stream(io.BytesIO(bytes([10]))))
self.assertEqual(actual_value, 10)
async def do_data(self, name):
data = await self.read_sized_block()
if not data:
return
if name not in self._telemetry_records:
return
record = self._telemetry_records[name]
if record:
struct = record.archive.read(reader.Stream(io.BytesIO(data)))
record.update(struct)
_set_tree_widget_data(record.tree_item, struct)
def _handle_data(self):
data = self._handle_sized_block()
if not data:
return
name, self._schema_name = self._schema_name, None
if name not in self._telemetry_records:
return
record = self._telemetry_records[name]
if record:
struct = record.archive.read(reader.Stream(io.BytesIO(data)))
record.update(struct)
_set_tree_widget_data(record.tree_item, struct)
self._serial_state = self.STATE_LINE
def test_basic(self):
_TESTS = [
(bytes([1]), reader.NullType, bytes([]), None),
(bytes([2]), reader.BooleanType, bytes([0]), False),
(bytes([3, 2]), reader.FixedIntType, bytes([1, 2]), 0x201),
(bytes([4, 4]), reader.FixedUIntType, bytes([1, 2, 3,
4]), 0x04030201),
(bytes([6]), reader.VaruintType, bytes([3]), 3),
(bytes([7]), reader.Float32Type, bytes([0, 0, 0, 0]), 0.0),
(bytes([8]), reader.Float64Type, bytes([0] * 8), 0.0),
(bytes([9]), reader.BytesType, bytes([2, 4, 5]), bytes([4, 5])),
(bytes([10]), reader.StringType, bytes([3, 97, 98, 99]), 'abc'),
(bytes([
16,
0,
0,
2,
102,
49,
0,
2,
0,
0,
0,
0,
0,
0,
]), reader.ObjectType, bytes([0]),
_TestType(lambda type_class, value: value == type_class.
namedtuple(False))),
(
bytes([
17,
4,
1,
2, # nvalues
1,
3,
101,
110,
49,
2,
3,
101,
110,
50,
]),
reader.EnumType,
bytes([1]),
_TestType(lambda type_class, value: value == type_class.
enum_class.en1)),
(bytes([18, 4, 1]), reader.ArrayType, bytes([3, 7, 8,
9]), [7, 8, 9]),
(bytes([19, 2, 4, 1]), reader.FixedArrayType, bytes([4,
5]), [4, 5]),
(bytes([20, 4, 1]), reader.MapType,
bytes([
2,
2,
105,
49,
10,
2,
105,
50,
11,
]), {
'i1': 10,
'i2': 11
}),
(bytes([21, 4, 1, 10,
0]), reader.UnionType, bytes([1, 3, 116, 115, 49]), 'ts1'),
# (bytes([22]), reader.TimestampType, bytes([0, 0, 0, 0, 0, 0, 0, 0]), ?),
(bytes([23]), reader.DurationType, bytes([0, 0, 0, 0, 0, 0, 0,
0]), 0.0),
]
for (schema_data, expected_type, data_data, data_value) in _TESTS:
actual_type = reader.Type.from_binary(io.BytesIO(schema_data))
self.assertTrue(isinstance(actual_type, expected_type))
actual_value = actual_type.read(
reader.Stream(io.BytesIO(data_data)))
if isinstance(data_value, _TestType):
self.assertTrue(data_value(actual_type, actual_value))
else:
self.assertEqual(actual_value, data_value)