|
@@ -1737,3 +1737,151 @@ def test_parse_advertisement_data_relay_switch_1():
|
|
|
rssi=-67,
|
|
|
active=True,
|
|
|
)
|
|
|
+
|
|
|
+
|
|
|
+def test_leak_active():
|
|
|
+ """Test parse_advertisement_data for the leak detector."""
|
|
|
+ ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
|
|
|
+ adv_data = generate_advertisement_data(
|
|
|
+ manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
|
|
|
+ service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00N"},
|
|
|
+ rssi=-72,
|
|
|
+ )
|
|
|
+ result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
|
|
|
+ assert result == SwitchBotAdvertisement(
|
|
|
+ address="aa:bb:cc:dd:ee:ff",
|
|
|
+ data={
|
|
|
+ "data": {
|
|
|
+ "leak": False,
|
|
|
+ "tampered": False,
|
|
|
+ "battery": 78,
|
|
|
+ "low_battery": False,
|
|
|
+ },
|
|
|
+ "isEncrypted": False,
|
|
|
+ "model": "3",
|
|
|
+ "modelFriendlyName": "Leak Detector",
|
|
|
+ "modelName": SwitchbotModel.LEAK,
|
|
|
+ "rawAdvData": b"&\x00N",
|
|
|
+ },
|
|
|
+ device=ble_device,
|
|
|
+ rssi=-72,
|
|
|
+ active=True,
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_leak_passive():
|
|
|
+ """Test parse_advertisement_data for the leak detector."""
|
|
|
+ ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
|
|
|
+ adv_data = generate_advertisement_data(
|
|
|
+ manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
|
|
|
+ rssi=-72,
|
|
|
+ )
|
|
|
+ result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
|
|
|
+ assert result == SwitchBotAdvertisement(
|
|
|
+ address="aa:bb:cc:dd:ee:ff",
|
|
|
+ data={
|
|
|
+ "data": {},
|
|
|
+ "isEncrypted": False,
|
|
|
+ "model": "3",
|
|
|
+ "rawAdvData": None,
|
|
|
+ },
|
|
|
+ device=ble_device,
|
|
|
+ rssi=-72,
|
|
|
+ active=False,
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_leak_no_leak_detected():
|
|
|
+ """Test parse_advertisement_data for the leak detector."""
|
|
|
+ ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
|
|
|
+ adv_data = generate_advertisement_data(
|
|
|
+ manufacturer_data={
|
|
|
+ 2409: b"\xc4407LzJd\x98ga\xc4\n<\x05\x00\x00\x00\x00"
|
|
|
+ }, # no leak, batt
|
|
|
+ service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
|
|
|
+ rssi=-73,
|
|
|
+ )
|
|
|
+ result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
|
|
|
+ assert result == SwitchBotAdvertisement(
|
|
|
+ address="aa:bb:cc:dd:ee:ff",
|
|
|
+ data={
|
|
|
+ "data": {
|
|
|
+ "leak": False,
|
|
|
+ "tampered": False,
|
|
|
+ "battery": 100,
|
|
|
+ "low_battery": False,
|
|
|
+ },
|
|
|
+ "isEncrypted": False,
|
|
|
+ "model": "3",
|
|
|
+ "modelFriendlyName": "Leak Detector",
|
|
|
+ "modelName": SwitchbotModel.LEAK,
|
|
|
+ "rawAdvData": b"&\x00d",
|
|
|
+ },
|
|
|
+ device=ble_device,
|
|
|
+ rssi=-73,
|
|
|
+ active=True,
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_leak_leak_detected():
|
|
|
+ """Test parse_advertisement_data for the leak detector."""
|
|
|
+ ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
|
|
|
+ adv_data = generate_advertisement_data(
|
|
|
+ manufacturer_data={
|
|
|
+ 2409: b"\xc4407LzGd\xf9ga\xc4\x08<\x05\x00\x00\x00\x00"
|
|
|
+ }, # leak, batt
|
|
|
+ service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
|
|
|
+ rssi=-73,
|
|
|
+ )
|
|
|
+ result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
|
|
|
+ assert result == SwitchBotAdvertisement(
|
|
|
+ address="aa:bb:cc:dd:ee:ff",
|
|
|
+ data={
|
|
|
+ "data": {
|
|
|
+ "leak": True,
|
|
|
+ "tampered": False,
|
|
|
+ "battery": 100,
|
|
|
+ "low_battery": False,
|
|
|
+ },
|
|
|
+ "isEncrypted": False,
|
|
|
+ "model": "3",
|
|
|
+ "modelFriendlyName": "Leak Detector",
|
|
|
+ "modelName": SwitchbotModel.LEAK,
|
|
|
+ "rawAdvData": b"&\x00d",
|
|
|
+ },
|
|
|
+ device=ble_device,
|
|
|
+ rssi=-73,
|
|
|
+ active=True,
|
|
|
+ )
|
|
|
+
|
|
|
+
|
|
|
+def test_leak_low_battery():
|
|
|
+ """Test parse_advertisement_data for the leak detector."""
|
|
|
+ ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
|
|
|
+ adv_data = generate_advertisement_data(
|
|
|
+ manufacturer_data={
|
|
|
+ 2409: b"\xc4407Lz\x02\t\x98\x00\x00\x00\x00<\x05\x00\x00\x00\x00"
|
|
|
+ }, # no leak, low battery
|
|
|
+ service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
|
|
|
+ rssi=-73,
|
|
|
+ )
|
|
|
+ result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
|
|
|
+ assert result == SwitchBotAdvertisement(
|
|
|
+ address="aa:bb:cc:dd:ee:ff",
|
|
|
+ data={
|
|
|
+ "data": {
|
|
|
+ "leak": False,
|
|
|
+ "tampered": False,
|
|
|
+ "battery": 9,
|
|
|
+ "low_battery": False,
|
|
|
+ },
|
|
|
+ "isEncrypted": False,
|
|
|
+ "model": "3",
|
|
|
+ "modelFriendlyName": "Leak Detector",
|
|
|
+ "modelName": SwitchbotModel.LEAK,
|
|
|
+ "rawAdvData": b"&\x00d",
|
|
|
+ },
|
|
|
+ device=ble_device,
|
|
|
+ rssi=-73,
|
|
|
+ active=True,
|
|
|
+ )
|