test_adv_parser.py 168 KB

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