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
|
from __future__ import annotations
from typing import Any
from unittest.mock import MagicMock
from bleak.backends.device import BLEDevice
from bleak.backends.scanner import AdvertisementData
import pytest
from aiohomekit.characteristic_cache import CharacteristicCacheMemory
from aiohomekit.controller.ble.controller import BleController
ADVERTISEMENT_DATA_DEFAULTS = {
"local_name": "",
"manufacturer_data": {},
"service_data": {},
"service_uuids": [],
"rssi": -127,
"platform_data": ((),),
"tx_power": -127,
}
BLE_DEVICE_DEFAULTS = {
"name": None,
"rssi": -127,
"details": None,
}
def generate_advertisement_data(**kwargs: Any) -> AdvertisementData:
"""Generate advertisement data with defaults."""
new = kwargs.copy()
for key, value in ADVERTISEMENT_DATA_DEFAULTS.items():
new.setdefault(key, value)
return AdvertisementData(**new)
def generate_ble_device(
address: str | None = None,
name: str | None = None,
details: Any | None = None,
rssi: int | None = None,
**kwargs: Any,
) -> BLEDevice:
"""Generate a BLEDevice with defaults."""
new = kwargs.copy()
if address is not None:
new["address"] = address
if name is not None:
new["name"] = name
if details is not None:
new["details"] = details
if rssi is not None:
new["rssi"] = rssi
for key, value in BLE_DEVICE_DEFAULTS.items():
new.setdefault(key, value)
return BLEDevice(**new)
@pytest.fixture
def mock_bleak_scanner() -> MagicMock:
return MagicMock()
@pytest.fixture
def ble_controller(mock_bleak_scanner: MagicMock) -> BleController:
controller = BleController(CharacteristicCacheMemory(), mock_bleak_scanner)
return controller
def test_discovery_with_none_name(
mock_bleak_scanner: MagicMock, ble_controller: BleController
) -> None:
ble_device_with_short_name = generate_ble_device(
name="Nam", address="00:00:00:00:00:00"
)
ble_device_with_name = generate_ble_device(
name="Name in Full", address="00:00:00:00:00:00"
)
ble_device = generate_ble_device(
name=None,
address="00:00:00:00:00:00",
)
adv = generate_advertisement_data(
local_name=None,
manufacturer_data={
76: b"\x061\x00\x80\xe7\x14j74\x06\x00\x9f!\x04\x02<\xb9\xeb\x0e"
},
)
ble_controller._device_detected(ble_device, adv)
assert "80:e7:14:6a:37:34" in ble_controller.discoveries
ble_controller._device_detected(ble_device_with_short_name, adv)
assert "80:e7:14:6a:37:34" in ble_controller.discoveries
assert (
ble_controller.discoveries["80:e7:14:6a:37:34"].name
== "Nam (00:00:00:00:00:00)"
)
ble_controller._device_detected(ble_device, adv)
assert (
ble_controller.discoveries["80:e7:14:6a:37:34"].name
== "Nam (00:00:00:00:00:00)"
)
ble_controller._device_detected(ble_device_with_name, adv)
assert (
ble_controller.discoveries["80:e7:14:6a:37:34"].name
== "Name in Full (00:00:00:00:00:00)"
)
ble_controller._device_detected(ble_device_with_short_name, adv)
assert (
ble_controller.discoveries["80:e7:14:6a:37:34"].name
== "Name in Full (00:00:00:00:00:00)"
)
|
