test_adv_parser.py 71 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199
  1. from __future__ import annotations
  2. from typing import Any
  3. from bleak.backends.device import BLEDevice
  4. from bleak.backends.scanner import AdvertisementData
  5. from switchbot import SwitchbotModel
  6. from switchbot.adv_parser import parse_advertisement_data
  7. from switchbot.const.lock import LockStatus
  8. from switchbot.models import SwitchBotAdvertisement
  9. ADVERTISEMENT_DATA_DEFAULTS = {
  10. "local_name": "",
  11. "manufacturer_data": {},
  12. "service_data": {},
  13. "service_uuids": [],
  14. "rssi": -127,
  15. "platform_data": ((),),
  16. "tx_power": -127,
  17. }
  18. BLE_DEVICE_DEFAULTS = {
  19. "name": None,
  20. "rssi": -127,
  21. "details": None,
  22. }
  23. def generate_ble_device(
  24. address: str | None = None,
  25. name: str | None = None,
  26. details: Any | None = None,
  27. rssi: int | None = None,
  28. **kwargs: Any,
  29. ) -> BLEDevice:
  30. """Generate a BLEDevice with defaults."""
  31. new = kwargs.copy()
  32. if address is not None:
  33. new["address"] = address
  34. if name is not None:
  35. new["name"] = name
  36. if details is not None:
  37. new["details"] = details
  38. if rssi is not None:
  39. new["rssi"] = rssi
  40. for key, value in BLE_DEVICE_DEFAULTS.items():
  41. new.setdefault(key, value)
  42. return BLEDevice(**new)
  43. def generate_advertisement_data(**kwargs: Any) -> AdvertisementData:
  44. """Generate advertisement data with defaults."""
  45. new = kwargs.copy()
  46. for key, value in ADVERTISEMENT_DATA_DEFAULTS.items():
  47. new.setdefault(key, value)
  48. return AdvertisementData(**new)
  49. def test_parse_advertisement_data_curtain():
  50. """Test parse_advertisement_data for curtain."""
  51. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  52. adv_data = generate_advertisement_data(
  53. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  54. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0X\x00\x11\x04"},
  55. rssi=-80,
  56. )
  57. result = parse_advertisement_data(ble_device, adv_data)
  58. assert result == SwitchBotAdvertisement(
  59. address="aa:bb:cc:dd:ee:ff",
  60. data={
  61. "rawAdvData": b"c\xc0X\x00\x11\x04",
  62. "data": {
  63. "calibration": True,
  64. "battery": 88,
  65. "inMotion": False,
  66. "position": 100,
  67. "lightLevel": 1,
  68. "deviceChain": 1,
  69. },
  70. "isEncrypted": False,
  71. "model": "c",
  72. "modelFriendlyName": "Curtain",
  73. "modelName": SwitchbotModel.CURTAIN,
  74. },
  75. device=ble_device,
  76. rssi=-80,
  77. active=True,
  78. )
  79. def test_parse_advertisement_data_curtain_passive():
  80. """Test parse_advertisement_data for curtain passive."""
  81. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  82. adv_data = generate_advertisement_data(
  83. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  84. service_data={},
  85. rssi=-80,
  86. )
  87. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  88. assert result == SwitchBotAdvertisement(
  89. address="aa:bb:cc:dd:ee:ff",
  90. data={
  91. "rawAdvData": None,
  92. "data": {
  93. "calibration": None,
  94. "battery": None,
  95. "inMotion": False,
  96. "position": 100,
  97. "lightLevel": 1,
  98. "deviceChain": 1,
  99. },
  100. "isEncrypted": False,
  101. "model": "c",
  102. "modelFriendlyName": "Curtain",
  103. "modelName": SwitchbotModel.CURTAIN,
  104. },
  105. device=ble_device,
  106. rssi=-80,
  107. active=False,
  108. )
  109. def test_parse_advertisement_data_curtain_passive_12_bytes():
  110. """Test parse_advertisement_data for curtain passive."""
  111. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  112. adv_data = generate_advertisement_data(
  113. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04\x00"},
  114. service_data={},
  115. rssi=-80,
  116. )
  117. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  118. assert result == SwitchBotAdvertisement(
  119. address="aa:bb:cc:dd:ee:ff",
  120. data={
  121. "rawAdvData": None,
  122. "data": {
  123. "calibration": None,
  124. "battery": None,
  125. "inMotion": False,
  126. "position": 100,
  127. "lightLevel": 1,
  128. "deviceChain": 1,
  129. },
  130. "isEncrypted": False,
  131. "model": "c",
  132. "modelFriendlyName": "Curtain",
  133. "modelName": SwitchbotModel.CURTAIN,
  134. },
  135. device=ble_device,
  136. rssi=-80,
  137. active=False,
  138. )
  139. def test_parse_advertisement_data_curtain_position_zero():
  140. """Test parse_advertisement_data for curtain position zero."""
  141. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  142. adv_data = generate_advertisement_data(
  143. local_name="WoCurtain",
  144. manufacturer_data={89: b"\xc1\xc7'}U\xab"},
  145. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0\xced\x11\x04"},
  146. service_uuids=[
  147. "00001800-0000-1000-8000-00805f9b34fb",
  148. "00001801-0000-1000-8000-00805f9b34fb",
  149. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  150. ],
  151. rssi=-52,
  152. )
  153. result = parse_advertisement_data(ble_device, adv_data)
  154. assert result == SwitchBotAdvertisement(
  155. address="aa:bb:cc:dd:ee:ff",
  156. data={
  157. "rawAdvData": b"c\xd0\xced\x11\x04",
  158. "data": {
  159. "calibration": True,
  160. "battery": 78,
  161. "inMotion": False,
  162. "position": 0,
  163. "lightLevel": 1,
  164. "deviceChain": 1,
  165. },
  166. "isEncrypted": False,
  167. "model": "c",
  168. "modelFriendlyName": "Curtain",
  169. "modelName": SwitchbotModel.CURTAIN,
  170. },
  171. device=ble_device,
  172. rssi=-52,
  173. active=True,
  174. )
  175. def test_parse_advertisement_data_curtain_firmware_six_position_100():
  176. """Test parse_advertisement_data with firmware six for curtain position 100."""
  177. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  178. adv_data = generate_advertisement_data(
  179. local_name="WoCurtain",
  180. manufacturer_data={
  181. 89: b"\xf5\x98\x94\x08\xa0\xe7",
  182. 2409: b'\xf5\x98\x94\x08\xa0\xe7\x9b\x0f\x00"\x04',
  183. },
  184. service_data={
  185. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0H\x00\x12\x04",
  186. "0000fd3d-0000-1000-8000-00805f9b34fb": b'c\xc0G\x00"\x04',
  187. },
  188. service_uuids=[
  189. "00001800-0000-1000-8000-00805f9b34fb",
  190. "00001801-0000-1000-8000-00805f9b34fb",
  191. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  192. ],
  193. rssi=-62,
  194. )
  195. result = parse_advertisement_data(ble_device, adv_data)
  196. assert result == SwitchBotAdvertisement(
  197. address="aa:bb:cc:dd:ee:ff",
  198. data={
  199. "rawAdvData": b'c\xc0G\x00"\x04',
  200. "data": {
  201. "calibration": True,
  202. "battery": 71,
  203. "inMotion": False,
  204. "position": 100,
  205. "lightLevel": 2,
  206. "deviceChain": 2,
  207. },
  208. "isEncrypted": False,
  209. "model": "c",
  210. "modelFriendlyName": "Curtain",
  211. "modelName": SwitchbotModel.CURTAIN,
  212. },
  213. device=ble_device,
  214. rssi=-62,
  215. active=True,
  216. )
  217. def test_parse_advertisement_data_curtain_firmware_six_position_100_other_rssi():
  218. """Test parse_advertisement_data with firmware six for curtain position 100 other rssi."""
  219. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  220. adv_data = generate_advertisement_data(
  221. local_name="WoCurtain",
  222. manufacturer_data={
  223. 89: b"\xf5\x98\x94\x08\xa0\xe7",
  224. 2409: b'\xf5\x98\x94\x08\xa0\xe7\xa5\x0fc"\x04',
  225. },
  226. service_data={
  227. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0H\x00\x12\x04",
  228. "0000fd3d-0000-1000-8000-00805f9b34fb": b'c\xc0Gc"\x04',
  229. },
  230. service_uuids=[
  231. "00001800-0000-1000-8000-00805f9b34fb",
  232. "00001801-0000-1000-8000-00805f9b34fb",
  233. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  234. ],
  235. rssi=-67,
  236. )
  237. result = parse_advertisement_data(ble_device, adv_data)
  238. assert result == SwitchBotAdvertisement(
  239. address="aa:bb:cc:dd:ee:ff",
  240. data={
  241. "rawAdvData": b'c\xc0Gc"\x04',
  242. "data": {
  243. "calibration": True,
  244. "battery": 71,
  245. "inMotion": False,
  246. "position": 1,
  247. "lightLevel": 2,
  248. "deviceChain": 2,
  249. },
  250. "isEncrypted": False,
  251. "model": "c",
  252. "modelFriendlyName": "Curtain",
  253. "modelName": SwitchbotModel.CURTAIN,
  254. },
  255. device=ble_device,
  256. rssi=-67,
  257. active=True,
  258. )
  259. def test_parse_advertisement_data_curtain_fully_closed():
  260. """Test parse_advertisement_data with firmware six fully closed."""
  261. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  262. adv_data = generate_advertisement_data(
  263. local_name="WoCurtain",
  264. manufacturer_data={2409: b"\xc1\xc7'}U\xab\"\x0fd\x11\x04"},
  265. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0Sd\x11\x04"},
  266. service_uuids=[
  267. "00001800-0000-1000-8000-00805f9b34fb",
  268. "00001801-0000-1000-8000-00805f9b34fb",
  269. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  270. ],
  271. rssi=1,
  272. )
  273. result = parse_advertisement_data(ble_device, adv_data)
  274. assert result == SwitchBotAdvertisement(
  275. address="aa:bb:cc:dd:ee:ff",
  276. data={
  277. "rawAdvData": b"c\xc0Sd\x11\x04",
  278. "data": {
  279. "calibration": True,
  280. "battery": 83,
  281. "inMotion": False,
  282. "position": 0,
  283. "lightLevel": 1,
  284. "deviceChain": 1,
  285. },
  286. "isEncrypted": False,
  287. "model": "c",
  288. "modelFriendlyName": "Curtain",
  289. "modelName": SwitchbotModel.CURTAIN,
  290. },
  291. device=ble_device,
  292. rssi=1,
  293. active=True,
  294. )
  295. def test_parse_advertisement_data_curtain_fully_open():
  296. """Test parse_advertisement_data with firmware six fully open."""
  297. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  298. adv_data = generate_advertisement_data(
  299. local_name="WoCurtain",
  300. manufacturer_data={2409: b"\xc1\xc7'}U\xab%\x0f\x00\x11\x04"},
  301. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0S\x00\x11\x04"},
  302. service_uuids=[
  303. "00001800-0000-1000-8000-00805f9b34fb",
  304. "00001801-0000-1000-8000-00805f9b34fb",
  305. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  306. ],
  307. rssi=1,
  308. )
  309. result = parse_advertisement_data(ble_device, adv_data)
  310. assert result == SwitchBotAdvertisement(
  311. address="aa:bb:cc:dd:ee:ff",
  312. data={
  313. "rawAdvData": b"c\xc0S\x00\x11\x04",
  314. "data": {
  315. "calibration": True,
  316. "battery": 83,
  317. "inMotion": False,
  318. "position": 100,
  319. "lightLevel": 1,
  320. "deviceChain": 1,
  321. },
  322. "isEncrypted": False,
  323. "model": "c",
  324. "modelFriendlyName": "Curtain",
  325. "modelName": SwitchbotModel.CURTAIN,
  326. },
  327. device=ble_device,
  328. rssi=1,
  329. active=True,
  330. )
  331. def test_parse_advertisement_data_curtain3():
  332. """Test parse_advertisement_data for curtain 3."""
  333. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  334. adv_data = generate_advertisement_data(
  335. manufacturer_data={
  336. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\xf7\x07\x00\x11\x04\x00\x49"
  337. },
  338. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"{\xc0\x49\x00\x11\x04"},
  339. rssi=-80,
  340. )
  341. result = parse_advertisement_data(ble_device, adv_data)
  342. assert result == SwitchBotAdvertisement(
  343. address="aa:bb:cc:dd:ee:ff",
  344. data={
  345. "rawAdvData": b"{\xc0\x49\x00\x11\x04",
  346. "data": {
  347. "calibration": True,
  348. "battery": 73,
  349. "inMotion": False,
  350. "position": 100,
  351. "lightLevel": 1,
  352. "deviceChain": 1,
  353. },
  354. "isEncrypted": False,
  355. "model": "{",
  356. "modelFriendlyName": "Curtain 3",
  357. "modelName": SwitchbotModel.CURTAIN,
  358. },
  359. device=ble_device,
  360. rssi=-80,
  361. active=True,
  362. )
  363. def test_parse_advertisement_data_curtain3_passive():
  364. """Test parse_advertisement_data for curtain passive."""
  365. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  366. adv_data = generate_advertisement_data(
  367. manufacturer_data={
  368. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\xf7\x07\x00\x11\x04\x00\x49"
  369. },
  370. service_data={},
  371. rssi=-80,
  372. )
  373. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  374. assert result == SwitchBotAdvertisement(
  375. address="aa:bb:cc:dd:ee:ff",
  376. data={
  377. "rawAdvData": None,
  378. "data": {
  379. "calibration": None,
  380. "battery": 73,
  381. "inMotion": False,
  382. "position": 100,
  383. "lightLevel": 1,
  384. "deviceChain": 1,
  385. },
  386. "isEncrypted": False,
  387. "model": "c",
  388. "modelFriendlyName": "Curtain",
  389. "modelName": SwitchbotModel.CURTAIN,
  390. },
  391. device=ble_device,
  392. rssi=-80,
  393. active=False,
  394. )
  395. def test_parse_advertisement_data_contact():
  396. """Test parse_advertisement_data for the contact sensor."""
  397. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  398. adv_data = generate_advertisement_data(
  399. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  400. service_data={
  401. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@d\x05\x00u\x00\xf8\x12"
  402. },
  403. rssi=-80,
  404. )
  405. result = parse_advertisement_data(ble_device, adv_data)
  406. assert result == SwitchBotAdvertisement(
  407. address="aa:bb:cc:dd:ee:ff",
  408. data={
  409. "rawAdvData": b"d@d\x05\x00u\x00\xf8\x12",
  410. "data": {
  411. "button_count": 2,
  412. "contact_open": True,
  413. "contact_timeout": True,
  414. "is_light": True,
  415. "battery": 100,
  416. "motion_detected": True,
  417. "tested": False,
  418. },
  419. "isEncrypted": False,
  420. "model": "d",
  421. "modelFriendlyName": "Contact Sensor",
  422. "modelName": SwitchbotModel.CONTACT_SENSOR,
  423. },
  424. device=ble_device,
  425. rssi=-80,
  426. active=True,
  427. )
  428. def test_parse_advertisement_data_empty():
  429. """Test parse_advertisement_data with empty data does not blow up."""
  430. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  431. adv_data = generate_advertisement_data(
  432. manufacturer_data={2403: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  433. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b""},
  434. )
  435. result = parse_advertisement_data(ble_device, adv_data)
  436. assert result is None
  437. def test_new_bot_firmware():
  438. """Test parsing adv data from new bot firmware."""
  439. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  440. adv_data = generate_advertisement_data(
  441. manufacturer_data={89: b"\xd8.\xad\xcd\r\x85"},
  442. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"H\x10\xe1"},
  443. service_uuids=["CBA20D00-224D-11E6-9FB8-0002A5D5C51B"],
  444. rssi=-90,
  445. )
  446. result = parse_advertisement_data(ble_device, adv_data)
  447. assert result == SwitchBotAdvertisement(
  448. address="aa:bb:cc:dd:ee:ff",
  449. data={
  450. "rawAdvData": b"H\x10\xe1",
  451. "data": {"switchMode": False, "isOn": False, "battery": 97},
  452. "model": "H",
  453. "isEncrypted": False,
  454. "modelFriendlyName": "Bot",
  455. "modelName": SwitchbotModel.BOT,
  456. },
  457. device=ble_device,
  458. rssi=-90,
  459. active=True,
  460. )
  461. def test_parse_advertisement_data_curtain_firmware_six_fully_closed():
  462. """Test parse_advertisement_data with firmware six fully closed."""
  463. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  464. adv_data = generate_advertisement_data(
  465. local_name="WoCurtain",
  466. manufacturer_data={
  467. 89: b"\xcc\xf4\xc4\xf9\xacl",
  468. 2409: b"\xcc\xf4\xc4\xf9\xacl\xeb\x0fd\x12\x04",
  469. },
  470. service_data={
  471. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0Yd\x11\x04",
  472. "0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0dd\x12\x04",
  473. },
  474. service_uuids=[
  475. "00001800-0000-1000-8000-00805f9b34fb",
  476. "00001801-0000-1000-8000-00805f9b34fb",
  477. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  478. ],
  479. rssi=-2,
  480. )
  481. result = parse_advertisement_data(ble_device, adv_data)
  482. assert result == SwitchBotAdvertisement(
  483. address="aa:bb:cc:dd:ee:ff",
  484. data={
  485. "rawAdvData": b"c\xc0dd\x12\x04",
  486. "data": {
  487. "calibration": True,
  488. "battery": 100,
  489. "inMotion": False,
  490. "position": 0,
  491. "lightLevel": 1,
  492. "deviceChain": 2,
  493. },
  494. "isEncrypted": False,
  495. "model": "c",
  496. "modelFriendlyName": "Curtain",
  497. "modelName": SwitchbotModel.CURTAIN,
  498. },
  499. device=ble_device,
  500. rssi=-2,
  501. active=True,
  502. )
  503. def test_parse_advertisement_data_curtain_firmware_six_fully_open():
  504. """Test parse_advertisement_data with firmware six fully open."""
  505. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  506. adv_data = generate_advertisement_data(
  507. local_name="WoCurtain",
  508. manufacturer_data={
  509. 89: b"\xcc\xf4\xc4\xf9\xacl",
  510. 2409: b"\xcc\xf4\xc4\xf9\xacl\xe2\x0f\x00\x12\x04",
  511. },
  512. service_data={
  513. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0Yd\x11\x04",
  514. "0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0d\x00\x12\x04",
  515. },
  516. service_uuids=[
  517. "00001800-0000-1000-8000-00805f9b34fb",
  518. "00001801-0000-1000-8000-00805f9b34fb",
  519. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  520. ],
  521. rssi=-2,
  522. )
  523. result = parse_advertisement_data(ble_device, adv_data)
  524. assert result == SwitchBotAdvertisement(
  525. address="aa:bb:cc:dd:ee:ff",
  526. data={
  527. "rawAdvData": b"c\xc0d\x00\x12\x04",
  528. "data": {
  529. "calibration": True,
  530. "battery": 100,
  531. "inMotion": False,
  532. "position": 100,
  533. "lightLevel": 1,
  534. "deviceChain": 2,
  535. },
  536. "isEncrypted": False,
  537. "model": "c",
  538. "modelFriendlyName": "Curtain",
  539. "modelName": SwitchbotModel.CURTAIN,
  540. },
  541. device=ble_device,
  542. rssi=-2,
  543. active=True,
  544. )
  545. def test_contact_sensor_mfr():
  546. """Test parsing adv data from new bot firmware."""
  547. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  548. adv_data = generate_advertisement_data(
  549. manufacturer_data={2409: b"\xcb9\xcd\xc4=FA,\x00F\x01\x8f\xc4"},
  550. service_data={
  551. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d\x00\xda\x04\x00F\x01\x8f\xc4"
  552. },
  553. tx_power=-127,
  554. rssi=-70,
  555. )
  556. result = parse_advertisement_data(ble_device, adv_data)
  557. assert result == SwitchBotAdvertisement(
  558. address="aa:bb:cc:dd:ee:ff",
  559. data={
  560. "data": {
  561. "battery": 90,
  562. "button_count": 4,
  563. "contact_open": True,
  564. "contact_timeout": True,
  565. "is_light": False,
  566. "motion_detected": False,
  567. "tested": False,
  568. },
  569. "isEncrypted": False,
  570. "model": "d",
  571. "modelFriendlyName": "Contact Sensor",
  572. "modelName": SwitchbotModel.CONTACT_SENSOR,
  573. "rawAdvData": b"d\x00\xda\x04\x00F\x01\x8f\xc4",
  574. },
  575. device=ble_device,
  576. rssi=-70,
  577. active=True,
  578. )
  579. def test_contact_sensor_mfr_no_service_data():
  580. """Test contact sensor with passive data only."""
  581. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  582. adv_data = generate_advertisement_data(
  583. manufacturer_data={2409: b"\xcb9\xcd\xc4=FA,\x00F\x01\x8f\xc4"},
  584. service_data={},
  585. tx_power=-127,
  586. rssi=-70,
  587. )
  588. result = parse_advertisement_data(ble_device, adv_data)
  589. # Passive detection of contact sensor is not supported
  590. # anymore since the Switchbot Curtain v3 was released
  591. # which uses the heuristics for the contact sensor.
  592. assert result is None
  593. def test_contact_sensor_srv():
  594. """Test parsing adv data from new bot firmware."""
  595. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  596. adv_data = generate_advertisement_data(
  597. service_data={
  598. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d\x00\xda\x04\x00F\x01\x8f\xc4"
  599. },
  600. tx_power=-127,
  601. rssi=-70,
  602. )
  603. result = parse_advertisement_data(ble_device, adv_data)
  604. assert result == SwitchBotAdvertisement(
  605. address="aa:bb:cc:dd:ee:ff",
  606. data={
  607. "data": {
  608. "battery": 90,
  609. "button_count": 4,
  610. "contact_open": True,
  611. "contact_timeout": True,
  612. "is_light": False,
  613. "motion_detected": False,
  614. "tested": False,
  615. },
  616. "isEncrypted": False,
  617. "model": "d",
  618. "modelFriendlyName": "Contact Sensor",
  619. "modelName": SwitchbotModel.CONTACT_SENSOR,
  620. "rawAdvData": b"d\x00\xda\x04\x00F\x01\x8f\xc4",
  621. },
  622. device=ble_device,
  623. rssi=-70,
  624. active=True,
  625. )
  626. def test_contact_sensor_open():
  627. """Test parsing mfr adv data from new bot firmware."""
  628. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  629. adv_data = generate_advertisement_data(
  630. manufacturer_data={2409: b"\xcb9\xcd\xc4=F\x84\x9c\x00\x17\x00QD"},
  631. service_data={
  632. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@\xda\x02\x00\x17\x00QD"
  633. },
  634. tx_power=-127,
  635. rssi=-59,
  636. )
  637. result = parse_advertisement_data(ble_device, adv_data)
  638. assert result == SwitchBotAdvertisement(
  639. address="aa:bb:cc:dd:ee:ff",
  640. data={
  641. "data": {
  642. "battery": 90,
  643. "button_count": 4,
  644. "contact_open": True,
  645. "contact_timeout": False,
  646. "is_light": False,
  647. "motion_detected": True,
  648. "tested": False,
  649. },
  650. "isEncrypted": False,
  651. "model": "d",
  652. "modelFriendlyName": "Contact Sensor",
  653. "modelName": SwitchbotModel.CONTACT_SENSOR,
  654. "rawAdvData": b"d@\xda\x02\x00\x17\x00QD",
  655. },
  656. device=ble_device,
  657. rssi=-59,
  658. active=True,
  659. )
  660. def test_contact_sensor_closed():
  661. """Test parsing mfr adv data from new bot firmware."""
  662. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  663. adv_data = generate_advertisement_data(
  664. manufacturer_data={2409: b"\xcb9\xcd\xc4=F\x89\x8c\x00+\x00\x19\x84"},
  665. service_data={
  666. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@\xda\x00\x00+\x00\x19\x84"
  667. },
  668. tx_power=-127,
  669. rssi=-50,
  670. )
  671. result = parse_advertisement_data(ble_device, adv_data)
  672. assert result == SwitchBotAdvertisement(
  673. address="aa:bb:cc:dd:ee:ff",
  674. data={
  675. "data": {
  676. "battery": 90,
  677. "button_count": 4,
  678. "contact_open": False,
  679. "contact_timeout": False,
  680. "is_light": False,
  681. "motion_detected": True,
  682. "tested": False,
  683. },
  684. "isEncrypted": False,
  685. "model": "d",
  686. "modelFriendlyName": "Contact Sensor",
  687. "modelName": SwitchbotModel.CONTACT_SENSOR,
  688. "rawAdvData": b"d@\xda\x00\x00+\x00\x19\x84",
  689. },
  690. device=ble_device,
  691. rssi=-50,
  692. active=True,
  693. )
  694. def test_switchbot_passive():
  695. """Test parsing switchbot as passive."""
  696. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  697. adv_data = generate_advertisement_data(
  698. manufacturer_data={89: bytes.fromhex("d51cfb397856")},
  699. service_data={},
  700. tx_power=-127,
  701. rssi=-50,
  702. )
  703. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.BOT)
  704. assert result == SwitchBotAdvertisement(
  705. address="aa:bb:cc:dd:ee:ff",
  706. data={
  707. "data": {
  708. "battery": None,
  709. "switchMode": None,
  710. "isOn": None,
  711. },
  712. "isEncrypted": False,
  713. "model": "H",
  714. "modelFriendlyName": "Bot",
  715. "modelName": SwitchbotModel.BOT,
  716. "rawAdvData": None,
  717. },
  718. device=ble_device,
  719. rssi=-50,
  720. active=False,
  721. )
  722. def test_bulb_active():
  723. """Test parsing bulb as active."""
  724. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  725. adv_data = generate_advertisement_data(
  726. manufacturer_data={2409: b"\x84\xf7\x03\xb4\xcbz\x03\xe4!\x00\x00"},
  727. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"u\x00d"},
  728. tx_power=-127,
  729. rssi=-50,
  730. )
  731. result = parse_advertisement_data(ble_device, adv_data)
  732. assert result == SwitchBotAdvertisement(
  733. address="aa:bb:cc:dd:ee:ff",
  734. data={
  735. "data": {
  736. "brightness": 100,
  737. "color_mode": 1,
  738. "delay": False,
  739. "isOn": True,
  740. "loop_index": 0,
  741. "preset": False,
  742. "sequence_number": 3,
  743. "speed": 0,
  744. },
  745. "isEncrypted": False,
  746. "model": "u",
  747. "modelFriendlyName": "Color Bulb",
  748. "modelName": SwitchbotModel.COLOR_BULB,
  749. "rawAdvData": b"u\x00d",
  750. },
  751. device=ble_device,
  752. rssi=-50,
  753. active=True,
  754. )
  755. def test_wosensor_passive_and_active():
  756. """Test parsing wosensor as passive with active data as well."""
  757. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  758. adv_data = generate_advertisement_data(
  759. manufacturer_data={2409: b"\xd7\xc1}]\xebC\xde\x03\x06\x985"},
  760. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\xe4\x06\x985"},
  761. tx_power=-127,
  762. rssi=-50,
  763. )
  764. result = parse_advertisement_data(ble_device, adv_data)
  765. assert result == SwitchBotAdvertisement(
  766. address="aa:bb:cc:dd:ee:ff",
  767. data={
  768. "data": {
  769. "battery": 100,
  770. "fahrenheit": False,
  771. "humidity": 53,
  772. "temp": {"c": 24.6, "f": 76.28},
  773. "temperature": 24.6,
  774. },
  775. "isEncrypted": False,
  776. "model": "T",
  777. "modelFriendlyName": "Meter",
  778. "modelName": SwitchbotModel.METER,
  779. "rawAdvData": b"T\x00\xe4\x06\x985",
  780. },
  781. device=ble_device,
  782. rssi=-50,
  783. active=True,
  784. )
  785. def test_wosensor_active():
  786. """Test parsing wosensor with active data as well."""
  787. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  788. adv_data = generate_advertisement_data(
  789. manufacturer_data={},
  790. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\xe4\x06\x985"},
  791. tx_power=-127,
  792. rssi=-50,
  793. )
  794. result = parse_advertisement_data(ble_device, adv_data)
  795. assert result == SwitchBotAdvertisement(
  796. address="aa:bb:cc:dd:ee:ff",
  797. data={
  798. "data": {
  799. "battery": 100,
  800. "fahrenheit": False,
  801. "humidity": 53,
  802. "temp": {"c": 24.6, "f": 76.28},
  803. "temperature": 24.6,
  804. },
  805. "isEncrypted": False,
  806. "model": "T",
  807. "modelFriendlyName": "Meter",
  808. "modelName": SwitchbotModel.METER,
  809. "rawAdvData": b"T\x00\xe4\x06\x985",
  810. },
  811. device=ble_device,
  812. rssi=-50,
  813. active=True,
  814. )
  815. def test_wosensor_passive_only():
  816. """Test parsing wosensor with only passive data."""
  817. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  818. adv_data = generate_advertisement_data(
  819. manufacturer_data={2409: b"\xd7\xc1}]\xebC\xde\x03\x06\x985"},
  820. service_data={},
  821. tx_power=-127,
  822. rssi=-50,
  823. )
  824. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER)
  825. assert result == SwitchBotAdvertisement(
  826. address="aa:bb:cc:dd:ee:ff",
  827. data={
  828. "data": {
  829. "battery": None,
  830. "fahrenheit": False,
  831. "humidity": 53,
  832. "temp": {"c": 24.6, "f": 76.28},
  833. "temperature": 24.6,
  834. },
  835. "isEncrypted": False,
  836. "model": "T",
  837. "modelFriendlyName": "Meter",
  838. "modelName": SwitchbotModel.METER,
  839. "rawAdvData": None,
  840. },
  841. device=ble_device,
  842. rssi=-50,
  843. active=False,
  844. )
  845. def test_wosensor_active_zero_data():
  846. """Test parsing wosensor with active data but all values are zero."""
  847. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  848. adv_data = generate_advertisement_data(
  849. manufacturer_data={},
  850. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\x00\x00\x00\x00"},
  851. tx_power=-127,
  852. rssi=-50,
  853. )
  854. result = parse_advertisement_data(ble_device, adv_data)
  855. assert result == SwitchBotAdvertisement(
  856. address="aa:bb:cc:dd:ee:ff",
  857. data={
  858. "data": {},
  859. "isEncrypted": False,
  860. "model": "T",
  861. "rawAdvData": b"T\x00\x00\x00\x00\x00",
  862. },
  863. device=ble_device,
  864. rssi=-50,
  865. active=True,
  866. )
  867. def test_wohub2_passive_and_active():
  868. """Test parsing wosensor as passive with active data as well."""
  869. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  870. adv_data = generate_advertisement_data(
  871. manufacturer_data={
  872. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x00\xfffT\x1a\xf1\x82\x07\x9a2\x00"
  873. },
  874. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"v\x00"},
  875. tx_power=-127,
  876. rssi=-50,
  877. )
  878. result = parse_advertisement_data(ble_device, adv_data)
  879. assert result == SwitchBotAdvertisement(
  880. address="aa:bb:cc:dd:ee:ff",
  881. data={
  882. "data": {
  883. "fahrenheit": False,
  884. "humidity": 50,
  885. "lightLevel": 2,
  886. "illuminance": 10,
  887. "temp": {"c": 26.7, "f": 80.06},
  888. "temperature": 26.7,
  889. },
  890. "isEncrypted": False,
  891. "model": "v",
  892. "modelFriendlyName": "Hub 2",
  893. "modelName": SwitchbotModel.HUB2,
  894. "rawAdvData": b"v\x00",
  895. },
  896. device=ble_device,
  897. rssi=-50,
  898. active=True,
  899. )
  900. def test_woiosensor_passive_and_active():
  901. """Test parsing woiosensor as passive with active data as well."""
  902. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  903. adv_data = generate_advertisement_data(
  904. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee\xff\xe0\x0f\x06\x985\x00"},
  905. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"w\x00\xe4"},
  906. tx_power=-127,
  907. rssi=-50,
  908. )
  909. result = parse_advertisement_data(ble_device, adv_data)
  910. assert result == SwitchBotAdvertisement(
  911. address="aa:bb:cc:dd:ee:ff",
  912. data={
  913. "data": {
  914. "battery": 100,
  915. "fahrenheit": False,
  916. "humidity": 53,
  917. "temp": {"c": 24.6, "f": 76.28},
  918. "temperature": 24.6,
  919. },
  920. "isEncrypted": False,
  921. "model": "w",
  922. "modelFriendlyName": "Indoor/Outdoor Meter",
  923. "modelName": SwitchbotModel.IO_METER,
  924. "rawAdvData": b"w\x00\xe4",
  925. },
  926. device=ble_device,
  927. rssi=-50,
  928. active=True,
  929. )
  930. def test_woiosensor_passive_only():
  931. """Test parsing woiosensor with only passive data."""
  932. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  933. adv_data = generate_advertisement_data(
  934. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee\xff\xe0\x0f\x06\x985\x00"},
  935. service_data={},
  936. tx_power=-127,
  937. rssi=-50,
  938. )
  939. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.IO_METER)
  940. assert result == SwitchBotAdvertisement(
  941. address="aa:bb:cc:dd:ee:ff",
  942. data={
  943. "data": {
  944. "battery": None,
  945. "fahrenheit": False,
  946. "humidity": 53,
  947. "temp": {"c": 24.6, "f": 76.28},
  948. "temperature": 24.6,
  949. },
  950. "isEncrypted": False,
  951. "model": "w",
  952. "modelFriendlyName": "Indoor/Outdoor Meter",
  953. "modelName": SwitchbotModel.IO_METER,
  954. "rawAdvData": None,
  955. },
  956. device=ble_device,
  957. rssi=-50,
  958. active=False,
  959. )
  960. def test_motion_sensor_clear():
  961. """Test parsing motion sensor with clear data."""
  962. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  963. adv_data = generate_advertisement_data(
  964. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIj\x1c\x00f"},
  965. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x00f\x01"},
  966. tx_power=-127,
  967. rssi=-87,
  968. )
  969. result = parse_advertisement_data(
  970. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  971. )
  972. assert result == SwitchBotAdvertisement(
  973. address="aa:bb:cc:dd:ee:ff",
  974. data={
  975. "data": {
  976. "battery": 98,
  977. "iot": 0,
  978. "is_light": False,
  979. "led": 0,
  980. "light_intensity": 1,
  981. "motion_detected": False,
  982. "sense_distance": 0,
  983. "tested": False,
  984. },
  985. "isEncrypted": False,
  986. "model": "s",
  987. "modelFriendlyName": "Motion Sensor",
  988. "modelName": SwitchbotModel.MOTION_SENSOR,
  989. "rawAdvData": b"s\x00\xe2\x00f\x01",
  990. },
  991. device=ble_device,
  992. rssi=-87,
  993. active=True,
  994. )
  995. def test_motion_sensor_clear_passive():
  996. """Test parsing motion sensor with clear data."""
  997. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  998. adv_data = generate_advertisement_data(
  999. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIj\x1c\x00f"},
  1000. service_data={},
  1001. tx_power=-127,
  1002. rssi=-87,
  1003. )
  1004. result = parse_advertisement_data(
  1005. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1006. )
  1007. assert result == SwitchBotAdvertisement(
  1008. address="aa:bb:cc:dd:ee:ff",
  1009. data={
  1010. "data": {
  1011. "battery": None,
  1012. "iot": None,
  1013. "is_light": False,
  1014. "led": None,
  1015. "light_intensity": None,
  1016. "motion_detected": False,
  1017. "sense_distance": None,
  1018. "tested": None,
  1019. },
  1020. "isEncrypted": False,
  1021. "model": "s",
  1022. "modelFriendlyName": "Motion Sensor",
  1023. "modelName": SwitchbotModel.MOTION_SENSOR,
  1024. "rawAdvData": None,
  1025. },
  1026. device=ble_device,
  1027. rssi=-87,
  1028. active=False,
  1029. )
  1030. def test_motion_sensor_motion():
  1031. """Test parsing motion sensor with motion data."""
  1032. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1033. adv_data = generate_advertisement_data(
  1034. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIi\\\x008"},
  1035. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s@\xe2\x008\x01"},
  1036. tx_power=-127,
  1037. rssi=-87,
  1038. )
  1039. result = parse_advertisement_data(
  1040. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1041. )
  1042. assert result == SwitchBotAdvertisement(
  1043. address="aa:bb:cc:dd:ee:ff",
  1044. data={
  1045. "data": {
  1046. "battery": 98,
  1047. "iot": 0,
  1048. "is_light": False,
  1049. "led": 0,
  1050. "light_intensity": 1,
  1051. "motion_detected": True,
  1052. "sense_distance": 0,
  1053. "tested": False,
  1054. },
  1055. "isEncrypted": False,
  1056. "model": "s",
  1057. "modelFriendlyName": "Motion Sensor",
  1058. "modelName": SwitchbotModel.MOTION_SENSOR,
  1059. "rawAdvData": b"s@\xe2\x008\x01",
  1060. },
  1061. device=ble_device,
  1062. rssi=-87,
  1063. active=True,
  1064. )
  1065. def test_motion_sensor_motion_passive():
  1066. """Test parsing motion sensor with motion data."""
  1067. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1068. adv_data = generate_advertisement_data(
  1069. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIi\\\x008"},
  1070. service_data={},
  1071. tx_power=-127,
  1072. rssi=-87,
  1073. )
  1074. result = parse_advertisement_data(
  1075. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1076. )
  1077. assert result == SwitchBotAdvertisement(
  1078. address="aa:bb:cc:dd:ee:ff",
  1079. data={
  1080. "data": {
  1081. "battery": None,
  1082. "iot": None,
  1083. "is_light": False,
  1084. "led": None,
  1085. "light_intensity": None,
  1086. "motion_detected": True,
  1087. "sense_distance": None,
  1088. "tested": None,
  1089. },
  1090. "isEncrypted": False,
  1091. "model": "s",
  1092. "modelFriendlyName": "Motion Sensor",
  1093. "modelName": SwitchbotModel.MOTION_SENSOR,
  1094. "rawAdvData": None,
  1095. },
  1096. device=ble_device,
  1097. rssi=-87,
  1098. active=False,
  1099. )
  1100. def test_motion_sensor_is_light_passive():
  1101. """Test parsing motion sensor with motion data."""
  1102. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1103. adv_data = generate_advertisement_data(
  1104. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIs,\x04g"},
  1105. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x04g\x02"},
  1106. tx_power=-127,
  1107. rssi=-93,
  1108. )
  1109. result = parse_advertisement_data(
  1110. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1111. )
  1112. assert result == SwitchBotAdvertisement(
  1113. address="aa:bb:cc:dd:ee:ff",
  1114. data={
  1115. "data": {
  1116. "battery": 98,
  1117. "iot": 0,
  1118. "is_light": True,
  1119. "led": 0,
  1120. "light_intensity": 2,
  1121. "motion_detected": False,
  1122. "sense_distance": 0,
  1123. "tested": False,
  1124. },
  1125. "isEncrypted": False,
  1126. "model": "s",
  1127. "modelFriendlyName": "Motion Sensor",
  1128. "modelName": SwitchbotModel.MOTION_SENSOR,
  1129. "rawAdvData": b"s\x00\xe2\x04g\x02",
  1130. },
  1131. device=ble_device,
  1132. rssi=-93,
  1133. active=True,
  1134. )
  1135. def test_motion_sensor_is_light_active():
  1136. """Test parsing motion sensor with motion data."""
  1137. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1138. adv_data = generate_advertisement_data(
  1139. manufacturer_data={},
  1140. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x04g\x02"},
  1141. tx_power=-127,
  1142. rssi=-93,
  1143. )
  1144. result = parse_advertisement_data(
  1145. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1146. )
  1147. assert result == SwitchBotAdvertisement(
  1148. address="aa:bb:cc:dd:ee:ff",
  1149. data={
  1150. "data": {
  1151. "battery": 98,
  1152. "iot": 0,
  1153. "is_light": True,
  1154. "led": 0,
  1155. "light_intensity": 2,
  1156. "motion_detected": False,
  1157. "sense_distance": 0,
  1158. "tested": False,
  1159. },
  1160. "isEncrypted": False,
  1161. "model": "s",
  1162. "modelFriendlyName": "Motion Sensor",
  1163. "modelName": SwitchbotModel.MOTION_SENSOR,
  1164. "rawAdvData": b"s\x00\xe2\x04g\x02",
  1165. },
  1166. device=ble_device,
  1167. rssi=-93,
  1168. active=True,
  1169. )
  1170. def test_motion_with_light_detected():
  1171. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1172. adv_data = generate_advertisement_data(
  1173. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIvl\x00,"},
  1174. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s@\xe2\x00,\x02"},
  1175. tx_power=-127,
  1176. rssi=-84,
  1177. )
  1178. result = parse_advertisement_data(
  1179. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1180. )
  1181. assert result == SwitchBotAdvertisement(
  1182. address="aa:bb:cc:dd:ee:ff",
  1183. data={
  1184. "data": {
  1185. "battery": 98,
  1186. "iot": 0,
  1187. "is_light": True,
  1188. "led": 0,
  1189. "light_intensity": 2,
  1190. "motion_detected": True,
  1191. "sense_distance": 0,
  1192. "tested": False,
  1193. },
  1194. "isEncrypted": False,
  1195. "model": "s",
  1196. "modelFriendlyName": "Motion Sensor",
  1197. "modelName": SwitchbotModel.MOTION_SENSOR,
  1198. "rawAdvData": b"s@\xe2\x00,\x02",
  1199. },
  1200. device=ble_device,
  1201. rssi=-84,
  1202. active=True,
  1203. )
  1204. def test_parsing_lock_active():
  1205. """Test parsing lock with active data."""
  1206. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1207. adv_data = generate_advertisement_data(
  1208. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00 "},
  1209. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"o\x80d"},
  1210. rssi=-67,
  1211. )
  1212. result = parse_advertisement_data(ble_device, adv_data)
  1213. assert result == SwitchBotAdvertisement(
  1214. address="aa:bb:cc:dd:ee:ff",
  1215. data={
  1216. "data": {
  1217. "auto_lock_paused": False,
  1218. "battery": 100,
  1219. "calibration": True,
  1220. "door_open": False,
  1221. "double_lock_mode": False,
  1222. "night_latch": False,
  1223. "status": LockStatus.LOCKED,
  1224. "unclosed_alarm": False,
  1225. "unlocked_alarm": False,
  1226. "update_from_secondary_lock": False,
  1227. },
  1228. "isEncrypted": False,
  1229. "model": "o",
  1230. "modelFriendlyName": "Lock",
  1231. "modelName": SwitchbotModel.LOCK,
  1232. "rawAdvData": b"o\x80d",
  1233. },
  1234. device=ble_device,
  1235. rssi=-67,
  1236. active=True,
  1237. )
  1238. def test_parsing_lock_passive():
  1239. """Test parsing lock with active data."""
  1240. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1241. adv_data = generate_advertisement_data(
  1242. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00 "}, rssi=-67
  1243. )
  1244. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK)
  1245. assert result == SwitchBotAdvertisement(
  1246. address="aa:bb:cc:dd:ee:ff",
  1247. data={
  1248. "data": {
  1249. "auto_lock_paused": False,
  1250. "battery": None,
  1251. "calibration": True,
  1252. "door_open": False,
  1253. "double_lock_mode": False,
  1254. "night_latch": False,
  1255. "status": LockStatus.LOCKED,
  1256. "unclosed_alarm": False,
  1257. "unlocked_alarm": False,
  1258. "update_from_secondary_lock": False,
  1259. },
  1260. "isEncrypted": False,
  1261. "model": "o",
  1262. "modelFriendlyName": "Lock",
  1263. "modelName": SwitchbotModel.LOCK,
  1264. "rawAdvData": None,
  1265. },
  1266. device=ble_device,
  1267. rssi=-67,
  1268. active=False,
  1269. )
  1270. def test_parsing_lock_pro_active():
  1271. """Test parsing lock pro with active data."""
  1272. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1273. adv_data = generate_advertisement_data(
  1274. manufacturer_data={2409: b"\xc8\xf5,\xd9-V\x07\x82\x00d\x00\x00"},
  1275. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"$\x80d"},
  1276. rssi=-80,
  1277. )
  1278. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK_PRO)
  1279. assert result == SwitchBotAdvertisement(
  1280. address="aa:bb:cc:dd:ee:ff",
  1281. data={
  1282. "data": {
  1283. "battery": 100,
  1284. "calibration": True,
  1285. "status": LockStatus.LOCKED,
  1286. "update_from_secondary_lock": False,
  1287. "door_open": False,
  1288. "double_lock_mode": False,
  1289. "unclosed_alarm": False,
  1290. "unlocked_alarm": False,
  1291. "auto_lock_paused": False,
  1292. "night_latch": False,
  1293. },
  1294. "model": "$",
  1295. "isEncrypted": False,
  1296. "modelFriendlyName": "Lock Pro",
  1297. "modelName": SwitchbotModel.LOCK_PRO,
  1298. "rawAdvData": b"$\x80d",
  1299. },
  1300. device=ble_device,
  1301. rssi=-80,
  1302. active=True,
  1303. )
  1304. def test_parsing_lock_pro_passive():
  1305. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1306. adv_data = generate_advertisement_data(
  1307. manufacturer_data={2409: bytes.fromhex("aabbccddeeff208200640000")}, rssi=-67
  1308. )
  1309. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK_PRO)
  1310. assert result == SwitchBotAdvertisement(
  1311. address="aa:bb:cc:dd:ee:ff",
  1312. data={
  1313. "data": {
  1314. "battery": None,
  1315. "calibration": True,
  1316. "status": LockStatus.LOCKED,
  1317. "update_from_secondary_lock": False,
  1318. "door_open": False,
  1319. "double_lock_mode": False,
  1320. "unclosed_alarm": False,
  1321. "unlocked_alarm": False,
  1322. "auto_lock_paused": False,
  1323. "night_latch": False,
  1324. },
  1325. "model": "$",
  1326. "isEncrypted": False,
  1327. "modelFriendlyName": "Lock Pro",
  1328. "modelName": SwitchbotModel.LOCK_PRO,
  1329. "rawAdvData": None,
  1330. },
  1331. device=ble_device,
  1332. rssi=-67,
  1333. active=False,
  1334. )
  1335. def test_parsing_lock_pro_passive_nightlatch_disabled():
  1336. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1337. adv_data = generate_advertisement_data(
  1338. manufacturer_data={2409: bytes.fromhex("aabbccddeeff0a8200630000")}, rssi=-67
  1339. )
  1340. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK_PRO)
  1341. assert result == SwitchBotAdvertisement(
  1342. address="aa:bb:cc:dd:ee:ff",
  1343. data={
  1344. "data": {
  1345. "battery": None,
  1346. "calibration": True,
  1347. "status": LockStatus.LOCKED,
  1348. "update_from_secondary_lock": False,
  1349. "door_open": False,
  1350. "double_lock_mode": False,
  1351. "unclosed_alarm": False,
  1352. "unlocked_alarm": False,
  1353. "auto_lock_paused": False,
  1354. "night_latch": False,
  1355. },
  1356. "model": "$",
  1357. "isEncrypted": False,
  1358. "modelFriendlyName": "Lock Pro",
  1359. "modelName": SwitchbotModel.LOCK_PRO,
  1360. "rawAdvData": None,
  1361. },
  1362. device=ble_device,
  1363. rssi=-67,
  1364. active=False,
  1365. )
  1366. def test_parsing_lock_active_old_firmware():
  1367. """Test parsing lock with active data. Old firmware."""
  1368. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1369. adv_data = generate_advertisement_data(
  1370. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00"},
  1371. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"o\x80d"},
  1372. rssi=-67,
  1373. )
  1374. result = parse_advertisement_data(ble_device, adv_data)
  1375. assert result == SwitchBotAdvertisement(
  1376. address="aa:bb:cc:dd:ee:ff",
  1377. data={
  1378. "data": {
  1379. "auto_lock_paused": False,
  1380. "battery": 100,
  1381. "calibration": True,
  1382. "door_open": False,
  1383. "double_lock_mode": False,
  1384. "night_latch": False,
  1385. "status": LockStatus.LOCKED,
  1386. "unclosed_alarm": False,
  1387. "unlocked_alarm": False,
  1388. "update_from_secondary_lock": False,
  1389. },
  1390. "isEncrypted": False,
  1391. "model": "o",
  1392. "modelFriendlyName": "Lock",
  1393. "modelName": SwitchbotModel.LOCK,
  1394. "rawAdvData": b"o\x80d",
  1395. },
  1396. device=ble_device,
  1397. rssi=-67,
  1398. active=True,
  1399. )
  1400. def test_parsing_lock_passive_old_firmware():
  1401. """Test parsing lock with active data. Old firmware."""
  1402. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1403. adv_data = generate_advertisement_data(
  1404. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00"}, rssi=-67
  1405. )
  1406. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK)
  1407. assert result == SwitchBotAdvertisement(
  1408. address="aa:bb:cc:dd:ee:ff",
  1409. data={
  1410. "data": {
  1411. "auto_lock_paused": False,
  1412. "battery": None,
  1413. "calibration": True,
  1414. "door_open": False,
  1415. "double_lock_mode": False,
  1416. "night_latch": False,
  1417. "status": LockStatus.LOCKED,
  1418. "unclosed_alarm": False,
  1419. "unlocked_alarm": False,
  1420. "update_from_secondary_lock": False,
  1421. },
  1422. "isEncrypted": False,
  1423. "model": "o",
  1424. "modelFriendlyName": "Lock",
  1425. "modelName": SwitchbotModel.LOCK,
  1426. "rawAdvData": None,
  1427. },
  1428. device=ble_device,
  1429. rssi=-67,
  1430. active=False,
  1431. )
  1432. def test_meter_pro_active() -> None:
  1433. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1434. adv_data = generate_advertisement_data(
  1435. manufacturer_data={2409: b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05"},
  1436. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"4\x00d"},
  1437. rssi=-67,
  1438. )
  1439. result = parse_advertisement_data(ble_device, adv_data)
  1440. assert result == SwitchBotAdvertisement(
  1441. address="aa:bb:cc:dd:ee:ff",
  1442. data={
  1443. "data": {
  1444. "battery": 100,
  1445. "fahrenheit": False,
  1446. "humidity": 44,
  1447. "temp": {"c": 23.8, "f": 74.84},
  1448. "temperature": 23.8,
  1449. },
  1450. "isEncrypted": False,
  1451. "model": "4",
  1452. "modelFriendlyName": "Meter",
  1453. "modelName": SwitchbotModel.METER_PRO,
  1454. "rawAdvData": b"4\x00d",
  1455. },
  1456. device=ble_device,
  1457. rssi=-67,
  1458. active=True,
  1459. )
  1460. def test_meter_pro_passive() -> None:
  1461. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1462. adv_data = generate_advertisement_data(
  1463. manufacturer_data={2409: b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05"},
  1464. rssi=-67,
  1465. )
  1466. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER_PRO)
  1467. assert result == SwitchBotAdvertisement(
  1468. address="aa:bb:cc:dd:ee:ff",
  1469. data={
  1470. "data": {
  1471. "battery": None,
  1472. "fahrenheit": False,
  1473. "humidity": 44,
  1474. "temp": {"c": 23.8, "f": 74.84},
  1475. "temperature": 23.8,
  1476. },
  1477. "isEncrypted": False,
  1478. "model": "4",
  1479. "modelFriendlyName": "Meter",
  1480. "modelName": SwitchbotModel.METER_PRO,
  1481. "rawAdvData": None,
  1482. },
  1483. device=ble_device,
  1484. rssi=-67,
  1485. active=False,
  1486. )
  1487. def test_meter_pro_c_active() -> None:
  1488. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1489. adv_data = generate_advertisement_data(
  1490. manufacturer_data={
  1491. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x02\xd5\x00"
  1492. },
  1493. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"5\x00d"},
  1494. rssi=-67,
  1495. )
  1496. result = parse_advertisement_data(ble_device, adv_data)
  1497. assert result == SwitchBotAdvertisement(
  1498. address="aa:bb:cc:dd:ee:ff",
  1499. data={
  1500. "data": {
  1501. "battery": 100,
  1502. "fahrenheit": True,
  1503. "humidity": 36,
  1504. "temp": {"c": 27.7, "f": 81.86},
  1505. "temperature": 27.7,
  1506. "co2": 725,
  1507. },
  1508. "isEncrypted": False,
  1509. "model": "5",
  1510. "modelFriendlyName": "Meter",
  1511. "modelName": SwitchbotModel.METER_PRO_C,
  1512. "rawAdvData": b"5\x00d",
  1513. },
  1514. device=ble_device,
  1515. rssi=-67,
  1516. active=True,
  1517. )
  1518. def test_meter_pro_c_passive() -> None:
  1519. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1520. adv_data = generate_advertisement_data(
  1521. manufacturer_data={
  1522. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x02\xd5\x00"
  1523. },
  1524. rssi=-67,
  1525. )
  1526. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER_PRO_C)
  1527. assert result == SwitchBotAdvertisement(
  1528. address="aa:bb:cc:dd:ee:ff",
  1529. data={
  1530. "data": {
  1531. "battery": None,
  1532. "fahrenheit": True,
  1533. "humidity": 36,
  1534. "temp": {"c": 27.7, "f": 81.86},
  1535. "temperature": 27.7,
  1536. "co2": 725,
  1537. },
  1538. "isEncrypted": False,
  1539. "model": "5",
  1540. "modelFriendlyName": "Meter",
  1541. "modelName": SwitchbotModel.METER_PRO_C,
  1542. "rawAdvData": None,
  1543. },
  1544. device=ble_device,
  1545. rssi=-67,
  1546. active=False,
  1547. )
  1548. def test_parse_advertisement_data_keypad():
  1549. """Test parse_advertisement_data for the keypad."""
  1550. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1551. adv_data = generate_advertisement_data(
  1552. manufacturer_data={2409: b"\xeb\x13\x02\xe6#\x0f\x8fd\x00\x00\x00\x00"},
  1553. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"y\x00d"},
  1554. rssi=-67,
  1555. )
  1556. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.KEYPAD)
  1557. assert result == SwitchBotAdvertisement(
  1558. address="aa:bb:cc:dd:ee:ff",
  1559. data={
  1560. "data": {"attempt_state": 143, "battery": 100},
  1561. "isEncrypted": False,
  1562. "model": "y",
  1563. "modelFriendlyName": "Keypad",
  1564. "modelName": SwitchbotModel.KEYPAD,
  1565. "rawAdvData": b"y\x00d",
  1566. },
  1567. device=ble_device,
  1568. rssi=-67,
  1569. active=True,
  1570. )
  1571. def test_parse_advertisement_data_relay_switch_1pm():
  1572. """Test parse_advertisement_data for the keypad."""
  1573. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1574. adv_data = generate_advertisement_data(
  1575. manufacturer_data={2409: b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00"},
  1576. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"<\x00\x00\x00"},
  1577. rssi=-67,
  1578. )
  1579. result = parse_advertisement_data(
  1580. ble_device, adv_data, SwitchbotModel.RELAY_SWITCH_1PM
  1581. )
  1582. assert result == SwitchBotAdvertisement(
  1583. address="aa:bb:cc:dd:ee:ff",
  1584. data={
  1585. "data": {
  1586. "switchMode": True,
  1587. "sequence_number": 71,
  1588. "isOn": True,
  1589. "power": 4.9,
  1590. "voltage": 0,
  1591. "current": 0,
  1592. },
  1593. "isEncrypted": False,
  1594. "model": "<",
  1595. "modelFriendlyName": "Relay Switch 1PM",
  1596. "modelName": SwitchbotModel.RELAY_SWITCH_1PM,
  1597. "rawAdvData": b"<\x00\x00\x00",
  1598. },
  1599. device=ble_device,
  1600. rssi=-67,
  1601. active=True,
  1602. )
  1603. def test_parse_advertisement_data_relay_switch_1():
  1604. """Test parse_advertisement_data for the keypad."""
  1605. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1606. adv_data = generate_advertisement_data(
  1607. manufacturer_data={2409: b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00"},
  1608. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b";\x00\x00\x00"},
  1609. rssi=-67,
  1610. )
  1611. result = parse_advertisement_data(
  1612. ble_device, adv_data, SwitchbotModel.RELAY_SWITCH_1
  1613. )
  1614. assert result == SwitchBotAdvertisement(
  1615. address="aa:bb:cc:dd:ee:ff",
  1616. data={
  1617. "data": {
  1618. "switchMode": True,
  1619. "sequence_number": 71,
  1620. "isOn": True,
  1621. },
  1622. "isEncrypted": False,
  1623. "model": ";",
  1624. "modelFriendlyName": "Relay Switch 1",
  1625. "modelName": SwitchbotModel.RELAY_SWITCH_1,
  1626. "rawAdvData": b";\x00\x00\x00",
  1627. },
  1628. device=ble_device,
  1629. rssi=-67,
  1630. active=True,
  1631. )
  1632. def test_leak_active():
  1633. """Test parse_advertisement_data for the leak detector."""
  1634. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1635. adv_data = generate_advertisement_data(
  1636. manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
  1637. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00N"},
  1638. rssi=-72,
  1639. )
  1640. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1641. assert result == SwitchBotAdvertisement(
  1642. address="aa:bb:cc:dd:ee:ff",
  1643. data={
  1644. "data": {
  1645. "leak": False,
  1646. "tampered": False,
  1647. "battery": 78,
  1648. "low_battery": False,
  1649. },
  1650. "isEncrypted": False,
  1651. "model": "&",
  1652. "modelFriendlyName": "Leak Detector",
  1653. "modelName": SwitchbotModel.LEAK,
  1654. "rawAdvData": b"&\x00N",
  1655. },
  1656. device=ble_device,
  1657. rssi=-72,
  1658. active=True,
  1659. )
  1660. def test_leak_passive():
  1661. """Test parse_advertisement_data for the leak detector."""
  1662. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1663. adv_data = generate_advertisement_data(
  1664. manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
  1665. rssi=-72,
  1666. )
  1667. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1668. assert result == SwitchBotAdvertisement(
  1669. address="aa:bb:cc:dd:ee:ff",
  1670. data={
  1671. "data": {},
  1672. "isEncrypted": False,
  1673. "model": "&",
  1674. "rawAdvData": None,
  1675. },
  1676. device=ble_device,
  1677. rssi=-72,
  1678. active=False,
  1679. )
  1680. def test_leak_no_leak_detected():
  1681. """Test parse_advertisement_data for the leak detector."""
  1682. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1683. adv_data = generate_advertisement_data(
  1684. manufacturer_data={
  1685. 2409: b"\xc4407LzJd\x98ga\xc4\n<\x05\x00\x00\x00\x00"
  1686. }, # no leak, batt
  1687. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1688. rssi=-73,
  1689. )
  1690. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1691. assert result == SwitchBotAdvertisement(
  1692. address="aa:bb:cc:dd:ee:ff",
  1693. data={
  1694. "data": {
  1695. "leak": False,
  1696. "tampered": False,
  1697. "battery": 100,
  1698. "low_battery": False,
  1699. },
  1700. "isEncrypted": False,
  1701. "model": "&",
  1702. "modelFriendlyName": "Leak Detector",
  1703. "modelName": SwitchbotModel.LEAK,
  1704. "rawAdvData": b"&\x00d",
  1705. },
  1706. device=ble_device,
  1707. rssi=-73,
  1708. active=True,
  1709. )
  1710. def test_leak_leak_detected():
  1711. """Test parse_advertisement_data for the leak detector."""
  1712. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1713. adv_data = generate_advertisement_data(
  1714. manufacturer_data={
  1715. 2409: b"\xc4407LzGd\xf9ga\xc4\x08<\x05\x00\x00\x00\x00"
  1716. }, # leak, batt
  1717. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1718. rssi=-73,
  1719. )
  1720. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1721. assert result == SwitchBotAdvertisement(
  1722. address="aa:bb:cc:dd:ee:ff",
  1723. data={
  1724. "data": {
  1725. "leak": True,
  1726. "tampered": False,
  1727. "battery": 100,
  1728. "low_battery": False,
  1729. },
  1730. "isEncrypted": False,
  1731. "model": "&",
  1732. "modelFriendlyName": "Leak Detector",
  1733. "modelName": SwitchbotModel.LEAK,
  1734. "rawAdvData": b"&\x00d",
  1735. },
  1736. device=ble_device,
  1737. rssi=-73,
  1738. active=True,
  1739. )
  1740. def test_leak_low_battery():
  1741. """Test parse_advertisement_data for the leak detector."""
  1742. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1743. adv_data = generate_advertisement_data(
  1744. manufacturer_data={
  1745. 2409: b"\xc4407Lz\x02\t\x98\x00\x00\x00\x00<\x05\x00\x00\x00\x00"
  1746. }, # no leak, low battery
  1747. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1748. rssi=-73,
  1749. )
  1750. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1751. assert result == SwitchBotAdvertisement(
  1752. address="aa:bb:cc:dd:ee:ff",
  1753. data={
  1754. "data": {
  1755. "leak": False,
  1756. "tampered": False,
  1757. "battery": 9,
  1758. "low_battery": False,
  1759. },
  1760. "isEncrypted": False,
  1761. "model": "&",
  1762. "modelFriendlyName": "Leak Detector",
  1763. "modelName": SwitchbotModel.LEAK,
  1764. "rawAdvData": b"&\x00d",
  1765. },
  1766. device=ble_device,
  1767. rssi=-73,
  1768. active=True,
  1769. )
  1770. def test_leak_real_data_from_ha():
  1771. """Test parse_advertisement_data for the leak detector."""
  1772. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1773. adv_data = generate_advertisement_data(
  1774. manufacturer_data={
  1775. 2409: b"\\xd6407D1\\x02V\\x90\\x00\\x00\\x00\\x00\\x1e\\x05\\x00\\x00\\x00\\x00"
  1776. }, # no leak, low battery
  1777. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\\x00V"},
  1778. rssi=-73,
  1779. )
  1780. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1781. assert result == SwitchBotAdvertisement(
  1782. address="aa:bb:cc:dd:ee:ff",
  1783. data={
  1784. "data": {
  1785. "leak": True,
  1786. "tampered": False,
  1787. "battery": 68,
  1788. "low_battery": False,
  1789. },
  1790. "isEncrypted": False,
  1791. "model": "&",
  1792. "modelFriendlyName": "Leak Detector",
  1793. "modelName": SwitchbotModel.LEAK,
  1794. "rawAdvData": b"&\\x00V",
  1795. },
  1796. device=ble_device,
  1797. rssi=-73,
  1798. active=True,
  1799. )
  1800. def test_remote_active() -> None:
  1801. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1802. adv_data = generate_advertisement_data(
  1803. manufacturer_data={89: b"\xaa\xbb\xcc\xdd\xee\xff"},
  1804. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"b V\x00"},
  1805. service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
  1806. rssi=-95,
  1807. )
  1808. result = parse_advertisement_data(ble_device, adv_data)
  1809. assert result == SwitchBotAdvertisement(
  1810. address="aa:bb:cc:dd:ee:ff",
  1811. data={
  1812. "data": {
  1813. "battery": 86,
  1814. },
  1815. "isEncrypted": False,
  1816. "model": "b",
  1817. "modelFriendlyName": "Remote",
  1818. "modelName": SwitchbotModel.REMOTE,
  1819. "rawAdvData": b"b V\x00",
  1820. },
  1821. device=ble_device,
  1822. rssi=-95,
  1823. active=True,
  1824. )
  1825. def test_remote_passive() -> None:
  1826. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1827. adv_data = generate_advertisement_data(
  1828. manufacturer_data={89: b"\xaa\xbb\xcc\xdd\xee\xff"},
  1829. rssi=-97,
  1830. )
  1831. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.REMOTE)
  1832. assert result == SwitchBotAdvertisement(
  1833. address="aa:bb:cc:dd:ee:ff",
  1834. data={
  1835. "data": {
  1836. "battery": None,
  1837. },
  1838. "isEncrypted": False,
  1839. "model": "b",
  1840. "modelFriendlyName": "Remote",
  1841. "modelName": SwitchbotModel.REMOTE,
  1842. "rawAdvData": None,
  1843. },
  1844. device=ble_device,
  1845. rssi=-97,
  1846. active=False,
  1847. )
  1848. def test_parse_advertisement_data_hubmini_matter():
  1849. """Test parse_advertisement_data for the HubMini Matter."""
  1850. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1851. adv_data = generate_advertisement_data(
  1852. manufacturer_data={
  1853. 2409: b"\xe6\xa1\xcd\x1f[e\x00\x00\x00\x00\x00\x00\x14\x01\x985\x00"
  1854. },
  1855. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"%\x00"},
  1856. rssi=-67,
  1857. )
  1858. result = parse_advertisement_data(
  1859. ble_device, adv_data, SwitchbotModel.HUBMINI_MATTER
  1860. )
  1861. assert result == SwitchBotAdvertisement(
  1862. address="aa:bb:cc:dd:ee:ff",
  1863. data={
  1864. "data": {
  1865. "fahrenheit": False,
  1866. "humidity": 53,
  1867. "temp": {"c": 24.1, "f": 75.38},
  1868. "temperature": 24.1,
  1869. },
  1870. "isEncrypted": False,
  1871. "model": "%",
  1872. "modelFriendlyName": "HubMini Matter",
  1873. "modelName": SwitchbotModel.HUBMINI_MATTER,
  1874. "rawAdvData": b"%\x00",
  1875. },
  1876. device=ble_device,
  1877. rssi=-67,
  1878. active=True,
  1879. )
  1880. def test_parse_advertisement_data_roller_shade():
  1881. """Test parse_advertisement_data for roller shade."""
  1882. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1883. adv_data = generate_advertisement_data(
  1884. manufacturer_data={2409: b"\xb0\xe9\xfeT\x90\x1b,\x08\x9f\x11\x04'\x00"},
  1885. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b",\x00'\x9f\x11\x04"},
  1886. rssi=-80,
  1887. )
  1888. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.ROLLER_SHADE)
  1889. assert result == SwitchBotAdvertisement(
  1890. address="aa:bb:cc:dd:ee:ff",
  1891. data={
  1892. "rawAdvData": b",\x00'\x9f\x11\x04",
  1893. "data": {
  1894. "battery": 39,
  1895. "calibration": True,
  1896. "deviceChain": 1,
  1897. "inMotion": False,
  1898. "lightLevel": 1,
  1899. "position": 69,
  1900. "sequence_number": 44,
  1901. },
  1902. "isEncrypted": False,
  1903. "model": ",",
  1904. "modelFriendlyName": "Roller Shade",
  1905. "modelName": SwitchbotModel.ROLLER_SHADE,
  1906. },
  1907. device=ble_device,
  1908. rssi=-80,
  1909. active=True,
  1910. )
  1911. def test_hubmini_matter_passive() -> None:
  1912. """Test parsing hubmini matter with passive data."""
  1913. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1914. adv_data = generate_advertisement_data(
  1915. manufacturer_data={
  1916. 2409: b"\xe6\xa1\xcd\x1f[e\x00\x00\x00\x00\x00\x00\x14\x01\x985\x00"
  1917. },
  1918. rssi=-97,
  1919. )
  1920. result = parse_advertisement_data(
  1921. ble_device, adv_data, SwitchbotModel.HUBMINI_MATTER
  1922. )
  1923. assert result == SwitchBotAdvertisement(
  1924. address="aa:bb:cc:dd:ee:ff",
  1925. data={
  1926. "data": {
  1927. "fahrenheit": False,
  1928. "humidity": 53,
  1929. "temp": {"c": 24.1, "f": 75.38},
  1930. "temperature": 24.1,
  1931. },
  1932. "isEncrypted": False,
  1933. "model": "%",
  1934. "modelFriendlyName": "HubMini Matter",
  1935. "modelName": SwitchbotModel.HUBMINI_MATTER,
  1936. "rawAdvData": None,
  1937. },
  1938. device=ble_device,
  1939. rssi=-97,
  1940. active=False,
  1941. )
  1942. def test_roller_shade_passive() -> None:
  1943. """Test parsing roller_shade with passive data."""
  1944. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1945. adv_data = generate_advertisement_data(
  1946. manufacturer_data={2409: b"\xb0\xe9\xfeT\x90\x1b,\x08\x9f\x11\x04'\x00"},
  1947. rssi=-97,
  1948. )
  1949. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.ROLLER_SHADE)
  1950. assert result == SwitchBotAdvertisement(
  1951. address="aa:bb:cc:dd:ee:ff",
  1952. data={
  1953. "rawAdvData": None,
  1954. "data": {
  1955. "battery": None,
  1956. "calibration": True,
  1957. "deviceChain": 1,
  1958. "inMotion": False,
  1959. "lightLevel": 1,
  1960. "position": 69,
  1961. "sequence_number": 44,
  1962. },
  1963. "isEncrypted": False,
  1964. "model": ",",
  1965. "modelFriendlyName": "Roller Shade",
  1966. "modelName": SwitchbotModel.ROLLER_SHADE,
  1967. },
  1968. device=ble_device,
  1969. rssi=-97,
  1970. active=False,
  1971. )
  1972. def test_circulator_fan_active() -> None:
  1973. """Test parsing circulator fan with active data."""
  1974. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1975. adv_data = generate_advertisement_data(
  1976. manufacturer_data={2409: b"\xb0\xe9\xfeXY\xa8~LR9"},
  1977. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"~\x00R"},
  1978. rssi=-97,
  1979. )
  1980. result = parse_advertisement_data(
  1981. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  1982. )
  1983. assert result == SwitchBotAdvertisement(
  1984. address="aa:bb:cc:dd:ee:ff",
  1985. data={
  1986. "rawAdvData": b"~\x00R",
  1987. "data": {
  1988. "sequence_number": 126,
  1989. "isOn": False,
  1990. "mode": "BABY",
  1991. "nightLight": 3,
  1992. "oscillating": False,
  1993. "battery": 82,
  1994. "speed": 57,
  1995. },
  1996. "isEncrypted": False,
  1997. "model": "~",
  1998. "modelFriendlyName": "Circulator Fan",
  1999. "modelName": SwitchbotModel.CIRCULATOR_FAN,
  2000. },
  2001. device=ble_device,
  2002. rssi=-97,
  2003. active=True,
  2004. )
  2005. def test_circulator_fan_passive() -> None:
  2006. """Test parsing circulator fan with passive data."""
  2007. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2008. adv_data = generate_advertisement_data(
  2009. manufacturer_data={2409: b"\xb0\xe9\xfeXY\xa8~LR9"},
  2010. rssi=-97,
  2011. )
  2012. result = parse_advertisement_data(
  2013. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  2014. )
  2015. assert result == SwitchBotAdvertisement(
  2016. address="aa:bb:cc:dd:ee:ff",
  2017. data={
  2018. "rawAdvData": None,
  2019. "data": {
  2020. "sequence_number": 126,
  2021. "isOn": False,
  2022. "mode": "BABY",
  2023. "nightLight": 3,
  2024. "oscillating": False,
  2025. "battery": 82,
  2026. "speed": 57,
  2027. },
  2028. "isEncrypted": False,
  2029. "model": "~",
  2030. "modelFriendlyName": "Circulator Fan",
  2031. "modelName": SwitchbotModel.CIRCULATOR_FAN,
  2032. },
  2033. device=ble_device,
  2034. rssi=-97,
  2035. active=False,
  2036. )
  2037. def test_circulator_fan_with_empty_data() -> None:
  2038. """Test parsing circulator fan with empty data."""
  2039. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2040. adv_data = generate_advertisement_data(
  2041. manufacturer_data={2409: None},
  2042. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"~\x00R"},
  2043. rssi=-97,
  2044. )
  2045. result = parse_advertisement_data(
  2046. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  2047. )
  2048. assert result == SwitchBotAdvertisement(
  2049. address="aa:bb:cc:dd:ee:ff",
  2050. data={
  2051. "rawAdvData": b"~\x00R",
  2052. "data": {},
  2053. "isEncrypted": False,
  2054. "model": "~",
  2055. },
  2056. device=ble_device,
  2057. rssi=-97,
  2058. active=True,
  2059. )