58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
import unittest
|
|
|
|
from bulltron_gui import DeviceState, bulltron_match_reason, decode_response, read_frame, split_frames
|
|
|
|
|
|
class BullTronParserTest(unittest.TestCase):
|
|
def test_read_frame_crc(self):
|
|
self.assertEqual(read_frame(0, 62).hex().upper(), "D2030000003ED7B9")
|
|
|
|
def test_split_and_decode_live_frame(self):
|
|
words = [0] * 62
|
|
words[0:4] = [3306, 3312, 3308, 3311]
|
|
words[40] = 132
|
|
words[41] = 29925
|
|
words[42] = 652
|
|
words[43] = 3312
|
|
words[44] = 3306
|
|
words[47] = 2
|
|
words[48] = 1075
|
|
words[51] = 3
|
|
words[53] = 1
|
|
words[54] = 1
|
|
words[56] = 6
|
|
words[61] = 0x0010
|
|
payload = b"".join(word.to_bytes(2, "big") for word in words)
|
|
body = bytes([0xD2, 0x03, len(payload)]) + payload
|
|
from bulltron_gui import crc16_modbus_swapped
|
|
|
|
frame = body + crc16_modbus_swapped(body).to_bytes(2, "big")
|
|
buffer = bytearray(b"\x00" + frame + frame)
|
|
frames = split_frames(buffer)
|
|
self.assertEqual(len(frames), 2)
|
|
|
|
state = DeviceState()
|
|
decode_response(frames[0], state)
|
|
self.assertAlmostEqual(state.telemetry.voltage, 13.2)
|
|
self.assertAlmostEqual(state.telemetry.current, -7.5)
|
|
self.assertAlmostEqual(state.telemetry.soc, 65.2)
|
|
self.assertEqual(state.telemetry.cycle_count, 3)
|
|
self.assertTrue(state.telemetry.charge_mos)
|
|
self.assertTrue(state.telemetry.discharge_mos)
|
|
self.assertEqual(state.telemetry.cell_count, 4)
|
|
|
|
def test_bulltron_scan_filter_matches_apk_markers(self):
|
|
self.assertEqual(bulltron_match_reason("DL-1234", [], []), "name:DL")
|
|
self.assertEqual(bulltron_match_reason("B35-1234", [], []), "name:B35")
|
|
self.assertEqual(
|
|
bulltron_match_reason("unknown", ["0000fff0-0000-1000-8000-00805f9b34fb"], []),
|
|
"service:fff0",
|
|
)
|
|
payload = bytes.fromhex("00" * 14 + "4A4842" + "00")
|
|
self.assertEqual(bulltron_match_reason("unknown", [], [payload]), "adv:JHB")
|
|
self.assertIsNone(bulltron_match_reason("keyboard", [], [b"\x01\x02\x03"]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|