test_adv_parser.py 64 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973
  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. )