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
|
import logging
import struct
import pytest
from home_assistant_bluetooth import BluetoothServiceInfo
from sensor_state_data import (
DeviceClass,
DeviceKey,
SensorDescription,
SensorDeviceInfo,
SensorUpdate,
SensorValue,
Units,
)
from rapt_ble.parser import RAPTPillBluetoothDeviceData
@pytest.fixture(autouse=True)
def logging_config(caplog):
caplog.set_level(logging.DEBUG)
def bytes_to_service_info(payload: bytes) -> BluetoothServiceInfo:
manufacturer_data = {}
(manufacturer_id,) = struct.unpack("<H", payload[:2])
manufacturer_data[manufacturer_id] = payload[2:]
return BluetoothServiceInfo(
name="",
address="00:11:22:33:44:55",
rssi=-60,
manufacturer_data=manufacturer_data,
service_data={},
service_uuids=[],
source="local",
)
@pytest.mark.parametrize(
"data_bytes",
[
# payload v1
b"RAPT\x01x\xe3m<\xb9\x94\x94\x8bD|\xb9\xf64E\x02b&w*\xac",
# payload v2 - invalid gravity velocity
b"RAPT\x02\x00\x00\x00\x00\x00\x00\x94\x8bD|\xb9\xf64E\x02b&w*\xac",
# payload v2 - valid gravity velocity
b"RAPT\x02\x00\x01\x3e\x9d\xd1\xab\x94\x8bD|\xb9\xf64E\x02b&w*\xac",
],
)
def test_device_supported(data_bytes):
device = RAPTPillBluetoothDeviceData()
data = bytes_to_service_info(data_bytes)
assert device.supported(data)
def test_parse_version():
device = RAPTPillBluetoothDeviceData()
data = bytes_to_service_info(b"KEG20220612_050156_81c6d1")
device.update(data)
assert device._get_device_info(None).sw_version == "20220612_050156_81c6d1"
@pytest.mark.parametrize(
"data_bytes",
[
# payload v1
b"RAPT\x01x\xe3m<\xb9\x94\x94\x8bD|\xb9\xf64E\x02b&w*\xac",
# payload v2 - invalid gravity velocity
b"RAPT\x02\x00\x00\x00\x00\x00\x00\x94\x8bD|\xb9\xf64E\x02b&w*\xac",
# payload v2 - valid gravity velocity
b"RAPT\x02\x00\x01\x3e\x9d\xd1\xab\x94\x8bD|\xb9\xf64E\x02b&w*\xac",
],
)
def test_parse_metrics(data_bytes):
device = RAPTPillBluetoothDeviceData()
data = bytes_to_service_info(data_bytes)
result = device.update(data)
assert result == SensorUpdate(
title="RAPT Pill 4455",
devices={
None: SensorDeviceInfo(
name="RAPT Pill 4455",
manufacturer="RAPT",
model="RAPT Pill hydrometer",
hw_version=None,
sw_version=None,
),
},
entity_descriptions={
DeviceKey(key="specific_gravity", device_id=None): SensorDescription(
device_key=DeviceKey(key="specific_gravity", device_id=None),
device_class=DeviceClass.SPECIFIC_GRAVITY,
native_unit_of_measurement=Units.SPECIFIC_GRAVITY,
),
DeviceKey(key="temperature", device_id=None): SensorDescription(
device_key=DeviceKey(key="temperature", device_id=None),
device_class=DeviceClass.TEMPERATURE,
native_unit_of_measurement=Units.TEMP_CELSIUS,
),
DeviceKey(key="battery", device_id=None): SensorDescription(
device_key=DeviceKey(key="battery", device_id=None),
device_class=DeviceClass.BATTERY,
native_unit_of_measurement=Units.PERCENTAGE,
),
DeviceKey(key="signal_strength", device_id=None): SensorDescription(
device_key=DeviceKey(key="signal_strength", device_id=None),
device_class=DeviceClass.SIGNAL_STRENGTH,
native_unit_of_measurement=Units.SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
),
},
entity_values={
DeviceKey(key="specific_gravity", device_id=None): SensorValue(
device_key=DeviceKey("specific_gravity", device_id=None),
name="Specific Gravity",
native_value=1.0109,
),
DeviceKey(key="temperature", device_id=None): SensorValue(
device_key=DeviceKey(key="temperature", device_id=None),
name="Temperature",
native_value=23.94,
),
DeviceKey(key="battery", device_id=None): SensorValue(
device_key=DeviceKey(key="battery", device_id=None),
name="Battery",
native_value=43,
),
DeviceKey(key="signal_strength", device_id=None): SensorValue(
device_key=DeviceKey(key="signal_strength", device_id=None),
name="Signal Strength",
native_value=-60,
),
},
)
|
