test_adv_parser.py 85 KB

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