test_adv_parser.py 152 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791
  1. from __future__ import annotations
  2. import datetime
  3. from typing import Any
  4. import pytest
  5. from bleak.backends.device import BLEDevice
  6. from bleak.backends.scanner import AdvertisementData
  7. from switchbot import HumidifierMode, SwitchbotModel
  8. from switchbot.adv_parser import (
  9. _MODEL_TO_MAC_CACHE,
  10. parse_advertisement_data,
  11. populate_model_to_mac_cache,
  12. )
  13. from switchbot.const.lock import LockStatus
  14. from switchbot.models import SwitchBotAdvertisement
  15. from . import AdvTestCase
  16. ADVERTISEMENT_DATA_DEFAULTS = {
  17. "local_name": "",
  18. "manufacturer_data": {},
  19. "service_data": {},
  20. "service_uuids": [],
  21. "rssi": -127,
  22. "platform_data": ((),),
  23. "tx_power": -127,
  24. }
  25. BLE_DEVICE_DEFAULTS = {
  26. "name": None,
  27. "rssi": -127,
  28. "details": None,
  29. }
  30. def generate_ble_device(
  31. address: str | None = None,
  32. name: str | None = None,
  33. details: Any | None = None,
  34. rssi: int | None = None,
  35. **kwargs: Any,
  36. ) -> BLEDevice:
  37. """Generate a BLEDevice with defaults."""
  38. new = kwargs.copy()
  39. if address is not None:
  40. new["address"] = address
  41. if name is not None:
  42. new["name"] = name
  43. if details is not None:
  44. new["details"] = details
  45. if rssi is not None:
  46. new["rssi"] = rssi
  47. for key, value in BLE_DEVICE_DEFAULTS.items():
  48. new.setdefault(key, value)
  49. return BLEDevice(**new)
  50. def generate_advertisement_data(**kwargs: Any) -> AdvertisementData:
  51. """Generate advertisement data with defaults."""
  52. new = kwargs.copy()
  53. for key, value in ADVERTISEMENT_DATA_DEFAULTS.items():
  54. new.setdefault(key, value)
  55. return AdvertisementData(**new)
  56. def test_parse_advertisement_data_curtain():
  57. """Test parse_advertisement_data for curtain."""
  58. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  59. adv_data = generate_advertisement_data(
  60. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  61. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0X\x00\x11\x04"},
  62. rssi=-80,
  63. )
  64. result = parse_advertisement_data(ble_device, adv_data)
  65. assert result == SwitchBotAdvertisement(
  66. address="aa:bb:cc:dd:ee:ff",
  67. data={
  68. "rawAdvData": b"c\xc0X\x00\x11\x04",
  69. "data": {
  70. "calibration": True,
  71. "battery": 88,
  72. "inMotion": False,
  73. "position": 100,
  74. "lightLevel": 1,
  75. "deviceChain": 1,
  76. },
  77. "isEncrypted": False,
  78. "model": "c",
  79. "modelFriendlyName": "Curtain",
  80. "modelName": SwitchbotModel.CURTAIN,
  81. },
  82. device=ble_device,
  83. rssi=-80,
  84. active=True,
  85. )
  86. def test_parse_advertisement_data_curtain_passive():
  87. """Test parse_advertisement_data for curtain passive."""
  88. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  89. adv_data = generate_advertisement_data(
  90. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  91. service_data={},
  92. rssi=-80,
  93. )
  94. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  95. assert result == SwitchBotAdvertisement(
  96. address="aa:bb:cc:dd:ee:ff",
  97. data={
  98. "rawAdvData": None,
  99. "data": {
  100. "calibration": None,
  101. "battery": None,
  102. "inMotion": False,
  103. "position": 100,
  104. "lightLevel": 1,
  105. "deviceChain": 1,
  106. },
  107. "isEncrypted": False,
  108. "model": "{",
  109. "modelFriendlyName": "Curtain 3",
  110. "modelName": SwitchbotModel.CURTAIN,
  111. },
  112. device=ble_device,
  113. rssi=-80,
  114. active=False,
  115. )
  116. def test_parse_advertisement_data_curtain_passive_12_bytes():
  117. """Test parse_advertisement_data for curtain passive."""
  118. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  119. adv_data = generate_advertisement_data(
  120. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04\x00"},
  121. service_data={},
  122. rssi=-80,
  123. )
  124. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  125. assert result == SwitchBotAdvertisement(
  126. address="aa:bb:cc:dd:ee:ff",
  127. data={
  128. "rawAdvData": None,
  129. "data": {
  130. "calibration": None,
  131. "battery": None,
  132. "inMotion": False,
  133. "position": 100,
  134. "lightLevel": 1,
  135. "deviceChain": 1,
  136. },
  137. "isEncrypted": False,
  138. "model": "{",
  139. "modelFriendlyName": "Curtain 3",
  140. "modelName": SwitchbotModel.CURTAIN,
  141. },
  142. device=ble_device,
  143. rssi=-80,
  144. active=False,
  145. )
  146. def test_parse_advertisement_data_curtain_position_zero():
  147. """Test parse_advertisement_data for curtain position zero."""
  148. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  149. adv_data = generate_advertisement_data(
  150. local_name="WoCurtain",
  151. manufacturer_data={89: b"\xc1\xc7'}U\xab"},
  152. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0\xced\x11\x04"},
  153. service_uuids=[
  154. "00001800-0000-1000-8000-00805f9b34fb",
  155. "00001801-0000-1000-8000-00805f9b34fb",
  156. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  157. ],
  158. rssi=-52,
  159. )
  160. result = parse_advertisement_data(ble_device, adv_data)
  161. assert result == SwitchBotAdvertisement(
  162. address="aa:bb:cc:dd:ee:ff",
  163. data={
  164. "rawAdvData": b"c\xd0\xced\x11\x04",
  165. "data": {
  166. "calibration": True,
  167. "battery": 78,
  168. "inMotion": False,
  169. "position": 0,
  170. "lightLevel": 1,
  171. "deviceChain": 1,
  172. },
  173. "isEncrypted": False,
  174. "model": "c",
  175. "modelFriendlyName": "Curtain",
  176. "modelName": SwitchbotModel.CURTAIN,
  177. },
  178. device=ble_device,
  179. rssi=-52,
  180. active=True,
  181. )
  182. def test_parse_advertisement_data_curtain_firmware_six_position_100():
  183. """Test parse_advertisement_data with firmware six for curtain position 100."""
  184. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  185. adv_data = generate_advertisement_data(
  186. local_name="WoCurtain",
  187. manufacturer_data={
  188. 89: b"\xf5\x98\x94\x08\xa0\xe7",
  189. 2409: b'\xf5\x98\x94\x08\xa0\xe7\x9b\x0f\x00"\x04',
  190. },
  191. service_data={
  192. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0H\x00\x12\x04",
  193. "0000fd3d-0000-1000-8000-00805f9b34fb": b'c\xc0G\x00"\x04',
  194. },
  195. service_uuids=[
  196. "00001800-0000-1000-8000-00805f9b34fb",
  197. "00001801-0000-1000-8000-00805f9b34fb",
  198. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  199. ],
  200. rssi=-62,
  201. )
  202. result = parse_advertisement_data(ble_device, adv_data)
  203. assert result == SwitchBotAdvertisement(
  204. address="aa:bb:cc:dd:ee:ff",
  205. data={
  206. "rawAdvData": b'c\xc0G\x00"\x04',
  207. "data": {
  208. "calibration": True,
  209. "battery": 71,
  210. "inMotion": False,
  211. "position": 100,
  212. "lightLevel": 2,
  213. "deviceChain": 2,
  214. },
  215. "isEncrypted": False,
  216. "model": "c",
  217. "modelFriendlyName": "Curtain",
  218. "modelName": SwitchbotModel.CURTAIN,
  219. },
  220. device=ble_device,
  221. rssi=-62,
  222. active=True,
  223. )
  224. def test_parse_advertisement_data_curtain_firmware_six_position_100_other_rssi():
  225. """Test parse_advertisement_data with firmware six for curtain position 100 other rssi."""
  226. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  227. adv_data = generate_advertisement_data(
  228. local_name="WoCurtain",
  229. manufacturer_data={
  230. 89: b"\xf5\x98\x94\x08\xa0\xe7",
  231. 2409: b'\xf5\x98\x94\x08\xa0\xe7\xa5\x0fc"\x04',
  232. },
  233. service_data={
  234. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0H\x00\x12\x04",
  235. "0000fd3d-0000-1000-8000-00805f9b34fb": b'c\xc0Gc"\x04',
  236. },
  237. service_uuids=[
  238. "00001800-0000-1000-8000-00805f9b34fb",
  239. "00001801-0000-1000-8000-00805f9b34fb",
  240. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  241. ],
  242. rssi=-67,
  243. )
  244. result = parse_advertisement_data(ble_device, adv_data)
  245. assert result == SwitchBotAdvertisement(
  246. address="aa:bb:cc:dd:ee:ff",
  247. data={
  248. "rawAdvData": b'c\xc0Gc"\x04',
  249. "data": {
  250. "calibration": True,
  251. "battery": 71,
  252. "inMotion": False,
  253. "position": 1,
  254. "lightLevel": 2,
  255. "deviceChain": 2,
  256. },
  257. "isEncrypted": False,
  258. "model": "c",
  259. "modelFriendlyName": "Curtain",
  260. "modelName": SwitchbotModel.CURTAIN,
  261. },
  262. device=ble_device,
  263. rssi=-67,
  264. active=True,
  265. )
  266. def test_parse_advertisement_data_curtain_fully_closed():
  267. """Test parse_advertisement_data with firmware six fully closed."""
  268. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  269. adv_data = generate_advertisement_data(
  270. local_name="WoCurtain",
  271. manufacturer_data={2409: b"\xc1\xc7'}U\xab\"\x0fd\x11\x04"},
  272. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0Sd\x11\x04"},
  273. service_uuids=[
  274. "00001800-0000-1000-8000-00805f9b34fb",
  275. "00001801-0000-1000-8000-00805f9b34fb",
  276. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  277. ],
  278. rssi=1,
  279. )
  280. result = parse_advertisement_data(ble_device, adv_data)
  281. assert result == SwitchBotAdvertisement(
  282. address="aa:bb:cc:dd:ee:ff",
  283. data={
  284. "rawAdvData": b"c\xc0Sd\x11\x04",
  285. "data": {
  286. "calibration": True,
  287. "battery": 83,
  288. "inMotion": False,
  289. "position": 0,
  290. "lightLevel": 1,
  291. "deviceChain": 1,
  292. },
  293. "isEncrypted": False,
  294. "model": "c",
  295. "modelFriendlyName": "Curtain",
  296. "modelName": SwitchbotModel.CURTAIN,
  297. },
  298. device=ble_device,
  299. rssi=1,
  300. active=True,
  301. )
  302. def test_parse_advertisement_data_curtain_fully_open():
  303. """Test parse_advertisement_data with firmware six fully open."""
  304. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  305. adv_data = generate_advertisement_data(
  306. local_name="WoCurtain",
  307. manufacturer_data={2409: b"\xc1\xc7'}U\xab%\x0f\x00\x11\x04"},
  308. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0S\x00\x11\x04"},
  309. service_uuids=[
  310. "00001800-0000-1000-8000-00805f9b34fb",
  311. "00001801-0000-1000-8000-00805f9b34fb",
  312. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  313. ],
  314. rssi=1,
  315. )
  316. result = parse_advertisement_data(ble_device, adv_data)
  317. assert result == SwitchBotAdvertisement(
  318. address="aa:bb:cc:dd:ee:ff",
  319. data={
  320. "rawAdvData": b"c\xc0S\x00\x11\x04",
  321. "data": {
  322. "calibration": True,
  323. "battery": 83,
  324. "inMotion": False,
  325. "position": 100,
  326. "lightLevel": 1,
  327. "deviceChain": 1,
  328. },
  329. "isEncrypted": False,
  330. "model": "c",
  331. "modelFriendlyName": "Curtain",
  332. "modelName": SwitchbotModel.CURTAIN,
  333. },
  334. device=ble_device,
  335. rssi=1,
  336. active=True,
  337. )
  338. def test_parse_advertisement_data_curtain3():
  339. """Test parse_advertisement_data for curtain 3."""
  340. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  341. adv_data = generate_advertisement_data(
  342. manufacturer_data={
  343. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\xf7\x07\x00\x11\x04\x00\x49"
  344. },
  345. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"{\xc0\x49\x00\x11\x04"},
  346. rssi=-80,
  347. )
  348. result = parse_advertisement_data(ble_device, adv_data)
  349. assert result == SwitchBotAdvertisement(
  350. address="aa:bb:cc:dd:ee:ff",
  351. data={
  352. "rawAdvData": b"{\xc0\x49\x00\x11\x04",
  353. "data": {
  354. "calibration": True,
  355. "battery": 73,
  356. "inMotion": False,
  357. "position": 100,
  358. "lightLevel": 1,
  359. "deviceChain": 1,
  360. },
  361. "isEncrypted": False,
  362. "model": "{",
  363. "modelFriendlyName": "Curtain 3",
  364. "modelName": SwitchbotModel.CURTAIN,
  365. },
  366. device=ble_device,
  367. rssi=-80,
  368. active=True,
  369. )
  370. def test_parse_advertisement_data_curtain3_passive():
  371. """Test parse_advertisement_data for curtain passive."""
  372. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  373. adv_data = generate_advertisement_data(
  374. manufacturer_data={
  375. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\xf7\x07\x00\x11\x04\x00\x49"
  376. },
  377. service_data={},
  378. rssi=-80,
  379. )
  380. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.CURTAIN)
  381. assert result == SwitchBotAdvertisement(
  382. address="aa:bb:cc:dd:ee:ff",
  383. data={
  384. "rawAdvData": None,
  385. "data": {
  386. "calibration": None,
  387. "battery": 73,
  388. "inMotion": False,
  389. "position": 100,
  390. "lightLevel": 1,
  391. "deviceChain": 1,
  392. },
  393. "isEncrypted": False,
  394. "model": "{",
  395. "modelFriendlyName": "Curtain 3",
  396. "modelName": SwitchbotModel.CURTAIN,
  397. },
  398. device=ble_device,
  399. rssi=-80,
  400. active=False,
  401. )
  402. def test_parse_advertisement_data_contact():
  403. """Test parse_advertisement_data for the contact sensor."""
  404. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  405. adv_data = generate_advertisement_data(
  406. manufacturer_data={2409: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  407. service_data={
  408. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@d\x05\x00u\x00\xf8\x12"
  409. },
  410. rssi=-80,
  411. )
  412. result = parse_advertisement_data(ble_device, adv_data)
  413. assert result == SwitchBotAdvertisement(
  414. address="aa:bb:cc:dd:ee:ff",
  415. data={
  416. "rawAdvData": b"d@d\x05\x00u\x00\xf8\x12",
  417. "data": {
  418. "button_count": 2,
  419. "contact_open": True,
  420. "contact_timeout": True,
  421. "is_light": True,
  422. "battery": 100,
  423. "motion_detected": True,
  424. "tested": False,
  425. },
  426. "isEncrypted": False,
  427. "model": "d",
  428. "modelFriendlyName": "Contact Sensor",
  429. "modelName": SwitchbotModel.CONTACT_SENSOR,
  430. },
  431. device=ble_device,
  432. rssi=-80,
  433. active=True,
  434. )
  435. def test_parse_advertisement_data_empty():
  436. """Test parse_advertisement_data with empty data does not blow up."""
  437. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  438. adv_data = generate_advertisement_data(
  439. manufacturer_data={2403: b"\xe7\xabF\xac\x8f\x92|\x0f\x00\x11\x04"},
  440. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b""},
  441. )
  442. result = parse_advertisement_data(ble_device, adv_data)
  443. assert result is None
  444. def test_new_bot_firmware():
  445. """Test parsing adv data from new bot firmware."""
  446. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  447. adv_data = generate_advertisement_data(
  448. manufacturer_data={89: b"\xd8.\xad\xcd\r\x85"},
  449. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"H\x10\xe1"},
  450. service_uuids=["CBA20D00-224D-11E6-9FB8-0002A5D5C51B"],
  451. rssi=-90,
  452. )
  453. result = parse_advertisement_data(ble_device, adv_data)
  454. assert result == SwitchBotAdvertisement(
  455. address="aa:bb:cc:dd:ee:ff",
  456. data={
  457. "rawAdvData": b"H\x10\xe1",
  458. "data": {"switchMode": False, "isOn": False, "battery": 97},
  459. "model": "H",
  460. "isEncrypted": False,
  461. "modelFriendlyName": "Bot",
  462. "modelName": SwitchbotModel.BOT,
  463. },
  464. device=ble_device,
  465. rssi=-90,
  466. active=True,
  467. )
  468. def test_parse_advertisement_data_curtain_firmware_six_fully_closed():
  469. """Test parse_advertisement_data with firmware six fully closed."""
  470. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  471. adv_data = generate_advertisement_data(
  472. local_name="WoCurtain",
  473. manufacturer_data={
  474. 89: b"\xcc\xf4\xc4\xf9\xacl",
  475. 2409: b"\xcc\xf4\xc4\xf9\xacl\xeb\x0fd\x12\x04",
  476. },
  477. service_data={
  478. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0Yd\x11\x04",
  479. "0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0dd\x12\x04",
  480. },
  481. service_uuids=[
  482. "00001800-0000-1000-8000-00805f9b34fb",
  483. "00001801-0000-1000-8000-00805f9b34fb",
  484. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  485. ],
  486. rssi=-2,
  487. )
  488. result = parse_advertisement_data(ble_device, adv_data)
  489. assert result == SwitchBotAdvertisement(
  490. address="aa:bb:cc:dd:ee:ff",
  491. data={
  492. "rawAdvData": b"c\xc0dd\x12\x04",
  493. "data": {
  494. "calibration": True,
  495. "battery": 100,
  496. "inMotion": False,
  497. "position": 0,
  498. "lightLevel": 1,
  499. "deviceChain": 2,
  500. },
  501. "isEncrypted": False,
  502. "model": "c",
  503. "modelFriendlyName": "Curtain",
  504. "modelName": SwitchbotModel.CURTAIN,
  505. },
  506. device=ble_device,
  507. rssi=-2,
  508. active=True,
  509. )
  510. def test_parse_advertisement_data_curtain_firmware_six_fully_open():
  511. """Test parse_advertisement_data with firmware six fully open."""
  512. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  513. adv_data = generate_advertisement_data(
  514. local_name="WoCurtain",
  515. manufacturer_data={
  516. 89: b"\xcc\xf4\xc4\xf9\xacl",
  517. 2409: b"\xcc\xf4\xc4\xf9\xacl\xe2\x0f\x00\x12\x04",
  518. },
  519. service_data={
  520. "00000d00-0000-1000-8000-00805f9b34fb": b"c\xd0Yd\x11\x04",
  521. "0000fd3d-0000-1000-8000-00805f9b34fb": b"c\xc0d\x00\x12\x04",
  522. },
  523. service_uuids=[
  524. "00001800-0000-1000-8000-00805f9b34fb",
  525. "00001801-0000-1000-8000-00805f9b34fb",
  526. "cba20d00-224d-11e6-9fb8-0002a5d5c51b",
  527. ],
  528. rssi=-2,
  529. )
  530. result = parse_advertisement_data(ble_device, adv_data)
  531. assert result == SwitchBotAdvertisement(
  532. address="aa:bb:cc:dd:ee:ff",
  533. data={
  534. "rawAdvData": b"c\xc0d\x00\x12\x04",
  535. "data": {
  536. "calibration": True,
  537. "battery": 100,
  538. "inMotion": False,
  539. "position": 100,
  540. "lightLevel": 1,
  541. "deviceChain": 2,
  542. },
  543. "isEncrypted": False,
  544. "model": "c",
  545. "modelFriendlyName": "Curtain",
  546. "modelName": SwitchbotModel.CURTAIN,
  547. },
  548. device=ble_device,
  549. rssi=-2,
  550. active=True,
  551. )
  552. def test_contact_sensor_mfr():
  553. """Test parsing adv data from new bot firmware."""
  554. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  555. adv_data = generate_advertisement_data(
  556. manufacturer_data={2409: b"\xcb9\xcd\xc4=FA,\x00F\x01\x8f\xc4"},
  557. service_data={
  558. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d\x00\xda\x04\x00F\x01\x8f\xc4"
  559. },
  560. tx_power=-127,
  561. rssi=-70,
  562. )
  563. result = parse_advertisement_data(ble_device, adv_data)
  564. assert result == SwitchBotAdvertisement(
  565. address="aa:bb:cc:dd:ee:ff",
  566. data={
  567. "data": {
  568. "battery": 90,
  569. "button_count": 4,
  570. "contact_open": True,
  571. "contact_timeout": True,
  572. "is_light": False,
  573. "motion_detected": False,
  574. "tested": False,
  575. },
  576. "isEncrypted": False,
  577. "model": "d",
  578. "modelFriendlyName": "Contact Sensor",
  579. "modelName": SwitchbotModel.CONTACT_SENSOR,
  580. "rawAdvData": b"d\x00\xda\x04\x00F\x01\x8f\xc4",
  581. },
  582. device=ble_device,
  583. rssi=-70,
  584. active=True,
  585. )
  586. def test_contact_sensor_mfr_no_service_data():
  587. """Test contact sensor with passive data only."""
  588. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  589. adv_data = generate_advertisement_data(
  590. manufacturer_data={2409: b"\xcb9\xcd\xc4=FA,\x00F\x01\x8f\xc4"},
  591. service_data={},
  592. tx_power=-127,
  593. rssi=-70,
  594. )
  595. result = parse_advertisement_data(ble_device, adv_data)
  596. # Passive detection of contact sensor is not supported
  597. # anymore since the Switchbot Curtain v3 was released
  598. # which uses the heuristics for the contact sensor.
  599. assert result is None
  600. def test_contact_sensor_srv():
  601. """Test parsing adv data from new bot firmware."""
  602. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  603. adv_data = generate_advertisement_data(
  604. service_data={
  605. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d\x00\xda\x04\x00F\x01\x8f\xc4"
  606. },
  607. tx_power=-127,
  608. rssi=-70,
  609. )
  610. result = parse_advertisement_data(ble_device, adv_data)
  611. assert result == SwitchBotAdvertisement(
  612. address="aa:bb:cc:dd:ee:ff",
  613. data={
  614. "data": {
  615. "battery": 90,
  616. "button_count": 4,
  617. "contact_open": True,
  618. "contact_timeout": True,
  619. "is_light": False,
  620. "motion_detected": False,
  621. "tested": False,
  622. },
  623. "isEncrypted": False,
  624. "model": "d",
  625. "modelFriendlyName": "Contact Sensor",
  626. "modelName": SwitchbotModel.CONTACT_SENSOR,
  627. "rawAdvData": b"d\x00\xda\x04\x00F\x01\x8f\xc4",
  628. },
  629. device=ble_device,
  630. rssi=-70,
  631. active=True,
  632. )
  633. def test_contact_sensor_open():
  634. """Test parsing mfr adv data from new bot firmware."""
  635. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  636. adv_data = generate_advertisement_data(
  637. manufacturer_data={2409: b"\xcb9\xcd\xc4=F\x84\x9c\x00\x17\x00QD"},
  638. service_data={
  639. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@\xda\x02\x00\x17\x00QD"
  640. },
  641. tx_power=-127,
  642. rssi=-59,
  643. )
  644. result = parse_advertisement_data(ble_device, adv_data)
  645. assert result == SwitchBotAdvertisement(
  646. address="aa:bb:cc:dd:ee:ff",
  647. data={
  648. "data": {
  649. "battery": 90,
  650. "button_count": 4,
  651. "contact_open": True,
  652. "contact_timeout": False,
  653. "is_light": False,
  654. "motion_detected": True,
  655. "tested": False,
  656. },
  657. "isEncrypted": False,
  658. "model": "d",
  659. "modelFriendlyName": "Contact Sensor",
  660. "modelName": SwitchbotModel.CONTACT_SENSOR,
  661. "rawAdvData": b"d@\xda\x02\x00\x17\x00QD",
  662. },
  663. device=ble_device,
  664. rssi=-59,
  665. active=True,
  666. )
  667. def test_contact_sensor_closed():
  668. """Test parsing mfr adv data from new bot firmware."""
  669. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  670. adv_data = generate_advertisement_data(
  671. manufacturer_data={2409: b"\xcb9\xcd\xc4=F\x89\x8c\x00+\x00\x19\x84"},
  672. service_data={
  673. "0000fd3d-0000-1000-8000-00805f9b34fb": b"d@\xda\x00\x00+\x00\x19\x84"
  674. },
  675. tx_power=-127,
  676. rssi=-50,
  677. )
  678. result = parse_advertisement_data(ble_device, adv_data)
  679. assert result == SwitchBotAdvertisement(
  680. address="aa:bb:cc:dd:ee:ff",
  681. data={
  682. "data": {
  683. "battery": 90,
  684. "button_count": 4,
  685. "contact_open": False,
  686. "contact_timeout": False,
  687. "is_light": False,
  688. "motion_detected": True,
  689. "tested": False,
  690. },
  691. "isEncrypted": False,
  692. "model": "d",
  693. "modelFriendlyName": "Contact Sensor",
  694. "modelName": SwitchbotModel.CONTACT_SENSOR,
  695. "rawAdvData": b"d@\xda\x00\x00+\x00\x19\x84",
  696. },
  697. device=ble_device,
  698. rssi=-50,
  699. active=True,
  700. )
  701. def test_switchbot_passive():
  702. """Test parsing switchbot as passive."""
  703. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  704. adv_data = generate_advertisement_data(
  705. manufacturer_data={89: bytes.fromhex("d51cfb397856")},
  706. service_data={},
  707. tx_power=-127,
  708. rssi=-50,
  709. )
  710. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.BOT)
  711. assert result == SwitchBotAdvertisement(
  712. address="aa:bb:cc:dd:ee:ff",
  713. data={
  714. "data": {
  715. "battery": None,
  716. "switchMode": None,
  717. "isOn": None,
  718. },
  719. "isEncrypted": False,
  720. "model": "H",
  721. "modelFriendlyName": "Bot",
  722. "modelName": SwitchbotModel.BOT,
  723. "rawAdvData": None,
  724. },
  725. device=ble_device,
  726. rssi=-50,
  727. active=False,
  728. )
  729. def test_bulb_active():
  730. """Test parsing bulb as active."""
  731. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  732. adv_data = generate_advertisement_data(
  733. manufacturer_data={2409: b"\x84\xf7\x03\xb4\xcbz\x03\xe4!\x00\x00"},
  734. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"u\x00d"},
  735. tx_power=-127,
  736. rssi=-50,
  737. )
  738. result = parse_advertisement_data(ble_device, adv_data)
  739. assert result == SwitchBotAdvertisement(
  740. address="aa:bb:cc:dd:ee:ff",
  741. data={
  742. "data": {
  743. "brightness": 100,
  744. "color_mode": 1,
  745. "delay": False,
  746. "isOn": True,
  747. "loop_index": 0,
  748. "preset": False,
  749. "sequence_number": 3,
  750. "speed": 0,
  751. },
  752. "isEncrypted": False,
  753. "model": "u",
  754. "modelFriendlyName": "Color Bulb",
  755. "modelName": SwitchbotModel.COLOR_BULB,
  756. "rawAdvData": b"u\x00d",
  757. },
  758. device=ble_device,
  759. rssi=-50,
  760. active=True,
  761. )
  762. def test_wosensor_passive_and_active():
  763. """Test parsing wosensor as passive with active data as well."""
  764. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  765. adv_data = generate_advertisement_data(
  766. manufacturer_data={2409: b"\xd7\xc1}]\xebC\xde\x03\x06\x985"},
  767. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\xe4\x06\x985"},
  768. tx_power=-127,
  769. rssi=-50,
  770. )
  771. result = parse_advertisement_data(ble_device, adv_data)
  772. assert result == SwitchBotAdvertisement(
  773. address="aa:bb:cc:dd:ee:ff",
  774. data={
  775. "data": {
  776. "battery": 100,
  777. "fahrenheit": False,
  778. "humidity": 53,
  779. "temp": {"c": 24.6, "f": 76.28},
  780. "temperature": 24.6,
  781. },
  782. "isEncrypted": False,
  783. "model": "T",
  784. "modelFriendlyName": "Meter",
  785. "modelName": SwitchbotModel.METER,
  786. "rawAdvData": b"T\x00\xe4\x06\x985",
  787. },
  788. device=ble_device,
  789. rssi=-50,
  790. active=True,
  791. )
  792. def test_wosensor_active():
  793. """Test parsing wosensor with active data as well."""
  794. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  795. adv_data = generate_advertisement_data(
  796. manufacturer_data={},
  797. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\xe4\x06\x985"},
  798. tx_power=-127,
  799. rssi=-50,
  800. )
  801. result = parse_advertisement_data(ble_device, adv_data)
  802. assert result == SwitchBotAdvertisement(
  803. address="aa:bb:cc:dd:ee:ff",
  804. data={
  805. "data": {
  806. "battery": 100,
  807. "fahrenheit": False,
  808. "humidity": 53,
  809. "temp": {"c": 24.6, "f": 76.28},
  810. "temperature": 24.6,
  811. },
  812. "isEncrypted": False,
  813. "model": "T",
  814. "modelFriendlyName": "Meter",
  815. "modelName": SwitchbotModel.METER,
  816. "rawAdvData": b"T\x00\xe4\x06\x985",
  817. },
  818. device=ble_device,
  819. rssi=-50,
  820. active=True,
  821. )
  822. def test_wosensor_passive_only():
  823. """Test parsing wosensor with only passive data."""
  824. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  825. adv_data = generate_advertisement_data(
  826. manufacturer_data={2409: b"\xd7\xc1}]\xebC\xde\x03\x06\x985"},
  827. service_data={},
  828. tx_power=-127,
  829. rssi=-50,
  830. )
  831. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER)
  832. assert result == SwitchBotAdvertisement(
  833. address="aa:bb:cc:dd:ee:ff",
  834. data={
  835. "data": {
  836. "battery": None,
  837. "fahrenheit": False,
  838. "humidity": 53,
  839. "temp": {"c": 24.6, "f": 76.28},
  840. "temperature": 24.6,
  841. },
  842. "isEncrypted": False,
  843. "model": "i",
  844. "modelFriendlyName": "Meter Plus",
  845. "modelName": SwitchbotModel.METER,
  846. "rawAdvData": None,
  847. },
  848. device=ble_device,
  849. rssi=-50,
  850. active=False,
  851. )
  852. def test_wosensor_active_zero_data():
  853. """Test parsing wosensor with active data but all values are zero."""
  854. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  855. adv_data = generate_advertisement_data(
  856. manufacturer_data={},
  857. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"T\x00\x00\x00\x00\x00"},
  858. tx_power=-127,
  859. rssi=-50,
  860. )
  861. result = parse_advertisement_data(ble_device, adv_data)
  862. assert result == SwitchBotAdvertisement(
  863. address="aa:bb:cc:dd:ee:ff",
  864. data={
  865. "data": {},
  866. "isEncrypted": False,
  867. "model": "T",
  868. "rawAdvData": b"T\x00\x00\x00\x00\x00",
  869. },
  870. device=ble_device,
  871. rssi=-50,
  872. active=True,
  873. )
  874. def test_wohub2_passive_and_active():
  875. """Test parsing wosensor as passive with active data as well."""
  876. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  877. adv_data = generate_advertisement_data(
  878. manufacturer_data={
  879. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x00\xfffT\x1a\xf1\x82\x07\x9a2\x00"
  880. },
  881. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"v\x00"},
  882. tx_power=-127,
  883. rssi=-50,
  884. )
  885. result = parse_advertisement_data(ble_device, adv_data)
  886. assert result == SwitchBotAdvertisement(
  887. address="aa:bb:cc:dd:ee:ff",
  888. data={
  889. "data": {
  890. "fahrenheit": False,
  891. "humidity": 50,
  892. "lightLevel": 2,
  893. "illuminance": 10,
  894. "temp": {"c": 26.7, "f": 80.06},
  895. "temperature": 26.7,
  896. },
  897. "isEncrypted": False,
  898. "model": "v",
  899. "modelFriendlyName": "Hub 2",
  900. "modelName": SwitchbotModel.HUB2,
  901. "rawAdvData": b"v\x00",
  902. },
  903. device=ble_device,
  904. rssi=-50,
  905. active=True,
  906. )
  907. def test_woiosensor_passive_and_active():
  908. """Test parsing woiosensor as passive with active data as well."""
  909. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  910. adv_data = generate_advertisement_data(
  911. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee\xff\xe0\x0f\x06\x985\x00"},
  912. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"w\x00\xe4"},
  913. tx_power=-127,
  914. rssi=-50,
  915. )
  916. result = parse_advertisement_data(ble_device, adv_data)
  917. assert result == SwitchBotAdvertisement(
  918. address="aa:bb:cc:dd:ee:ff",
  919. data={
  920. "data": {
  921. "battery": 100,
  922. "fahrenheit": False,
  923. "humidity": 53,
  924. "temp": {"c": 24.6, "f": 76.28},
  925. "temperature": 24.6,
  926. },
  927. "isEncrypted": False,
  928. "model": "w",
  929. "modelFriendlyName": "Indoor/Outdoor Meter",
  930. "modelName": SwitchbotModel.IO_METER,
  931. "rawAdvData": b"w\x00\xe4",
  932. },
  933. device=ble_device,
  934. rssi=-50,
  935. active=True,
  936. )
  937. def test_woiosensor_passive_only():
  938. """Test parsing woiosensor with only passive data."""
  939. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  940. adv_data = generate_advertisement_data(
  941. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee\xff\xe0\x0f\x06\x985\x00"},
  942. service_data={},
  943. tx_power=-127,
  944. rssi=-50,
  945. )
  946. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.IO_METER)
  947. assert result == SwitchBotAdvertisement(
  948. address="aa:bb:cc:dd:ee:ff",
  949. data={
  950. "data": {
  951. "battery": None,
  952. "fahrenheit": False,
  953. "humidity": 53,
  954. "temp": {"c": 24.6, "f": 76.28},
  955. "temperature": 24.6,
  956. },
  957. "isEncrypted": False,
  958. "model": "w",
  959. "modelFriendlyName": "Indoor/Outdoor Meter",
  960. "modelName": SwitchbotModel.IO_METER,
  961. "rawAdvData": None,
  962. },
  963. device=ble_device,
  964. rssi=-50,
  965. active=False,
  966. )
  967. def test_motion_sensor_clear():
  968. """Test parsing motion sensor with clear data."""
  969. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  970. adv_data = generate_advertisement_data(
  971. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIj\x1c\x00f"},
  972. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x00f\x01"},
  973. tx_power=-127,
  974. rssi=-87,
  975. )
  976. result = parse_advertisement_data(
  977. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  978. )
  979. assert result == SwitchBotAdvertisement(
  980. address="aa:bb:cc:dd:ee:ff",
  981. data={
  982. "data": {
  983. "battery": 98,
  984. "iot": 0,
  985. "is_light": False,
  986. "led": 0,
  987. "light_intensity": 1,
  988. "motion_detected": False,
  989. "sense_distance": 0,
  990. "tested": False,
  991. },
  992. "isEncrypted": False,
  993. "model": "s",
  994. "modelFriendlyName": "Motion Sensor",
  995. "modelName": SwitchbotModel.MOTION_SENSOR,
  996. "rawAdvData": b"s\x00\xe2\x00f\x01",
  997. },
  998. device=ble_device,
  999. rssi=-87,
  1000. active=True,
  1001. )
  1002. def test_motion_sensor_clear_passive():
  1003. """Test parsing motion sensor with clear data."""
  1004. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1005. adv_data = generate_advertisement_data(
  1006. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIj\x1c\x00f"},
  1007. service_data={},
  1008. tx_power=-127,
  1009. rssi=-87,
  1010. )
  1011. result = parse_advertisement_data(
  1012. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1013. )
  1014. assert result == SwitchBotAdvertisement(
  1015. address="aa:bb:cc:dd:ee:ff",
  1016. data={
  1017. "data": {
  1018. "battery": None,
  1019. "iot": None,
  1020. "is_light": False,
  1021. "led": None,
  1022. "light_intensity": None,
  1023. "motion_detected": False,
  1024. "sense_distance": None,
  1025. "tested": None,
  1026. },
  1027. "isEncrypted": False,
  1028. "model": "s",
  1029. "modelFriendlyName": "Motion Sensor",
  1030. "modelName": SwitchbotModel.MOTION_SENSOR,
  1031. "rawAdvData": None,
  1032. },
  1033. device=ble_device,
  1034. rssi=-87,
  1035. active=False,
  1036. )
  1037. def test_motion_sensor_motion():
  1038. """Test parsing motion sensor with motion data."""
  1039. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1040. adv_data = generate_advertisement_data(
  1041. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIi\\\x008"},
  1042. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s@\xe2\x008\x01"},
  1043. tx_power=-127,
  1044. rssi=-87,
  1045. )
  1046. result = parse_advertisement_data(
  1047. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1048. )
  1049. assert result == SwitchBotAdvertisement(
  1050. address="aa:bb:cc:dd:ee:ff",
  1051. data={
  1052. "data": {
  1053. "battery": 98,
  1054. "iot": 0,
  1055. "is_light": False,
  1056. "led": 0,
  1057. "light_intensity": 1,
  1058. "motion_detected": True,
  1059. "sense_distance": 0,
  1060. "tested": False,
  1061. },
  1062. "isEncrypted": False,
  1063. "model": "s",
  1064. "modelFriendlyName": "Motion Sensor",
  1065. "modelName": SwitchbotModel.MOTION_SENSOR,
  1066. "rawAdvData": b"s@\xe2\x008\x01",
  1067. },
  1068. device=ble_device,
  1069. rssi=-87,
  1070. active=True,
  1071. )
  1072. def test_motion_sensor_motion_passive():
  1073. """Test parsing motion sensor with motion data."""
  1074. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1075. adv_data = generate_advertisement_data(
  1076. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIi\\\x008"},
  1077. service_data={},
  1078. tx_power=-127,
  1079. rssi=-87,
  1080. )
  1081. result = parse_advertisement_data(
  1082. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1083. )
  1084. assert result == SwitchBotAdvertisement(
  1085. address="aa:bb:cc:dd:ee:ff",
  1086. data={
  1087. "data": {
  1088. "battery": None,
  1089. "iot": None,
  1090. "is_light": False,
  1091. "led": None,
  1092. "light_intensity": None,
  1093. "motion_detected": True,
  1094. "sense_distance": None,
  1095. "tested": None,
  1096. },
  1097. "isEncrypted": False,
  1098. "model": "s",
  1099. "modelFriendlyName": "Motion Sensor",
  1100. "modelName": SwitchbotModel.MOTION_SENSOR,
  1101. "rawAdvData": None,
  1102. },
  1103. device=ble_device,
  1104. rssi=-87,
  1105. active=False,
  1106. )
  1107. def test_motion_sensor_is_light_passive():
  1108. """Test parsing motion sensor with motion data."""
  1109. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1110. adv_data = generate_advertisement_data(
  1111. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIs,\x04g"},
  1112. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x04g\x02"},
  1113. tx_power=-127,
  1114. rssi=-93,
  1115. )
  1116. result = parse_advertisement_data(
  1117. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1118. )
  1119. assert result == SwitchBotAdvertisement(
  1120. address="aa:bb:cc:dd:ee:ff",
  1121. data={
  1122. "data": {
  1123. "battery": 98,
  1124. "iot": 0,
  1125. "is_light": True,
  1126. "led": 0,
  1127. "light_intensity": 2,
  1128. "motion_detected": False,
  1129. "sense_distance": 0,
  1130. "tested": False,
  1131. },
  1132. "isEncrypted": False,
  1133. "model": "s",
  1134. "modelFriendlyName": "Motion Sensor",
  1135. "modelName": SwitchbotModel.MOTION_SENSOR,
  1136. "rawAdvData": b"s\x00\xe2\x04g\x02",
  1137. },
  1138. device=ble_device,
  1139. rssi=-93,
  1140. active=True,
  1141. )
  1142. def test_motion_sensor_is_light_active():
  1143. """Test parsing motion sensor with motion data."""
  1144. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1145. adv_data = generate_advertisement_data(
  1146. manufacturer_data={},
  1147. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s\x00\xe2\x04g\x02"},
  1148. tx_power=-127,
  1149. rssi=-93,
  1150. )
  1151. result = parse_advertisement_data(
  1152. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1153. )
  1154. assert result == SwitchBotAdvertisement(
  1155. address="aa:bb:cc:dd:ee:ff",
  1156. data={
  1157. "data": {
  1158. "battery": 98,
  1159. "iot": 0,
  1160. "is_light": True,
  1161. "led": 0,
  1162. "light_intensity": 2,
  1163. "motion_detected": False,
  1164. "sense_distance": 0,
  1165. "tested": False,
  1166. },
  1167. "isEncrypted": False,
  1168. "model": "s",
  1169. "modelFriendlyName": "Motion Sensor",
  1170. "modelName": SwitchbotModel.MOTION_SENSOR,
  1171. "rawAdvData": b"s\x00\xe2\x04g\x02",
  1172. },
  1173. device=ble_device,
  1174. rssi=-93,
  1175. active=True,
  1176. )
  1177. def test_motion_with_light_detected():
  1178. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1179. adv_data = generate_advertisement_data(
  1180. manufacturer_data={2409: b"\xc0!\x9a\xe8\xbcIvl\x00,"},
  1181. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"s@\xe2\x00,\x02"},
  1182. tx_power=-127,
  1183. rssi=-84,
  1184. )
  1185. result = parse_advertisement_data(
  1186. ble_device, adv_data, SwitchbotModel.MOTION_SENSOR
  1187. )
  1188. assert result == SwitchBotAdvertisement(
  1189. address="aa:bb:cc:dd:ee:ff",
  1190. data={
  1191. "data": {
  1192. "battery": 98,
  1193. "iot": 0,
  1194. "is_light": True,
  1195. "led": 0,
  1196. "light_intensity": 2,
  1197. "motion_detected": True,
  1198. "sense_distance": 0,
  1199. "tested": False,
  1200. },
  1201. "isEncrypted": False,
  1202. "model": "s",
  1203. "modelFriendlyName": "Motion Sensor",
  1204. "modelName": SwitchbotModel.MOTION_SENSOR,
  1205. "rawAdvData": b"s@\xe2\x00,\x02",
  1206. },
  1207. device=ble_device,
  1208. rssi=-84,
  1209. active=True,
  1210. )
  1211. def test_meter_pro_active() -> None:
  1212. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1213. adv_data = generate_advertisement_data(
  1214. manufacturer_data={2409: b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05"},
  1215. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"4\x00d"},
  1216. rssi=-67,
  1217. )
  1218. result = parse_advertisement_data(ble_device, adv_data)
  1219. assert result == SwitchBotAdvertisement(
  1220. address="aa:bb:cc:dd:ee:ff",
  1221. data={
  1222. "data": {
  1223. "battery": 100,
  1224. "fahrenheit": False,
  1225. "humidity": 44,
  1226. "temp": {"c": 23.8, "f": 74.84},
  1227. "temperature": 23.8,
  1228. },
  1229. "isEncrypted": False,
  1230. "model": "4",
  1231. "modelFriendlyName": "Meter Pro",
  1232. "modelName": SwitchbotModel.METER_PRO,
  1233. "rawAdvData": b"4\x00d",
  1234. },
  1235. device=ble_device,
  1236. rssi=-67,
  1237. active=True,
  1238. )
  1239. def test_meter_pro_passive() -> None:
  1240. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1241. adv_data = generate_advertisement_data(
  1242. manufacturer_data={2409: b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05"},
  1243. rssi=-67,
  1244. )
  1245. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER_PRO)
  1246. assert result == SwitchBotAdvertisement(
  1247. address="aa:bb:cc:dd:ee:ff",
  1248. data={
  1249. "data": {
  1250. "battery": None,
  1251. "fahrenheit": False,
  1252. "humidity": 44,
  1253. "temp": {"c": 23.8, "f": 74.84},
  1254. "temperature": 23.8,
  1255. },
  1256. "isEncrypted": False,
  1257. "model": "4",
  1258. "modelFriendlyName": "Meter Pro",
  1259. "modelName": SwitchbotModel.METER_PRO,
  1260. "rawAdvData": None,
  1261. },
  1262. device=ble_device,
  1263. rssi=-67,
  1264. active=False,
  1265. )
  1266. def test_meter_pro_c_active() -> None:
  1267. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1268. adv_data = generate_advertisement_data(
  1269. manufacturer_data={
  1270. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x02\xd5\x00"
  1271. },
  1272. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"5\x00d"},
  1273. rssi=-67,
  1274. )
  1275. result = parse_advertisement_data(ble_device, adv_data)
  1276. assert result == SwitchBotAdvertisement(
  1277. address="aa:bb:cc:dd:ee:ff",
  1278. data={
  1279. "data": {
  1280. "battery": 100,
  1281. "fahrenheit": True,
  1282. "humidity": 36,
  1283. "temp": {"c": 27.7, "f": 81.86},
  1284. "temperature": 27.7,
  1285. "co2": 725,
  1286. },
  1287. "isEncrypted": False,
  1288. "model": "5",
  1289. "modelFriendlyName": "Meter Pro CO2",
  1290. "modelName": SwitchbotModel.METER_PRO_C,
  1291. "rawAdvData": b"5\x00d",
  1292. },
  1293. device=ble_device,
  1294. rssi=-67,
  1295. active=True,
  1296. )
  1297. def test_meter_pro_c_passive() -> None:
  1298. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1299. adv_data = generate_advertisement_data(
  1300. manufacturer_data={
  1301. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x02\xd5\x00"
  1302. },
  1303. rssi=-67,
  1304. )
  1305. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.METER_PRO_C)
  1306. assert result == SwitchBotAdvertisement(
  1307. address="aa:bb:cc:dd:ee:ff",
  1308. data={
  1309. "data": {
  1310. "battery": None,
  1311. "fahrenheit": True,
  1312. "humidity": 36,
  1313. "temp": {"c": 27.7, "f": 81.86},
  1314. "temperature": 27.7,
  1315. "co2": 725,
  1316. },
  1317. "isEncrypted": False,
  1318. "model": "5",
  1319. "modelFriendlyName": "Meter Pro CO2",
  1320. "modelName": SwitchbotModel.METER_PRO_C,
  1321. "rawAdvData": None,
  1322. },
  1323. device=ble_device,
  1324. rssi=-67,
  1325. active=False,
  1326. )
  1327. def test_meter_pro_c_co2_out_of_range_dropped() -> None:
  1328. """CO2 readings above sensor spec range (9999 ppm) are dropped as spurious."""
  1329. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1330. adv_data = generate_advertisement_data(
  1331. manufacturer_data={
  1332. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x9c\x40\x00"
  1333. },
  1334. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"5\x00d"},
  1335. rssi=-67,
  1336. )
  1337. result = parse_advertisement_data(ble_device, adv_data)
  1338. assert "co2" not in result.data["data"]
  1339. assert result.data["data"]["temperature"] == 27.7
  1340. assert result.data["data"]["humidity"] == 36
  1341. def test_meter_pro_c_co2_boundary_accepted() -> None:
  1342. """CO2 at the upper bound (9999 ppm) is accepted."""
  1343. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1344. adv_data = generate_advertisement_data(
  1345. manufacturer_data={
  1346. 2409: b"\xb0\xe9\xfeT2\x15\xb7\xe4\x07\x9b\xa4\x007\x27\x0f\x00"
  1347. },
  1348. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"5\x00d"},
  1349. rssi=-67,
  1350. )
  1351. result = parse_advertisement_data(ble_device, adv_data)
  1352. assert result.data["data"]["co2"] == 9999
  1353. def test_parse_advertisement_data_keypad():
  1354. """Test parse_advertisement_data for the keypad."""
  1355. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1356. adv_data = generate_advertisement_data(
  1357. manufacturer_data={2409: b"\xeb\x13\x02\xe6#\x0f\x8fd\x00\x00\x00\x00"},
  1358. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"y\x00d"},
  1359. rssi=-67,
  1360. )
  1361. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.KEYPAD)
  1362. assert result == SwitchBotAdvertisement(
  1363. address="aa:bb:cc:dd:ee:ff",
  1364. data={
  1365. "data": {"attempt_state": 143, "battery": 100},
  1366. "isEncrypted": False,
  1367. "model": "y",
  1368. "modelFriendlyName": "Keypad",
  1369. "modelName": SwitchbotModel.KEYPAD,
  1370. "rawAdvData": b"y\x00d",
  1371. },
  1372. device=ble_device,
  1373. rssi=-67,
  1374. active=True,
  1375. )
  1376. def test_leak_active():
  1377. """Test parse_advertisement_data for the leak detector."""
  1378. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1379. adv_data = generate_advertisement_data(
  1380. manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
  1381. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00N"},
  1382. rssi=-72,
  1383. )
  1384. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1385. assert result == SwitchBotAdvertisement(
  1386. address="aa:bb:cc:dd:ee:ff",
  1387. data={
  1388. "data": {
  1389. "leak": False,
  1390. "tampered": False,
  1391. "battery": 78,
  1392. "low_battery": False,
  1393. },
  1394. "isEncrypted": False,
  1395. "model": "&",
  1396. "modelFriendlyName": "Leak Detector",
  1397. "modelName": SwitchbotModel.LEAK,
  1398. "rawAdvData": b"&\x00N",
  1399. },
  1400. device=ble_device,
  1401. rssi=-72,
  1402. active=True,
  1403. )
  1404. def test_leak_passive():
  1405. """Test parse_advertisement_data for the leak detector."""
  1406. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1407. adv_data = generate_advertisement_data(
  1408. manufacturer_data={2409: b"\xc4407Lz\x18N\x98g^\x94Q<\x05\x00\x00\x00\x00"},
  1409. rssi=-72,
  1410. )
  1411. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1412. assert result == SwitchBotAdvertisement(
  1413. address="aa:bb:cc:dd:ee:ff",
  1414. data={
  1415. "data": {
  1416. "leak": False,
  1417. "tampered": False,
  1418. "battery": 78,
  1419. "low_battery": False,
  1420. },
  1421. "isEncrypted": False,
  1422. "model": "&",
  1423. "modelFriendlyName": "Leak Detector",
  1424. "modelName": SwitchbotModel.LEAK,
  1425. "rawAdvData": None,
  1426. },
  1427. device=ble_device,
  1428. rssi=-72,
  1429. active=False,
  1430. )
  1431. def test_leak_no_leak_detected():
  1432. """Test parse_advertisement_data for the leak detector."""
  1433. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1434. adv_data = generate_advertisement_data(
  1435. manufacturer_data={
  1436. 2409: b"\xc4407LzJd\x98ga\xc4\n<\x05\x00\x00\x00\x00"
  1437. }, # no leak, batt
  1438. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1439. rssi=-73,
  1440. )
  1441. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1442. assert result == SwitchBotAdvertisement(
  1443. address="aa:bb:cc:dd:ee:ff",
  1444. data={
  1445. "data": {
  1446. "leak": False,
  1447. "tampered": False,
  1448. "battery": 100,
  1449. "low_battery": False,
  1450. },
  1451. "isEncrypted": False,
  1452. "model": "&",
  1453. "modelFriendlyName": "Leak Detector",
  1454. "modelName": SwitchbotModel.LEAK,
  1455. "rawAdvData": b"&\x00d",
  1456. },
  1457. device=ble_device,
  1458. rssi=-73,
  1459. active=True,
  1460. )
  1461. def test_leak_leak_detected():
  1462. """Test parse_advertisement_data for the leak detector."""
  1463. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1464. adv_data = generate_advertisement_data(
  1465. manufacturer_data={
  1466. 2409: b"\xc4407LzGd\xf9ga\xc4\x08<\x05\x00\x00\x00\x00"
  1467. }, # leak, batt
  1468. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1469. rssi=-73,
  1470. )
  1471. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1472. assert result == SwitchBotAdvertisement(
  1473. address="aa:bb:cc:dd:ee:ff",
  1474. data={
  1475. "data": {
  1476. "leak": True,
  1477. "tampered": False,
  1478. "battery": 100,
  1479. "low_battery": False,
  1480. },
  1481. "isEncrypted": False,
  1482. "model": "&",
  1483. "modelFriendlyName": "Leak Detector",
  1484. "modelName": SwitchbotModel.LEAK,
  1485. "rawAdvData": b"&\x00d",
  1486. },
  1487. device=ble_device,
  1488. rssi=-73,
  1489. active=True,
  1490. )
  1491. def test_leak_low_battery():
  1492. """Test parse_advertisement_data for the leak detector."""
  1493. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1494. adv_data = generate_advertisement_data(
  1495. manufacturer_data={
  1496. 2409: b"\xc4407Lz\x02\t\x98\x00\x00\x00\x00<\x05\x00\x00\x00\x00"
  1497. }, # no leak, low battery
  1498. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\x00d"},
  1499. rssi=-73,
  1500. )
  1501. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1502. assert result == SwitchBotAdvertisement(
  1503. address="aa:bb:cc:dd:ee:ff",
  1504. data={
  1505. "data": {
  1506. "leak": False,
  1507. "tampered": False,
  1508. "battery": 9,
  1509. "low_battery": False,
  1510. },
  1511. "isEncrypted": False,
  1512. "model": "&",
  1513. "modelFriendlyName": "Leak Detector",
  1514. "modelName": SwitchbotModel.LEAK,
  1515. "rawAdvData": b"&\x00d",
  1516. },
  1517. device=ble_device,
  1518. rssi=-73,
  1519. active=True,
  1520. )
  1521. def test_leak_real_data_from_ha():
  1522. """Test parse_advertisement_data for the leak detector."""
  1523. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "Any")
  1524. adv_data = generate_advertisement_data(
  1525. manufacturer_data={
  1526. 2409: b"\\xd6407D1\\x02V\\x90\\x00\\x00\\x00\\x00\\x1e\\x05\\x00\\x00\\x00\\x00"
  1527. }, # no leak, low battery
  1528. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"&\\x00V"},
  1529. rssi=-73,
  1530. )
  1531. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.LEAK)
  1532. assert result == SwitchBotAdvertisement(
  1533. address="aa:bb:cc:dd:ee:ff",
  1534. data={
  1535. "data": {
  1536. "leak": True,
  1537. "tampered": False,
  1538. "battery": 68,
  1539. "low_battery": False,
  1540. },
  1541. "isEncrypted": False,
  1542. "model": "&",
  1543. "modelFriendlyName": "Leak Detector",
  1544. "modelName": SwitchbotModel.LEAK,
  1545. "rawAdvData": b"&\\x00V",
  1546. },
  1547. device=ble_device,
  1548. rssi=-73,
  1549. active=True,
  1550. )
  1551. def test_remote_active() -> None:
  1552. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1553. adv_data = generate_advertisement_data(
  1554. manufacturer_data={89: b"\xaa\xbb\xcc\xdd\xee\xff"},
  1555. service_data={"00000d00-0000-1000-8000-00805f9b34fb": b"b V\x00"},
  1556. service_uuids=["cba20d00-224d-11e6-9fb8-0002a5d5c51b"],
  1557. rssi=-95,
  1558. )
  1559. result = parse_advertisement_data(ble_device, adv_data)
  1560. assert result == SwitchBotAdvertisement(
  1561. address="aa:bb:cc:dd:ee:ff",
  1562. data={
  1563. "data": {
  1564. "battery": 86,
  1565. },
  1566. "isEncrypted": False,
  1567. "model": "b",
  1568. "modelFriendlyName": "Remote",
  1569. "modelName": SwitchbotModel.REMOTE,
  1570. "rawAdvData": b"b V\x00",
  1571. },
  1572. device=ble_device,
  1573. rssi=-95,
  1574. active=True,
  1575. )
  1576. def test_remote_passive() -> None:
  1577. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1578. adv_data = generate_advertisement_data(
  1579. manufacturer_data={89: b"\xaa\xbb\xcc\xdd\xee\xff"},
  1580. rssi=-97,
  1581. )
  1582. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.REMOTE)
  1583. assert result == SwitchBotAdvertisement(
  1584. address="aa:bb:cc:dd:ee:ff",
  1585. data={
  1586. "data": {
  1587. "battery": None,
  1588. },
  1589. "isEncrypted": False,
  1590. "model": "b",
  1591. "modelFriendlyName": "Remote",
  1592. "modelName": SwitchbotModel.REMOTE,
  1593. "rawAdvData": None,
  1594. },
  1595. device=ble_device,
  1596. rssi=-97,
  1597. active=False,
  1598. )
  1599. def test_parse_advertisement_data_hubmini_matter():
  1600. """Test parse_advertisement_data for the HubMini Matter."""
  1601. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1602. adv_data = generate_advertisement_data(
  1603. manufacturer_data={
  1604. 2409: b"\xe6\xa1\xcd\x1f[e\x00\x00\x00\x00\x00\x00\x14\x01\x985\x00"
  1605. },
  1606. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"%\x00"},
  1607. rssi=-67,
  1608. )
  1609. result = parse_advertisement_data(
  1610. ble_device, adv_data, SwitchbotModel.HUBMINI_MATTER
  1611. )
  1612. assert result == SwitchBotAdvertisement(
  1613. address="aa:bb:cc:dd:ee:ff",
  1614. data={
  1615. "data": {
  1616. "fahrenheit": False,
  1617. "humidity": 53,
  1618. "temp": {"c": 24.1, "f": 75.38},
  1619. "temperature": 24.1,
  1620. },
  1621. "isEncrypted": False,
  1622. "model": "%",
  1623. "modelFriendlyName": "HubMini Matter",
  1624. "modelName": SwitchbotModel.HUBMINI_MATTER,
  1625. "rawAdvData": b"%\x00",
  1626. },
  1627. device=ble_device,
  1628. rssi=-67,
  1629. active=True,
  1630. )
  1631. def test_parse_advertisement_data_roller_shade():
  1632. """Test parse_advertisement_data for roller shade."""
  1633. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1634. adv_data = generate_advertisement_data(
  1635. manufacturer_data={2409: b"\xb0\xe9\xfeT\x90\x1b,\x08\x9f\x11\x04'\x00"},
  1636. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b",\x00'\x9f\x11\x04"},
  1637. rssi=-80,
  1638. )
  1639. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.ROLLER_SHADE)
  1640. assert result == SwitchBotAdvertisement(
  1641. address="aa:bb:cc:dd:ee:ff",
  1642. data={
  1643. "rawAdvData": b",\x00'\x9f\x11\x04",
  1644. "data": {
  1645. "battery": 39,
  1646. "calibration": True,
  1647. "deviceChain": 1,
  1648. "inMotion": False,
  1649. "lightLevel": 1,
  1650. "position": 69,
  1651. "sequence_number": 44,
  1652. },
  1653. "isEncrypted": False,
  1654. "model": ",",
  1655. "modelFriendlyName": "Roller Shade",
  1656. "modelName": SwitchbotModel.ROLLER_SHADE,
  1657. },
  1658. device=ble_device,
  1659. rssi=-80,
  1660. active=True,
  1661. )
  1662. def test_hubmini_matter_passive() -> None:
  1663. """Test parsing hubmini matter with passive data."""
  1664. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1665. adv_data = generate_advertisement_data(
  1666. manufacturer_data={
  1667. 2409: b"\xe6\xa1\xcd\x1f[e\x00\x00\x00\x00\x00\x00\x14\x01\x985\x00"
  1668. },
  1669. rssi=-97,
  1670. )
  1671. result = parse_advertisement_data(
  1672. ble_device, adv_data, SwitchbotModel.HUBMINI_MATTER
  1673. )
  1674. assert result == SwitchBotAdvertisement(
  1675. address="aa:bb:cc:dd:ee:ff",
  1676. data={
  1677. "data": {
  1678. "fahrenheit": False,
  1679. "humidity": 53,
  1680. "temp": {"c": 24.1, "f": 75.38},
  1681. "temperature": 24.1,
  1682. },
  1683. "isEncrypted": False,
  1684. "model": "%",
  1685. "modelFriendlyName": "HubMini Matter",
  1686. "modelName": SwitchbotModel.HUBMINI_MATTER,
  1687. "rawAdvData": None,
  1688. },
  1689. device=ble_device,
  1690. rssi=-97,
  1691. active=False,
  1692. )
  1693. def test_roller_shade_passive() -> None:
  1694. """Test parsing roller_shade with passive data."""
  1695. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1696. adv_data = generate_advertisement_data(
  1697. manufacturer_data={2409: b"\xb0\xe9\xfeT\x90\x1b,\x08\x9f\x11\x04'\x00"},
  1698. rssi=-97,
  1699. )
  1700. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.ROLLER_SHADE)
  1701. assert result == SwitchBotAdvertisement(
  1702. address="aa:bb:cc:dd:ee:ff",
  1703. data={
  1704. "rawAdvData": None,
  1705. "data": {
  1706. "battery": None,
  1707. "calibration": True,
  1708. "deviceChain": 1,
  1709. "inMotion": False,
  1710. "lightLevel": 1,
  1711. "position": 69,
  1712. "sequence_number": 44,
  1713. },
  1714. "isEncrypted": False,
  1715. "model": ",",
  1716. "modelFriendlyName": "Roller Shade",
  1717. "modelName": SwitchbotModel.ROLLER_SHADE,
  1718. },
  1719. device=ble_device,
  1720. rssi=-97,
  1721. active=False,
  1722. )
  1723. def test_circulator_fan_active() -> None:
  1724. """Test parsing circulator fan with active data."""
  1725. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1726. adv_data = generate_advertisement_data(
  1727. manufacturer_data={2409: b"\xb0\xe9\xfeXY\xa8~LR9"},
  1728. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"~\x00R"},
  1729. rssi=-97,
  1730. )
  1731. result = parse_advertisement_data(
  1732. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  1733. )
  1734. assert result == SwitchBotAdvertisement(
  1735. address="aa:bb:cc:dd:ee:ff",
  1736. data={
  1737. "rawAdvData": b"~\x00R",
  1738. "data": {
  1739. "sequence_number": 126,
  1740. "isOn": False,
  1741. "mode": "baby",
  1742. "nightLight": 3,
  1743. "oscillating": False,
  1744. "oscillating_horizontal": False,
  1745. "oscillating_vertical": False,
  1746. "battery": 82,
  1747. "speed": 57,
  1748. },
  1749. "isEncrypted": False,
  1750. "model": "~",
  1751. "modelFriendlyName": "Circulator Fan",
  1752. "modelName": SwitchbotModel.CIRCULATOR_FAN,
  1753. },
  1754. device=ble_device,
  1755. rssi=-97,
  1756. active=True,
  1757. )
  1758. def test_circulator_fan_passive() -> None:
  1759. """Test parsing circulator fan with passive data."""
  1760. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1761. adv_data = generate_advertisement_data(
  1762. manufacturer_data={2409: b"\xb0\xe9\xfeXY\xa8~LR9"},
  1763. rssi=-97,
  1764. )
  1765. result = parse_advertisement_data(
  1766. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  1767. )
  1768. assert result == SwitchBotAdvertisement(
  1769. address="aa:bb:cc:dd:ee:ff",
  1770. data={
  1771. "rawAdvData": None,
  1772. "data": {
  1773. "sequence_number": 126,
  1774. "isOn": False,
  1775. "mode": "baby",
  1776. "nightLight": 3,
  1777. "oscillating": False,
  1778. "oscillating_horizontal": False,
  1779. "oscillating_vertical": False,
  1780. "battery": 82,
  1781. "speed": 57,
  1782. },
  1783. "isEncrypted": False,
  1784. "model": "~",
  1785. "modelFriendlyName": "Circulator Fan",
  1786. "modelName": SwitchbotModel.CIRCULATOR_FAN,
  1787. },
  1788. device=ble_device,
  1789. rssi=-97,
  1790. active=False,
  1791. )
  1792. def test_circulator_fan_with_empty_data() -> None:
  1793. """Test parsing circulator fan with empty data."""
  1794. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1795. adv_data = generate_advertisement_data(
  1796. manufacturer_data={2409: None},
  1797. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"~\x00R"},
  1798. rssi=-97,
  1799. )
  1800. result = parse_advertisement_data(
  1801. ble_device, adv_data, SwitchbotModel.CIRCULATOR_FAN
  1802. )
  1803. assert result == SwitchBotAdvertisement(
  1804. address="aa:bb:cc:dd:ee:ff",
  1805. data={
  1806. "rawAdvData": b"~\x00R",
  1807. "data": {},
  1808. "isEncrypted": False,
  1809. "model": "~",
  1810. },
  1811. device=ble_device,
  1812. rssi=-97,
  1813. active=True,
  1814. )
  1815. def test_standing_fan_custom_natural_with_per_axis_oscillation() -> None:
  1816. """
  1817. Standing Fan in CUSTOM_NATURAL (mode 5) + both axes oscillating.
  1818. Uses a realistic 7-byte service_data ending with the 4-byte
  1819. STANDING_FAN suffix so auto-detection (no explicit model arg)
  1820. resolves the model via `_find_model_from_service_data_suffix`.
  1821. """
  1822. # Mode field is bits [6:4] of device_data[1].
  1823. # Mode 5 (custom_natural) = 0b0101 0000, plus isOn=0b1000_0000, plus
  1824. # horizontal=0b10, vertical=0b01 → 0b1101 0011 = 0xD3.
  1825. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1826. adv_data = generate_advertisement_data(
  1827. manufacturer_data={2409: b"\xb0\xe9\xfe\x01\x02\x03~\xd3R9"},
  1828. service_data={
  1829. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x00\x00\x11\x07\x60"
  1830. },
  1831. rssi=-97,
  1832. )
  1833. # No explicit model — exercises suffix-based auto-detection.
  1834. result = parse_advertisement_data(ble_device, adv_data)
  1835. assert result is not None
  1836. assert result.data["modelName"] == SwitchbotModel.STANDING_FAN
  1837. data = result.data["data"]
  1838. assert data["mode"] == "custom_natural"
  1839. assert data["oscillating"] is True
  1840. assert data["oscillating_horizontal"] is True
  1841. assert data["oscillating_vertical"] is True
  1842. def test_k20_active() -> None:
  1843. """Test parsing k20 with active data."""
  1844. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1845. adv_data = generate_advertisement_data(
  1846. manufacturer_data={2409: b"\xb0\xe9\xfe\x01\xf3\x8f'\x01\x11S\x00\x10d\x0f"},
  1847. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b".\x00d"},
  1848. rssi=-97,
  1849. )
  1850. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K20_VACUUM)
  1851. assert result == SwitchBotAdvertisement(
  1852. address="aa:bb:cc:dd:ee:ff",
  1853. data={
  1854. "rawAdvData": b".\x00d",
  1855. "data": {
  1856. "sequence_number": 39,
  1857. "soc_version": "1.1.083",
  1858. "step": 0,
  1859. "mqtt_connected": True,
  1860. "battery": 100,
  1861. "work_status": 15,
  1862. },
  1863. "isEncrypted": False,
  1864. "model": ".",
  1865. "modelFriendlyName": "K20 Vacuum",
  1866. "modelName": SwitchbotModel.K20_VACUUM,
  1867. },
  1868. device=ble_device,
  1869. rssi=-97,
  1870. active=True,
  1871. )
  1872. def test_k20_passive() -> None:
  1873. """Test parsing k20 with passive data."""
  1874. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1875. adv_data = generate_advertisement_data(
  1876. manufacturer_data={2409: b"\xb0\xe9\xfe\x01\xf3\x8f'\x01\x11S\x00\x10d\x0f"},
  1877. rssi=-97,
  1878. )
  1879. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K20_VACUUM)
  1880. assert result == SwitchBotAdvertisement(
  1881. address="aa:bb:cc:dd:ee:ff",
  1882. data={
  1883. "rawAdvData": None,
  1884. "data": {
  1885. "sequence_number": 39,
  1886. "soc_version": "1.1.083",
  1887. "step": 0,
  1888. "mqtt_connected": True,
  1889. "battery": 100,
  1890. "work_status": 15,
  1891. },
  1892. "isEncrypted": False,
  1893. "model": ".",
  1894. "modelFriendlyName": "K20 Vacuum",
  1895. "modelName": SwitchbotModel.K20_VACUUM,
  1896. },
  1897. device=ble_device,
  1898. rssi=-97,
  1899. active=False,
  1900. )
  1901. def test_k20_with_empty_data() -> None:
  1902. """Test parsing k20 with empty data."""
  1903. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1904. adv_data = generate_advertisement_data(
  1905. manufacturer_data={2409: None},
  1906. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b".\x00d"},
  1907. rssi=-97,
  1908. )
  1909. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K20_VACUUM)
  1910. assert result == SwitchBotAdvertisement(
  1911. address="aa:bb:cc:dd:ee:ff",
  1912. data={
  1913. "rawAdvData": b".\x00d",
  1914. "data": {},
  1915. "isEncrypted": False,
  1916. "model": ".",
  1917. },
  1918. device=ble_device,
  1919. rssi=-97,
  1920. active=True,
  1921. )
  1922. def test_k10_pro_active() -> None:
  1923. """Test parsing k10 pro with active data."""
  1924. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1925. adv_data = generate_advertisement_data(
  1926. manufacturer_data={2409: b"\xb0\xe9\xfeP\x8d\x8d\x02 d"},
  1927. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"(\x00"},
  1928. rssi=-97,
  1929. )
  1930. result = parse_advertisement_data(
  1931. ble_device, adv_data, SwitchbotModel.K10_PRO_VACUUM
  1932. )
  1933. assert result == SwitchBotAdvertisement(
  1934. address="aa:bb:cc:dd:ee:ff",
  1935. data={
  1936. "rawAdvData": b"(\x00",
  1937. "data": {
  1938. "sequence_number": 2,
  1939. "dusbin_connected": False,
  1940. "dustbin_bound": False,
  1941. "network_connected": True,
  1942. "battery": 100,
  1943. "work_status": 0,
  1944. },
  1945. "isEncrypted": False,
  1946. "model": "(",
  1947. "modelFriendlyName": "K10+ Pro Vacuum",
  1948. "modelName": SwitchbotModel.K10_PRO_VACUUM,
  1949. },
  1950. device=ble_device,
  1951. rssi=-97,
  1952. active=True,
  1953. )
  1954. def test_k10_pro_passive() -> None:
  1955. """Test parsing k10 pro with passive data."""
  1956. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1957. adv_data = generate_advertisement_data(
  1958. manufacturer_data={2409: b"\xb0\xe9\xfeP\x8d\x8d\x02 d"},
  1959. rssi=-97,
  1960. )
  1961. result = parse_advertisement_data(
  1962. ble_device, adv_data, SwitchbotModel.K10_PRO_VACUUM
  1963. )
  1964. assert result == SwitchBotAdvertisement(
  1965. address="aa:bb:cc:dd:ee:ff",
  1966. data={
  1967. "rawAdvData": None,
  1968. "data": {
  1969. "sequence_number": 2,
  1970. "dusbin_connected": False,
  1971. "dustbin_bound": False,
  1972. "network_connected": True,
  1973. "battery": 100,
  1974. "work_status": 0,
  1975. },
  1976. "isEncrypted": False,
  1977. "model": "(",
  1978. "modelFriendlyName": "K10+ Pro Vacuum",
  1979. "modelName": SwitchbotModel.K10_PRO_VACUUM,
  1980. },
  1981. device=ble_device,
  1982. rssi=-97,
  1983. active=False,
  1984. )
  1985. def test_k10_pro_with_empty_data() -> None:
  1986. """Test parsing k10 pro with empty data."""
  1987. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  1988. adv_data = generate_advertisement_data(
  1989. manufacturer_data={2409: None},
  1990. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"(\x00"},
  1991. rssi=-97,
  1992. )
  1993. result = parse_advertisement_data(
  1994. ble_device, adv_data, SwitchbotModel.K10_PRO_VACUUM
  1995. )
  1996. assert result == SwitchBotAdvertisement(
  1997. address="aa:bb:cc:dd:ee:ff",
  1998. data={
  1999. "rawAdvData": b"(\x00",
  2000. "data": {},
  2001. "isEncrypted": False,
  2002. "model": "(",
  2003. },
  2004. device=ble_device,
  2005. rssi=-97,
  2006. active=True,
  2007. )
  2008. def test_k10_active() -> None:
  2009. """Test parsing k10+ with active data."""
  2010. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2011. adv_data = generate_advertisement_data(
  2012. manufacturer_data={2409: b"\xca8\x06\xa9_\xf1\x02 d"},
  2013. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"}\x00"},
  2014. rssi=-97,
  2015. )
  2016. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K10_VACUUM)
  2017. assert result == SwitchBotAdvertisement(
  2018. address="aa:bb:cc:dd:ee:ff",
  2019. data={
  2020. "rawAdvData": b"}\x00",
  2021. "data": {
  2022. "sequence_number": 2,
  2023. "dusbin_connected": False,
  2024. "dustbin_bound": False,
  2025. "network_connected": True,
  2026. "battery": 100,
  2027. "work_status": 0,
  2028. },
  2029. "isEncrypted": False,
  2030. "model": "}",
  2031. "modelFriendlyName": "K10+ Vacuum",
  2032. "modelName": SwitchbotModel.K10_VACUUM,
  2033. },
  2034. device=ble_device,
  2035. rssi=-97,
  2036. active=True,
  2037. )
  2038. def test_k10_passive() -> None:
  2039. """Test parsing k10+ with passive data."""
  2040. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2041. adv_data = generate_advertisement_data(
  2042. manufacturer_data={2409: b"\xca8\x06\xa9_\xf1\x02 d"},
  2043. rssi=-97,
  2044. )
  2045. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K10_VACUUM)
  2046. assert result == SwitchBotAdvertisement(
  2047. address="aa:bb:cc:dd:ee:ff",
  2048. data={
  2049. "rawAdvData": None,
  2050. "data": {
  2051. "sequence_number": 2,
  2052. "dusbin_connected": False,
  2053. "dustbin_bound": False,
  2054. "network_connected": True,
  2055. "battery": 100,
  2056. "work_status": 0,
  2057. },
  2058. "isEncrypted": False,
  2059. "model": "}",
  2060. "modelFriendlyName": "K10+ Vacuum",
  2061. "modelName": SwitchbotModel.K10_VACUUM,
  2062. },
  2063. device=ble_device,
  2064. rssi=-97,
  2065. active=False,
  2066. )
  2067. def test_k10_with_empty_data() -> None:
  2068. """Test parsing k10+ with empty data."""
  2069. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2070. adv_data = generate_advertisement_data(
  2071. manufacturer_data={2409: None},
  2072. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"}\x00"},
  2073. rssi=-97,
  2074. )
  2075. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.K10_VACUUM)
  2076. assert result == SwitchBotAdvertisement(
  2077. address="aa:bb:cc:dd:ee:ff",
  2078. data={
  2079. "rawAdvData": b"}\x00",
  2080. "data": {},
  2081. "isEncrypted": False,
  2082. "model": "}",
  2083. },
  2084. device=ble_device,
  2085. rssi=-97,
  2086. active=True,
  2087. )
  2088. def test_k10_pro_combo_active() -> None:
  2089. """Test parsing k10+ pro combo with active data."""
  2090. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2091. adv_data = generate_advertisement_data(
  2092. manufacturer_data={
  2093. 2409: b"\xb0\xe9\xfe\x01\xf4\x1d\x0b\x01\x01\xb1\x03\x118\x01"
  2094. },
  2095. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"3\x00\x00"},
  2096. rssi=-97,
  2097. )
  2098. result = parse_advertisement_data(
  2099. ble_device, adv_data, SwitchbotModel.K10_PRO_COMBO_VACUUM
  2100. )
  2101. assert result == SwitchBotAdvertisement(
  2102. address="aa:bb:cc:dd:ee:ff",
  2103. data={
  2104. "rawAdvData": b"3\x00\x00",
  2105. "data": {
  2106. "sequence_number": 11,
  2107. "soc_version": "1.0.945",
  2108. "step": 1,
  2109. "mqtt_connected": True,
  2110. "battery": 56,
  2111. "work_status": 1,
  2112. },
  2113. "isEncrypted": False,
  2114. "model": "3",
  2115. "modelFriendlyName": "K10+ Pro Combo Vacuum",
  2116. "modelName": SwitchbotModel.K10_PRO_COMBO_VACUUM,
  2117. },
  2118. device=ble_device,
  2119. rssi=-97,
  2120. active=True,
  2121. )
  2122. def test_k10_pro_combo_passive() -> None:
  2123. """Test parsing k10+ pro combo with passive data."""
  2124. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2125. adv_data = generate_advertisement_data(
  2126. manufacturer_data={
  2127. 2409: b"\xb0\xe9\xfe\x01\xf4\x1d\x0b\x01\x01\xb1\x03\x118\x01"
  2128. },
  2129. rssi=-97,
  2130. )
  2131. result = parse_advertisement_data(
  2132. ble_device, adv_data, SwitchbotModel.K10_PRO_COMBO_VACUUM
  2133. )
  2134. assert result == SwitchBotAdvertisement(
  2135. address="aa:bb:cc:dd:ee:ff",
  2136. data={
  2137. "rawAdvData": None,
  2138. "data": {
  2139. "sequence_number": 11,
  2140. "soc_version": "1.0.945",
  2141. "step": 1,
  2142. "mqtt_connected": True,
  2143. "battery": 56,
  2144. "work_status": 1,
  2145. },
  2146. "isEncrypted": False,
  2147. "model": "3",
  2148. "modelFriendlyName": "K10+ Pro Combo Vacuum",
  2149. "modelName": SwitchbotModel.K10_PRO_COMBO_VACUUM,
  2150. },
  2151. device=ble_device,
  2152. rssi=-97,
  2153. active=False,
  2154. )
  2155. def test_k10_pro_combo_with_empty_data() -> None:
  2156. """Test parsing k10+ pro combo with empty data."""
  2157. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2158. adv_data = generate_advertisement_data(
  2159. manufacturer_data={2409: None},
  2160. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"3\x00\x00"},
  2161. rssi=-97,
  2162. )
  2163. result = parse_advertisement_data(
  2164. ble_device, adv_data, SwitchbotModel.K10_PRO_COMBO_VACUUM
  2165. )
  2166. assert result == SwitchBotAdvertisement(
  2167. address="aa:bb:cc:dd:ee:ff",
  2168. data={
  2169. "rawAdvData": b"3\x00\x00",
  2170. "data": {},
  2171. "isEncrypted": False,
  2172. "model": "3",
  2173. },
  2174. device=ble_device,
  2175. rssi=-97,
  2176. active=True,
  2177. )
  2178. def test_s10_active() -> None:
  2179. """Test parsing s10 with active 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\xfe\x00\x08|\n\x01\x11\x05\x00\x10M\x02"},
  2183. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"z\x00\x00"},
  2184. rssi=-97,
  2185. )
  2186. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.S10_VACUUM)
  2187. assert result == SwitchBotAdvertisement(
  2188. address="aa:bb:cc:dd:ee:ff",
  2189. data={
  2190. "rawAdvData": b"z\x00\x00",
  2191. "data": {
  2192. "sequence_number": 10,
  2193. "soc_version": "1.1.005",
  2194. "step": 0,
  2195. "mqtt_connected": True,
  2196. "battery": 77,
  2197. "work_status": 2,
  2198. },
  2199. "isEncrypted": False,
  2200. "model": "z",
  2201. "modelFriendlyName": "S10 Vacuum",
  2202. "modelName": SwitchbotModel.S10_VACUUM,
  2203. },
  2204. device=ble_device,
  2205. rssi=-97,
  2206. active=True,
  2207. )
  2208. def test_s10_passive() -> None:
  2209. """Test parsing s10 with passive data."""
  2210. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2211. adv_data = generate_advertisement_data(
  2212. manufacturer_data={2409: b"\xb0\xe9\xfe\x00\x08|\n\x01\x11\x05\x00\x10M\x02"},
  2213. rssi=-97,
  2214. )
  2215. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.S10_VACUUM)
  2216. assert result == SwitchBotAdvertisement(
  2217. address="aa:bb:cc:dd:ee:ff",
  2218. data={
  2219. "rawAdvData": None,
  2220. "data": {
  2221. "sequence_number": 10,
  2222. "soc_version": "1.1.005",
  2223. "step": 0,
  2224. "mqtt_connected": True,
  2225. "battery": 77,
  2226. "work_status": 2,
  2227. },
  2228. "isEncrypted": False,
  2229. "model": "z",
  2230. "modelFriendlyName": "S10 Vacuum",
  2231. "modelName": SwitchbotModel.S10_VACUUM,
  2232. },
  2233. device=ble_device,
  2234. rssi=-97,
  2235. active=False,
  2236. )
  2237. def test_s10_with_empty_data() -> None:
  2238. """Test parsing s10 with empty data."""
  2239. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2240. adv_data = generate_advertisement_data(
  2241. manufacturer_data={2409: None},
  2242. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"z\x00\x00"},
  2243. rssi=-97,
  2244. )
  2245. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.S10_VACUUM)
  2246. assert result == SwitchBotAdvertisement(
  2247. address="aa:bb:cc:dd:ee:ff",
  2248. data={
  2249. "rawAdvData": b"z\x00\x00",
  2250. "data": {},
  2251. "isEncrypted": False,
  2252. "model": "z",
  2253. },
  2254. device=ble_device,
  2255. rssi=-97,
  2256. active=True,
  2257. )
  2258. @pytest.mark.parametrize(
  2259. "test_case",
  2260. [
  2261. AdvTestCase(
  2262. b"\xf0\x9e\x9e\x96j\xd6\xa1\x81\x88\xe4\x00\x01\x95\x00\x00",
  2263. b"7\x00\x00\x95-\x00",
  2264. {
  2265. "isOn": True,
  2266. "mode": "level_3",
  2267. "isAqiValid": False,
  2268. "child_lock": False,
  2269. "speed": 100,
  2270. "aqi_level": "excellent",
  2271. "filter element working time": 405,
  2272. "err_code": 0,
  2273. "sequence_number": 161,
  2274. },
  2275. "7",
  2276. "Air Purifier Table US",
  2277. SwitchbotModel.AIR_PURIFIER_TABLE_US,
  2278. ),
  2279. AdvTestCase(
  2280. b'\xcc\x8d\xa2\xa7\x92>\t"\x80\x000\x00\x0f\x00\x00',
  2281. b"*\x00\x00\x15\x04\x00",
  2282. {
  2283. "isOn": False,
  2284. "mode": "auto",
  2285. "isAqiValid": False,
  2286. "child_lock": False,
  2287. "speed": 0,
  2288. "aqi_level": "excellent",
  2289. "filter element working time": 15,
  2290. "err_code": 0,
  2291. "sequence_number": 9,
  2292. },
  2293. "*",
  2294. "Air Purifier US",
  2295. SwitchbotModel.AIR_PURIFIER_US,
  2296. ),
  2297. AdvTestCase(
  2298. b"\xcc\x8d\xa2\xa7\xe4\xa6\x0b\x83\x88d\x00\xea`\x00\x00",
  2299. b"+\x00\x00\x15\x04\x00",
  2300. {
  2301. "isOn": True,
  2302. "mode": "sleep",
  2303. "isAqiValid": False,
  2304. "child_lock": False,
  2305. "speed": 100,
  2306. "aqi_level": "excellent",
  2307. "filter element working time": 60000,
  2308. "err_code": 0,
  2309. "sequence_number": 11,
  2310. },
  2311. "+",
  2312. "Air Purifier JP",
  2313. SwitchbotModel.AIR_PURIFIER_JP,
  2314. ),
  2315. AdvTestCase(
  2316. b"\xcc\x8d\xa2\xa7\xc1\xae\x9b\x81\x8c\xb2\x00\x01\x94\x00\x00",
  2317. b"8\x00\x00\x95-\x00",
  2318. {
  2319. "isOn": True,
  2320. "mode": "level_2",
  2321. "isAqiValid": True,
  2322. "child_lock": False,
  2323. "speed": 50,
  2324. "aqi_level": "excellent",
  2325. "filter element working time": 404,
  2326. "err_code": 0,
  2327. "sequence_number": 155,
  2328. },
  2329. "8",
  2330. "Air Purifier Table JP",
  2331. SwitchbotModel.AIR_PURIFIER_TABLE_JP,
  2332. ),
  2333. AdvTestCase(
  2334. b"\xcc\x8d\xa2\xa7\xc1\xae\x9e\xa1\x8c\x800\x01\x95\x00\x00",
  2335. b"8\x00\x00\x95-\x00",
  2336. {
  2337. "isOn": True,
  2338. "mode": "level_1",
  2339. "isAqiValid": True,
  2340. "child_lock": False,
  2341. "speed": 0,
  2342. "aqi_level": "excellent",
  2343. "filter element working time": 405,
  2344. "err_code": 0,
  2345. "sequence_number": 158,
  2346. },
  2347. "8",
  2348. "Air Purifier Table JP",
  2349. SwitchbotModel.AIR_PURIFIER_TABLE_JP,
  2350. ),
  2351. AdvTestCase(
  2352. b"\xcc\x8d\xa2\xa7\xc1\xae\x9e\x05\x8c\x800\x01\x95\x00\x00",
  2353. b"8\x00\x00\x95-\x00",
  2354. {
  2355. "isOn": False,
  2356. "mode": None,
  2357. "isAqiValid": True,
  2358. "child_lock": False,
  2359. "speed": 0,
  2360. "aqi_level": "excellent",
  2361. "filter element working time": 405,
  2362. "err_code": 0,
  2363. "sequence_number": 158,
  2364. },
  2365. "8",
  2366. "Air Purifier Table JP",
  2367. SwitchbotModel.AIR_PURIFIER_TABLE_JP,
  2368. ),
  2369. ],
  2370. )
  2371. def test_air_purifier_active(test_case: AdvTestCase) -> None:
  2372. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2373. adv_data = generate_advertisement_data(
  2374. manufacturer_data={2409: test_case.manufacturer_data},
  2375. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  2376. rssi=-97,
  2377. )
  2378. result = parse_advertisement_data(ble_device, adv_data)
  2379. assert result == SwitchBotAdvertisement(
  2380. address="aa:bb:cc:dd:ee:ff",
  2381. data={
  2382. "rawAdvData": test_case.service_data,
  2383. "data": test_case.data,
  2384. "isEncrypted": False,
  2385. "model": test_case.model,
  2386. "modelFriendlyName": test_case.modelFriendlyName,
  2387. "modelName": test_case.modelName,
  2388. },
  2389. device=ble_device,
  2390. rssi=-97,
  2391. active=True,
  2392. )
  2393. def test_air_purifier_passive() -> None:
  2394. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2395. adv_data = generate_advertisement_data(
  2396. manufacturer_data={
  2397. 2409: b"\xf0\x9e\x9e\x96j\xd6\xa1\x81\x88\xe4\x00\x01\x95\x00\x00"
  2398. },
  2399. rssi=-97,
  2400. )
  2401. result = parse_advertisement_data(
  2402. ble_device, adv_data, SwitchbotModel.AIR_PURIFIER_US
  2403. )
  2404. assert result == SwitchBotAdvertisement(
  2405. address="aa:bb:cc:dd:ee:ff",
  2406. data={
  2407. "rawAdvData": None,
  2408. "data": {
  2409. "isOn": True,
  2410. "mode": "level_3",
  2411. "isAqiValid": False,
  2412. "child_lock": False,
  2413. "speed": 100,
  2414. "aqi_level": "excellent",
  2415. "filter element working time": 405,
  2416. "err_code": 0,
  2417. "sequence_number": 161,
  2418. },
  2419. "isEncrypted": False,
  2420. "model": "*",
  2421. "modelFriendlyName": "Air Purifier US",
  2422. "modelName": SwitchbotModel.AIR_PURIFIER_US,
  2423. },
  2424. device=ble_device,
  2425. rssi=-97,
  2426. active=False,
  2427. )
  2428. def test_air_purifier_with_empty_data() -> None:
  2429. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2430. adv_data = generate_advertisement_data(
  2431. manufacturer_data={2409: None},
  2432. service_data={
  2433. "0000fd3d-0000-1000-8000-00805f9b34fb": b"+\x00\x00\x15\x04\x00",
  2434. },
  2435. rssi=-97,
  2436. )
  2437. result = parse_advertisement_data(ble_device, adv_data)
  2438. assert result == SwitchBotAdvertisement(
  2439. address="aa:bb:cc:dd:ee:ff",
  2440. data={
  2441. "rawAdvData": b"+\x00\x00\x15\x04\x00",
  2442. "data": {},
  2443. "isEncrypted": False,
  2444. "model": "+",
  2445. },
  2446. device=ble_device,
  2447. rssi=-97,
  2448. active=True,
  2449. )
  2450. def test_hub3_active() -> None:
  2451. """Test parsing hub3 with active data."""
  2452. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2453. adv_data = generate_advertisement_data(
  2454. manufacturer_data={2409: b"\xb0\xe9\xfen^)\x00\xffh&\xd6d\x83\x03\x994\x80"},
  2455. service_data={
  2456. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00d\x00\x10\xb9@"
  2457. },
  2458. rssi=-97,
  2459. )
  2460. result = parse_advertisement_data(ble_device, adv_data)
  2461. assert result == SwitchBotAdvertisement(
  2462. address="aa:bb:cc:dd:ee:ff",
  2463. data={
  2464. "rawAdvData": b"\x00\x00d\x00\x10\xb9@",
  2465. "data": {
  2466. "sequence_number": 0,
  2467. "network_state": 2,
  2468. "sensor_inserted": True,
  2469. "lightLevel": 3,
  2470. "illuminance": 90,
  2471. "temperature_alarm": False,
  2472. "humidity_alarm": False,
  2473. "temp": {"c": 25.3, "f": 77.5},
  2474. "temperature": 25.3,
  2475. "humidity": 52,
  2476. "motion_detected": True,
  2477. },
  2478. "isEncrypted": False,
  2479. "model": b"\x00\x10\xb9@",
  2480. "modelFriendlyName": "Hub3",
  2481. "modelName": SwitchbotModel.HUB3,
  2482. },
  2483. device=ble_device,
  2484. rssi=-97,
  2485. active=True,
  2486. )
  2487. def test_hub3_passive() -> None:
  2488. """Test parsing hub3 with passive data."""
  2489. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2490. adv_data = generate_advertisement_data(
  2491. manufacturer_data={2409: b"\xb0\xe9\xfen^)\x00\xffh&\xd6d\x83\x03\x994\x80"},
  2492. rssi=-97,
  2493. )
  2494. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.HUB3)
  2495. assert result == SwitchBotAdvertisement(
  2496. address="aa:bb:cc:dd:ee:ff",
  2497. data={
  2498. "rawAdvData": None,
  2499. "data": {
  2500. "sequence_number": 0,
  2501. "network_state": 2,
  2502. "sensor_inserted": True,
  2503. "lightLevel": 3,
  2504. "illuminance": 90,
  2505. "temperature_alarm": False,
  2506. "humidity_alarm": False,
  2507. "temp": {"c": 25.3, "f": 77.5},
  2508. "temperature": 25.3,
  2509. "humidity": 52,
  2510. "motion_detected": True,
  2511. },
  2512. "isEncrypted": False,
  2513. "model": b"\x00\x10\xb9@",
  2514. "modelFriendlyName": "Hub3",
  2515. "modelName": SwitchbotModel.HUB3,
  2516. },
  2517. device=ble_device,
  2518. rssi=-97,
  2519. active=False,
  2520. )
  2521. def test_hub3_with_empty_data() -> None:
  2522. """Test parsing hub3 with empty data."""
  2523. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2524. adv_data = generate_advertisement_data(
  2525. manufacturer_data={2409: None},
  2526. service_data={
  2527. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00d\x00\x10\xb9@"
  2528. },
  2529. rssi=-97,
  2530. )
  2531. result = parse_advertisement_data(ble_device, adv_data)
  2532. assert result == SwitchBotAdvertisement(
  2533. address="aa:bb:cc:dd:ee:ff",
  2534. data={
  2535. "rawAdvData": b"\x00\x00d\x00\x10\xb9@",
  2536. "data": {},
  2537. "isEncrypted": False,
  2538. "model": b"\x00\x10\xb9@",
  2539. },
  2540. device=ble_device,
  2541. rssi=-97,
  2542. active=True,
  2543. )
  2544. @pytest.mark.parametrize(
  2545. "test_case",
  2546. [
  2547. AdvTestCase(
  2548. b"\xe9\xd5\x11\xb2kS\x17\x93\x08 ",
  2549. b"-\x80d",
  2550. {
  2551. "sequence_number": 23,
  2552. "battery": 100,
  2553. "calibration": True,
  2554. "status": LockStatus.UNLOCKED,
  2555. "update_from_secondary_lock": False,
  2556. "double_lock_mode": False,
  2557. "unlocked_alarm": False,
  2558. "night_latch": False,
  2559. },
  2560. "-",
  2561. "Lock Lite",
  2562. SwitchbotModel.LOCK_LITE,
  2563. ),
  2564. AdvTestCase(
  2565. b"\xee\xf5\xe6\t\x8f\xe8\x11\x97\x08 ",
  2566. b"o\x80d",
  2567. {
  2568. "sequence_number": 17,
  2569. "battery": 100,
  2570. "calibration": True,
  2571. "status": LockStatus.UNLOCKED,
  2572. "update_from_secondary_lock": False,
  2573. "door_open": True,
  2574. "double_lock_mode": False,
  2575. "unclosed_alarm": False,
  2576. "unlocked_alarm": False,
  2577. "auto_lock_paused": False,
  2578. "night_latch": False,
  2579. },
  2580. "o",
  2581. "Lock",
  2582. SwitchbotModel.LOCK,
  2583. ),
  2584. AdvTestCase(
  2585. b"\xf7a\x07H\xe6\xe8:\x8a\x00d\x00\x00",
  2586. b"$\x80d",
  2587. {
  2588. "sequence_number": 58,
  2589. "battery": 100,
  2590. "calibration": True,
  2591. "status": LockStatus.UNLOCKED,
  2592. "update_from_secondary_lock": False,
  2593. "door_open": False,
  2594. "door_open_from_secondary_lock": False,
  2595. "double_lock_mode": False,
  2596. "is_secondary_lock": False,
  2597. "left_battery_compartment_alarm": 0,
  2598. "right_battery_compartment_alarm": 0,
  2599. "low_temperature_alarm": False,
  2600. "manual_unlock_linkage": False,
  2601. "unclosed_alarm": False,
  2602. "unlocked_alarm": False,
  2603. "auto_lock_paused": False,
  2604. "night_latch": False,
  2605. },
  2606. "$",
  2607. "Lock Pro",
  2608. SwitchbotModel.LOCK_PRO,
  2609. ),
  2610. AdvTestCase(
  2611. b"\xb0\xe9\xfe\xb6j=%\x8204\x00\x04",
  2612. b"\x00\x804\x00\x10\xa5\xb8",
  2613. {
  2614. "sequence_number": 37,
  2615. "battery": 52,
  2616. "calibration": True,
  2617. "status": LockStatus.LOCKED,
  2618. "update_from_secondary_lock": False,
  2619. "door_open": True,
  2620. "door_open_from_secondary_lock": True,
  2621. "double_lock_mode": False,
  2622. "is_secondary_lock": False,
  2623. "manual_unlock_linkage": False,
  2624. "unclosed_alarm": False,
  2625. "unlocked_alarm": False,
  2626. "auto_lock_paused": False,
  2627. "night_latch": False,
  2628. "power_alarm": False,
  2629. "battery_status": 4,
  2630. },
  2631. b"\x00\x10\xa5\xb8",
  2632. "Lock Ultra",
  2633. SwitchbotModel.LOCK_ULTRA,
  2634. ),
  2635. AdvTestCase(
  2636. b"\xb0\xe9\xfe\x6f\xc9\xa6\x0a\x00\x00\x2e\x00\x0c\x00\x00\x00\x00",
  2637. b"\x00\x80.\x00\x11i\x08",
  2638. {
  2639. "sequence_number": 10,
  2640. "battery": 46,
  2641. "calibration": False,
  2642. "status": LockStatus.LOCKED,
  2643. "update_from_secondary_lock": False,
  2644. "double_lock_mode": False,
  2645. "unlocked_alarm": False,
  2646. "night_latch": False,
  2647. },
  2648. b"\x00\x11\x69\x08",
  2649. "Lock Vision",
  2650. SwitchbotModel.LOCK_VISION,
  2651. ),
  2652. AdvTestCase(
  2653. b"\xb0\xe9\xfe\xe6\x1aq\x03\x00\x003\x00\x0c\x00\x00\x00\x00",
  2654. b"\x00\x803\x00\x11i\t",
  2655. {
  2656. "sequence_number": 3,
  2657. "battery": 51,
  2658. "calibration": False,
  2659. "status": LockStatus.LOCKED,
  2660. "update_from_secondary_lock": False,
  2661. "door_open": False,
  2662. "door_open_from_secondary_lock": False,
  2663. "double_lock_mode": False,
  2664. "is_secondary_lock": False,
  2665. "manual_unlock_linkage": False,
  2666. "unclosed_alarm": False,
  2667. "unlocked_alarm": False,
  2668. "auto_lock_paused": False,
  2669. "night_latch": False,
  2670. "power_alarm": False,
  2671. "battery_status": 4,
  2672. },
  2673. b"\x00\x11\x69\x09",
  2674. "Lock Vision Pro",
  2675. SwitchbotModel.LOCK_VISION_PRO,
  2676. ),
  2677. AdvTestCase(
  2678. b"\xb0\xe9\xfe\xb0^\xfe\x0e\x02\x00=\x00\x08",
  2679. b"\x00\x80=\x00\x10\xff\x90",
  2680. {
  2681. "sequence_number": 14,
  2682. "battery": 61,
  2683. "calibration": False,
  2684. "status": LockStatus.LOCKED,
  2685. "update_from_secondary_lock": False,
  2686. "door_open": False,
  2687. "door_open_from_secondary_lock": False,
  2688. "double_lock_mode": False,
  2689. "is_secondary_lock": False,
  2690. "left_battery_compartment_alarm": 0,
  2691. "right_battery_compartment_alarm": 0,
  2692. "low_temperature_alarm": False,
  2693. "manual_unlock_linkage": False,
  2694. "unclosed_alarm": False,
  2695. "unlocked_alarm": False,
  2696. "auto_lock_paused": False,
  2697. "night_latch": False,
  2698. },
  2699. b"\x00\x10\xff\x90",
  2700. "Lock Pro Wifi",
  2701. SwitchbotModel.LOCK_PRO_WIFI,
  2702. ),
  2703. ],
  2704. )
  2705. def test_lock_active(test_case: AdvTestCase) -> None:
  2706. """Test lokc series with active data."""
  2707. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2708. adv_data = generate_advertisement_data(
  2709. manufacturer_data={2409: test_case.manufacturer_data},
  2710. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  2711. rssi=-97,
  2712. )
  2713. result = parse_advertisement_data(ble_device, adv_data)
  2714. assert result == SwitchBotAdvertisement(
  2715. address="aa:bb:cc:dd:ee:ff",
  2716. data={
  2717. "rawAdvData": test_case.service_data,
  2718. "data": test_case.data,
  2719. "isEncrypted": False,
  2720. "model": test_case.model,
  2721. "modelFriendlyName": test_case.modelFriendlyName,
  2722. "modelName": test_case.modelName,
  2723. },
  2724. device=ble_device,
  2725. rssi=-97,
  2726. active=True,
  2727. )
  2728. @pytest.mark.parametrize(
  2729. "test_case",
  2730. [
  2731. AdvTestCase(
  2732. b"\xe9\xd5\x11\xb2kS\x17\x93\x08 ",
  2733. b"-\x80d",
  2734. {
  2735. "sequence_number": 23,
  2736. "battery": None,
  2737. "calibration": True,
  2738. "status": LockStatus.UNLOCKED,
  2739. "update_from_secondary_lock": False,
  2740. "double_lock_mode": False,
  2741. "unlocked_alarm": False,
  2742. "night_latch": False,
  2743. },
  2744. "-",
  2745. "Lock Lite",
  2746. SwitchbotModel.LOCK_LITE,
  2747. ),
  2748. AdvTestCase(
  2749. b"\xee\xf5\xe6\t\x8f\xe8\x11\x97\x08 ",
  2750. b"o\x80d",
  2751. {
  2752. "sequence_number": 17,
  2753. "battery": None,
  2754. "calibration": True,
  2755. "status": LockStatus.UNLOCKED,
  2756. "update_from_secondary_lock": False,
  2757. "door_open": True,
  2758. "double_lock_mode": False,
  2759. "unclosed_alarm": False,
  2760. "unlocked_alarm": False,
  2761. "auto_lock_paused": False,
  2762. "night_latch": False,
  2763. },
  2764. "o",
  2765. "Lock",
  2766. SwitchbotModel.LOCK,
  2767. ),
  2768. AdvTestCase(
  2769. b"\xf7a\x07H\xe6\xe8:\x8a\x00d\x00\x00",
  2770. b"$\x80d",
  2771. {
  2772. "sequence_number": 58,
  2773. "battery": 100,
  2774. "calibration": True,
  2775. "status": LockStatus.UNLOCKED,
  2776. "update_from_secondary_lock": False,
  2777. "door_open": False,
  2778. "door_open_from_secondary_lock": False,
  2779. "double_lock_mode": False,
  2780. "is_secondary_lock": False,
  2781. "left_battery_compartment_alarm": 0,
  2782. "right_battery_compartment_alarm": 0,
  2783. "low_temperature_alarm": False,
  2784. "manual_unlock_linkage": False,
  2785. "unclosed_alarm": False,
  2786. "unlocked_alarm": False,
  2787. "auto_lock_paused": False,
  2788. "night_latch": False,
  2789. },
  2790. "$",
  2791. "Lock Pro",
  2792. SwitchbotModel.LOCK_PRO,
  2793. ),
  2794. AdvTestCase(
  2795. b"\xb0\xe9\xfe\xb6j=%\x8204\x00\x04",
  2796. b"\x00\x804\x00\x10\xa5\xb8",
  2797. {
  2798. "sequence_number": 37,
  2799. "battery": 52,
  2800. "calibration": True,
  2801. "status": LockStatus.LOCKED,
  2802. "update_from_secondary_lock": False,
  2803. "door_open": True,
  2804. "door_open_from_secondary_lock": True,
  2805. "double_lock_mode": False,
  2806. "is_secondary_lock": False,
  2807. "manual_unlock_linkage": False,
  2808. "unclosed_alarm": False,
  2809. "unlocked_alarm": False,
  2810. "auto_lock_paused": False,
  2811. "night_latch": False,
  2812. "power_alarm": False,
  2813. "battery_status": 4,
  2814. },
  2815. b"\x00\x10\xa5\xb8",
  2816. "Lock Ultra",
  2817. SwitchbotModel.LOCK_ULTRA,
  2818. ),
  2819. AdvTestCase(
  2820. b"\xb0\xe9\xfe\x6f\xc9\xa6\x04\x00\x00\x2e\x00\x0c\x00\x00\x00\x00",
  2821. b"\x00\x80.\x01\x11i\x08",
  2822. {
  2823. "sequence_number": 4,
  2824. "battery": None,
  2825. "calibration": False,
  2826. "status": LockStatus.LOCKED,
  2827. "update_from_secondary_lock": False,
  2828. "double_lock_mode": False,
  2829. "unlocked_alarm": False,
  2830. "night_latch": False,
  2831. },
  2832. b"\x00\x11\x69\x08",
  2833. "Lock Vision",
  2834. SwitchbotModel.LOCK_VISION,
  2835. ),
  2836. AdvTestCase(
  2837. b"\xb0\xe9\xfe\xe6\x1aq\x04\x00\x003\x00\x0c\x00\x00\x00\x00",
  2838. b"\x00\x803\x01\x11i\t",
  2839. {
  2840. "sequence_number": 4,
  2841. "battery": 51,
  2842. "calibration": False,
  2843. "status": LockStatus.LOCKED,
  2844. "update_from_secondary_lock": False,
  2845. "door_open": False,
  2846. "door_open_from_secondary_lock": False,
  2847. "double_lock_mode": False,
  2848. "is_secondary_lock": False,
  2849. "manual_unlock_linkage": False,
  2850. "unclosed_alarm": False,
  2851. "unlocked_alarm": False,
  2852. "auto_lock_paused": False,
  2853. "night_latch": False,
  2854. "power_alarm": False,
  2855. "battery_status": 4,
  2856. },
  2857. b"\x00\x11\x69\x09",
  2858. "Lock Vision Pro",
  2859. SwitchbotModel.LOCK_VISION_PRO,
  2860. ),
  2861. AdvTestCase(
  2862. b"\xb0\xe9\xfe\xb0^\xfe\x0e\x02\x00=\x00\x08",
  2863. b"\x00\x80=\x00\x10\xff\x90",
  2864. {
  2865. "sequence_number": 14,
  2866. "battery": 61,
  2867. "calibration": False,
  2868. "status": LockStatus.LOCKED,
  2869. "update_from_secondary_lock": False,
  2870. "door_open": False,
  2871. "door_open_from_secondary_lock": False,
  2872. "double_lock_mode": False,
  2873. "is_secondary_lock": False,
  2874. "left_battery_compartment_alarm": 0,
  2875. "right_battery_compartment_alarm": 0,
  2876. "low_temperature_alarm": False,
  2877. "manual_unlock_linkage": False,
  2878. "unclosed_alarm": False,
  2879. "unlocked_alarm": False,
  2880. "auto_lock_paused": False,
  2881. "night_latch": False,
  2882. },
  2883. b"\x00\x10\xff\x90",
  2884. "Lock Pro Wifi",
  2885. SwitchbotModel.LOCK_PRO_WIFI,
  2886. ),
  2887. ],
  2888. )
  2889. def test_lock_passive(test_case: AdvTestCase) -> None:
  2890. """Test lokc series with passive data."""
  2891. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2892. adv_data = generate_advertisement_data(
  2893. manufacturer_data={2409: test_case.manufacturer_data},
  2894. rssi=-97,
  2895. )
  2896. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  2897. assert result == SwitchBotAdvertisement(
  2898. address="aa:bb:cc:dd:ee:ff",
  2899. data={
  2900. "rawAdvData": None,
  2901. "data": test_case.data,
  2902. "isEncrypted": False,
  2903. "model": test_case.model,
  2904. "modelFriendlyName": test_case.modelFriendlyName,
  2905. "modelName": test_case.modelName,
  2906. },
  2907. device=ble_device,
  2908. rssi=-97,
  2909. active=False,
  2910. )
  2911. @pytest.mark.parametrize(
  2912. "test_case",
  2913. [
  2914. AdvTestCase(
  2915. None,
  2916. b"-\x80d",
  2917. {},
  2918. "-",
  2919. "Lock Lite",
  2920. SwitchbotModel.LOCK_LITE,
  2921. ),
  2922. AdvTestCase(
  2923. None,
  2924. b"o\x80d",
  2925. {},
  2926. "o",
  2927. "Lock",
  2928. SwitchbotModel.LOCK,
  2929. ),
  2930. AdvTestCase(
  2931. None,
  2932. b"$\x80d",
  2933. {},
  2934. "$",
  2935. "Lock Pro",
  2936. SwitchbotModel.LOCK_PRO,
  2937. ),
  2938. AdvTestCase(
  2939. None,
  2940. b"\x00\x804\x00\x10\xa5\xb8",
  2941. {},
  2942. b"\x00\x10\xa5\xb8",
  2943. "Lock Ultra",
  2944. SwitchbotModel.LOCK_ULTRA,
  2945. ),
  2946. AdvTestCase(
  2947. None,
  2948. b"\x00\x80\x2e\x00\x11\x69\x08",
  2949. {},
  2950. b"\x00\x11\x69\x08",
  2951. "Lock Vision",
  2952. SwitchbotModel.LOCK_VISION,
  2953. ),
  2954. AdvTestCase(
  2955. None,
  2956. b"\x00\x803\x00\x11i\t",
  2957. {},
  2958. b"\x00\x11\x69\x09",
  2959. "Lock Vision Pro",
  2960. SwitchbotModel.LOCK_VISION_PRO,
  2961. ),
  2962. AdvTestCase(
  2963. None,
  2964. b"\x00\x80=\x00\x10\xff\x90",
  2965. {},
  2966. b"\x00\x10\xff\x90",
  2967. "Lock Pro Wifi",
  2968. SwitchbotModel.LOCK_PRO_WIFI,
  2969. ),
  2970. ],
  2971. )
  2972. def test_lock_with_empty_data(test_case: AdvTestCase) -> None:
  2973. """Test lokc series with empty data."""
  2974. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2975. adv_data = generate_advertisement_data(
  2976. manufacturer_data={2409: test_case.manufacturer_data},
  2977. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  2978. rssi=-97,
  2979. )
  2980. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  2981. assert result == SwitchBotAdvertisement(
  2982. address="aa:bb:cc:dd:ee:ff",
  2983. data={
  2984. "rawAdvData": test_case.service_data,
  2985. "data": {},
  2986. "isEncrypted": False,
  2987. "model": test_case.model,
  2988. },
  2989. device=ble_device,
  2990. rssi=-97,
  2991. active=True,
  2992. )
  2993. def test_blind_tilt_active() -> None:
  2994. """Test parsing blind tilt with active data."""
  2995. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  2996. adv_data = generate_advertisement_data(
  2997. manufacturer_data={2409: b"\xfc(\\6l\x7f\x0b'\x00\xa1\x84"},
  2998. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"x\x00H"},
  2999. rssi=-97,
  3000. )
  3001. result = parse_advertisement_data(ble_device, adv_data)
  3002. assert result == SwitchBotAdvertisement(
  3003. address="aa:bb:cc:dd:ee:ff",
  3004. data={
  3005. "rawAdvData": b"x\x00H",
  3006. "data": {
  3007. "sequence_number": 11,
  3008. "battery": 72,
  3009. "tilt": 0,
  3010. "inMotion": False,
  3011. "calibration": True,
  3012. "lightLevel": 10,
  3013. },
  3014. "isEncrypted": False,
  3015. "model": "x",
  3016. "modelFriendlyName": "Blind Tilt",
  3017. "modelName": SwitchbotModel.BLIND_TILT,
  3018. },
  3019. device=ble_device,
  3020. rssi=-97,
  3021. active=True,
  3022. )
  3023. def test_blind_tilt_passive() -> None:
  3024. """Test parsing blind tilt with passive data."""
  3025. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3026. adv_data = generate_advertisement_data(
  3027. manufacturer_data={2409: b"\xfc(\\6l\x7f\x0b'\x00\xa1\x84"},
  3028. rssi=-97,
  3029. )
  3030. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.BLIND_TILT)
  3031. assert result == SwitchBotAdvertisement(
  3032. address="aa:bb:cc:dd:ee:ff",
  3033. data={
  3034. "rawAdvData": None,
  3035. "data": {
  3036. "sequence_number": 11,
  3037. "battery": None,
  3038. "tilt": 0,
  3039. "inMotion": False,
  3040. "calibration": True,
  3041. "lightLevel": 10,
  3042. },
  3043. "isEncrypted": False,
  3044. "model": "x",
  3045. "modelFriendlyName": "Blind Tilt",
  3046. "modelName": SwitchbotModel.BLIND_TILT,
  3047. },
  3048. device=ble_device,
  3049. rssi=-97,
  3050. active=False,
  3051. )
  3052. def test_blind_tilt_with_empty_data() -> None:
  3053. """Test parsing blind tilt with empty data."""
  3054. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3055. adv_data = generate_advertisement_data(
  3056. manufacturer_data={2409: None},
  3057. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"x\x00H"},
  3058. rssi=-97,
  3059. )
  3060. result = parse_advertisement_data(ble_device, adv_data)
  3061. assert result == SwitchBotAdvertisement(
  3062. address="aa:bb:cc:dd:ee:ff",
  3063. data={
  3064. "rawAdvData": b"x\x00H",
  3065. "data": {},
  3066. "isEncrypted": False,
  3067. "model": "x",
  3068. },
  3069. device=ble_device,
  3070. rssi=-97,
  3071. active=True,
  3072. )
  3073. @pytest.mark.parametrize(
  3074. "test_case",
  3075. [
  3076. AdvTestCase(
  3077. b"\xa0\xa3\xb3,\x9c\xe68\x86\x88\xb5\x99\x12\x10\x1b\x00\x85]",
  3078. b"#\x00\x00\x15\x1c\x00",
  3079. {
  3080. "seq_number": 56,
  3081. "isOn": True,
  3082. "mode": HumidifierMode(6),
  3083. "over_humidify_protection": True,
  3084. "child_lock": False,
  3085. "tank_removed": False,
  3086. "tilted_alert": False,
  3087. "filter_missing": False,
  3088. "is_meter_binded": True,
  3089. "humidity": 53,
  3090. "temperature": 25.1,
  3091. "temp": {"c": 25.1, "f": 77.18},
  3092. "water_level": "medium",
  3093. "filter_run_time": datetime.timedelta(days=1, seconds=10800),
  3094. "filter_alert": False,
  3095. "target_humidity": 93,
  3096. },
  3097. "#",
  3098. "Evaporative Humidifier",
  3099. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3100. ),
  3101. AdvTestCase(
  3102. b"\xa0\xa3\xb3,\x9c\xe6H\x86\x80\x7f\xff\xf2\x10\x1d\x00\x874",
  3103. b"#\x00\x00\x15\x1c\x00",
  3104. {
  3105. "seq_number": 72,
  3106. "isOn": True,
  3107. "mode": HumidifierMode(6),
  3108. "over_humidify_protection": True,
  3109. "child_lock": False,
  3110. "tank_removed": False,
  3111. "tilted_alert": False,
  3112. "filter_missing": False,
  3113. "is_meter_binded": False,
  3114. "humidity": None,
  3115. "temperature": None,
  3116. "temp": {"c": None, "f": None},
  3117. "water_level": "medium",
  3118. "filter_run_time": datetime.timedelta(days=1, seconds=18000),
  3119. "filter_alert": False,
  3120. "target_humidity": 52,
  3121. },
  3122. "#",
  3123. "Evaporative Humidifier",
  3124. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3125. ),
  3126. AdvTestCase(
  3127. b"\xa0\xa3\xb3,\x9c\xe6H\x86\x80\xff\xff\xf2\x10\x1d\x00\x874",
  3128. b"#\x00\x00\x15\x1c\x00",
  3129. {
  3130. "seq_number": 72,
  3131. "isOn": True,
  3132. "mode": HumidifierMode(6),
  3133. "over_humidify_protection": True,
  3134. "child_lock": False,
  3135. "tank_removed": False,
  3136. "tilted_alert": False,
  3137. "filter_missing": False,
  3138. "is_meter_binded": True,
  3139. "humidity": None,
  3140. "temperature": None,
  3141. "temp": {"c": None, "f": None},
  3142. "water_level": "medium",
  3143. "filter_run_time": datetime.timedelta(days=1, seconds=18000),
  3144. "filter_alert": False,
  3145. "target_humidity": 52,
  3146. },
  3147. "#",
  3148. "Evaporative Humidifier",
  3149. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3150. ),
  3151. AdvTestCase(
  3152. b"\xacg\xb2\xcd\xfa\xbe",
  3153. b"e\x80\x00\xf9\x80Bc\x00",
  3154. {
  3155. "isOn": True,
  3156. "level": 128,
  3157. "switchMode": True,
  3158. },
  3159. "e",
  3160. "Humidifier",
  3161. SwitchbotModel.HUMIDIFIER,
  3162. ),
  3163. ],
  3164. )
  3165. def test_humidifer_active(test_case: AdvTestCase) -> None:
  3166. """Test humidifier with active data."""
  3167. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3168. adv_data = generate_advertisement_data(
  3169. manufacturer_data={2409: test_case.manufacturer_data},
  3170. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  3171. rssi=-97,
  3172. )
  3173. result = parse_advertisement_data(ble_device, adv_data)
  3174. assert result == SwitchBotAdvertisement(
  3175. address="aa:bb:cc:dd:ee:ff",
  3176. data={
  3177. "rawAdvData": test_case.service_data,
  3178. "data": test_case.data,
  3179. "isEncrypted": False,
  3180. "model": test_case.model,
  3181. "modelFriendlyName": test_case.modelFriendlyName,
  3182. "modelName": test_case.modelName,
  3183. },
  3184. device=ble_device,
  3185. rssi=-97,
  3186. active=True,
  3187. )
  3188. @pytest.mark.parametrize(
  3189. "test_case",
  3190. [
  3191. AdvTestCase(
  3192. b"\xa0\xa3\xb3,\x9c\xe68\x86\x88\xb5\x99\x12\x10\x1b\x00\x85]",
  3193. b"#\x00\x00\x15\x1c\x00",
  3194. {
  3195. "seq_number": 56,
  3196. "isOn": True,
  3197. "mode": HumidifierMode(6),
  3198. "over_humidify_protection": True,
  3199. "child_lock": False,
  3200. "tank_removed": False,
  3201. "tilted_alert": False,
  3202. "filter_missing": False,
  3203. "is_meter_binded": True,
  3204. "humidity": 53,
  3205. "temperature": 25.1,
  3206. "temp": {"c": 25.1, "f": 77.18},
  3207. "water_level": "medium",
  3208. "filter_run_time": datetime.timedelta(days=1, seconds=10800),
  3209. "filter_alert": False,
  3210. "target_humidity": 93,
  3211. },
  3212. "#",
  3213. "Evaporative Humidifier",
  3214. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3215. ),
  3216. AdvTestCase(
  3217. b"\xacg\xb2\xcd\xfa\xbe",
  3218. b"e\x80\x00\xf9\x80Bc\x00",
  3219. {
  3220. "isOn": None,
  3221. "level": None,
  3222. "switchMode": True,
  3223. },
  3224. "e",
  3225. "Humidifier",
  3226. SwitchbotModel.HUMIDIFIER,
  3227. ),
  3228. ],
  3229. )
  3230. def test_humidifer_passive(test_case: AdvTestCase) -> None:
  3231. """Test humidifier with passive data."""
  3232. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3233. adv_data = generate_advertisement_data(
  3234. manufacturer_data={2409: test_case.manufacturer_data},
  3235. rssi=-97,
  3236. )
  3237. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  3238. assert result == SwitchBotAdvertisement(
  3239. address="aa:bb:cc:dd:ee:ff",
  3240. data={
  3241. "rawAdvData": None,
  3242. "data": test_case.data,
  3243. "isEncrypted": False,
  3244. "model": test_case.model,
  3245. "modelFriendlyName": test_case.modelFriendlyName,
  3246. "modelName": test_case.modelName,
  3247. },
  3248. device=ble_device,
  3249. rssi=-97,
  3250. active=False,
  3251. )
  3252. def test_humidifer_with_empty_data() -> None:
  3253. """Test humidifier with empty data."""
  3254. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3255. adv_data = generate_advertisement_data(
  3256. manufacturer_data={2409: None},
  3257. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"#\x00\x00\x15\x1c\x00"},
  3258. rssi=-97,
  3259. )
  3260. result = parse_advertisement_data(ble_device, adv_data)
  3261. assert result == SwitchBotAdvertisement(
  3262. address="aa:bb:cc:dd:ee:ff",
  3263. data={
  3264. "rawAdvData": b"#\x00\x00\x15\x1c\x00",
  3265. "data": {},
  3266. "isEncrypted": False,
  3267. "model": "#",
  3268. },
  3269. device=ble_device,
  3270. rssi=-97,
  3271. active=True,
  3272. )
  3273. @pytest.mark.parametrize(
  3274. "test_case",
  3275. [
  3276. AdvTestCase(
  3277. b"\xc0N0\xdd\xb9\xf2\x8a\xc1\x00\x00\x00\x00\x00F\x00\x00",
  3278. b"=\x00\x00\x00",
  3279. {
  3280. 1: {
  3281. "isOn": True,
  3282. "sequence_number": 138,
  3283. "switchMode": True,
  3284. "power": 0.0,
  3285. },
  3286. 2: {
  3287. "isOn": True,
  3288. "sequence_number": 138,
  3289. "switchMode": True,
  3290. "power": 70.0,
  3291. },
  3292. "sequence_number": 138,
  3293. },
  3294. "=",
  3295. "Relay Switch 2PM",
  3296. SwitchbotModel.RELAY_SWITCH_2PM,
  3297. ),
  3298. AdvTestCase(
  3299. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  3300. b";\x00\x00\x00",
  3301. {
  3302. "isOn": True,
  3303. "sequence_number": 71,
  3304. "switchMode": True,
  3305. },
  3306. ";",
  3307. "Relay Switch 1",
  3308. SwitchbotModel.RELAY_SWITCH_1,
  3309. ),
  3310. AdvTestCase(
  3311. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  3312. b"<\x00\x00\x00",
  3313. {
  3314. "isOn": True,
  3315. "sequence_number": 71,
  3316. "switchMode": True,
  3317. "power": 49.0,
  3318. },
  3319. "<",
  3320. "Relay Switch 1PM",
  3321. SwitchbotModel.RELAY_SWITCH_1PM,
  3322. ),
  3323. AdvTestCase(
  3324. b"$X|\x05BN\x0f\x00\x00\x03\x00\x00\x00\x00\x00\x00",
  3325. b">\x00\x00\x00",
  3326. {
  3327. "door_open": True,
  3328. "isOn": False,
  3329. "sequence_number": 15,
  3330. "switchMode": True,
  3331. },
  3332. ">",
  3333. "Garage Door Opener",
  3334. SwitchbotModel.GARAGE_DOOR_OPENER,
  3335. ),
  3336. AdvTestCase(
  3337. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  3338. b"\x00\x00\x00\x00\x10\xd0\xb1",
  3339. {
  3340. "sequence_number": 133,
  3341. "isOn": True,
  3342. "brightness": 30,
  3343. "delay": False,
  3344. "network_state": 2,
  3345. "color_mode": 2,
  3346. "cw": 4753,
  3347. },
  3348. b"\x00\x10\xd0\xb1",
  3349. "Strip Light 3",
  3350. SwitchbotModel.STRIP_LIGHT_3,
  3351. ),
  3352. AdvTestCase(
  3353. b'\xa0\x85\xe3e,\x06P\xaa"\xd4\x00\x00\x00\x00\x00\x00\r\x93\x00',
  3354. b"\x00\x00\x00\x00\x10\xd0\xb0",
  3355. {
  3356. "sequence_number": 80,
  3357. "isOn": True,
  3358. "brightness": 42,
  3359. "delay": False,
  3360. "network_state": 2,
  3361. "color_mode": 2,
  3362. "cw": 3475,
  3363. },
  3364. b"\x00\x10\xd0\xb0",
  3365. "Floor Lamp",
  3366. SwitchbotModel.FLOOR_LAMP,
  3367. ),
  3368. AdvTestCase(
  3369. b"\x90\xe5\xb1h\xda\xaa\n\xb0 \x00",
  3370. b"\x00\x00\x00\x00\x11\x22\xb8",
  3371. {
  3372. "brightness": 48,
  3373. "delay": False,
  3374. "isOn": True,
  3375. "network_state": 2,
  3376. "sequence_number": 10,
  3377. },
  3378. b"\x00\x11\x22\xb8",
  3379. "Candle Warmer Lamp",
  3380. SwitchbotModel.CANDLE_WARMER_LAMP,
  3381. ),
  3382. AdvTestCase(
  3383. b"\xef\xfe\xfb\x9d\x10\xfe\n\x01\x18\xf3$",
  3384. b"q\x00",
  3385. {
  3386. "brightness": 1,
  3387. "color_mode": 1,
  3388. "cw": 6387,
  3389. "isOn": False,
  3390. "sequence_number": 10,
  3391. },
  3392. "q",
  3393. "Ceiling Light",
  3394. SwitchbotModel.CEILING_LIGHT,
  3395. ),
  3396. AdvTestCase(
  3397. b'|,g\xc8\x15&jR"l\x00\x00\x00\x00\x00\x00',
  3398. b"r\x00d",
  3399. {
  3400. "brightness": 82,
  3401. "color_mode": 2,
  3402. "delay": False,
  3403. "isOn": False,
  3404. "sequence_number": 106,
  3405. "network_state": 2,
  3406. },
  3407. "r",
  3408. "Light Strip",
  3409. SwitchbotModel.LIGHT_STRIP,
  3410. ),
  3411. AdvTestCase(
  3412. b"@L\xca\xa7_\x12\x02\x81\x12\x00\x00",
  3413. b"u\x00d",
  3414. {
  3415. "brightness": 1,
  3416. "color_mode": 2,
  3417. "delay": False,
  3418. "isOn": True,
  3419. "loop_index": 0,
  3420. "preset": False,
  3421. "sequence_number": 2,
  3422. "speed": 0,
  3423. },
  3424. "u",
  3425. "Color Bulb",
  3426. SwitchbotModel.COLOR_BULB,
  3427. ),
  3428. AdvTestCase(
  3429. b"\x94\xa9\x90T\x85^?\xa1\x00\x00\x04\xe6\x00\x00\x00\x00",
  3430. b"?\x00\x00\x00",
  3431. {
  3432. "isOn": True,
  3433. "power": 1254.0,
  3434. "sequence_number": 63,
  3435. "switchMode": True,
  3436. },
  3437. "?",
  3438. "Plug Mini (EU)",
  3439. SwitchbotModel.PLUG_MINI_EU,
  3440. ),
  3441. AdvTestCase(
  3442. b'\xdc\x06u\xa6\xfb\xb2y\x9e"\x00\x11\xb8\x00',
  3443. b"\x00\x00\x00\x00\x10\xd0\xb4",
  3444. {
  3445. "sequence_number": 121,
  3446. "isOn": True,
  3447. "brightness": 30,
  3448. "delay": False,
  3449. "network_state": 2,
  3450. "color_mode": 2,
  3451. "cw": 4536,
  3452. },
  3453. b"\x00\x10\xd0\xb4",
  3454. "RGBICWW Floor Lamp",
  3455. SwitchbotModel.RGBICWW_FLOOR_LAMP,
  3456. ),
  3457. AdvTestCase(
  3458. b'(7/L\x94\xb2\x0c\x9e"\x00\x11:\x00',
  3459. b"\x00\x00\x00\x00\x10\xd0\xb3",
  3460. {
  3461. "sequence_number": 12,
  3462. "isOn": True,
  3463. "brightness": 30,
  3464. "delay": False,
  3465. "network_state": 2,
  3466. "color_mode": 2,
  3467. "cw": 4410,
  3468. },
  3469. b"\x00\x10\xd0\xb3",
  3470. "RGBICWW Strip Light",
  3471. SwitchbotModel.RGBICWW_STRIP_LIGHT,
  3472. ),
  3473. AdvTestCase(
  3474. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  3475. b"\x00\x00\x00\x00\x10\xd0\xb7",
  3476. {
  3477. "sequence_number": 133,
  3478. "isOn": True,
  3479. "brightness": 30,
  3480. "delay": False,
  3481. "network_state": 2,
  3482. "color_mode": 2,
  3483. "cw": 0,
  3484. },
  3485. b"\x00\x10\xd0\xb7",
  3486. "Permanent Outdoor Light",
  3487. SwitchbotModel.PERMANENT_OUTDOOR_LIGHT,
  3488. ),
  3489. AdvTestCase(
  3490. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  3491. b"\x00\x00\x00\x00\x10\xd0\xb6",
  3492. {
  3493. "sequence_number": 47,
  3494. "isOn": True,
  3495. "brightness": 11,
  3496. "delay": False,
  3497. "network_state": 2,
  3498. "color_mode": 7,
  3499. },
  3500. b"\x00\x10\xd0\xb6",
  3501. "RGBIC Neon Rope Light",
  3502. SwitchbotModel.RGBIC_NEON_ROPE_LIGHT,
  3503. ),
  3504. AdvTestCase(
  3505. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  3506. b"\x00\x00\x00\x00\x10\xd0\xb5",
  3507. {
  3508. "sequence_number": 47,
  3509. "isOn": True,
  3510. "brightness": 11,
  3511. "delay": False,
  3512. "network_state": 2,
  3513. "color_mode": 7,
  3514. },
  3515. b"\x00\x10\xd0\xb5",
  3516. "RGBIC Neon Wire Rope Light",
  3517. SwitchbotModel.RGBIC_NEON_WIRE_ROPE_LIGHT,
  3518. ),
  3519. AdvTestCase(
  3520. b"\xb0\xe9\xfe\xe4\xbf\xd8\x0b\x01\x11f\x00\x16M\x15",
  3521. b"\x00\x00M\x00\x10\xfb\xa8",
  3522. {
  3523. "battery": 77,
  3524. "mqtt_connected": True,
  3525. "sequence_number": 11,
  3526. "soc_version": "1.1.102",
  3527. "step": 6,
  3528. "work_status": 21,
  3529. },
  3530. b"\x00\x10\xfb\xa8",
  3531. "K11+ Vacuum",
  3532. SwitchbotModel.K11_VACUUM,
  3533. ),
  3534. AdvTestCase(
  3535. b"\xb0\xe9\xfe\x848\x05\x06\x14\x05\x99-\x00\x00\x00\x00\xd4\x00\x0c\x04\x00",
  3536. b"\x00 _\x00\x10\xf3\xd8@",
  3537. {
  3538. "battery": 20,
  3539. "humidity": 45,
  3540. "sequence_number": 6,
  3541. "humidity_alarm": 0,
  3542. "isOn": False,
  3543. "is_light": True,
  3544. "motion_detected": True,
  3545. "temp_alarm": 0,
  3546. "temperature": 25.5,
  3547. },
  3548. b"\x00\x10\xf3\xd8",
  3549. "Climate Panel",
  3550. SwitchbotModel.CLIMATE_PANEL,
  3551. ),
  3552. AdvTestCase(
  3553. b"\xb0\xe9\xfe\x848\x05\x13\x14\t\x99-\x00\x00\x00\x00\xd2\x01\x14\x04\x00",
  3554. b"\x00 _\x00\x10\xf3\xd8@",
  3555. {
  3556. "battery": 20,
  3557. "humidity": 45,
  3558. "sequence_number": 19,
  3559. "humidity_alarm": 0,
  3560. "isOn": False,
  3561. "is_light": False,
  3562. "motion_detected": True,
  3563. "temp_alarm": 0,
  3564. "temperature": 25.9,
  3565. },
  3566. b"\x00\x10\xf3\xd8",
  3567. "Climate Panel",
  3568. SwitchbotModel.CLIMATE_PANEL,
  3569. ),
  3570. AdvTestCase(
  3571. b"\xb0\xe9\xfe\xa2T|6\xe4\x00\x9c\xa3A\x00",
  3572. b"\x00 d\x00\x116@",
  3573. {
  3574. "battery": 100,
  3575. "door_open": False,
  3576. "fault_code": 0,
  3577. "isOn": True,
  3578. "last_mode": "comfort",
  3579. "mode": "manual",
  3580. "sequence_number": 54,
  3581. "need_update_temp": False,
  3582. "restarted": False,
  3583. "target_temperature": 35.0,
  3584. "temperature": 28.0,
  3585. },
  3586. b"\x00\x116@",
  3587. "Smart Thermostat Radiator",
  3588. SwitchbotModel.SMART_THERMOSTAT_RADIATOR,
  3589. ),
  3590. AdvTestCase(
  3591. b"\xb0\xe9\xfe\xc3\x1a!:\x01\x11\x1e\x00\x00d\x03",
  3592. b"\x00\x00d\x00\x10\xe0P",
  3593. {
  3594. "battery": 100,
  3595. "mqtt_connected": False,
  3596. "sequence_number": 58,
  3597. "soc_version": "1.1.030",
  3598. "step": 0,
  3599. "work_status": 3,
  3600. },
  3601. b"\x00\x10\xe0P",
  3602. "S20 Vacuum",
  3603. SwitchbotModel.S20_VACUUM,
  3604. ),
  3605. AdvTestCase(
  3606. b"\xb0\xe9\xfelf\xaa\x06\xcc\x04V\x00\x8c",
  3607. b"\x00 d\x00\x10\xcc\xc8",
  3608. {
  3609. "adaptive_state": True,
  3610. "battery": 100,
  3611. "battery_range": ">=60%",
  3612. "duration": 1110,
  3613. "led_state": True,
  3614. "lightLevel": 12,
  3615. "motion_detected": True,
  3616. "sequence_number": 6,
  3617. "trigger_flag": 0,
  3618. },
  3619. b"\x00\x10\xcc\xc8",
  3620. "Presence Sensor",
  3621. SwitchbotModel.PRESENCE_SENSOR,
  3622. ),
  3623. AdvTestCase(
  3624. b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05",
  3625. b"\x14\x00d",
  3626. {
  3627. "battery": 100,
  3628. "fahrenheit": False,
  3629. "humidity": 44,
  3630. "temp": {"c": 23.8, "f": 74.84},
  3631. "temperature": 23.8,
  3632. },
  3633. b"\x14",
  3634. "Meter Pro",
  3635. SwitchbotModel.METER_PRO,
  3636. ),
  3637. AdvTestCase(
  3638. b"\xb0\xe9\xfe\xe2\xfa8\x157\x03\x08",
  3639. b"\x00\x007\x01\x11>\x10",
  3640. {
  3641. "battery": 55,
  3642. "battery_charging": False,
  3643. "display_mode": 1,
  3644. "display_size": 0,
  3645. "image_index": 3,
  3646. "last_network_status": 0,
  3647. "sequence_number": 21,
  3648. },
  3649. b"\x01\x11>\x10",
  3650. "Art Frame",
  3651. SwitchbotModel.ART_FRAME,
  3652. ),
  3653. AdvTestCase(
  3654. b"\xb0\xe9\xfe\xe5\x04\x1e\xac\xdf\x00\x00\x00\x00\x00\x02",
  3655. b"\x00\x00_\x01\x11\x03x",
  3656. {
  3657. "battery": 95,
  3658. "battery_charging": True,
  3659. "doorbell": False,
  3660. "doorbell_seq": 0,
  3661. "duress_alarm": False,
  3662. "high_temperature": False,
  3663. "lockout_alarm": False,
  3664. "low_temperature": False,
  3665. "pir_triggered_level": 2,
  3666. "sequence_number": 172,
  3667. "tamper_alarm": False,
  3668. },
  3669. b"\x01\x11\x03x",
  3670. "Keypad Vision",
  3671. SwitchbotModel.KEYPAD_VISION,
  3672. ),
  3673. AdvTestCase(
  3674. b"\xb0\xe9\xfe\xde\xb6\x8c+`\x00\x00\x00\x00\x00\x002",
  3675. b"\x00\x00`\x01\x11Q\x98",
  3676. {
  3677. "battery": 96,
  3678. "battery_charging": False,
  3679. "doorbell": False,
  3680. "doorbell_seq": 0,
  3681. "duress_alarm": False,
  3682. "high_temperature": False,
  3683. "lockout_alarm": False,
  3684. "low_temperature": False,
  3685. "radar_triggered_distance": 0,
  3686. "radar_triggered_level": 0,
  3687. "sequence_number": 43,
  3688. "tamper_alarm": False,
  3689. },
  3690. b"\x01\x11Q\x98",
  3691. "Keypad Vision Pro",
  3692. SwitchbotModel.KEYPAD_VISION_PRO,
  3693. ),
  3694. ],
  3695. )
  3696. def test_adv_active(test_case: AdvTestCase) -> None:
  3697. """Test with active data."""
  3698. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3699. adv_data = generate_advertisement_data(
  3700. manufacturer_data={2409: test_case.manufacturer_data},
  3701. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  3702. rssi=-97,
  3703. )
  3704. result = parse_advertisement_data(ble_device, adv_data)
  3705. assert result == SwitchBotAdvertisement(
  3706. address="aa:bb:cc:dd:ee:ff",
  3707. data={
  3708. "rawAdvData": test_case.service_data,
  3709. "data": test_case.data,
  3710. "isEncrypted": False,
  3711. "model": test_case.model,
  3712. "modelFriendlyName": test_case.modelFriendlyName,
  3713. "modelName": test_case.modelName,
  3714. },
  3715. device=ble_device,
  3716. rssi=-97,
  3717. active=True,
  3718. )
  3719. @pytest.mark.parametrize(
  3720. "test_case",
  3721. [
  3722. AdvTestCase(
  3723. b"\xc0N0\xdd\xb9\xf2\x8a\xc1\x00\x00\x00\x00\x00F\x00\x00",
  3724. b"=\x00\x00\x00",
  3725. {
  3726. 1: {
  3727. "isOn": True,
  3728. "sequence_number": 138,
  3729. "switchMode": True,
  3730. "power": 0.0,
  3731. },
  3732. 2: {
  3733. "isOn": True,
  3734. "sequence_number": 138,
  3735. "switchMode": True,
  3736. "power": 70.0,
  3737. },
  3738. "sequence_number": 138,
  3739. },
  3740. "=",
  3741. "Relay Switch 2PM",
  3742. SwitchbotModel.RELAY_SWITCH_2PM,
  3743. ),
  3744. AdvTestCase(
  3745. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  3746. b";\x00\x00\x00",
  3747. {
  3748. "isOn": True,
  3749. "sequence_number": 71,
  3750. "switchMode": True,
  3751. },
  3752. ";",
  3753. "Relay Switch 1",
  3754. SwitchbotModel.RELAY_SWITCH_1,
  3755. ),
  3756. AdvTestCase(
  3757. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  3758. b"<\x00\x00\x00",
  3759. {
  3760. "isOn": True,
  3761. "sequence_number": 71,
  3762. "switchMode": True,
  3763. "power": 49.0,
  3764. },
  3765. "<",
  3766. "Relay Switch 1PM",
  3767. SwitchbotModel.RELAY_SWITCH_1PM,
  3768. ),
  3769. AdvTestCase(
  3770. b"$X|\x05BN\x0f\x00\x00\x03\x00\x00\x00\x00\x00\x00",
  3771. b">\x00\x00\x00",
  3772. {
  3773. "door_open": True,
  3774. "isOn": False,
  3775. "sequence_number": 15,
  3776. "switchMode": True,
  3777. },
  3778. ">",
  3779. "Garage Door Opener",
  3780. SwitchbotModel.GARAGE_DOOR_OPENER,
  3781. ),
  3782. AdvTestCase(
  3783. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  3784. None,
  3785. {
  3786. "sequence_number": 133,
  3787. "isOn": True,
  3788. "brightness": 30,
  3789. "delay": False,
  3790. "network_state": 2,
  3791. "color_mode": 2,
  3792. "cw": 4753,
  3793. },
  3794. b"\x00\x10\xd0\xb1",
  3795. "Strip Light 3",
  3796. SwitchbotModel.STRIP_LIGHT_3,
  3797. ),
  3798. AdvTestCase(
  3799. b'\xa0\x85\xe3e,\x06P\xaa"\xd4\x00\x00\x00\x00\x00\x00\r\x93\x00',
  3800. None,
  3801. {
  3802. "sequence_number": 80,
  3803. "isOn": True,
  3804. "brightness": 42,
  3805. "delay": False,
  3806. "network_state": 2,
  3807. "color_mode": 2,
  3808. "cw": 3475,
  3809. },
  3810. b"\x00\x10\xd0\xb0",
  3811. "Floor Lamp",
  3812. SwitchbotModel.FLOOR_LAMP,
  3813. ),
  3814. AdvTestCase(
  3815. b'|,g\xc8\x15&jR"l\x00\x00\x00\x00\x00\x00',
  3816. None,
  3817. {
  3818. "brightness": 82,
  3819. "color_mode": 2,
  3820. "delay": False,
  3821. "isOn": False,
  3822. "sequence_number": 106,
  3823. "network_state": 2,
  3824. },
  3825. "r",
  3826. "Light Strip",
  3827. SwitchbotModel.LIGHT_STRIP,
  3828. ),
  3829. AdvTestCase(
  3830. b"@L\xca\xa7_\x12\x02\x81\x12\x00\x00",
  3831. None,
  3832. {
  3833. "brightness": 1,
  3834. "color_mode": 2,
  3835. "delay": False,
  3836. "isOn": True,
  3837. "loop_index": 0,
  3838. "preset": False,
  3839. "sequence_number": 2,
  3840. "speed": 0,
  3841. },
  3842. "u",
  3843. "Color Bulb",
  3844. SwitchbotModel.COLOR_BULB,
  3845. ),
  3846. AdvTestCase(
  3847. b"\x94\xa9\x90T\x85^?\xa1\x00\x00\x04\xe6\x00\x00\x00\x00",
  3848. None,
  3849. {
  3850. "isOn": True,
  3851. "power": 1254.0,
  3852. "sequence_number": 63,
  3853. "switchMode": True,
  3854. },
  3855. "?",
  3856. "Plug Mini (EU)",
  3857. SwitchbotModel.PLUG_MINI_EU,
  3858. ),
  3859. AdvTestCase(
  3860. b'\xdc\x06u\xa6\xfb\xb2y\x9e"\x00\x11\xb8\x00',
  3861. None,
  3862. {
  3863. "sequence_number": 121,
  3864. "isOn": True,
  3865. "brightness": 30,
  3866. "delay": False,
  3867. "network_state": 2,
  3868. "color_mode": 2,
  3869. "cw": 4536,
  3870. },
  3871. b"\x00\x10\xd0\xb4",
  3872. "RGBICWW Floor Lamp",
  3873. SwitchbotModel.RGBICWW_FLOOR_LAMP,
  3874. ),
  3875. AdvTestCase(
  3876. b"\x90\xe5\xb1h\xda\xaa\n\xb0 \x00",
  3877. None,
  3878. {
  3879. "brightness": 48,
  3880. "delay": False,
  3881. "isOn": True,
  3882. "network_state": 2,
  3883. "sequence_number": 10,
  3884. },
  3885. b"\x00\x11\x22\xb8",
  3886. "Candle Warmer Lamp",
  3887. SwitchbotModel.CANDLE_WARMER_LAMP,
  3888. ),
  3889. AdvTestCase(
  3890. b'(7/L\x94\xb2\x0c\x9e"\x00\x11:\x00',
  3891. None,
  3892. {
  3893. "sequence_number": 12,
  3894. "isOn": True,
  3895. "brightness": 30,
  3896. "delay": False,
  3897. "network_state": 2,
  3898. "color_mode": 2,
  3899. "cw": 4410,
  3900. },
  3901. b"\x00\x10\xd0\xb3",
  3902. "RGBICWW Strip Light",
  3903. SwitchbotModel.RGBICWW_STRIP_LIGHT,
  3904. ),
  3905. AdvTestCase(
  3906. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  3907. None,
  3908. {
  3909. "sequence_number": 133,
  3910. "isOn": True,
  3911. "brightness": 30,
  3912. "delay": False,
  3913. "network_state": 2,
  3914. "color_mode": 2,
  3915. "cw": 0,
  3916. },
  3917. b"\x00\x10\xd0\xb7",
  3918. "Permanent Outdoor Light",
  3919. SwitchbotModel.PERMANENT_OUTDOOR_LIGHT,
  3920. ),
  3921. AdvTestCase(
  3922. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  3923. None,
  3924. {
  3925. "sequence_number": 47,
  3926. "isOn": True,
  3927. "brightness": 11,
  3928. "delay": False,
  3929. "network_state": 2,
  3930. "color_mode": 7,
  3931. },
  3932. b"\x00\x10\xd0\xb6",
  3933. "RGBIC Neon Rope Light",
  3934. SwitchbotModel.RGBIC_NEON_ROPE_LIGHT,
  3935. ),
  3936. AdvTestCase(
  3937. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  3938. None,
  3939. {
  3940. "sequence_number": 47,
  3941. "isOn": True,
  3942. "brightness": 11,
  3943. "delay": False,
  3944. "network_state": 2,
  3945. "color_mode": 7,
  3946. },
  3947. b"\x00\x10\xd0\xb5",
  3948. "RGBIC Neon Wire Rope Light",
  3949. SwitchbotModel.RGBIC_NEON_WIRE_ROPE_LIGHT,
  3950. ),
  3951. AdvTestCase(
  3952. b"\xb0\xe9\xfe\xe4\xbf\xd8\x0b\x01\x11f\x00\x16M\x15",
  3953. None,
  3954. {
  3955. "battery": 77,
  3956. "mqtt_connected": True,
  3957. "sequence_number": 11,
  3958. "soc_version": "1.1.102",
  3959. "step": 6,
  3960. "work_status": 21,
  3961. },
  3962. b"\x00\x10\xfb\xa8",
  3963. "K11+ Vacuum",
  3964. SwitchbotModel.K11_VACUUM,
  3965. ),
  3966. AdvTestCase(
  3967. b"\xb0\xe9\xfe\x8e\x98Oi_\x06\x9a,\x00\x00\x00\x00\xe4\x00\x08\x04\x00\x01\x00\x00",
  3968. None,
  3969. {
  3970. "battery": 95,
  3971. "humidity": 44,
  3972. "sequence_number": 105,
  3973. "humidity_alarm": 0,
  3974. "isOn": False,
  3975. "is_light": True,
  3976. "motion_detected": True,
  3977. "temp_alarm": 0,
  3978. "temperature": 26.6,
  3979. },
  3980. b"\x00\x10\xf3\xd8",
  3981. "Climate Panel",
  3982. SwitchbotModel.CLIMATE_PANEL,
  3983. ),
  3984. AdvTestCase(
  3985. b"\xb0\xe9\xfe\xa2T|6\xe4\x00\x9c\xa3A\x00",
  3986. None,
  3987. {
  3988. "battery": 100,
  3989. "door_open": False,
  3990. "fault_code": 0,
  3991. "isOn": True,
  3992. "last_mode": "comfort",
  3993. "mode": "manual",
  3994. "sequence_number": 54,
  3995. "need_update_temp": False,
  3996. "restarted": False,
  3997. "target_temperature": 35.0,
  3998. "temperature": 28.0,
  3999. },
  4000. b"\x00\x116@",
  4001. "Smart Thermostat Radiator",
  4002. SwitchbotModel.SMART_THERMOSTAT_RADIATOR,
  4003. ),
  4004. AdvTestCase(
  4005. b"\xb0\xe9\xfe\xc3\x1a!:\x01\x11\x1e\x00\x00d\x03",
  4006. None,
  4007. {
  4008. "battery": 100,
  4009. "mqtt_connected": False,
  4010. "sequence_number": 58,
  4011. "soc_version": "1.1.030",
  4012. "step": 0,
  4013. "work_status": 3,
  4014. },
  4015. b"\x00\x10\xe0P",
  4016. "S20 Vacuum",
  4017. SwitchbotModel.S20_VACUUM,
  4018. ),
  4019. AdvTestCase(
  4020. b"\xb0\xe9\xfelf\xaa\x06\xcc\x04V\x00\x8c",
  4021. None,
  4022. {
  4023. "adaptive_state": True,
  4024. "battery_range": ">=60%",
  4025. "duration": 1110,
  4026. "led_state": True,
  4027. "lightLevel": 12,
  4028. "motion_detected": True,
  4029. "sequence_number": 6,
  4030. "trigger_flag": 0,
  4031. },
  4032. b"\x00\x10\xcc\xc8",
  4033. "Presence Sensor",
  4034. SwitchbotModel.PRESENCE_SENSOR,
  4035. ),
  4036. AdvTestCase(
  4037. b"\xb0\xe9\xfe\xe2\xfa8\x157\x03\x08",
  4038. None,
  4039. {
  4040. "battery": 55,
  4041. "battery_charging": False,
  4042. "display_mode": 1,
  4043. "display_size": 0,
  4044. "image_index": 3,
  4045. "last_network_status": 0,
  4046. "sequence_number": 21,
  4047. },
  4048. b"\x00\x11>\x10",
  4049. "Art Frame",
  4050. SwitchbotModel.ART_FRAME,
  4051. ),
  4052. AdvTestCase(
  4053. b"\xb0\xe9\xfe\xe5\x04\x1e\xac\xdf\x00\x00\x00\x00\x00\x02",
  4054. None,
  4055. {
  4056. "battery": 95,
  4057. "battery_charging": True,
  4058. "doorbell": False,
  4059. "doorbell_seq": 0,
  4060. "duress_alarm": False,
  4061. "high_temperature": False,
  4062. "lockout_alarm": False,
  4063. "low_temperature": False,
  4064. "pir_triggered_level": 2,
  4065. "sequence_number": 172,
  4066. "tamper_alarm": False,
  4067. },
  4068. b"\x00\x11\x03x",
  4069. "Keypad Vision",
  4070. SwitchbotModel.KEYPAD_VISION,
  4071. ),
  4072. AdvTestCase(
  4073. b"\xb0\xe9\xfe\xde\xb6\x8c+`\x00\x00\x00\x00\x00\x002",
  4074. None,
  4075. {
  4076. "battery": 96,
  4077. "battery_charging": False,
  4078. "doorbell": False,
  4079. "doorbell_seq": 0,
  4080. "duress_alarm": False,
  4081. "high_temperature": False,
  4082. "lockout_alarm": False,
  4083. "low_temperature": False,
  4084. "radar_triggered_distance": 0,
  4085. "radar_triggered_level": 0,
  4086. "sequence_number": 43,
  4087. "tamper_alarm": False,
  4088. },
  4089. b"\x00\x11Q\x98",
  4090. "Keypad Vision Pro",
  4091. SwitchbotModel.KEYPAD_VISION_PRO,
  4092. ),
  4093. ],
  4094. )
  4095. def test_adv_passive(test_case: AdvTestCase) -> None:
  4096. """Test with passive data."""
  4097. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4098. adv_data = generate_advertisement_data(
  4099. manufacturer_data={2409: test_case.manufacturer_data},
  4100. rssi=-97,
  4101. )
  4102. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  4103. assert result == SwitchBotAdvertisement(
  4104. address="aa:bb:cc:dd:ee:ff",
  4105. data={
  4106. "rawAdvData": None,
  4107. "data": test_case.data,
  4108. "isEncrypted": False,
  4109. "model": test_case.model,
  4110. "modelFriendlyName": test_case.modelFriendlyName,
  4111. "modelName": test_case.modelName,
  4112. },
  4113. device=ble_device,
  4114. rssi=-97,
  4115. active=False,
  4116. )
  4117. @pytest.mark.parametrize(
  4118. "test_case",
  4119. [
  4120. AdvTestCase(
  4121. None,
  4122. b"=\x00\x00\x00",
  4123. {
  4124. 1: {
  4125. "isOn": True,
  4126. "power": 0.0,
  4127. "sequence_number": 138,
  4128. "switchMode": True,
  4129. },
  4130. 2: {
  4131. "isOn": True,
  4132. "power": 7.0,
  4133. "sequence_number": 138,
  4134. "switchMode": True,
  4135. },
  4136. },
  4137. "=",
  4138. "Relay Switch 2PM",
  4139. SwitchbotModel.RELAY_SWITCH_2PM,
  4140. ),
  4141. AdvTestCase(
  4142. None,
  4143. b";\x00\x00\x00",
  4144. {
  4145. "isOn": True,
  4146. "sequence_number": 71,
  4147. "switchMode": True,
  4148. },
  4149. ";",
  4150. "Relay Switch 1",
  4151. SwitchbotModel.RELAY_SWITCH_1,
  4152. ),
  4153. AdvTestCase(
  4154. None,
  4155. b"<\x00\x00\x00",
  4156. {
  4157. "isOn": True,
  4158. "power": 4.9,
  4159. "sequence_number": 71,
  4160. "switchMode": True,
  4161. },
  4162. "<",
  4163. "Relay Switch 1PM",
  4164. SwitchbotModel.RELAY_SWITCH_1PM,
  4165. ),
  4166. AdvTestCase(
  4167. None,
  4168. b">\x00\x00\x00",
  4169. {
  4170. "door_open": True,
  4171. "isOn": False,
  4172. "sequence_number": 15,
  4173. "switchMode": True,
  4174. },
  4175. ">",
  4176. "Garage Door Opener",
  4177. SwitchbotModel.GARAGE_DOOR_OPENER,
  4178. ),
  4179. AdvTestCase(
  4180. None,
  4181. b"\x00\x00\x00\x00\x10\xd0\xb1",
  4182. {},
  4183. b"\x00\x10\xd0\xb1",
  4184. "Strip Light 3",
  4185. SwitchbotModel.STRIP_LIGHT_3,
  4186. ),
  4187. AdvTestCase(
  4188. None,
  4189. b"\x00\x00\x00\x00\x10\xd0\xb0",
  4190. {},
  4191. b"\x00\x10\xd0\xb0",
  4192. "Floor Lamp",
  4193. SwitchbotModel.FLOOR_LAMP,
  4194. ),
  4195. AdvTestCase(
  4196. None,
  4197. b"\x00\x00\x00\x00\x11\x22\xb8",
  4198. {},
  4199. b"\x00\x11\x22\xb8",
  4200. "Candle Warmer Lamp",
  4201. SwitchbotModel.CANDLE_WARMER_LAMP,
  4202. ),
  4203. AdvTestCase(
  4204. None,
  4205. b"q\x00",
  4206. {},
  4207. "q",
  4208. "Ceiling Light",
  4209. SwitchbotModel.CEILING_LIGHT,
  4210. ),
  4211. AdvTestCase(
  4212. None,
  4213. b"r\x00d",
  4214. {},
  4215. "r",
  4216. "Light Strip",
  4217. SwitchbotModel.LIGHT_STRIP,
  4218. ),
  4219. AdvTestCase(
  4220. None,
  4221. b"u\x00d",
  4222. {},
  4223. "u",
  4224. "Color Bulb",
  4225. SwitchbotModel.COLOR_BULB,
  4226. ),
  4227. AdvTestCase(
  4228. None,
  4229. b"?\x00\x00\x00",
  4230. {
  4231. "isOn": True,
  4232. "power": 1254.0,
  4233. "sequence_number": 63,
  4234. "switchMode": True,
  4235. },
  4236. "?",
  4237. "Plug Mini (EU)",
  4238. SwitchbotModel.PLUG_MINI_EU,
  4239. ),
  4240. AdvTestCase(
  4241. None,
  4242. b"\x00\x00\x00\x00\x10\xd0\xb4",
  4243. {},
  4244. b"\x00\x10\xd0\xb4",
  4245. "RGBICWW Floor Lamp",
  4246. SwitchbotModel.RGBICWW_FLOOR_LAMP,
  4247. ),
  4248. AdvTestCase(
  4249. None,
  4250. b"\x00\x00\x00\x00\x10\xd0\xb3",
  4251. {},
  4252. b"\x00\x10\xd0\xb3",
  4253. "RGBICWW Strip Light",
  4254. SwitchbotModel.RGBICWW_STRIP_LIGHT,
  4255. ),
  4256. AdvTestCase(
  4257. None,
  4258. b"\x00\x00\x00\x00\x10\xd0\xb7",
  4259. {},
  4260. b"\x00\x10\xd0\xb7",
  4261. "Permanent Outdoor Light",
  4262. SwitchbotModel.PERMANENT_OUTDOOR_LIGHT,
  4263. ),
  4264. AdvTestCase(
  4265. None,
  4266. b"\x00\x00M\x00\x10\xfb\xa8",
  4267. {},
  4268. b"\x00\x10\xfb\xa8",
  4269. "K11+ Vacuum",
  4270. SwitchbotModel.K11_VACUUM,
  4271. ),
  4272. AdvTestCase(
  4273. None,
  4274. b"\x00 _\x00\x10\xf3\xd8@",
  4275. {},
  4276. b"\x00\x10\xf3\xd8",
  4277. "Climate Panel",
  4278. SwitchbotModel.CLIMATE_PANEL,
  4279. ),
  4280. AdvTestCase(
  4281. None,
  4282. b"\x00 d\x00\x116@",
  4283. {},
  4284. b"\x00\x116@",
  4285. "Smart Thermostat Radiator",
  4286. SwitchbotModel.SMART_THERMOSTAT_RADIATOR,
  4287. ),
  4288. AdvTestCase(
  4289. None,
  4290. b"\x00\x00d\x00\x10\xe0P",
  4291. {},
  4292. b"\x00\x10\xe0P",
  4293. "S20 Vacuum",
  4294. SwitchbotModel.S20_VACUUM,
  4295. ),
  4296. AdvTestCase(
  4297. None,
  4298. b"\x00 d\x00\x10\xcc\xc8",
  4299. {},
  4300. b"\x00\x10\xcc\xc8",
  4301. "Presence Sensor",
  4302. SwitchbotModel.PRESENCE_SENSOR,
  4303. ),
  4304. AdvTestCase(
  4305. None,
  4306. b"\x00\x007\x01\x11>\x10",
  4307. {},
  4308. b"\x01\x11>\x10",
  4309. "Art Frame",
  4310. SwitchbotModel.ART_FRAME,
  4311. ),
  4312. AdvTestCase(
  4313. None,
  4314. b"\x00\x00_\x01\x11\x03x",
  4315. {},
  4316. b"\x01\x11\x03x",
  4317. "Keypad Vision",
  4318. SwitchbotModel.KEYPAD_VISION,
  4319. ),
  4320. AdvTestCase(
  4321. None,
  4322. b"\x00\x00\x00\x00\x10\xd0\xb6",
  4323. {},
  4324. b"\x00\x10\xd0\xb6",
  4325. "RGBIC Neon Rope Light",
  4326. SwitchbotModel.RGBIC_NEON_ROPE_LIGHT,
  4327. ),
  4328. AdvTestCase(
  4329. None,
  4330. b"\x00\x00`\x01\x11Q\x98",
  4331. {},
  4332. b"\x01\x11Q\x98",
  4333. "Keypad Vision Pro",
  4334. SwitchbotModel.KEYPAD_VISION_PRO,
  4335. ),
  4336. ],
  4337. )
  4338. def test_adv_with_empty_data(test_case: AdvTestCase) -> None:
  4339. """Test with empty data."""
  4340. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4341. adv_data = generate_advertisement_data(
  4342. manufacturer_data={2409: None},
  4343. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  4344. rssi=-97,
  4345. )
  4346. result = parse_advertisement_data(ble_device, adv_data)
  4347. assert result == SwitchBotAdvertisement(
  4348. address="aa:bb:cc:dd:ee:ff",
  4349. data={
  4350. "rawAdvData": test_case.service_data,
  4351. "data": {},
  4352. "isEncrypted": False,
  4353. "model": test_case.model,
  4354. },
  4355. device=ble_device,
  4356. rssi=-97,
  4357. active=True,
  4358. )
  4359. def test_parse_advertisement_with_mac_cache() -> None:
  4360. """Test that populating the MAC cache helps identify unknown passive devices."""
  4361. # Clear the cache to ensure clean test
  4362. _MODEL_TO_MAC_CACHE.clear()
  4363. # Create a passive lock device with manufacturer data only (no service data)
  4364. # This would normally not be identifiable without the cache
  4365. mac_address = "C1:64:B8:7D:06:05"
  4366. ble_device = generate_ble_device(mac_address, "WoLock")
  4367. # Lock passive advertisement with manufacturer data
  4368. # This is real data from a WoLock device in passive mode
  4369. adv_data = generate_advertisement_data(
  4370. manufacturer_data={2409: b"\xc1d\xb8}\x06\x05\x00\x00\x00\x00\x00"},
  4371. service_data={},
  4372. rssi=-70,
  4373. )
  4374. # First attempt: Without cache, parser cannot identify the device model
  4375. result_without_cache = parse_advertisement_data(ble_device, adv_data)
  4376. assert result_without_cache is None, "Should not decode without model hint"
  4377. # Now populate the cache with the device's MAC and model
  4378. populate_model_to_mac_cache(mac_address, SwitchbotModel.LOCK)
  4379. # Second attempt: With cache, parser can now identify and decode the device
  4380. result_with_cache = parse_advertisement_data(ble_device, adv_data)
  4381. assert result_with_cache is not None, "Should decode with MAC cache"
  4382. assert result_with_cache.data["modelName"] == SwitchbotModel.LOCK
  4383. assert result_with_cache.data["modelFriendlyName"] == "Lock"
  4384. assert result_with_cache.active is False # Passive advertisement
  4385. # Clean up
  4386. _MODEL_TO_MAC_CACHE.clear()
  4387. def test_parse_advertisement_with_mac_cache_curtain() -> None:
  4388. """Test MAC cache with a passive curtain device."""
  4389. # Clear the cache
  4390. _MODEL_TO_MAC_CACHE.clear()
  4391. # Create a passive curtain device
  4392. mac_address = "CC:F4:C4:F9:AC:6C"
  4393. ble_device = generate_ble_device(mac_address, None)
  4394. # Curtain passive advertisement with only manufacturer data
  4395. adv_data = generate_advertisement_data(
  4396. manufacturer_data={2409: b"\xccOLG\x00c\x00\x00\x11\x00\x00"},
  4397. service_data={},
  4398. rssi=-85,
  4399. )
  4400. # Without cache, cannot identify
  4401. result_without_cache = parse_advertisement_data(ble_device, adv_data)
  4402. assert result_without_cache is None
  4403. # Populate cache
  4404. populate_model_to_mac_cache(mac_address, SwitchbotModel.CURTAIN)
  4405. # With cache, can identify and parse
  4406. result_with_cache = parse_advertisement_data(ble_device, adv_data)
  4407. assert result_with_cache is not None
  4408. assert result_with_cache.data["modelName"] == SwitchbotModel.CURTAIN
  4409. assert result_with_cache.data["modelFriendlyName"] == "Curtain 3"
  4410. assert result_with_cache.active is False
  4411. # Clean up
  4412. _MODEL_TO_MAC_CACHE.clear()
  4413. @pytest.mark.parametrize(
  4414. ("manufacturer_data", "service_data", "model"),
  4415. [
  4416. (b"\xff\xff\xff\xff", b"\xff\xff\xff\xff", None),
  4417. (b"\xff\xff\xff\xff", b"\xff\xff\xff\xff", "F"),
  4418. (b"\xff\xff\xff\xff\xff\xff\xff", b"\xff\xff\xff\xff\xff\xff\xff\xff", None),
  4419. (None, None, None),
  4420. ],
  4421. )
  4422. def test_with_invalid_advertisement(manufacturer_data, service_data, model) -> None:
  4423. """Test with invalid advertisement data."""
  4424. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4425. adv_data = generate_advertisement_data(
  4426. manufacturer_data={2409: manufacturer_data},
  4427. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": service_data},
  4428. rssi=-97,
  4429. )
  4430. result = parse_advertisement_data(ble_device, adv_data, model)
  4431. assert result is None
  4432. def test_weather_station_active() -> None:
  4433. """Test Weather Station active advertisement."""
  4434. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4435. adv_data = generate_advertisement_data(
  4436. manufacturer_data={
  4437. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01\x50\x06\x9a\x23\x00\x00\x00\x00\x00"
  4438. },
  4439. service_data={
  4440. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x50\x00\x10\x53\xb0"
  4441. },
  4442. rssi=-67,
  4443. )
  4444. result = parse_advertisement_data(ble_device, adv_data)
  4445. assert result == SwitchBotAdvertisement(
  4446. address="aa:bb:cc:dd:ee:ff",
  4447. data={
  4448. "data": {
  4449. "battery": 80,
  4450. "fahrenheit": False,
  4451. "humidity": 35,
  4452. "temp": {"c": 26.6, "f": 79.88},
  4453. "temperature": 26.6,
  4454. },
  4455. "isEncrypted": False,
  4456. "model": b"\x00\x10\x53\xb0",
  4457. "modelFriendlyName": "Weather Station",
  4458. "modelName": SwitchbotModel.WEATHER_STATION,
  4459. "rawAdvData": b"\x00\x00\x50\x00\x10\x53\xb0",
  4460. },
  4461. device=ble_device,
  4462. rssi=-67,
  4463. active=True,
  4464. )
  4465. def test_weather_station_passive() -> None:
  4466. """Test Weather Station passive advertisement."""
  4467. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4468. adv_data = generate_advertisement_data(
  4469. manufacturer_data={
  4470. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01\x50\x06\x9a\x23\x00\x00\x00\x00\x00"
  4471. },
  4472. rssi=-67,
  4473. )
  4474. result = parse_advertisement_data(
  4475. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  4476. )
  4477. assert result == SwitchBotAdvertisement(
  4478. address="aa:bb:cc:dd:ee:ff",
  4479. data={
  4480. "data": {
  4481. "battery": 80,
  4482. "fahrenheit": False,
  4483. "humidity": 35,
  4484. "temp": {"c": 26.6, "f": 79.88},
  4485. "temperature": 26.6,
  4486. },
  4487. "isEncrypted": False,
  4488. "model": b"\x00\x10\x53\xb0",
  4489. "modelFriendlyName": "Weather Station",
  4490. "modelName": SwitchbotModel.WEATHER_STATION,
  4491. "rawAdvData": None,
  4492. },
  4493. device=ble_device,
  4494. rssi=-67,
  4495. active=False,
  4496. )
  4497. def test_weather_station_service_data_only() -> None:
  4498. """Test Weather Station with service data only (no manufacturer data)."""
  4499. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4500. adv_data = generate_advertisement_data(
  4501. service_data={
  4502. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x50\x06\x9a\x23\x00\x10\x53\xb0"
  4503. },
  4504. rssi=-67,
  4505. )
  4506. result = parse_advertisement_data(
  4507. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  4508. )
  4509. assert result is not None
  4510. assert result.data["data"]["temperature"] == 26.6
  4511. assert result.data["data"]["humidity"] == 35
  4512. assert result.data["data"]["battery"] == 80
  4513. def test_weather_station_empty_data() -> None:
  4514. """Test Weather Station with empty/zero data returns empty dict."""
  4515. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4516. adv_data = generate_advertisement_data(
  4517. manufacturer_data={
  4518. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01\x00\x00\x80\x00\x00\x00\x00\x00\x00"
  4519. },
  4520. service_data={
  4521. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x00\x00\x10\x53\xb0"
  4522. },
  4523. rssi=-67,
  4524. )
  4525. result = parse_advertisement_data(ble_device, adv_data)
  4526. assert result is not None
  4527. assert result.data["data"] == {}
  4528. def test_weather_station_short_service_data() -> None:
  4529. """Test Weather Station with too-short service data does not raise IndexError."""
  4530. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4531. adv_data = generate_advertisement_data(
  4532. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x01"},
  4533. rssi=-67,
  4534. )
  4535. result = parse_advertisement_data(
  4536. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  4537. )
  4538. assert result is not None
  4539. assert result.data["data"] == {}
  4540. def test_weather_station_no_data() -> None:
  4541. """Test Weather Station with no usable data."""
  4542. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4543. adv_data = generate_advertisement_data(
  4544. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee"},
  4545. rssi=-67,
  4546. )
  4547. result = parse_advertisement_data(
  4548. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  4549. )
  4550. assert result is not None
  4551. assert result.data["data"] == {}
  4552. def test_with_special_manufacturer_data_length() -> None:
  4553. """Test with special manufacturer data length."""
  4554. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4555. adv_data = generate_advertisement_data(
  4556. manufacturer_data={741: b"\xacg\xb2\xcd\xfa\xbe"},
  4557. service_data={
  4558. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\xff\x80\x00\xf9\x80Bc\x00"
  4559. },
  4560. rssi=-97,
  4561. )
  4562. result = parse_advertisement_data(ble_device, adv_data)
  4563. assert result.data["modelName"] == SwitchbotModel.HUMIDIFIER