test_adv_parser.py 91 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830
  1. from __future__ import annotations
  2. from typing import Any
  3. import pytest
  4. from bleak.backends.device import BLEDevice
  5. from bleak.backends.scanner import AdvertisementData
  6. from switchbot import SwitchbotModel
  7. from switchbot.adv_parser import parse_advertisement_data
  8. from switchbot.const.lock import LockStatus
  9. from switchbot.models import SwitchBotAdvertisement
  10. ADVERTISEMENT_DATA_DEFAULTS = {
  11. "local_name": "",
  12. "manufacturer_data": {},
  13. "service_data": {},
  14. "service_uuids": [],
  15. "rssi": -127,
  16. "platform_data": ((),),
  17. "tx_power": -127,
  18. }
  19. BLE_DEVICE_DEFAULTS = {
  20. "name": None,
  21. "rssi": -127,
  22. "details": None,
  23. }
  24. def generate_ble_device(
  25. address: str | None = None,
  26. name: str | None = None,
  27. details: Any | None = None,
  28. rssi: int | None = None,
  29. **kwargs: Any,
  30. ) -> BLEDevice:
  31. """Generate a BLEDevice with defaults."""
  32. new = kwargs.copy()
  33. if address is not None:
  34. new["address"] = address
  35. if name is not None:
  36. new["name"] = name
  37. if details is not None:
  38. new["details"] = details
  39. if rssi is not None:
  40. new["rssi"] = rssi
  41. for key, value in BLE_DEVICE_DEFAULTS.items():
  42. new.setdefault(key, value)
  43. return BLEDevice(**new)
  44. def generate_advertisement_data(**kwargs: Any) -> AdvertisementData:
  45. """Generate advertisement data with defaults."""
  46. new = kwargs.copy()
  47. for key, value in ADVERTISEMENT_DATA_DEFAULTS.items():
  48. new.setdefault(key, value)
  49. return AdvertisementData(**new)
  50. def test_parse_advertisement_data_curtain():
  51. """Test parse_advertisement_data for curtain."""
  52. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  53. adv_data = generate_advertisement_data(
  54. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  55. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0X\x00\x11\x04"},
  56. rssi=-80,
  57. )
  58. result = parse_advertisement_data(ble_device, adv_data)
  59. assert result == SwitchBotAdvertisement(
  60. address="aa:bb:cc:dd:ee:ff",
  61. data={
  62. "rawAdvData": b"c\xc0X\x00\x11\x04",
  63. "data": {
  64. "calibration": True,
  65. "battery": 88,
  66. "inMotion": False,
  67. "position": 100,
  68. "lightLevel": 1,
  69. "deviceChain": 1,
  70. },
  71. "isEncrypted": False,
  72. "model": "c",
  73. "modelFriendlyName": "Curtain",
  74. "modelName": SwitchbotModel.CURTAIN,
  75. },
  76. device=ble_device,
  77. rssi=-80,
  78. active=True,
  79. )
  80. def test_parse_advertisement_data_curtain_passive():
  81. """Test parse_advertisement_data for curtain passive."""
  82. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  83. adv_data = generate_advertisement_data(
  84. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  85. service_data={},
  86. rssi=-80,
  87. )
  88. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  89. assert result == SwitchBotAdvertisement(
  90. address="aa:bb:cc:dd:ee:ff",
  91. data={
  92. "rawAdvData": None,
  93. "data": {
  94. "calibration": None,
  95. "battery": None,
  96. "inMotion": False,
  97. "position": 100,
  98. "lightLevel": 1,
  99. "deviceChain": 1,
  100. },
  101. "isEncrypted": False,
  102. "model": "c",
  103. "modelFriendlyName": "Curtain",
  104. "modelName": SwitchbotModel.CURTAIN,
  105. },
  106. device=ble_device,
  107. rssi=-80,
  108. active=False,
  109. )
  110. def test_parse_advertisement_data_curtain_passive_12_bytes():
  111. """Test parse_advertisement_data for curtain passive."""
  112. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  113. adv_data = generate_advertisement_data(
  114. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04\x00"},
  115. service_data={},
  116. rssi=-80,
  117. )
  118. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  119. assert result == SwitchBotAdvertisement(
  120. address="aa:bb:cc:dd:ee:ff",
  121. data={
  122. "rawAdvData": None,
  123. "data": {
  124. "calibration": None,
  125. "battery": None,
  126. "inMotion": False,
  127. "position": 100,
  128. "lightLevel": 1,
  129. "deviceChain": 1,
  130. },
  131. "isEncrypted": False,
  132. "model": "c",
  133. "modelFriendlyName": "Curtain",
  134. "modelName": SwitchbotModel.CURTAIN,
  135. },
  136. device=ble_device,
  137. rssi=-80,
  138. active=False,
  139. )
  140. def test_parse_advertisement_data_curtain_position_zero():
  141. """Test parse_advertisement_data for curtain position zero."""
  142. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  143. adv_data = generate_advertisement_data(
  144. local_name="WoCurtain",
  145. manufacturer_data={89: b"\xc1\xc7'}U\xab"},
  146. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0\xced\x11\x04"},
  147. service_uuids=[
  148. "00001800-0000-1000-8000-00805f9b34fb",
  149. "00001801-0000-1000-8000-00805f9b34fb",
  150. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  151. ],
  152. rssi=-52,
  153. )
  154. result = parse_advertisement_data(ble_device, adv_data)
  155. assert result == SwitchBotAdvertisement(
  156. address="aa:bb:cc:dd:ee:ff",
  157. data={
  158. "rawAdvData": b"c\xd0\xced\x11\x04",
  159. "data": {
  160. "calibration": True,
  161. "battery": 78,
  162. "inMotion": False,
  163. "position": 0,
  164. "lightLevel": 1,
  165. "deviceChain": 1,
  166. },
  167. "isEncrypted": False,
  168. "model": "c",
  169. "modelFriendlyName": "Curtain",
  170. "modelName": SwitchbotModel.CURTAIN,
  171. },
  172. device=ble_device,
  173. rssi=-52,
  174. active=True,
  175. )
  176. def test_parse_advertisement_data_curtain_firmware_six_position_100():
  177. """Test parse_advertisement_data with firmware six for curtain position 100."""
  178. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  179. adv_data = generate_advertisement_data(
  180. local_name="WoCurtain",
  181. manufacturer_data={
  182. 89: b"\xf5\x98\x94\x08\xa0\xe7",
  183. 2409: b'\xf5\x98\x94\x08\xa0\xe7\x9b\x0f\x00"\x04',
  184. },
  185. service_data={
  186. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0H\x00\x12\x04",
  187. "0000fd3d-0000-1000-8000-00805f9b34fb": b'c\xc0G\x00"\x04',
  188. },
  189. service_uuids=[
  190. "00001800-0000-1000-8000-00805f9b34fb",
  191. "00001801-0000-1000-8000-00805f9b34fb",
  192. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  193. ],
  194. rssi=-62,
  195. )
  196. result = parse_advertisement_data(ble_device, adv_data)
  197. assert result == SwitchBotAdvertisement(
  198. address="aa:bb:cc:dd:ee:ff",
  199. data={
  200. "rawAdvData": b'c\xc0G\x00"\x04',
  201. "data": {
  202. "calibration": True,
  203. "battery": 71,
  204. "inMotion": False,
  205. "position": 100,
  206. "lightLevel": 2,
  207. "deviceChain": 2,
  208. },
  209. "isEncrypted": False,
  210. "model": "c",
  211. "modelFriendlyName": "Curtain",
  212. "modelName": SwitchbotModel.CURTAIN,
  213. },
  214. device=ble_device,
  215. rssi=-62,
  216. active=True,
  217. )
  218. def test_parse_advertisement_data_curtain_firmware_six_position_100_other_rssi():
  219. """Test parse_advertisement_data with firmware six for curtain position 100 other rssi."""
  220. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  221. adv_data = generate_advertisement_data(
  222. local_name="WoCurtain",
  223. manufacturer_data={
  224. 89: b"\xf5\x98\x94\x08\xa0\xe7",
  225. 2409: b'\xf5\x98\x94\x08\xa0\xe7\xa5\x0fc"\x04',
  226. },
  227. service_data={
  228. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0H\x00\x12\x04",
  229. "0000fd3d-0000-1000-8000-00805f9b34fb": b'c\xc0Gc"\x04',
  230. },
  231. service_uuids=[
  232. "00001800-0000-1000-8000-00805f9b34fb",
  233. "00001801-0000-1000-8000-00805f9b34fb",
  234. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  235. ],
  236. rssi=-67,
  237. )
  238. result = parse_advertisement_data(ble_device, adv_data)
  239. assert result == SwitchBotAdvertisement(
  240. address="aa:bb:cc:dd:ee:ff",
  241. data={
  242. "rawAdvData": b'c\xc0Gc"\x04',
  243. "data": {
  244. "calibration": True,
  245. "battery": 71,
  246. "inMotion": False,
  247. "position": 1,
  248. "lightLevel": 2,
  249. "deviceChain": 2,
  250. },
  251. "isEncrypted": False,
  252. "model": "c",
  253. "modelFriendlyName": "Curtain",
  254. "modelName": SwitchbotModel.CURTAIN,
  255. },
  256. device=ble_device,
  257. rssi=-67,
  258. active=True,
  259. )
  260. def test_parse_advertisement_data_curtain_fully_closed():
  261. """Test parse_advertisement_data with firmware six fully closed."""
  262. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  263. adv_data = generate_advertisement_data(
  264. local_name="WoCurtain",
  265. manufacturer_data={2409: b"\xc1\xc7'}U\xab\"\x0fd\x11\x04"},
  266. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0Sd\x11\x04"},
  267. service_uuids=[
  268. "00001800-0000-1000-8000-00805f9b34fb",
  269. "00001801-0000-1000-8000-00805f9b34fb",
  270. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  271. ],
  272. rssi=1,
  273. )
  274. result = parse_advertisement_data(ble_device, adv_data)
  275. assert result == SwitchBotAdvertisement(
  276. address="aa:bb:cc:dd:ee:ff",
  277. data={
  278. "rawAdvData": b"c\xc0Sd\x11\x04",
  279. "data": {
  280. "calibration": True,
  281. "battery": 83,
  282. "inMotion": False,
  283. "position": 0,
  284. "lightLevel": 1,
  285. "deviceChain": 1,
  286. },
  287. "isEncrypted": False,
  288. "model": "c",
  289. "modelFriendlyName": "Curtain",
  290. "modelName": SwitchbotModel.CURTAIN,
  291. },
  292. device=ble_device,
  293. rssi=1,
  294. active=True,
  295. )
  296. def test_parse_advertisement_data_curtain_fully_open():
  297. """Test parse_advertisement_data with firmware six fully open."""
  298. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  299. adv_data = generate_advertisement_data(
  300. local_name="WoCurtain",
  301. manufacturer_data={2409: b"\xc1\xc7'}U\xab%\x0f\x00\x11\x04"},
  302. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0S\x00\x11\x04"},
  303. service_uuids=[
  304. "00001800-0000-1000-8000-00805f9b34fb",
  305. "00001801-0000-1000-8000-00805f9b34fb",
  306. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  307. ],
  308. rssi=1,
  309. )
  310. result = parse_advertisement_data(ble_device, adv_data)
  311. assert result == SwitchBotAdvertisement(
  312. address="aa:bb:cc:dd:ee:ff",
  313. data={
  314. "rawAdvData": b"c\xc0S\x00\x11\x04",
  315. "data": {
  316. "calibration": True,
  317. "battery": 83,
  318. "inMotion": False,
  319. "position": 100,
  320. "lightLevel": 1,
  321. "deviceChain": 1,
  322. },
  323. "isEncrypted": False,
  324. "model": "c",
  325. "modelFriendlyName": "Curtain",
  326. "modelName": SwitchbotModel.CURTAIN,
  327. },
  328. device=ble_device,
  329. rssi=1,
  330. active=True,
  331. )
  332. def test_parse_advertisement_data_curtain3():
  333. """Test parse_advertisement_data for curtain 3."""
  334. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  335. adv_data = generate_advertisement_data(
  336. manufacturer_data={
  337. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\xf7\x07\x00\x11\x04\x00\x49"
  338. },
  339. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"{\xc0\x49\x00\x11\x04"},
  340. rssi=-80,
  341. )
  342. result = parse_advertisement_data(ble_device, adv_data)
  343. assert result == SwitchBotAdvertisement(
  344. address="aa:bb:cc:dd:ee:ff",
  345. data={
  346. "rawAdvData": b"{\xc0\x49\x00\x11\x04",
  347. "data": {
  348. "calibration": True,
  349. "battery": 73,
  350. "inMotion": False,
  351. "position": 100,
  352. "lightLevel": 1,
  353. "deviceChain": 1,
  354. },
  355. "isEncrypted": False,
  356. "model": "{",
  357. "modelFriendlyName": "Curtain 3",
  358. "modelName": SwitchbotModel.CURTAIN,
  359. },
  360. device=ble_device,
  361. rssi=-80,
  362. active=True,
  363. )
  364. def test_parse_advertisement_data_curtain3_passive():
  365. """Test parse_advertisement_data for curtain passive."""
  366. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  367. adv_data = generate_advertisement_data(
  368. manufacturer_data={
  369. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\xf7\x07\x00\x11\x04\x00\x49"
  370. },
  371. service_data={},
  372. rssi=-80,
  373. )
  374. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  375. assert result == SwitchBotAdvertisement(
  376. address="aa:bb:cc:dd:ee:ff",
  377. data={
  378. "rawAdvData": None,
  379. "data": {
  380. "calibration": None,
  381. "battery": 73,
  382. "inMotion": False,
  383. "position": 100,
  384. "lightLevel": 1,
  385. "deviceChain": 1,
  386. },
  387. "isEncrypted": False,
  388. "model": "c",
  389. "modelFriendlyName": "Curtain",
  390. "modelName": SwitchbotModel.CURTAIN,
  391. },
  392. device=ble_device,
  393. rssi=-80,
  394. active=False,
  395. )
  396. def test_parse_advertisement_data_contact():
  397. """Test parse_advertisement_data for the contact sensor."""
  398. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  399. adv_data = generate_advertisement_data(
  400. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  401. service_data={
  402. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@d\x05\x00u\x00\xf8\x12"
  403. },
  404. rssi=-80,
  405. )
  406. result = parse_advertisement_data(ble_device, adv_data)
  407. assert result == SwitchBotAdvertisement(
  408. address="aa:bb:cc:dd:ee:ff",
  409. data={
  410. "rawAdvData": b"d@d\x05\x00u\x00\xf8\x12",
  411. "data": {
  412. "button_count": 2,
  413. "contact_open": True,
  414. "contact_timeout": True,
  415. "is_light": True,
  416. "battery": 100,
  417. "motion_detected": True,
  418. "tested": False,
  419. },
  420. "isEncrypted": False,
  421. "model": "d",
  422. "modelFriendlyName": "Contact Sensor",
  423. "modelName": SwitchbotModel.CONTACT_SENSOR,
  424. },
  425. device=ble_device,
  426. rssi=-80,
  427. active=True,
  428. )
  429. def test_parse_advertisement_data_empty():
  430. """Test parse_advertisement_data with empty data does not blow up."""
  431. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  432. adv_data = generate_advertisement_data(
  433. manufacturer_data={2403: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  434. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b""},
  435. )
  436. result = parse_advertisement_data(ble_device, adv_data)
  437. assert result is None
  438. def test_new_bot_firmware():
  439. """Test parsing adv data from new bot firmware."""
  440. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  441. adv_data = generate_advertisement_data(
  442. manufacturer_data={89: b"\xd8.\xad\xcd\r\x85"},
  443. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"H\x10\xe1"},
  444. service_uuids=["CBA20D00-224D-11E6-9FB8-0002A5D5C51B"],
  445. rssi=-90,
  446. )
  447. result = parse_advertisement_data(ble_device, adv_data)
  448. assert result == SwitchBotAdvertisement(
  449. address="aa:bb:cc:dd:ee:ff",
  450. data={
  451. "rawAdvData": b"H\x10\xe1",
  452. "data": {"switchMode": False, "isOn": False, "battery": 97},
  453. "model": "H",
  454. "isEncrypted": False,
  455. "modelFriendlyName": "Bot",
  456. "modelName": SwitchbotModel.BOT,
  457. },
  458. device=ble_device,
  459. rssi=-90,
  460. active=True,
  461. )
  462. def test_parse_advertisement_data_curtain_firmware_six_fully_closed():
  463. """Test parse_advertisement_data with firmware six fully closed."""
  464. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  465. adv_data = generate_advertisement_data(
  466. local_name="WoCurtain",
  467. manufacturer_data={
  468. 89: b"\xcc\xf4\xc4\xf9\xacl",
  469. 2409: b"\xcc\xf4\xc4\xf9\xacl\xeb\x0fd\x12\x04",
  470. },
  471. service_data={
  472. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0Yd\x11\x04",
  473. "0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0dd\x12\x04",
  474. },
  475. service_uuids=[
  476. "00001800-0000-1000-8000-00805f9b34fb",
  477. "00001801-0000-1000-8000-00805f9b34fb",
  478. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  479. ],
  480. rssi=-2,
  481. )
  482. result = parse_advertisement_data(ble_device, adv_data)
  483. assert result == SwitchBotAdvertisement(
  484. address="aa:bb:cc:dd:ee:ff",
  485. data={
  486. "rawAdvData": b"c\xc0dd\x12\x04",
  487. "data": {
  488. "calibration": True,
  489. "battery": 100,
  490. "inMotion": False,
  491. "position": 0,
  492. "lightLevel": 1,
  493. "deviceChain": 2,
  494. },
  495. "isEncrypted": False,
  496. "model": "c",
  497. "modelFriendlyName": "Curtain",
  498. "modelName": SwitchbotModel.CURTAIN,
  499. },
  500. device=ble_device,
  501. rssi=-2,
  502. active=True,
  503. )
  504. def test_parse_advertisement_data_curtain_firmware_six_fully_open():
  505. """Test parse_advertisement_data with firmware six fully open."""
  506. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  507. adv_data = generate_advertisement_data(
  508. local_name="WoCurtain",
  509. manufacturer_data={
  510. 89: b"\xcc\xf4\xc4\xf9\xacl",
  511. 2409: b"\xcc\xf4\xc4\xf9\xacl\xe2\x0f\x00\x12\x04",
  512. },
  513. service_data={
  514. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0Yd\x11\x04",
  515. "0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0d\x00\x12\x04",
  516. },
  517. service_uuids=[
  518. "00001800-0000-1000-8000-00805f9b34fb",
  519. "00001801-0000-1000-8000-00805f9b34fb",
  520. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  521. ],
  522. rssi=-2,
  523. )
  524. result = parse_advertisement_data(ble_device, adv_data)
  525. assert result == SwitchBotAdvertisement(
  526. address="aa:bb:cc:dd:ee:ff",
  527. data={
  528. "rawAdvData": b"c\xc0d\x00\x12\x04",
  529. "data": {
  530. "calibration": True,
  531. "battery": 100,
  532. "inMotion": False,
  533. "position": 100,
  534. "lightLevel": 1,
  535. "deviceChain": 2,
  536. },
  537. "isEncrypted": False,
  538. "model": "c",
  539. "modelFriendlyName": "Curtain",
  540. "modelName": SwitchbotModel.CURTAIN,
  541. },
  542. device=ble_device,
  543. rssi=-2,
  544. active=True,
  545. )
  546. def test_contact_sensor_mfr():
  547. """Test parsing adv data from new bot firmware."""
  548. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  549. adv_data = generate_advertisement_data(
  550. manufacturer_data={2409: b"\xcb9\xcd\xc4=FA,\x00F\x01\x8f\xc4"},
  551. service_data={
  552. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d\x00\xda\x04\x00F\x01\x8f\xc4"
  553. },
  554. tx_power=-127,
  555. rssi=-70,
  556. )
  557. result = parse_advertisement_data(ble_device, adv_data)
  558. assert result == SwitchBotAdvertisement(
  559. address="aa:bb:cc:dd:ee:ff",
  560. data={
  561. "data": {
  562. "battery": 90,
  563. "button_count": 4,
  564. "contact_open": True,
  565. "contact_timeout": True,
  566. "is_light": False,
  567. "motion_detected": False,
  568. "tested": False,
  569. },
  570. "isEncrypted": False,
  571. "model": "d",
  572. "modelFriendlyName": "Contact Sensor",
  573. "modelName": SwitchbotModel.CONTACT_SENSOR,
  574. "rawAdvData": b"d\x00\xda\x04\x00F\x01\x8f\xc4",
  575. },
  576. device=ble_device,
  577. rssi=-70,
  578. active=True,
  579. )
  580. def test_contact_sensor_mfr_no_service_data():
  581. """Test contact sensor with passive data only."""
  582. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  583. adv_data = generate_advertisement_data(
  584. manufacturer_data={2409: b"\xcb9\xcd\xc4=FA,\x00F\x01\x8f\xc4"},
  585. service_data={},
  586. tx_power=-127,
  587. rssi=-70,
  588. )
  589. result = parse_advertisement_data(ble_device, adv_data)
  590. # Passive detection of contact sensor is not supported
  591. # anymore since the Switchbot Curtain v3 was released
  592. # which uses the heuristics for the contact sensor.
  593. assert result is None
  594. def test_contact_sensor_srv():
  595. """Test parsing adv data from new bot firmware."""
  596. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  597. adv_data = generate_advertisement_data(
  598. service_data={
  599. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d\x00\xda\x04\x00F\x01\x8f\xc4"
  600. },
  601. tx_power=-127,
  602. rssi=-70,
  603. )
  604. result = parse_advertisement_data(ble_device, adv_data)
  605. assert result == SwitchBotAdvertisement(
  606. address="aa:bb:cc:dd:ee:ff",
  607. data={
  608. "data": {
  609. "battery": 90,
  610. "button_count": 4,
  611. "contact_open": True,
  612. "contact_timeout": True,
  613. "is_light": False,
  614. "motion_detected": False,
  615. "tested": False,
  616. },
  617. "isEncrypted": False,
  618. "model": "d",
  619. "modelFriendlyName": "Contact Sensor",
  620. "modelName": SwitchbotModel.CONTACT_SENSOR,
  621. "rawAdvData": b"d\x00\xda\x04\x00F\x01\x8f\xc4",
  622. },
  623. device=ble_device,
  624. rssi=-70,
  625. active=True,
  626. )
  627. def test_contact_sensor_open():
  628. """Test parsing mfr adv data from new bot firmware."""
  629. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  630. adv_data = generate_advertisement_data(
  631. manufacturer_data={2409: b"\xcb9\xcd\xc4=F\x84\x9c\x00\x17\x00QD"},
  632. service_data={
  633. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@\xda\x02\x00\x17\x00QD"
  634. },
  635. tx_power=-127,
  636. rssi=-59,
  637. )
  638. result = parse_advertisement_data(ble_device, adv_data)
  639. assert result == SwitchBotAdvertisement(
  640. address="aa:bb:cc:dd:ee:ff",
  641. data={
  642. "data": {
  643. "battery": 90,
  644. "button_count": 4,
  645. "contact_open": True,
  646. "contact_timeout": False,
  647. "is_light": False,
  648. "motion_detected": True,
  649. "tested": False,
  650. },
  651. "isEncrypted": False,
  652. "model": "d",
  653. "modelFriendlyName": "Contact Sensor",
  654. "modelName": SwitchbotModel.CONTACT_SENSOR,
  655. "rawAdvData": b"d@\xda\x02\x00\x17\x00QD",
  656. },
  657. device=ble_device,
  658. rssi=-59,
  659. active=True,
  660. )
  661. def test_contact_sensor_closed():
  662. """Test parsing mfr adv data from new bot firmware."""
  663. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  664. adv_data = generate_advertisement_data(
  665. manufacturer_data={2409: b"\xcb9\xcd\xc4=F\x89\x8c\x00+\x00\x19\x84"},
  666. service_data={
  667. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@\xda\x00\x00+\x00\x19\x84"
  668. },
  669. tx_power=-127,
  670. rssi=-50,
  671. )
  672. result = parse_advertisement_data(ble_device, adv_data)
  673. assert result == SwitchBotAdvertisement(
  674. address="aa:bb:cc:dd:ee:ff",
  675. data={
  676. "data": {
  677. "battery": 90,
  678. "button_count": 4,
  679. "contact_open": False,
  680. "contact_timeout": False,
  681. "is_light": False,
  682. "motion_detected": True,
  683. "tested": False,
  684. },
  685. "isEncrypted": False,
  686. "model": "d",
  687. "modelFriendlyName": "Contact Sensor",
  688. "modelName": SwitchbotModel.CONTACT_SENSOR,
  689. "rawAdvData": b"d@\xda\x00\x00+\x00\x19\x84",
  690. },
  691. device=ble_device,
  692. rssi=-50,
  693. active=True,
  694. )
  695. def test_switchbot_passive():
  696. """Test parsing switchbot as passive."""
  697. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  698. adv_data = generate_advertisement_data(
  699. manufacturer_data={89: bytes.fromhex("d51cfb397856")},
  700. service_data={},
  701. tx_power=-127,
  702. rssi=-50,
  703. )
  704. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.BOT)
  705. assert result == SwitchBotAdvertisement(
  706. address="aa:bb:cc:dd:ee:ff",
  707. data={
  708. "data": {
  709. "battery": None,
  710. "switchMode": None,
  711. "isOn": None,
  712. },
  713. "isEncrypted": False,
  714. "model": "H",
  715. "modelFriendlyName": "Bot",
  716. "modelName": SwitchbotModel.BOT,
  717. "rawAdvData": None,
  718. },
  719. device=ble_device,
  720. rssi=-50,
  721. active=False,
  722. )
  723. def test_bulb_active():
  724. """Test parsing bulb as active."""
  725. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  726. adv_data = generate_advertisement_data(
  727. manufacturer_data={2409: b"\x84\xf7\x03\xb4\xcbz\x03\xe4!\x00\x00"},
  728. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"u\x00d"},
  729. tx_power=-127,
  730. rssi=-50,
  731. )
  732. result = parse_advertisement_data(ble_device, adv_data)
  733. assert result == SwitchBotAdvertisement(
  734. address="aa:bb:cc:dd:ee:ff",
  735. data={
  736. "data": {
  737. "brightness": 100,
  738. "color_mode": 1,
  739. "delay": False,
  740. "isOn": True,
  741. "loop_index": 0,
  742. "preset": False,
  743. "sequence_number": 3,
  744. "speed": 0,
  745. },
  746. "isEncrypted": False,
  747. "model": "u",
  748. "modelFriendlyName": "Color Bulb",
  749. "modelName": SwitchbotModel.COLOR_BULB,
  750. "rawAdvData": b"u\x00d",
  751. },
  752. device=ble_device,
  753. rssi=-50,
  754. active=True,
  755. )
  756. def test_wosensor_passive_and_active():
  757. """Test parsing wosensor as passive with active data as well."""
  758. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  759. adv_data = generate_advertisement_data(
  760. manufacturer_data={2409: b"\xd7\xc1}]\xebC\xde\x03\x06\x985"},
  761. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\xe4\x06\x985"},
  762. tx_power=-127,
  763. rssi=-50,
  764. )
  765. result = parse_advertisement_data(ble_device, adv_data)
  766. assert result == SwitchBotAdvertisement(
  767. address="aa:bb:cc:dd:ee:ff",
  768. data={
  769. "data": {
  770. "battery": 100,
  771. "fahrenheit": False,
  772. "humidity": 53,
  773. "temp": {"c": 24.6, "f": 76.28},
  774. "temperature": 24.6,
  775. },
  776. "isEncrypted": False,
  777. "model": "T",
  778. "modelFriendlyName": "Meter",
  779. "modelName": SwitchbotModel.METER,
  780. "rawAdvData": b"T\x00\xe4\x06\x985",
  781. },
  782. device=ble_device,
  783. rssi=-50,
  784. active=True,
  785. )
  786. def test_wosensor_active():
  787. """Test parsing wosensor with active data as well."""
  788. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  789. adv_data = generate_advertisement_data(
  790. manufacturer_data={},
  791. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\xe4\x06\x985"},
  792. tx_power=-127,
  793. rssi=-50,
  794. )
  795. result = parse_advertisement_data(ble_device, adv_data)
  796. assert result == SwitchBotAdvertisement(
  797. address="aa:bb:cc:dd:ee:ff",
  798. data={
  799. "data": {
  800. "battery": 100,
  801. "fahrenheit": False,
  802. "humidity": 53,
  803. "temp": {"c": 24.6, "f": 76.28},
  804. "temperature": 24.6,
  805. },
  806. "isEncrypted": False,
  807. "model": "T",
  808. "modelFriendlyName": "Meter",
  809. "modelName": SwitchbotModel.METER,
  810. "rawAdvData": b"T\x00\xe4\x06\x985",
  811. },
  812. device=ble_device,
  813. rssi=-50,
  814. active=True,
  815. )
  816. def test_wosensor_passive_only():
  817. """Test parsing wosensor with only passive data."""
  818. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  819. adv_data = generate_advertisement_data(
  820. manufacturer_data={2409: b"\xd7\xc1}]\xebC\xde\x03\x06\x985"},
  821. service_data={},
  822. tx_power=-127,
  823. rssi=-50,
  824. )
  825. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER)
  826. assert result == SwitchBotAdvertisement(
  827. address="aa:bb:cc:dd:ee:ff",
  828. data={
  829. "data": {
  830. "battery": None,
  831. "fahrenheit": False,
  832. "humidity": 53,
  833. "temp": {"c": 24.6, "f": 76.28},
  834. "temperature": 24.6,
  835. },
  836. "isEncrypted": False,
  837. "model": "T",
  838. "modelFriendlyName": "Meter",
  839. "modelName": SwitchbotModel.METER,
  840. "rawAdvData": None,
  841. },
  842. device=ble_device,
  843. rssi=-50,
  844. active=False,
  845. )
  846. def test_wosensor_active_zero_data():
  847. """Test parsing wosensor with active data but all values are zero."""
  848. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  849. adv_data = generate_advertisement_data(
  850. manufacturer_data={},
  851. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\x00\x00\x00\x00"},
  852. tx_power=-127,
  853. rssi=-50,
  854. )
  855. result = parse_advertisement_data(ble_device, adv_data)
  856. assert result == SwitchBotAdvertisement(
  857. address="aa:bb:cc:dd:ee:ff",
  858. data={
  859. "data": {},
  860. "isEncrypted": False,
  861. "model": "T",
  862. "rawAdvData": b"T\x00\x00\x00\x00\x00",
  863. },
  864. device=ble_device,
  865. rssi=-50,
  866. active=True,
  867. )
  868. def test_wohub2_passive_and_active():
  869. """Test parsing wosensor as passive with active data as well."""
  870. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  871. adv_data = generate_advertisement_data(
  872. manufacturer_data={
  873. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x00\xfffT\x1a\xf1\x82\x07\x9a2\x00"
  874. },
  875. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"v\x00"},
  876. tx_power=-127,
  877. rssi=-50,
  878. )
  879. result = parse_advertisement_data(ble_device, adv_data)
  880. assert result == SwitchBotAdvertisement(
  881. address="aa:bb:cc:dd:ee:ff",
  882. data={
  883. "data": {
  884. "fahrenheit": False,
  885. "humidity": 50,
  886. "lightLevel": 2,
  887. "illuminance": 10,
  888. "temp": {"c": 26.7, "f": 80.06},
  889. "temperature": 26.7,
  890. },
  891. "isEncrypted": False,
  892. "model": "v",
  893. "modelFriendlyName": "Hub 2",
  894. "modelName": SwitchbotModel.HUB2,
  895. "rawAdvData": b"v\x00",
  896. },
  897. device=ble_device,
  898. rssi=-50,
  899. active=True,
  900. )
  901. def test_woiosensor_passive_and_active():
  902. """Test parsing woiosensor as passive with active data as well."""
  903. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  904. adv_data = generate_advertisement_data(
  905. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee\xff\xe0\x0f\x06\x985\x00"},
  906. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"w\x00\xe4"},
  907. tx_power=-127,
  908. rssi=-50,
  909. )
  910. result = parse_advertisement_data(ble_device, adv_data)
  911. assert result == SwitchBotAdvertisement(
  912. address="aa:bb:cc:dd:ee:ff",
  913. data={
  914. "data": {
  915. "battery": 100,
  916. "fahrenheit": False,
  917. "humidity": 53,
  918. "temp": {"c": 24.6, "f": 76.28},
  919. "temperature": 24.6,
  920. },
  921. "isEncrypted": False,
  922. "model": "w",
  923. "modelFriendlyName": "Indoor/Outdoor Meter",
  924. "modelName": SwitchbotModel.IO_METER,
  925. "rawAdvData": b"w\x00\xe4",
  926. },
  927. device=ble_device,
  928. rssi=-50,
  929. active=True,
  930. )
  931. def test_woiosensor_passive_only():
  932. """Test parsing woiosensor with only passive data."""
  933. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  934. adv_data = generate_advertisement_data(
  935. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee\xff\xe0\x0f\x06\x985\x00"},
  936. service_data={},
  937. tx_power=-127,
  938. rssi=-50,
  939. )
  940. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.IO_METER)
  941. assert result == SwitchBotAdvertisement(
  942. address="aa:bb:cc:dd:ee:ff",
  943. data={
  944. "data": {
  945. "battery": None,
  946. "fahrenheit": False,
  947. "humidity": 53,
  948. "temp": {"c": 24.6, "f": 76.28},
  949. "temperature": 24.6,
  950. },
  951. "isEncrypted": False,
  952. "model": "w",
  953. "modelFriendlyName": "Indoor/Outdoor Meter",
  954. "modelName": SwitchbotModel.IO_METER,
  955. "rawAdvData": None,
  956. },
  957. device=ble_device,
  958. rssi=-50,
  959. active=False,
  960. )
  961. def test_motion_sensor_clear():
  962. """Test parsing motion sensor with clear data."""
  963. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  964. adv_data = generate_advertisement_data(
  965. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIj\x1c\x00f"},
  966. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x00f\x01"},
  967. tx_power=-127,
  968. rssi=-87,
  969. )
  970. result = parse_advertisement_data(
  971. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  972. )
  973. assert result == SwitchBotAdvertisement(
  974. address="aa:bb:cc:dd:ee:ff",
  975. data={
  976. "data": {
  977. "battery": 98,
  978. "iot": 0,
  979. "is_light": False,
  980. "led": 0,
  981. "light_intensity": 1,
  982. "motion_detected": False,
  983. "sense_distance": 0,
  984. "tested": False,
  985. },
  986. "isEncrypted": False,
  987. "model": "s",
  988. "modelFriendlyName": "Motion Sensor",
  989. "modelName": SwitchbotModel.MOTION_SENSOR,
  990. "rawAdvData": b"s\x00\xe2\x00f\x01",
  991. },
  992. device=ble_device,
  993. rssi=-87,
  994. active=True,
  995. )
  996. def test_motion_sensor_clear_passive():
  997. """Test parsing motion sensor with clear data."""
  998. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  999. adv_data = generate_advertisement_data(
  1000. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIj\x1c\x00f"},
  1001. service_data={},
  1002. tx_power=-127,
  1003. rssi=-87,
  1004. )
  1005. result = parse_advertisement_data(
  1006. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1007. )
  1008. assert result == SwitchBotAdvertisement(
  1009. address="aa:bb:cc:dd:ee:ff",
  1010. data={
  1011. "data": {
  1012. "battery": None,
  1013. "iot": None,
  1014. "is_light": False,
  1015. "led": None,
  1016. "light_intensity": None,
  1017. "motion_detected": False,
  1018. "sense_distance": None,
  1019. "tested": None,
  1020. },
  1021. "isEncrypted": False,
  1022. "model": "s",
  1023. "modelFriendlyName": "Motion Sensor",
  1024. "modelName": SwitchbotModel.MOTION_SENSOR,
  1025. "rawAdvData": None,
  1026. },
  1027. device=ble_device,
  1028. rssi=-87,
  1029. active=False,
  1030. )
  1031. def test_motion_sensor_motion():
  1032. """Test parsing motion sensor with motion data."""
  1033. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1034. adv_data = generate_advertisement_data(
  1035. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIi\\\x008"},
  1036. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s@\xe2\x008\x01"},
  1037. tx_power=-127,
  1038. rssi=-87,
  1039. )
  1040. result = parse_advertisement_data(
  1041. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1042. )
  1043. assert result == SwitchBotAdvertisement(
  1044. address="aa:bb:cc:dd:ee:ff",
  1045. data={
  1046. "data": {
  1047. "battery": 98,
  1048. "iot": 0,
  1049. "is_light": False,
  1050. "led": 0,
  1051. "light_intensity": 1,
  1052. "motion_detected": True,
  1053. "sense_distance": 0,
  1054. "tested": False,
  1055. },
  1056. "isEncrypted": False,
  1057. "model": "s",
  1058. "modelFriendlyName": "Motion Sensor",
  1059. "modelName": SwitchbotModel.MOTION_SENSOR,
  1060. "rawAdvData": b"s@\xe2\x008\x01",
  1061. },
  1062. device=ble_device,
  1063. rssi=-87,
  1064. active=True,
  1065. )
  1066. def test_motion_sensor_motion_passive():
  1067. """Test parsing motion sensor with motion data."""
  1068. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1069. adv_data = generate_advertisement_data(
  1070. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIi\\\x008"},
  1071. service_data={},
  1072. tx_power=-127,
  1073. rssi=-87,
  1074. )
  1075. result = parse_advertisement_data(
  1076. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1077. )
  1078. assert result == SwitchBotAdvertisement(
  1079. address="aa:bb:cc:dd:ee:ff",
  1080. data={
  1081. "data": {
  1082. "battery": None,
  1083. "iot": None,
  1084. "is_light": False,
  1085. "led": None,
  1086. "light_intensity": None,
  1087. "motion_detected": True,
  1088. "sense_distance": None,
  1089. "tested": None,
  1090. },
  1091. "isEncrypted": False,
  1092. "model": "s",
  1093. "modelFriendlyName": "Motion Sensor",
  1094. "modelName": SwitchbotModel.MOTION_SENSOR,
  1095. "rawAdvData": None,
  1096. },
  1097. device=ble_device,
  1098. rssi=-87,
  1099. active=False,
  1100. )
  1101. def test_motion_sensor_is_light_passive():
  1102. """Test parsing motion sensor with motion data."""
  1103. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1104. adv_data = generate_advertisement_data(
  1105. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIs,\x04g"},
  1106. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x04g\x02"},
  1107. tx_power=-127,
  1108. rssi=-93,
  1109. )
  1110. result = parse_advertisement_data(
  1111. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1112. )
  1113. assert result == SwitchBotAdvertisement(
  1114. address="aa:bb:cc:dd:ee:ff",
  1115. data={
  1116. "data": {
  1117. "battery": 98,
  1118. "iot": 0,
  1119. "is_light": True,
  1120. "led": 0,
  1121. "light_intensity": 2,
  1122. "motion_detected": False,
  1123. "sense_distance": 0,
  1124. "tested": False,
  1125. },
  1126. "isEncrypted": False,
  1127. "model": "s",
  1128. "modelFriendlyName": "Motion Sensor",
  1129. "modelName": SwitchbotModel.MOTION_SENSOR,
  1130. "rawAdvData": b"s\x00\xe2\x04g\x02",
  1131. },
  1132. device=ble_device,
  1133. rssi=-93,
  1134. active=True,
  1135. )
  1136. def test_motion_sensor_is_light_active():
  1137. """Test parsing motion sensor with motion data."""
  1138. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1139. adv_data = generate_advertisement_data(
  1140. manufacturer_data={},
  1141. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x04g\x02"},
  1142. tx_power=-127,
  1143. rssi=-93,
  1144. )
  1145. result = parse_advertisement_data(
  1146. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1147. )
  1148. assert result == SwitchBotAdvertisement(
  1149. address="aa:bb:cc:dd:ee:ff",
  1150. data={
  1151. "data": {
  1152. "battery": 98,
  1153. "iot": 0,
  1154. "is_light": True,
  1155. "led": 0,
  1156. "light_intensity": 2,
  1157. "motion_detected": False,
  1158. "sense_distance": 0,
  1159. "tested": False,
  1160. },
  1161. "isEncrypted": False,
  1162. "model": "s",
  1163. "modelFriendlyName": "Motion Sensor",
  1164. "modelName": SwitchbotModel.MOTION_SENSOR,
  1165. "rawAdvData": b"s\x00\xe2\x04g\x02",
  1166. },
  1167. device=ble_device,
  1168. rssi=-93,
  1169. active=True,
  1170. )
  1171. def test_motion_with_light_detected():
  1172. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1173. adv_data = generate_advertisement_data(
  1174. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIvl\x00,"},
  1175. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s@\xe2\x00,\x02"},
  1176. tx_power=-127,
  1177. rssi=-84,
  1178. )
  1179. result = parse_advertisement_data(
  1180. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1181. )
  1182. assert result == SwitchBotAdvertisement(
  1183. address="aa:bb:cc:dd:ee:ff",
  1184. data={
  1185. "data": {
  1186. "battery": 98,
  1187. "iot": 0,
  1188. "is_light": True,
  1189. "led": 0,
  1190. "light_intensity": 2,
  1191. "motion_detected": True,
  1192. "sense_distance": 0,
  1193. "tested": False,
  1194. },
  1195. "isEncrypted": False,
  1196. "model": "s",
  1197. "modelFriendlyName": "Motion Sensor",
  1198. "modelName": SwitchbotModel.MOTION_SENSOR,
  1199. "rawAdvData": b"s@\xe2\x00,\x02",
  1200. },
  1201. device=ble_device,
  1202. rssi=-84,
  1203. active=True,
  1204. )
  1205. def test_parsing_lock_active():
  1206. """Test parsing lock with active data."""
  1207. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1208. adv_data = generate_advertisement_data(
  1209. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00 "},
  1210. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"o\x80d"},
  1211. rssi=-67,
  1212. )
  1213. result = parse_advertisement_data(ble_device, adv_data)
  1214. assert result == SwitchBotAdvertisement(
  1215. address="aa:bb:cc:dd:ee:ff",
  1216. data={
  1217. "data": {
  1218. "auto_lock_paused": False,
  1219. "battery": 100,
  1220. "calibration": True,
  1221. "door_open": False,
  1222. "double_lock_mode": False,
  1223. "night_latch": False,
  1224. "status": LockStatus.LOCKED,
  1225. "unclosed_alarm": False,
  1226. "unlocked_alarm": False,
  1227. "update_from_secondary_lock": False,
  1228. },
  1229. "isEncrypted": False,
  1230. "model": "o",
  1231. "modelFriendlyName": "Lock",
  1232. "modelName": SwitchbotModel.LOCK,
  1233. "rawAdvData": b"o\x80d",
  1234. },
  1235. device=ble_device,
  1236. rssi=-67,
  1237. active=True,
  1238. )
  1239. def test_parsing_lock_passive():
  1240. """Test parsing lock with active data."""
  1241. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1242. adv_data = generate_advertisement_data(
  1243. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00 "}, rssi=-67
  1244. )
  1245. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK)
  1246. assert result == SwitchBotAdvertisement(
  1247. address="aa:bb:cc:dd:ee:ff",
  1248. data={
  1249. "data": {
  1250. "auto_lock_paused": False,
  1251. "battery": None,
  1252. "calibration": True,
  1253. "door_open": False,
  1254. "double_lock_mode": False,
  1255. "night_latch": False,
  1256. "status": LockStatus.LOCKED,
  1257. "unclosed_alarm": False,
  1258. "unlocked_alarm": False,
  1259. "update_from_secondary_lock": False,
  1260. },
  1261. "isEncrypted": False,
  1262. "model": "o",
  1263. "modelFriendlyName": "Lock",
  1264. "modelName": SwitchbotModel.LOCK,
  1265. "rawAdvData": None,
  1266. },
  1267. device=ble_device,
  1268. rssi=-67,
  1269. active=False,
  1270. )
  1271. def test_parsing_lock_pro_active():
  1272. """Test parsing lock pro with active data."""
  1273. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1274. adv_data = generate_advertisement_data(
  1275. manufacturer_data={2409: b"\xc8\xf5,\xd9-V\x07\x82\x00d\x00\x00"},
  1276. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"$\x80d"},
  1277. rssi=-80,
  1278. )
  1279. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK_PRO)
  1280. assert result == SwitchBotAdvertisement(
  1281. address="aa:bb:cc:dd:ee:ff",
  1282. data={
  1283. "data": {
  1284. "battery": 100,
  1285. "calibration": True,
  1286. "status": LockStatus.LOCKED,
  1287. "update_from_secondary_lock": False,
  1288. "door_open": False,
  1289. "double_lock_mode": False,
  1290. "unclosed_alarm": False,
  1291. "unlocked_alarm": False,
  1292. "auto_lock_paused": False,
  1293. "night_latch": False,
  1294. },
  1295. "model": "$",
  1296. "isEncrypted": False,
  1297. "modelFriendlyName": "Lock Pro",
  1298. "modelName": SwitchbotModel.LOCK_PRO,
  1299. "rawAdvData": b"$\x80d",
  1300. },
  1301. device=ble_device,
  1302. rssi=-80,
  1303. active=True,
  1304. )
  1305. def test_parsing_lock_pro_passive():
  1306. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1307. adv_data = generate_advertisement_data(
  1308. manufacturer_data={2409: bytes.fromhex("aabbccddeeff208200640000")}, rssi=-67
  1309. )
  1310. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK_PRO)
  1311. assert result == SwitchBotAdvertisement(
  1312. address="aa:bb:cc:dd:ee:ff",
  1313. data={
  1314. "data": {
  1315. "battery": None,
  1316. "calibration": True,
  1317. "status": LockStatus.LOCKED,
  1318. "update_from_secondary_lock": False,
  1319. "door_open": False,
  1320. "double_lock_mode": False,
  1321. "unclosed_alarm": False,
  1322. "unlocked_alarm": False,
  1323. "auto_lock_paused": False,
  1324. "night_latch": False,
  1325. },
  1326. "model": "$",
  1327. "isEncrypted": False,
  1328. "modelFriendlyName": "Lock Pro",
  1329. "modelName": SwitchbotModel.LOCK_PRO,
  1330. "rawAdvData": None,
  1331. },
  1332. device=ble_device,
  1333. rssi=-67,
  1334. active=False,
  1335. )
  1336. def test_parsing_lock_pro_passive_nightlatch_disabled():
  1337. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1338. adv_data = generate_advertisement_data(
  1339. manufacturer_data={2409: bytes.fromhex("aabbccddeeff0a8200630000")}, rssi=-67
  1340. )
  1341. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK_PRO)
  1342. assert result == SwitchBotAdvertisement(
  1343. address="aa:bb:cc:dd:ee:ff",
  1344. data={
  1345. "data": {
  1346. "battery": None,
  1347. "calibration": True,
  1348. "status": LockStatus.LOCKED,
  1349. "update_from_secondary_lock": False,
  1350. "door_open": False,
  1351. "double_lock_mode": False,
  1352. "unclosed_alarm": False,
  1353. "unlocked_alarm": False,
  1354. "auto_lock_paused": False,
  1355. "night_latch": False,
  1356. },
  1357. "model": "$",
  1358. "isEncrypted": False,
  1359. "modelFriendlyName": "Lock Pro",
  1360. "modelName": SwitchbotModel.LOCK_PRO,
  1361. "rawAdvData": None,
  1362. },
  1363. device=ble_device,
  1364. rssi=-67,
  1365. active=False,
  1366. )
  1367. def test_parsing_lock_active_old_firmware():
  1368. """Test parsing lock with active data. Old firmware."""
  1369. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1370. adv_data = generate_advertisement_data(
  1371. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00"},
  1372. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"o\x80d"},
  1373. rssi=-67,
  1374. )
  1375. result = parse_advertisement_data(ble_device, adv_data)
  1376. assert result == SwitchBotAdvertisement(
  1377. address="aa:bb:cc:dd:ee:ff",
  1378. data={
  1379. "data": {
  1380. "auto_lock_paused": False,
  1381. "battery": 100,
  1382. "calibration": True,
  1383. "door_open": False,
  1384. "double_lock_mode": False,
  1385. "night_latch": False,
  1386. "status": LockStatus.LOCKED,
  1387. "unclosed_alarm": False,
  1388. "unlocked_alarm": False,
  1389. "update_from_secondary_lock": False,
  1390. },
  1391. "isEncrypted": False,
  1392. "model": "o",
  1393. "modelFriendlyName": "Lock",
  1394. "modelName": SwitchbotModel.LOCK,
  1395. "rawAdvData": b"o\x80d",
  1396. },
  1397. device=ble_device,
  1398. rssi=-67,
  1399. active=True,
  1400. )
  1401. def test_parsing_lock_passive_old_firmware():
  1402. """Test parsing lock with active data. Old firmware."""
  1403. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1404. adv_data = generate_advertisement_data(
  1405. manufacturer_data={2409: b"\xf1\t\x9fE\x1a]\x07\x83\x00"}, rssi=-67
  1406. )
  1407. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LOCK)
  1408. assert result == SwitchBotAdvertisement(
  1409. address="aa:bb:cc:dd:ee:ff",
  1410. data={
  1411. "data": {
  1412. "auto_lock_paused": False,
  1413. "battery": None,
  1414. "calibration": True,
  1415. "door_open": False,
  1416. "double_lock_mode": False,
  1417. "night_latch": False,
  1418. "status": LockStatus.LOCKED,
  1419. "unclosed_alarm": False,
  1420. "unlocked_alarm": False,
  1421. "update_from_secondary_lock": False,
  1422. },
  1423. "isEncrypted": False,
  1424. "model": "o",
  1425. "modelFriendlyName": "Lock",
  1426. "modelName": SwitchbotModel.LOCK,
  1427. "rawAdvData": None,
  1428. },
  1429. device=ble_device,
  1430. rssi=-67,
  1431. active=False,
  1432. )
  1433. def test_meter_pro_active() -> None:
  1434. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1435. adv_data = generate_advertisement_data(
  1436. manufacturer_data={2409: b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05"},
  1437. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"4\x00d"},
  1438. rssi=-67,
  1439. )
  1440. result = parse_advertisement_data(ble_device, adv_data)
  1441. assert result == SwitchBotAdvertisement(
  1442. address="aa:bb:cc:dd:ee:ff",
  1443. data={
  1444. "data": {
  1445. "battery": 100,
  1446. "fahrenheit": False,
  1447. "humidity": 44,
  1448. "temp": {"c": 23.8, "f": 74.84},
  1449. "temperature": 23.8,
  1450. },
  1451. "isEncrypted": False,
  1452. "model": "4",
  1453. "modelFriendlyName": "Meter",
  1454. "modelName": SwitchbotModel.METER_PRO,
  1455. "rawAdvData": b"4\x00d",
  1456. },
  1457. device=ble_device,
  1458. rssi=-67,
  1459. active=True,
  1460. )
  1461. def test_meter_pro_passive() -> None:
  1462. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1463. adv_data = generate_advertisement_data(
  1464. manufacturer_data={2409: b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05"},
  1465. rssi=-67,
  1466. )
  1467. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER_PRO)
  1468. assert result == SwitchBotAdvertisement(
  1469. address="aa:bb:cc:dd:ee:ff",
  1470. data={
  1471. "data": {
  1472. "battery": None,
  1473. "fahrenheit": False,
  1474. "humidity": 44,
  1475. "temp": {"c": 23.8, "f": 74.84},
  1476. "temperature": 23.8,
  1477. },
  1478. "isEncrypted": False,
  1479. "model": "4",
  1480. "modelFriendlyName": "Meter",
  1481. "modelName": SwitchbotModel.METER_PRO,
  1482. "rawAdvData": None,
  1483. },
  1484. device=ble_device,
  1485. rssi=-67,
  1486. active=False,
  1487. )
  1488. def test_meter_pro_c_active() -> None:
  1489. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1490. adv_data = generate_advertisement_data(
  1491. manufacturer_data={
  1492. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x02\xd5\x00"
  1493. },
  1494. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"5\x00d"},
  1495. rssi=-67,
  1496. )
  1497. result = parse_advertisement_data(ble_device, adv_data)
  1498. assert result == SwitchBotAdvertisement(
  1499. address="aa:bb:cc:dd:ee:ff",
  1500. data={
  1501. "data": {
  1502. "battery": 100,
  1503. "fahrenheit": True,
  1504. "humidity": 36,
  1505. "temp": {"c": 27.7, "f": 81.86},
  1506. "temperature": 27.7,
  1507. "co2": 725,
  1508. },
  1509. "isEncrypted": False,
  1510. "model": "5",
  1511. "modelFriendlyName": "Meter",
  1512. "modelName": SwitchbotModel.METER_PRO_C,
  1513. "rawAdvData": b"5\x00d",
  1514. },
  1515. device=ble_device,
  1516. rssi=-67,
  1517. active=True,
  1518. )
  1519. def test_meter_pro_c_passive() -> None:
  1520. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1521. adv_data = generate_advertisement_data(
  1522. manufacturer_data={
  1523. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x02\xd5\x00"
  1524. },
  1525. rssi=-67,
  1526. )
  1527. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER_PRO_C)
  1528. assert result == SwitchBotAdvertisement(
  1529. address="aa:bb:cc:dd:ee:ff",
  1530. data={
  1531. "data": {
  1532. "battery": None,
  1533. "fahrenheit": True,
  1534. "humidity": 36,
  1535. "temp": {"c": 27.7, "f": 81.86},
  1536. "temperature": 27.7,
  1537. "co2": 725,
  1538. },
  1539. "isEncrypted": False,
  1540. "model": "5",
  1541. "modelFriendlyName": "Meter",
  1542. "modelName": SwitchbotModel.METER_PRO_C,
  1543. "rawAdvData": None,
  1544. },
  1545. device=ble_device,
  1546. rssi=-67,
  1547. active=False,
  1548. )
  1549. def test_parse_advertisement_data_keypad():
  1550. """Test parse_advertisement_data for the keypad."""
  1551. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1552. adv_data = generate_advertisement_data(
  1553. manufacturer_data={2409: b"\xeb\x13\x02\xe6#\x0f\x8fd\x00\x00\x00\x00"},
  1554. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"y\x00d"},
  1555. rssi=-67,
  1556. )
  1557. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.KEYPAD)
  1558. assert result == SwitchBotAdvertisement(
  1559. address="aa:bb:cc:dd:ee:ff",
  1560. data={
  1561. "data": {"attempt_state": 143, "battery": 100},
  1562. "isEncrypted": False,
  1563. "model": "y",
  1564. "modelFriendlyName": "Keypad",
  1565. "modelName": SwitchbotModel.KEYPAD,
  1566. "rawAdvData": b"y\x00d",
  1567. },
  1568. device=ble_device,
  1569. rssi=-67,
  1570. active=True,
  1571. )
  1572. def test_parse_advertisement_data_relay_switch_1pm():
  1573. """Test parse_advertisement_data for the keypad."""
  1574. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1575. adv_data = generate_advertisement_data(
  1576. manufacturer_data={2409: b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00"},
  1577. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"<\x00\x00\x00"},
  1578. rssi=-67,
  1579. )
  1580. result = parse_advertisement_data(
  1581. ble_device, adv_data, SwitchbotModel.RELAY_SWITCH_1PM
  1582. )
  1583. assert result == SwitchBotAdvertisement(
  1584. address="aa:bb:cc:dd:ee:ff",
  1585. data={
  1586. "data": {
  1587. "switchMode": True,
  1588. "sequence_number": 71,
  1589. "isOn": True,
  1590. "power": 4.9,
  1591. "voltage": 0,
  1592. "current": 0,
  1593. },
  1594. "isEncrypted": False,
  1595. "model": "<",
  1596. "modelFriendlyName": "Relay Switch 1PM",
  1597. "modelName": SwitchbotModel.RELAY_SWITCH_1PM,
  1598. "rawAdvData": b"<\x00\x00\x00",
  1599. },
  1600. device=ble_device,
  1601. rssi=-67,
  1602. active=True,
  1603. )
  1604. def test_parse_advertisement_data_relay_switch_1():
  1605. """Test parse_advertisement_data for the keypad."""
  1606. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1607. adv_data = generate_advertisement_data(
  1608. manufacturer_data={2409: b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00"},
  1609. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b";\x00\x00\x00"},
  1610. rssi=-67,
  1611. )
  1612. result = parse_advertisement_data(
  1613. ble_device, adv_data, SwitchbotModel.RELAY_SWITCH_1
  1614. )
  1615. assert result == SwitchBotAdvertisement(
  1616. address="aa:bb:cc:dd:ee:ff",
  1617. data={
  1618. "data": {
  1619. "switchMode": True,
  1620. "sequence_number": 71,
  1621. "isOn": True,
  1622. },
  1623. "isEncrypted": False,
  1624. "model": ";",
  1625. "modelFriendlyName": "Relay Switch 1",
  1626. "modelName": SwitchbotModel.RELAY_SWITCH_1,
  1627. "rawAdvData": b";\x00\x00\x00",
  1628. },
  1629. device=ble_device,
  1630. rssi=-67,
  1631. active=True,
  1632. )
  1633. def test_leak_active():
  1634. """Test parse_advertisement_data for the leak detector."""
  1635. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1636. adv_data = generate_advertisement_data(
  1637. manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
  1638. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00N"},
  1639. rssi=-72,
  1640. )
  1641. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1642. assert result == SwitchBotAdvertisement(
  1643. address="aa:bb:cc:dd:ee:ff",
  1644. data={
  1645. "data": {
  1646. "leak": False,
  1647. "tampered": False,
  1648. "battery": 78,
  1649. "low_battery": False,
  1650. },
  1651. "isEncrypted": False,
  1652. "model": "&",
  1653. "modelFriendlyName": "Leak Detector",
  1654. "modelName": SwitchbotModel.LEAK,
  1655. "rawAdvData": b"&\x00N",
  1656. },
  1657. device=ble_device,
  1658. rssi=-72,
  1659. active=True,
  1660. )
  1661. def test_leak_passive():
  1662. """Test parse_advertisement_data for the leak detector."""
  1663. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1664. adv_data = generate_advertisement_data(
  1665. manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
  1666. rssi=-72,
  1667. )
  1668. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1669. assert result == SwitchBotAdvertisement(
  1670. address="aa:bb:cc:dd:ee:ff",
  1671. data={
  1672. "data": {},
  1673. "isEncrypted": False,
  1674. "model": "&",
  1675. "rawAdvData": None,
  1676. },
  1677. device=ble_device,
  1678. rssi=-72,
  1679. active=False,
  1680. )
  1681. def test_leak_no_leak_detected():
  1682. """Test parse_advertisement_data for the leak detector."""
  1683. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1684. adv_data = generate_advertisement_data(
  1685. manufacturer_data={
  1686. 2409: b"\xc4407LzJd\x98ga\xc4\n<\x05\x00\x00\x00\x00"
  1687. }, # no leak, batt
  1688. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1689. rssi=-73,
  1690. )
  1691. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1692. assert result == SwitchBotAdvertisement(
  1693. address="aa:bb:cc:dd:ee:ff",
  1694. data={
  1695. "data": {
  1696. "leak": False,
  1697. "tampered": False,
  1698. "battery": 100,
  1699. "low_battery": False,
  1700. },
  1701. "isEncrypted": False,
  1702. "model": "&",
  1703. "modelFriendlyName": "Leak Detector",
  1704. "modelName": SwitchbotModel.LEAK,
  1705. "rawAdvData": b"&\x00d",
  1706. },
  1707. device=ble_device,
  1708. rssi=-73,
  1709. active=True,
  1710. )
  1711. def test_leak_leak_detected():
  1712. """Test parse_advertisement_data for the leak detector."""
  1713. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1714. adv_data = generate_advertisement_data(
  1715. manufacturer_data={
  1716. 2409: b"\xc4407LzGd\xf9ga\xc4\x08<\x05\x00\x00\x00\x00"
  1717. }, # leak, batt
  1718. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1719. rssi=-73,
  1720. )
  1721. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1722. assert result == SwitchBotAdvertisement(
  1723. address="aa:bb:cc:dd:ee:ff",
  1724. data={
  1725. "data": {
  1726. "leak": True,
  1727. "tampered": False,
  1728. "battery": 100,
  1729. "low_battery": False,
  1730. },
  1731. "isEncrypted": False,
  1732. "model": "&",
  1733. "modelFriendlyName": "Leak Detector",
  1734. "modelName": SwitchbotModel.LEAK,
  1735. "rawAdvData": b"&\x00d",
  1736. },
  1737. device=ble_device,
  1738. rssi=-73,
  1739. active=True,
  1740. )
  1741. def test_leak_low_battery():
  1742. """Test parse_advertisement_data for the leak detector."""
  1743. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1744. adv_data = generate_advertisement_data(
  1745. manufacturer_data={
  1746. 2409: b"\xc4407Lz\x02\t\x98\x00\x00\x00\x00<\x05\x00\x00\x00\x00"
  1747. }, # no leak, low battery
  1748. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1749. rssi=-73,
  1750. )
  1751. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1752. assert result == SwitchBotAdvertisement(
  1753. address="aa:bb:cc:dd:ee:ff",
  1754. data={
  1755. "data": {
  1756. "leak": False,
  1757. "tampered": False,
  1758. "battery": 9,
  1759. "low_battery": False,
  1760. },
  1761. "isEncrypted": False,
  1762. "model": "&",
  1763. "modelFriendlyName": "Leak Detector",
  1764. "modelName": SwitchbotModel.LEAK,
  1765. "rawAdvData": b"&\x00d",
  1766. },
  1767. device=ble_device,
  1768. rssi=-73,
  1769. active=True,
  1770. )
  1771. def test_leak_real_data_from_ha():
  1772. """Test parse_advertisement_data for the leak detector."""
  1773. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1774. adv_data = generate_advertisement_data(
  1775. manufacturer_data={
  1776. 2409: b"\\xd6407D1\\x02V\\x90\\x00\\x00\\x00\\x00\\x1e\\x05\\x00\\x00\\x00\\x00"
  1777. }, # no leak, low battery
  1778. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\\x00V"},
  1779. rssi=-73,
  1780. )
  1781. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1782. assert result == SwitchBotAdvertisement(
  1783. address="aa:bb:cc:dd:ee:ff",
  1784. data={
  1785. "data": {
  1786. "leak": True,
  1787. "tampered": False,
  1788. "battery": 68,
  1789. "low_battery": False,
  1790. },
  1791. "isEncrypted": False,
  1792. "model": "&",
  1793. "modelFriendlyName": "Leak Detector",
  1794. "modelName": SwitchbotModel.LEAK,
  1795. "rawAdvData": b"&\\x00V",
  1796. },
  1797. device=ble_device,
  1798. rssi=-73,
  1799. active=True,
  1800. )
  1801. def test_remote_active() -> None:
  1802. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1803. adv_data = generate_advertisement_data(
  1804. manufacturer_data={89: b"\xaa\xbb\xcc\xdd\xee\xff"},
  1805. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"b V\x00"},
  1806. service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
  1807. rssi=-95,
  1808. )
  1809. result = parse_advertisement_data(ble_device, adv_data)
  1810. assert result == SwitchBotAdvertisement(
  1811. address="aa:bb:cc:dd:ee:ff",
  1812. data={
  1813. "data": {
  1814. "battery": 86,
  1815. },
  1816. "isEncrypted": False,
  1817. "model": "b",
  1818. "modelFriendlyName": "Remote",
  1819. "modelName": SwitchbotModel.REMOTE,
  1820. "rawAdvData": b"b V\x00",
  1821. },
  1822. device=ble_device,
  1823. rssi=-95,
  1824. active=True,
  1825. )
  1826. def test_remote_passive() -> None:
  1827. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1828. adv_data = generate_advertisement_data(
  1829. manufacturer_data={89: b"\xaa\xbb\xcc\xdd\xee\xff"},
  1830. rssi=-97,
  1831. )
  1832. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.REMOTE)
  1833. assert result == SwitchBotAdvertisement(
  1834. address="aa:bb:cc:dd:ee:ff",
  1835. data={
  1836. "data": {
  1837. "battery": None,
  1838. },
  1839. "isEncrypted": False,
  1840. "model": "b",
  1841. "modelFriendlyName": "Remote",
  1842. "modelName": SwitchbotModel.REMOTE,
  1843. "rawAdvData": None,
  1844. },
  1845. device=ble_device,
  1846. rssi=-97,
  1847. active=False,
  1848. )
  1849. def test_parse_advertisement_data_hubmini_matter():
  1850. """Test parse_advertisement_data for the HubMini Matter."""
  1851. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1852. adv_data = generate_advertisement_data(
  1853. manufacturer_data={
  1854. 2409: b"\xe6\xa1\xcd\x1f[e\x00\x00\x00\x00\x00\x00\x14\x01\x985\x00"
  1855. },
  1856. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"%\x00"},
  1857. rssi=-67,
  1858. )
  1859. result = parse_advertisement_data(
  1860. ble_device, adv_data, SwitchbotModel.HUBMINI_MATTER
  1861. )
  1862. assert result == SwitchBotAdvertisement(
  1863. address="aa:bb:cc:dd:ee:ff",
  1864. data={
  1865. "data": {
  1866. "fahrenheit": False,
  1867. "humidity": 53,
  1868. "temp": {"c": 24.1, "f": 75.38},
  1869. "temperature": 24.1,
  1870. },
  1871. "isEncrypted": False,
  1872. "model": "%",
  1873. "modelFriendlyName": "HubMini Matter",
  1874. "modelName": SwitchbotModel.HUBMINI_MATTER,
  1875. "rawAdvData": b"%\x00",
  1876. },
  1877. device=ble_device,
  1878. rssi=-67,
  1879. active=True,
  1880. )
  1881. def test_parse_advertisement_data_roller_shade():
  1882. """Test parse_advertisement_data for roller shade."""
  1883. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1884. adv_data = generate_advertisement_data(
  1885. manufacturer_data={2409: b"\xb0\xe9\xfeT\x90\x1b,\x08\x9f\x11\x04'\x00"},
  1886. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b",\x00'\x9f\x11\x04"},
  1887. rssi=-80,
  1888. )
  1889. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.ROLLER_SHADE)
  1890. assert result == SwitchBotAdvertisement(
  1891. address="aa:bb:cc:dd:ee:ff",
  1892. data={
  1893. "rawAdvData": b",\x00'\x9f\x11\x04",
  1894. "data": {
  1895. "battery": 39,
  1896. "calibration": True,
  1897. "deviceChain": 1,
  1898. "inMotion": False,
  1899. "lightLevel": 1,
  1900. "position": 69,
  1901. "sequence_number": 44,
  1902. },
  1903. "isEncrypted": False,
  1904. "model": ",",
  1905. "modelFriendlyName": "Roller Shade",
  1906. "modelName": SwitchbotModel.ROLLER_SHADE,
  1907. },
  1908. device=ble_device,
  1909. rssi=-80,
  1910. active=True,
  1911. )
  1912. def test_hubmini_matter_passive() -> None:
  1913. """Test parsing hubmini matter with passive data."""
  1914. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1915. adv_data = generate_advertisement_data(
  1916. manufacturer_data={
  1917. 2409: b"\xe6\xa1\xcd\x1f[e\x00\x00\x00\x00\x00\x00\x14\x01\x985\x00"
  1918. },
  1919. rssi=-97,
  1920. )
  1921. result = parse_advertisement_data(
  1922. ble_device, adv_data, SwitchbotModel.HUBMINI_MATTER
  1923. )
  1924. assert result == SwitchBotAdvertisement(
  1925. address="aa:bb:cc:dd:ee:ff",
  1926. data={
  1927. "data": {
  1928. "fahrenheit": False,
  1929. "humidity": 53,
  1930. "temp": {"c": 24.1, "f": 75.38},
  1931. "temperature": 24.1,
  1932. },
  1933. "isEncrypted": False,
  1934. "model": "%",
  1935. "modelFriendlyName": "HubMini Matter",
  1936. "modelName": SwitchbotModel.HUBMINI_MATTER,
  1937. "rawAdvData": None,
  1938. },
  1939. device=ble_device,
  1940. rssi=-97,
  1941. active=False,
  1942. )
  1943. def test_roller_shade_passive() -> None:
  1944. """Test parsing roller_shade with passive data."""
  1945. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1946. adv_data = generate_advertisement_data(
  1947. manufacturer_data={2409: b"\xb0\xe9\xfeT\x90\x1b,\x08\x9f\x11\x04'\x00"},
  1948. rssi=-97,
  1949. )
  1950. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.ROLLER_SHADE)
  1951. assert result == SwitchBotAdvertisement(
  1952. address="aa:bb:cc:dd:ee:ff",
  1953. data={
  1954. "rawAdvData": None,
  1955. "data": {
  1956. "battery": None,
  1957. "calibration": True,
  1958. "deviceChain": 1,
  1959. "inMotion": False,
  1960. "lightLevel": 1,
  1961. "position": 69,
  1962. "sequence_number": 44,
  1963. },
  1964. "isEncrypted": False,
  1965. "model": ",",
  1966. "modelFriendlyName": "Roller Shade",
  1967. "modelName": SwitchbotModel.ROLLER_SHADE,
  1968. },
  1969. device=ble_device,
  1970. rssi=-97,
  1971. active=False,
  1972. )
  1973. def test_circulator_fan_active() -> None:
  1974. """Test parsing circulator fan with active data."""
  1975. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1976. adv_data = generate_advertisement_data(
  1977. manufacturer_data={2409: b"\xb0\xe9\xfeXY\xa8~LR9"},
  1978. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"~\x00R"},
  1979. rssi=-97,
  1980. )
  1981. result = parse_advertisement_data(
  1982. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  1983. )
  1984. assert result == SwitchBotAdvertisement(
  1985. address="aa:bb:cc:dd:ee:ff",
  1986. data={
  1987. "rawAdvData": b"~\x00R",
  1988. "data": {
  1989. "sequence_number": 126,
  1990. "isOn": False,
  1991. "mode": "baby",
  1992. "nightLight": 3,
  1993. "oscillating": False,
  1994. "battery": 82,
  1995. "speed": 57,
  1996. },
  1997. "isEncrypted": False,
  1998. "model": "~",
  1999. "modelFriendlyName": "Circulator Fan",
  2000. "modelName": SwitchbotModel.CIRCULATOR_FAN,
  2001. },
  2002. device=ble_device,
  2003. rssi=-97,
  2004. active=True,
  2005. )
  2006. def test_circulator_fan_passive() -> None:
  2007. """Test parsing circulator fan with passive data."""
  2008. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2009. adv_data = generate_advertisement_data(
  2010. manufacturer_data={2409: b"\xb0\xe9\xfeXY\xa8~LR9"},
  2011. rssi=-97,
  2012. )
  2013. result = parse_advertisement_data(
  2014. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  2015. )
  2016. assert result == SwitchBotAdvertisement(
  2017. address="aa:bb:cc:dd:ee:ff",
  2018. data={
  2019. "rawAdvData": None,
  2020. "data": {
  2021. "sequence_number": 126,
  2022. "isOn": False,
  2023. "mode": "baby",
  2024. "nightLight": 3,
  2025. "oscillating": False,
  2026. "battery": 82,
  2027. "speed": 57,
  2028. },
  2029. "isEncrypted": False,
  2030. "model": "~",
  2031. "modelFriendlyName": "Circulator Fan",
  2032. "modelName": SwitchbotModel.CIRCULATOR_FAN,
  2033. },
  2034. device=ble_device,
  2035. rssi=-97,
  2036. active=False,
  2037. )
  2038. def test_circulator_fan_with_empty_data() -> None:
  2039. """Test parsing circulator fan with empty data."""
  2040. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2041. adv_data = generate_advertisement_data(
  2042. manufacturer_data={2409: None},
  2043. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"~\x00R"},
  2044. rssi=-97,
  2045. )
  2046. result = parse_advertisement_data(
  2047. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  2048. )
  2049. assert result == SwitchBotAdvertisement(
  2050. address="aa:bb:cc:dd:ee:ff",
  2051. data={
  2052. "rawAdvData": b"~\x00R",
  2053. "data": {},
  2054. "isEncrypted": False,
  2055. "model": "~",
  2056. },
  2057. device=ble_device,
  2058. rssi=-97,
  2059. active=True,
  2060. )
  2061. def test_k20_active() -> None:
  2062. """Test parsing k20 with active data."""
  2063. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2064. adv_data = generate_advertisement_data(
  2065. manufacturer_data={2409: b"\xb0\xe9\xfe\x01\xf3\x8f'\x01\x11S\x00\x10d\x0f"},
  2066. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b".\x00d"},
  2067. rssi=-97,
  2068. )
  2069. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K20_VACUUM)
  2070. assert result == SwitchBotAdvertisement(
  2071. address="aa:bb:cc:dd:ee:ff",
  2072. data={
  2073. "rawAdvData": b".\x00d",
  2074. "data": {
  2075. "sequence_number": 39,
  2076. "soc_version": "1.1.083",
  2077. "step": 0,
  2078. "mqtt_connected": True,
  2079. "battery": 100,
  2080. "work_status": 15,
  2081. },
  2082. "isEncrypted": False,
  2083. "model": ".",
  2084. "modelFriendlyName": "K20 Vacuum",
  2085. "modelName": SwitchbotModel.K20_VACUUM,
  2086. },
  2087. device=ble_device,
  2088. rssi=-97,
  2089. active=True,
  2090. )
  2091. def test_k20_passive() -> None:
  2092. """Test parsing k20 with passive data."""
  2093. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2094. adv_data = generate_advertisement_data(
  2095. manufacturer_data={2409: b"\xb0\xe9\xfe\x01\xf3\x8f'\x01\x11S\x00\x10d\x0f"},
  2096. rssi=-97,
  2097. )
  2098. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K20_VACUUM)
  2099. assert result == SwitchBotAdvertisement(
  2100. address="aa:bb:cc:dd:ee:ff",
  2101. data={
  2102. "rawAdvData": None,
  2103. "data": {
  2104. "sequence_number": 39,
  2105. "soc_version": "1.1.083",
  2106. "step": 0,
  2107. "mqtt_connected": True,
  2108. "battery": 100,
  2109. "work_status": 15,
  2110. },
  2111. "isEncrypted": False,
  2112. "model": ".",
  2113. "modelFriendlyName": "K20 Vacuum",
  2114. "modelName": SwitchbotModel.K20_VACUUM,
  2115. },
  2116. device=ble_device,
  2117. rssi=-97,
  2118. active=False,
  2119. )
  2120. def test_k20_with_empty_data() -> None:
  2121. """Test parsing k20 with empty data."""
  2122. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2123. adv_data = generate_advertisement_data(
  2124. manufacturer_data={2409: None},
  2125. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b".\x00d"},
  2126. rssi=-97,
  2127. )
  2128. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K20_VACUUM)
  2129. assert result == SwitchBotAdvertisement(
  2130. address="aa:bb:cc:dd:ee:ff",
  2131. data={
  2132. "rawAdvData": b".\x00d",
  2133. "data": {},
  2134. "isEncrypted": False,
  2135. "model": ".",
  2136. },
  2137. device=ble_device,
  2138. rssi=-97,
  2139. active=True,
  2140. )
  2141. def test_k10_pro_active() -> None:
  2142. """Test parsing k10 pro with active data."""
  2143. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2144. adv_data = generate_advertisement_data(
  2145. manufacturer_data={2409: b"\xb0\xe9\xfeP\x8d\x8d\x02 d"},
  2146. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"(\x00"},
  2147. rssi=-97,
  2148. )
  2149. result = parse_advertisement_data(
  2150. ble_device, adv_data, SwitchbotModel.K10_PRO_VACUUM
  2151. )
  2152. assert result == SwitchBotAdvertisement(
  2153. address="aa:bb:cc:dd:ee:ff",
  2154. data={
  2155. "rawAdvData": b"(\x00",
  2156. "data": {
  2157. "sequence_number": 2,
  2158. "dusbin_connected": False,
  2159. "dustbin_bound": False,
  2160. "network_connected": True,
  2161. "battery": 100,
  2162. "work_status": 0,
  2163. },
  2164. "isEncrypted": False,
  2165. "model": "(",
  2166. "modelFriendlyName": "K10+ Pro Vacuum",
  2167. "modelName": SwitchbotModel.K10_PRO_VACUUM,
  2168. },
  2169. device=ble_device,
  2170. rssi=-97,
  2171. active=True,
  2172. )
  2173. def test_k10_pro_passive() -> None:
  2174. """Test parsing k10 pro with passive data."""
  2175. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2176. adv_data = generate_advertisement_data(
  2177. manufacturer_data={2409: b"\xb0\xe9\xfeP\x8d\x8d\x02 d"},
  2178. rssi=-97,
  2179. )
  2180. result = parse_advertisement_data(
  2181. ble_device, adv_data, SwitchbotModel.K10_PRO_VACUUM
  2182. )
  2183. assert result == SwitchBotAdvertisement(
  2184. address="aa:bb:cc:dd:ee:ff",
  2185. data={
  2186. "rawAdvData": None,
  2187. "data": {
  2188. "sequence_number": 2,
  2189. "dusbin_connected": False,
  2190. "dustbin_bound": False,
  2191. "network_connected": True,
  2192. "battery": 100,
  2193. "work_status": 0,
  2194. },
  2195. "isEncrypted": False,
  2196. "model": "(",
  2197. "modelFriendlyName": "K10+ Pro Vacuum",
  2198. "modelName": SwitchbotModel.K10_PRO_VACUUM,
  2199. },
  2200. device=ble_device,
  2201. rssi=-97,
  2202. active=False,
  2203. )
  2204. def test_k10_pro_with_empty_data() -> None:
  2205. """Test parsing k10 pro with empty data."""
  2206. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2207. adv_data = generate_advertisement_data(
  2208. manufacturer_data={2409: None},
  2209. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"(\x00"},
  2210. rssi=-97,
  2211. )
  2212. result = parse_advertisement_data(
  2213. ble_device, adv_data, SwitchbotModel.K10_PRO_VACUUM
  2214. )
  2215. assert result == SwitchBotAdvertisement(
  2216. address="aa:bb:cc:dd:ee:ff",
  2217. data={
  2218. "rawAdvData": b"(\x00",
  2219. "data": {},
  2220. "isEncrypted": False,
  2221. "model": "(",
  2222. },
  2223. device=ble_device,
  2224. rssi=-97,
  2225. active=True,
  2226. )
  2227. def test_k10_active() -> None:
  2228. """Test parsing k10+ with active data."""
  2229. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2230. adv_data = generate_advertisement_data(
  2231. manufacturer_data={2409: b"\xca8\x06\xa9_\xf1\x02 d"},
  2232. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"}\x00"},
  2233. rssi=-97,
  2234. )
  2235. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K10_VACUUM)
  2236. assert result == SwitchBotAdvertisement(
  2237. address="aa:bb:cc:dd:ee:ff",
  2238. data={
  2239. "rawAdvData": b"}\x00",
  2240. "data": {
  2241. "sequence_number": 2,
  2242. "dusbin_connected": False,
  2243. "dustbin_bound": False,
  2244. "network_connected": True,
  2245. "battery": 100,
  2246. "work_status": 0,
  2247. },
  2248. "isEncrypted": False,
  2249. "model": "}",
  2250. "modelFriendlyName": "K10+ Vacuum",
  2251. "modelName": SwitchbotModel.K10_VACUUM,
  2252. },
  2253. device=ble_device,
  2254. rssi=-97,
  2255. active=True,
  2256. )
  2257. def test_k10_passive() -> None:
  2258. """Test parsing k10+ with passive data."""
  2259. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2260. adv_data = generate_advertisement_data(
  2261. manufacturer_data={2409: b"\xca8\x06\xa9_\xf1\x02 d"},
  2262. rssi=-97,
  2263. )
  2264. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K10_VACUUM)
  2265. assert result == SwitchBotAdvertisement(
  2266. address="aa:bb:cc:dd:ee:ff",
  2267. data={
  2268. "rawAdvData": None,
  2269. "data": {
  2270. "sequence_number": 2,
  2271. "dusbin_connected": False,
  2272. "dustbin_bound": False,
  2273. "network_connected": True,
  2274. "battery": 100,
  2275. "work_status": 0,
  2276. },
  2277. "isEncrypted": False,
  2278. "model": "}",
  2279. "modelFriendlyName": "K10+ Vacuum",
  2280. "modelName": SwitchbotModel.K10_VACUUM,
  2281. },
  2282. device=ble_device,
  2283. rssi=-97,
  2284. active=False,
  2285. )
  2286. def test_k10_with_empty_data() -> None:
  2287. """Test parsing k10+ with empty data."""
  2288. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2289. adv_data = generate_advertisement_data(
  2290. manufacturer_data={2409: None},
  2291. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"}\x00"},
  2292. rssi=-97,
  2293. )
  2294. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K10_VACUUM)
  2295. assert result == SwitchBotAdvertisement(
  2296. address="aa:bb:cc:dd:ee:ff",
  2297. data={
  2298. "rawAdvData": b"}\x00",
  2299. "data": {},
  2300. "isEncrypted": False,
  2301. "model": "}",
  2302. },
  2303. device=ble_device,
  2304. rssi=-97,
  2305. active=True,
  2306. )
  2307. def test_k10_pro_combo_active() -> None:
  2308. """Test parsing k10+ pro combo with active data."""
  2309. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2310. adv_data = generate_advertisement_data(
  2311. manufacturer_data={
  2312. 2409: b"\xb0\xe9\xfe\x01\xf4\x1d\x0b\x01\x01\xb1\x03\x118\x01"
  2313. },
  2314. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"3\x00\x00"},
  2315. rssi=-97,
  2316. )
  2317. result = parse_advertisement_data(
  2318. ble_device, adv_data, SwitchbotModel.K10_PRO_COMBO_VACUUM
  2319. )
  2320. assert result == SwitchBotAdvertisement(
  2321. address="aa:bb:cc:dd:ee:ff",
  2322. data={
  2323. "rawAdvData": b"3\x00\x00",
  2324. "data": {
  2325. "sequence_number": 11,
  2326. "soc_version": "1.0.945",
  2327. "step": 1,
  2328. "mqtt_connected": True,
  2329. "battery": 56,
  2330. "work_status": 1,
  2331. },
  2332. "isEncrypted": False,
  2333. "model": "3",
  2334. "modelFriendlyName": "K10+ Pro Combo Vacuum",
  2335. "modelName": SwitchbotModel.K10_PRO_COMBO_VACUUM,
  2336. },
  2337. device=ble_device,
  2338. rssi=-97,
  2339. active=True,
  2340. )
  2341. def test_k10_pro_combo_passive() -> None:
  2342. """Test parsing k10+ pro combo with passive data."""
  2343. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2344. adv_data = generate_advertisement_data(
  2345. manufacturer_data={
  2346. 2409: b"\xb0\xe9\xfe\x01\xf4\x1d\x0b\x01\x01\xb1\x03\x118\x01"
  2347. },
  2348. rssi=-97,
  2349. )
  2350. result = parse_advertisement_data(
  2351. ble_device, adv_data, SwitchbotModel.K10_PRO_COMBO_VACUUM
  2352. )
  2353. assert result == SwitchBotAdvertisement(
  2354. address="aa:bb:cc:dd:ee:ff",
  2355. data={
  2356. "rawAdvData": None,
  2357. "data": {
  2358. "sequence_number": 11,
  2359. "soc_version": "1.0.945",
  2360. "step": 1,
  2361. "mqtt_connected": True,
  2362. "battery": 56,
  2363. "work_status": 1,
  2364. },
  2365. "isEncrypted": False,
  2366. "model": "3",
  2367. "modelFriendlyName": "K10+ Pro Combo Vacuum",
  2368. "modelName": SwitchbotModel.K10_PRO_COMBO_VACUUM,
  2369. },
  2370. device=ble_device,
  2371. rssi=-97,
  2372. active=False,
  2373. )
  2374. def test_k10_pro_combo_with_empty_data() -> None:
  2375. """Test parsing k10+ pro combo with empty data."""
  2376. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2377. adv_data = generate_advertisement_data(
  2378. manufacturer_data={2409: None},
  2379. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"3\x00\x00"},
  2380. rssi=-97,
  2381. )
  2382. result = parse_advertisement_data(
  2383. ble_device, adv_data, SwitchbotModel.K10_PRO_COMBO_VACUUM
  2384. )
  2385. assert result == SwitchBotAdvertisement(
  2386. address="aa:bb:cc:dd:ee:ff",
  2387. data={
  2388. "rawAdvData": b"3\x00\x00",
  2389. "data": {},
  2390. "isEncrypted": False,
  2391. "model": "3",
  2392. },
  2393. device=ble_device,
  2394. rssi=-97,
  2395. active=True,
  2396. )
  2397. def test_s10_active() -> None:
  2398. """Test parsing s10 with active data."""
  2399. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2400. adv_data = generate_advertisement_data(
  2401. manufacturer_data={2409: b"\xb0\xe9\xfe\x00\x08|\n\x01\x11\x05\x00\x10M\x02"},
  2402. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"z\x00\x00"},
  2403. rssi=-97,
  2404. )
  2405. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.S10_VACUUM)
  2406. assert result == SwitchBotAdvertisement(
  2407. address="aa:bb:cc:dd:ee:ff",
  2408. data={
  2409. "rawAdvData": b"z\x00\x00",
  2410. "data": {
  2411. "sequence_number": 10,
  2412. "soc_version": "1.1.005",
  2413. "step": 0,
  2414. "mqtt_connected": True,
  2415. "battery": 77,
  2416. "work_status": 2,
  2417. },
  2418. "isEncrypted": False,
  2419. "model": "z",
  2420. "modelFriendlyName": "S10 Vacuum",
  2421. "modelName": SwitchbotModel.S10_VACUUM,
  2422. },
  2423. device=ble_device,
  2424. rssi=-97,
  2425. active=True,
  2426. )
  2427. def test_s10_passive() -> None:
  2428. """Test parsing s10 with passive data."""
  2429. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2430. adv_data = generate_advertisement_data(
  2431. manufacturer_data={2409: b"\xb0\xe9\xfe\x00\x08|\n\x01\x11\x05\x00\x10M\x02"},
  2432. rssi=-97,
  2433. )
  2434. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.S10_VACUUM)
  2435. assert result == SwitchBotAdvertisement(
  2436. address="aa:bb:cc:dd:ee:ff",
  2437. data={
  2438. "rawAdvData": None,
  2439. "data": {
  2440. "sequence_number": 10,
  2441. "soc_version": "1.1.005",
  2442. "step": 0,
  2443. "mqtt_connected": True,
  2444. "battery": 77,
  2445. "work_status": 2,
  2446. },
  2447. "isEncrypted": False,
  2448. "model": "z",
  2449. "modelFriendlyName": "S10 Vacuum",
  2450. "modelName": SwitchbotModel.S10_VACUUM,
  2451. },
  2452. device=ble_device,
  2453. rssi=-97,
  2454. active=False,
  2455. )
  2456. def test_s10_with_empty_data() -> None:
  2457. """Test parsing s10 with empty data."""
  2458. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2459. adv_data = generate_advertisement_data(
  2460. manufacturer_data={2409: None},
  2461. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"z\x00\x00"},
  2462. rssi=-97,
  2463. )
  2464. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.S10_VACUUM)
  2465. assert result == SwitchBotAdvertisement(
  2466. address="aa:bb:cc:dd:ee:ff",
  2467. data={
  2468. "rawAdvData": b"z\x00\x00",
  2469. "data": {},
  2470. "isEncrypted": False,
  2471. "model": "z",
  2472. },
  2473. device=ble_device,
  2474. rssi=-97,
  2475. active=True,
  2476. )
  2477. @pytest.mark.parametrize(
  2478. ("manufacturer_data", "service_data", "data", "model"),
  2479. [
  2480. (
  2481. b"\xf0\x9e\x9e\x96j\xd6\xa1\x81\x88\xe4\x00\x01\x95\x00\x00",
  2482. b"7\x00\x00\x95-\x00",
  2483. {
  2484. "isOn": True,
  2485. "mode": "level_3",
  2486. "isAqiValid": False,
  2487. "child_lock": False,
  2488. "speed": 100,
  2489. "aqi_level": "excellent",
  2490. "filter element working time": 405,
  2491. "err_code": 0,
  2492. "sequence_number": 161,
  2493. },
  2494. "7",
  2495. ),
  2496. (
  2497. b'\xcc\x8d\xa2\xa7\x92>\t"\x80\x000\x00\x0f\x00\x00',
  2498. b"*\x00\x00\x15\x04\x00",
  2499. {
  2500. "isOn": False,
  2501. "mode": "auto",
  2502. "isAqiValid": False,
  2503. "child_lock": False,
  2504. "speed": 0,
  2505. "aqi_level": "excellent",
  2506. "filter element working time": 15,
  2507. "err_code": 0,
  2508. "sequence_number": 9,
  2509. },
  2510. "*",
  2511. ),
  2512. (
  2513. b"\xcc\x8d\xa2\xa7\xe4\xa6\x0b\x83\x88d\x00\xea`\x00\x00",
  2514. b"+\x00\x00\x15\x04\x00",
  2515. {
  2516. "isOn": True,
  2517. "mode": "pet",
  2518. "isAqiValid": False,
  2519. "child_lock": False,
  2520. "speed": 100,
  2521. "aqi_level": "excellent",
  2522. "filter element working time": 60000,
  2523. "err_code": 0,
  2524. "sequence_number": 11,
  2525. },
  2526. "+",
  2527. ),
  2528. (
  2529. b"\xcc\x8d\xa2\xa7\xc1\xae\x9b\x81\x8c\xb2\x00\x01\x94\x00\x00",
  2530. b"8\x00\x00\x95-\x00",
  2531. {
  2532. "isOn": True,
  2533. "mode": "level_2",
  2534. "isAqiValid": True,
  2535. "child_lock": False,
  2536. "speed": 50,
  2537. "aqi_level": "excellent",
  2538. "filter element working time": 404,
  2539. "err_code": 0,
  2540. "sequence_number": 155,
  2541. },
  2542. "8",
  2543. ),
  2544. (
  2545. b"\xcc\x8d\xa2\xa7\xc1\xae\x9e\xa1\x8c\x800\x01\x95\x00\x00",
  2546. b"8\x00\x00\x95-\x00",
  2547. {
  2548. "isOn": True,
  2549. "mode": "level_1",
  2550. "isAqiValid": True,
  2551. "child_lock": False,
  2552. "speed": 0,
  2553. "aqi_level": "excellent",
  2554. "filter element working time": 405,
  2555. "err_code": 0,
  2556. "sequence_number": 158,
  2557. },
  2558. "8",
  2559. ),
  2560. (
  2561. b"\xcc\x8d\xa2\xa7\xc1\xae\x9e\x05\x8c\x800\x01\x95\x00\x00",
  2562. b"8\x00\x00\x95-\x00",
  2563. {
  2564. "isOn": False,
  2565. "mode": None,
  2566. "isAqiValid": True,
  2567. "child_lock": False,
  2568. "speed": 0,
  2569. "aqi_level": "excellent",
  2570. "filter element working time": 405,
  2571. "err_code": 0,
  2572. "sequence_number": 158,
  2573. },
  2574. "8",
  2575. ),
  2576. ],
  2577. )
  2578. def test_air_purifier_active(manufacturer_data, service_data, data, model) -> None:
  2579. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2580. adv_data = generate_advertisement_data(
  2581. manufacturer_data={2409: manufacturer_data},
  2582. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": service_data},
  2583. rssi=-97,
  2584. )
  2585. result = parse_advertisement_data(ble_device, adv_data)
  2586. assert result == SwitchBotAdvertisement(
  2587. address="aa:bb:cc:dd:ee:ff",
  2588. data={
  2589. "rawAdvData": service_data,
  2590. "data": data,
  2591. "isEncrypted": False,
  2592. "model": model,
  2593. "modelFriendlyName": "Air Purifier",
  2594. "modelName": SwitchbotModel.AIR_PURIFIER,
  2595. },
  2596. device=ble_device,
  2597. rssi=-97,
  2598. active=True,
  2599. )
  2600. def test_air_purifier_passive() -> None:
  2601. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2602. adv_data = generate_advertisement_data(
  2603. manufacturer_data={
  2604. 2409: b"\xf0\x9e\x9e\x96j\xd6\xa1\x81\x88\xe4\x00\x01\x95\x00\x00"
  2605. },
  2606. rssi=-97,
  2607. )
  2608. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.AIR_PURIFIER)
  2609. assert result == SwitchBotAdvertisement(
  2610. address="aa:bb:cc:dd:ee:ff",
  2611. data={
  2612. "rawAdvData": None,
  2613. "data": {
  2614. "isOn": True,
  2615. "mode": "level_3",
  2616. "isAqiValid": False,
  2617. "child_lock": False,
  2618. "speed": 100,
  2619. "aqi_level": "excellent",
  2620. "filter element working time": 405,
  2621. "err_code": 0,
  2622. "sequence_number": 161,
  2623. },
  2624. "isEncrypted": False,
  2625. "model": "8",
  2626. "modelFriendlyName": "Air Purifier",
  2627. "modelName": SwitchbotModel.AIR_PURIFIER,
  2628. },
  2629. device=ble_device,
  2630. rssi=-97,
  2631. active=False,
  2632. )
  2633. def test_air_purifier_with_empty_data() -> None:
  2634. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2635. adv_data = generate_advertisement_data(
  2636. manufacturer_data={2409: None},
  2637. service_data={
  2638. "0000fd3d-0000-1000-8000-00805f9b34fb": b"+\x00\x00\x15\x04\x00",
  2639. },
  2640. rssi=-97,
  2641. )
  2642. result = parse_advertisement_data(ble_device, adv_data)
  2643. assert result == SwitchBotAdvertisement(
  2644. address="aa:bb:cc:dd:ee:ff",
  2645. data={
  2646. "rawAdvData": b"+\x00\x00\x15\x04\x00",
  2647. "data": {},
  2648. "isEncrypted": False,
  2649. "model": "+",
  2650. },
  2651. device=ble_device,
  2652. rssi=-97,
  2653. active=True,
  2654. )