def disconnect(self, appdata=None):
if self.state != STATE_FINISHED:
self.sendInternal(b'', tlvs=[proto.DisconnectTLV()], appdata=appdata)
self.setState(STATE_PLAINTEXT)
self.crypto.finished()
else:
self.setState(STATE_PLAINTEXT)
def testGenericTLV(self):
testtlvs = [
(proto.DisconnectTLV(), b'\0\1\0\0'),
(proto.SMP1TLV([1, 2, 3, 4, 5, 6]),
b'\0\2\0\x22\0\0\0\6\0\0\0\1\1\0\0\0\1\2\0\0\0\1\3\0\0\0\1\4\0\0\0\1\5\0\0\0\1\6'),
(proto.SMPABORTTLV(), b'\0\6\0\0')
]
for tlv, data in testtlvs:
self.assertEqual(tlv, proto.TLV.parse(data)[0])
self.assertEqual(data, bytes(tlv))
tlvs, datas = tuple(zip(*testtlvs))
self.assertEqual(list(tlvs), proto.TLV.parse(b''.join(datas)))
self.assertRaises(TypeError, lambda: proto.TLV.parse(b'\0\1\0\1x'))