test_adv_parser.py 171 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312
  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\x11\x22\x33\x2a\x88\x18\x64\x00\x91",
  2956. b"\x00\x80\x64\x00\x12\x11\x00",
  2957. {
  2958. "sequence_number": 42,
  2959. "battery": 100,
  2960. "calibration": True,
  2961. "status": LockStatus.UNLOCKED,
  2962. "update_from_secondary_lock": False,
  2963. "door_open": True,
  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": True,
  2969. "unlocked_alarm": False,
  2970. "auto_lock_paused": True,
  2971. "night_latch": False,
  2972. "power_alarm": True,
  2973. "battery_status": 1,
  2974. },
  2975. b"\x00\x12\x11\x00",
  2976. "UWB Lock",
  2977. SwitchbotModel.UWB_LOCK,
  2978. ),
  2979. AdvTestCase(
  2980. b"\xb0\xe9\xfe\x44\x55\x66\x2b\x10\x00\x50\x00\x00",
  2981. b"\x00\x80\x50\x01\x11\x9f\xb8",
  2982. {
  2983. "sequence_number": 43,
  2984. "battery": 80,
  2985. "calibration": False,
  2986. "status": LockStatus.LOCKING,
  2987. "update_from_secondary_lock": False,
  2988. "door_open": False,
  2989. "door_open_from_secondary_lock": False,
  2990. "double_lock_mode": False,
  2991. "is_secondary_lock": False,
  2992. "manual_unlock_linkage": False,
  2993. "unclosed_alarm": False,
  2994. "unlocked_alarm": False,
  2995. "auto_lock_paused": False,
  2996. "night_latch": False,
  2997. "power_alarm": False,
  2998. "battery_status": 0,
  2999. },
  3000. b"\x01\x11\x9f\xb8",
  3001. "Lock Ultra Max",
  3002. SwitchbotModel.LOCK_ULTRA_MAX,
  3003. ),
  3004. AdvTestCase(
  3005. b"\xb0\xe9\xfe\x44\x55\x66\x2b\x10\x00\x50\x00\x00",
  3006. b"\x00\x80\x50\x01\x12\x11\x00",
  3007. {
  3008. "sequence_number": 43,
  3009. "battery": 80,
  3010. "calibration": False,
  3011. "status": LockStatus.LOCKING,
  3012. "update_from_secondary_lock": False,
  3013. "door_open": False,
  3014. "door_open_from_secondary_lock": False,
  3015. "double_lock_mode": False,
  3016. "is_secondary_lock": False,
  3017. "manual_unlock_linkage": False,
  3018. "unclosed_alarm": False,
  3019. "unlocked_alarm": False,
  3020. "auto_lock_paused": False,
  3021. "night_latch": False,
  3022. "power_alarm": False,
  3023. "battery_status": 0,
  3024. },
  3025. b"\x01\x12\x11\x00",
  3026. "UWB Lock",
  3027. SwitchbotModel.UWB_LOCK,
  3028. ),
  3029. ],
  3030. )
  3031. def test_lock_active(test_case: AdvTestCase) -> None:
  3032. """Test lokc series with active data."""
  3033. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3034. adv_data = generate_advertisement_data(
  3035. manufacturer_data={2409: test_case.manufacturer_data},
  3036. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  3037. rssi=-97,
  3038. )
  3039. result = parse_advertisement_data(ble_device, adv_data)
  3040. assert result == SwitchBotAdvertisement(
  3041. address="aa:bb:cc:dd:ee:ff",
  3042. data={
  3043. "rawAdvData": test_case.service_data,
  3044. "data": test_case.data,
  3045. "isEncrypted": False,
  3046. "model": test_case.model,
  3047. "modelFriendlyName": test_case.modelFriendlyName,
  3048. "modelName": test_case.modelName,
  3049. },
  3050. device=ble_device,
  3051. rssi=-97,
  3052. active=True,
  3053. )
  3054. @pytest.mark.parametrize(
  3055. "test_case",
  3056. [
  3057. AdvTestCase(
  3058. b"\xe9\xd5\x11\xb2kS\x17\x93\x08 ",
  3059. b"-\x80d",
  3060. {
  3061. "sequence_number": 23,
  3062. "battery": None,
  3063. "calibration": True,
  3064. "status": LockStatus.UNLOCKED,
  3065. "update_from_secondary_lock": False,
  3066. "double_lock_mode": False,
  3067. "unlocked_alarm": False,
  3068. "night_latch": False,
  3069. },
  3070. "-",
  3071. "Lock Lite",
  3072. SwitchbotModel.LOCK_LITE,
  3073. ),
  3074. AdvTestCase(
  3075. b"\xee\xf5\xe6\t\x8f\xe8\x11\x97\x08 ",
  3076. b"o\x80d",
  3077. {
  3078. "sequence_number": 17,
  3079. "battery": None,
  3080. "calibration": True,
  3081. "status": LockStatus.UNLOCKED,
  3082. "update_from_secondary_lock": False,
  3083. "door_open": True,
  3084. "double_lock_mode": False,
  3085. "unclosed_alarm": False,
  3086. "unlocked_alarm": False,
  3087. "auto_lock_paused": False,
  3088. "night_latch": False,
  3089. },
  3090. "o",
  3091. "Lock",
  3092. SwitchbotModel.LOCK,
  3093. ),
  3094. AdvTestCase(
  3095. b"\xf7a\x07H\xe6\xe8:\x8a\x00d\x00\x00",
  3096. b"$\x80d",
  3097. {
  3098. "sequence_number": 58,
  3099. "battery": 100,
  3100. "calibration": True,
  3101. "status": LockStatus.UNLOCKED,
  3102. "update_from_secondary_lock": False,
  3103. "door_open": False,
  3104. "door_open_from_secondary_lock": False,
  3105. "double_lock_mode": False,
  3106. "is_secondary_lock": False,
  3107. "left_battery_compartment_alarm": 0,
  3108. "right_battery_compartment_alarm": 0,
  3109. "low_temperature_alarm": False,
  3110. "manual_unlock_linkage": False,
  3111. "unclosed_alarm": False,
  3112. "unlocked_alarm": False,
  3113. "auto_lock_paused": False,
  3114. "night_latch": False,
  3115. },
  3116. "$",
  3117. "Lock Pro",
  3118. SwitchbotModel.LOCK_PRO,
  3119. ),
  3120. AdvTestCase(
  3121. b"\xb0\xe9\xfe\xb6j=%\x8204\x00\x04",
  3122. b"\x00\x804\x00\x10\xa5\xb8",
  3123. {
  3124. "sequence_number": 37,
  3125. "battery": 52,
  3126. "calibration": True,
  3127. "status": LockStatus.LOCKED,
  3128. "update_from_secondary_lock": False,
  3129. "door_open": True,
  3130. "door_open_from_secondary_lock": True,
  3131. "double_lock_mode": False,
  3132. "is_secondary_lock": False,
  3133. "manual_unlock_linkage": False,
  3134. "unclosed_alarm": False,
  3135. "unlocked_alarm": False,
  3136. "auto_lock_paused": False,
  3137. "night_latch": False,
  3138. "power_alarm": False,
  3139. "battery_status": 4,
  3140. },
  3141. b"\x00\x10\xa5\xb8",
  3142. "Lock Ultra",
  3143. SwitchbotModel.LOCK_ULTRA,
  3144. ),
  3145. AdvTestCase(
  3146. b"\xb0\xe9\xfe\x6f\xc9\xa6\x04\x00\x00\x2e\x00\x0c\x00\x00\x00\x00",
  3147. b"\x00\x80.\x01\x11i\x08",
  3148. {
  3149. "sequence_number": 4,
  3150. "battery": None,
  3151. "calibration": False,
  3152. "status": LockStatus.LOCKED,
  3153. "update_from_secondary_lock": False,
  3154. "double_lock_mode": False,
  3155. "unlocked_alarm": False,
  3156. "night_latch": False,
  3157. },
  3158. b"\x00\x11\x69\x08",
  3159. "Lock Vision",
  3160. SwitchbotModel.LOCK_VISION,
  3161. ),
  3162. AdvTestCase(
  3163. b"\xb0\xe9\xfe\xe6\x1aq\x04\x00\x003\x00\x0c\x00\x00\x00\x00",
  3164. b"\x00\x803\x01\x11i\t",
  3165. {
  3166. "sequence_number": 4,
  3167. "battery": 51,
  3168. "calibration": False,
  3169. "status": LockStatus.LOCKED,
  3170. "update_from_secondary_lock": False,
  3171. "door_open": False,
  3172. "door_open_from_secondary_lock": False,
  3173. "double_lock_mode": False,
  3174. "is_secondary_lock": False,
  3175. "manual_unlock_linkage": False,
  3176. "unclosed_alarm": False,
  3177. "unlocked_alarm": False,
  3178. "auto_lock_paused": False,
  3179. "night_latch": False,
  3180. "power_alarm": False,
  3181. "battery_status": 4,
  3182. },
  3183. b"\x00\x11\x69\x09",
  3184. "Lock Vision Pro",
  3185. SwitchbotModel.LOCK_VISION_PRO,
  3186. ),
  3187. AdvTestCase(
  3188. b"\xb0\xe9\xfe\xb0^\xfe\x0e\x02\x00=\x00\x08",
  3189. b"\x00\x80=\x00\x10\xff\x90",
  3190. {
  3191. "sequence_number": 14,
  3192. "battery": 61,
  3193. "calibration": False,
  3194. "status": LockStatus.LOCKED,
  3195. "update_from_secondary_lock": False,
  3196. "door_open": False,
  3197. "door_open_from_secondary_lock": False,
  3198. "double_lock_mode": False,
  3199. "is_secondary_lock": False,
  3200. "left_battery_compartment_alarm": 0,
  3201. "right_battery_compartment_alarm": 0,
  3202. "low_temperature_alarm": False,
  3203. "manual_unlock_linkage": False,
  3204. "unclosed_alarm": False,
  3205. "unlocked_alarm": False,
  3206. "auto_lock_paused": False,
  3207. "night_latch": False,
  3208. },
  3209. b"\x00\x10\xff\x90",
  3210. "Lock Pro Wifi",
  3211. SwitchbotModel.LOCK_PRO_WIFI,
  3212. ),
  3213. AdvTestCase(
  3214. b"\xb0\xe9\xfe\x11\x22\x33\x2a\x88\x18\x64\x00\x91",
  3215. b"\x00\x80\x64\x00\x11\x9f\xb8",
  3216. {
  3217. "sequence_number": 42,
  3218. "battery": 100,
  3219. "calibration": True,
  3220. "status": LockStatus.UNLOCKED,
  3221. "update_from_secondary_lock": False,
  3222. "door_open": True,
  3223. "door_open_from_secondary_lock": False,
  3224. "double_lock_mode": False,
  3225. "is_secondary_lock": False,
  3226. "manual_unlock_linkage": False,
  3227. "unclosed_alarm": True,
  3228. "unlocked_alarm": False,
  3229. "auto_lock_paused": True,
  3230. "night_latch": False,
  3231. "power_alarm": True,
  3232. "battery_status": 1,
  3233. },
  3234. b"\x00\x11\x9f\xb8",
  3235. "Lock Ultra Max",
  3236. SwitchbotModel.LOCK_ULTRA_MAX,
  3237. ),
  3238. AdvTestCase(
  3239. b"\xb0\xe9\xfe\x11\x22\x33\x2a\x88\x18\x64\x00\x91",
  3240. b"\x00\x80\x64\x00\x12\x11\x00",
  3241. {
  3242. "sequence_number": 42,
  3243. "battery": 100,
  3244. "calibration": True,
  3245. "status": LockStatus.UNLOCKED,
  3246. "update_from_secondary_lock": False,
  3247. "door_open": True,
  3248. "door_open_from_secondary_lock": False,
  3249. "double_lock_mode": False,
  3250. "is_secondary_lock": False,
  3251. "manual_unlock_linkage": False,
  3252. "unclosed_alarm": True,
  3253. "unlocked_alarm": False,
  3254. "auto_lock_paused": True,
  3255. "night_latch": False,
  3256. "power_alarm": True,
  3257. "battery_status": 1,
  3258. },
  3259. b"\x00\x12\x11\x00",
  3260. "UWB Lock",
  3261. SwitchbotModel.UWB_LOCK,
  3262. ),
  3263. ],
  3264. )
  3265. def test_lock_passive(test_case: AdvTestCase) -> None:
  3266. """Test lokc series with passive data."""
  3267. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3268. adv_data = generate_advertisement_data(
  3269. manufacturer_data={2409: test_case.manufacturer_data},
  3270. rssi=-97,
  3271. )
  3272. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  3273. assert result == SwitchBotAdvertisement(
  3274. address="aa:bb:cc:dd:ee:ff",
  3275. data={
  3276. "rawAdvData": None,
  3277. "data": test_case.data,
  3278. "isEncrypted": False,
  3279. "model": test_case.model,
  3280. "modelFriendlyName": test_case.modelFriendlyName,
  3281. "modelName": test_case.modelName,
  3282. },
  3283. device=ble_device,
  3284. rssi=-97,
  3285. active=False,
  3286. )
  3287. @pytest.mark.parametrize(
  3288. "test_case",
  3289. [
  3290. AdvTestCase(
  3291. None,
  3292. b"-\x80d",
  3293. {},
  3294. "-",
  3295. "Lock Lite",
  3296. SwitchbotModel.LOCK_LITE,
  3297. ),
  3298. AdvTestCase(
  3299. None,
  3300. b"o\x80d",
  3301. {},
  3302. "o",
  3303. "Lock",
  3304. SwitchbotModel.LOCK,
  3305. ),
  3306. AdvTestCase(
  3307. None,
  3308. b"$\x80d",
  3309. {},
  3310. "$",
  3311. "Lock Pro",
  3312. SwitchbotModel.LOCK_PRO,
  3313. ),
  3314. AdvTestCase(
  3315. None,
  3316. b"\x00\x804\x00\x10\xa5\xb8",
  3317. {},
  3318. b"\x00\x10\xa5\xb8",
  3319. "Lock Ultra",
  3320. SwitchbotModel.LOCK_ULTRA,
  3321. ),
  3322. AdvTestCase(
  3323. None,
  3324. b"\x00\x80\x2e\x00\x11\x69\x08",
  3325. {},
  3326. b"\x00\x11\x69\x08",
  3327. "Lock Vision",
  3328. SwitchbotModel.LOCK_VISION,
  3329. ),
  3330. AdvTestCase(
  3331. None,
  3332. b"\x00\x803\x00\x11i\t",
  3333. {},
  3334. b"\x00\x11\x69\x09",
  3335. "Lock Vision Pro",
  3336. SwitchbotModel.LOCK_VISION_PRO,
  3337. ),
  3338. AdvTestCase(
  3339. None,
  3340. b"\x00\x80=\x00\x10\xff\x90",
  3341. {},
  3342. b"\x00\x10\xff\x90",
  3343. "Lock Pro Wifi",
  3344. SwitchbotModel.LOCK_PRO_WIFI,
  3345. ),
  3346. AdvTestCase(
  3347. None,
  3348. b"\x00\x80\x64\x00\x11\x9f\xb8",
  3349. {},
  3350. b"\x00\x11\x9f\xb8",
  3351. "Lock Ultra Max",
  3352. SwitchbotModel.LOCK_ULTRA_MAX,
  3353. ),
  3354. AdvTestCase(
  3355. None,
  3356. b"\x00\x80\x64\x00\x12\x11\x00",
  3357. {},
  3358. b"\x00\x12\x11\x00",
  3359. "UWB Lock",
  3360. SwitchbotModel.UWB_LOCK,
  3361. ),
  3362. ],
  3363. )
  3364. def test_lock_with_empty_data(test_case: AdvTestCase) -> None:
  3365. """Test lokc series with empty data."""
  3366. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3367. adv_data = generate_advertisement_data(
  3368. manufacturer_data={2409: test_case.manufacturer_data},
  3369. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  3370. rssi=-97,
  3371. )
  3372. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  3373. assert result == SwitchBotAdvertisement(
  3374. address="aa:bb:cc:dd:ee:ff",
  3375. data={
  3376. "rawAdvData": test_case.service_data,
  3377. "data": {},
  3378. "isEncrypted": False,
  3379. "model": test_case.model,
  3380. },
  3381. device=ble_device,
  3382. rssi=-97,
  3383. active=True,
  3384. )
  3385. def test_blind_tilt_active() -> None:
  3386. """Test parsing blind tilt with active data."""
  3387. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3388. adv_data = generate_advertisement_data(
  3389. manufacturer_data={2409: b"\xfc(\\6l\x7f\x0b'\x00\xa1\x84"},
  3390. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"x\x00H"},
  3391. rssi=-97,
  3392. )
  3393. result = parse_advertisement_data(ble_device, adv_data)
  3394. assert result == SwitchBotAdvertisement(
  3395. address="aa:bb:cc:dd:ee:ff",
  3396. data={
  3397. "rawAdvData": b"x\x00H",
  3398. "data": {
  3399. "sequence_number": 11,
  3400. "battery": 72,
  3401. "tilt": 0,
  3402. "inMotion": False,
  3403. "calibration": True,
  3404. "lightLevel": 10,
  3405. },
  3406. "isEncrypted": False,
  3407. "model": "x",
  3408. "modelFriendlyName": "Blind Tilt",
  3409. "modelName": SwitchbotModel.BLIND_TILT,
  3410. },
  3411. device=ble_device,
  3412. rssi=-97,
  3413. active=True,
  3414. )
  3415. def test_blind_tilt_passive() -> None:
  3416. """Test parsing blind tilt with passive data."""
  3417. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3418. adv_data = generate_advertisement_data(
  3419. manufacturer_data={2409: b"\xfc(\\6l\x7f\x0b'\x00\xa1\x84"},
  3420. rssi=-97,
  3421. )
  3422. result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.BLIND_TILT)
  3423. assert result == SwitchBotAdvertisement(
  3424. address="aa:bb:cc:dd:ee:ff",
  3425. data={
  3426. "rawAdvData": None,
  3427. "data": {
  3428. "sequence_number": 11,
  3429. "battery": None,
  3430. "tilt": 0,
  3431. "inMotion": False,
  3432. "calibration": True,
  3433. "lightLevel": 10,
  3434. },
  3435. "isEncrypted": False,
  3436. "model": "x",
  3437. "modelFriendlyName": "Blind Tilt",
  3438. "modelName": SwitchbotModel.BLIND_TILT,
  3439. },
  3440. device=ble_device,
  3441. rssi=-97,
  3442. active=False,
  3443. )
  3444. def test_blind_tilt_with_empty_data() -> None:
  3445. """Test parsing blind tilt with empty data."""
  3446. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3447. adv_data = generate_advertisement_data(
  3448. manufacturer_data={2409: None},
  3449. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"x\x00H"},
  3450. rssi=-97,
  3451. )
  3452. result = parse_advertisement_data(ble_device, adv_data)
  3453. assert result == SwitchBotAdvertisement(
  3454. address="aa:bb:cc:dd:ee:ff",
  3455. data={
  3456. "rawAdvData": b"x\x00H",
  3457. "data": {},
  3458. "isEncrypted": False,
  3459. "model": "x",
  3460. },
  3461. device=ble_device,
  3462. rssi=-97,
  3463. active=True,
  3464. )
  3465. @pytest.mark.parametrize(
  3466. "test_case",
  3467. [
  3468. AdvTestCase(
  3469. b"\xa0\xa3\xb3,\x9c\xe68\x86\x88\xb5\x99\x12\x10\x1b\x00\x85]",
  3470. b"#\x00\x00\x15\x1c\x00",
  3471. {
  3472. "seq_number": 56,
  3473. "isOn": True,
  3474. "mode": HumidifierMode(6),
  3475. "over_humidify_protection": True,
  3476. "child_lock": False,
  3477. "tank_removed": False,
  3478. "tilted_alert": False,
  3479. "filter_missing": False,
  3480. "is_meter_binded": True,
  3481. "humidity": 53,
  3482. "temperature": 25.1,
  3483. "temp": {"c": 25.1, "f": 77.18},
  3484. "water_level": "medium",
  3485. "filter_run_time": datetime.timedelta(days=1, seconds=10800),
  3486. "filter_alert": False,
  3487. "target_humidity": 93,
  3488. },
  3489. "#",
  3490. "Evaporative Humidifier",
  3491. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3492. ),
  3493. AdvTestCase(
  3494. b"\xa0\xa3\xb3,\x9c\xe6H\x86\x80\x7f\xff\xf2\x10\x1d\x00\x874",
  3495. b"#\x00\x00\x15\x1c\x00",
  3496. {
  3497. "seq_number": 72,
  3498. "isOn": True,
  3499. "mode": HumidifierMode(6),
  3500. "over_humidify_protection": True,
  3501. "child_lock": False,
  3502. "tank_removed": False,
  3503. "tilted_alert": False,
  3504. "filter_missing": False,
  3505. "is_meter_binded": False,
  3506. "humidity": None,
  3507. "temperature": None,
  3508. "temp": {"c": None, "f": None},
  3509. "water_level": "medium",
  3510. "filter_run_time": datetime.timedelta(days=1, seconds=18000),
  3511. "filter_alert": False,
  3512. "target_humidity": 52,
  3513. },
  3514. "#",
  3515. "Evaporative Humidifier",
  3516. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3517. ),
  3518. AdvTestCase(
  3519. b"\xa0\xa3\xb3,\x9c\xe6H\x86\x80\xff\xff\xf2\x10\x1d\x00\x874",
  3520. b"#\x00\x00\x15\x1c\x00",
  3521. {
  3522. "seq_number": 72,
  3523. "isOn": True,
  3524. "mode": HumidifierMode(6),
  3525. "over_humidify_protection": True,
  3526. "child_lock": False,
  3527. "tank_removed": False,
  3528. "tilted_alert": False,
  3529. "filter_missing": False,
  3530. "is_meter_binded": True,
  3531. "humidity": None,
  3532. "temperature": None,
  3533. "temp": {"c": None, "f": None},
  3534. "water_level": "medium",
  3535. "filter_run_time": datetime.timedelta(days=1, seconds=18000),
  3536. "filter_alert": False,
  3537. "target_humidity": 52,
  3538. },
  3539. "#",
  3540. "Evaporative Humidifier",
  3541. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3542. ),
  3543. AdvTestCase(
  3544. b"\xacg\xb2\xcd\xfa\xbe",
  3545. b"e\x80\x00\xf9\x80Bc\x00",
  3546. {
  3547. "isOn": True,
  3548. "level": 128,
  3549. "switchMode": True,
  3550. },
  3551. "e",
  3552. "Humidifier",
  3553. SwitchbotModel.HUMIDIFIER,
  3554. ),
  3555. ],
  3556. )
  3557. def test_humidifer_active(test_case: AdvTestCase) -> None:
  3558. """Test humidifier with active data."""
  3559. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3560. adv_data = generate_advertisement_data(
  3561. manufacturer_data={2409: test_case.manufacturer_data},
  3562. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  3563. rssi=-97,
  3564. )
  3565. result = parse_advertisement_data(ble_device, adv_data)
  3566. assert result == SwitchBotAdvertisement(
  3567. address="aa:bb:cc:dd:ee:ff",
  3568. data={
  3569. "rawAdvData": test_case.service_data,
  3570. "data": test_case.data,
  3571. "isEncrypted": False,
  3572. "model": test_case.model,
  3573. "modelFriendlyName": test_case.modelFriendlyName,
  3574. "modelName": test_case.modelName,
  3575. },
  3576. device=ble_device,
  3577. rssi=-97,
  3578. active=True,
  3579. )
  3580. @pytest.mark.parametrize(
  3581. "test_case",
  3582. [
  3583. AdvTestCase(
  3584. b"\xa0\xa3\xb3,\x9c\xe68\x86\x88\xb5\x99\x12\x10\x1b\x00\x85]",
  3585. b"#\x00\x00\x15\x1c\x00",
  3586. {
  3587. "seq_number": 56,
  3588. "isOn": True,
  3589. "mode": HumidifierMode(6),
  3590. "over_humidify_protection": True,
  3591. "child_lock": False,
  3592. "tank_removed": False,
  3593. "tilted_alert": False,
  3594. "filter_missing": False,
  3595. "is_meter_binded": True,
  3596. "humidity": 53,
  3597. "temperature": 25.1,
  3598. "temp": {"c": 25.1, "f": 77.18},
  3599. "water_level": "medium",
  3600. "filter_run_time": datetime.timedelta(days=1, seconds=10800),
  3601. "filter_alert": False,
  3602. "target_humidity": 93,
  3603. },
  3604. "#",
  3605. "Evaporative Humidifier",
  3606. SwitchbotModel.EVAPORATIVE_HUMIDIFIER,
  3607. ),
  3608. AdvTestCase(
  3609. b"\xacg\xb2\xcd\xfa\xbe",
  3610. b"e\x80\x00\xf9\x80Bc\x00",
  3611. {
  3612. "isOn": None,
  3613. "level": None,
  3614. "switchMode": True,
  3615. },
  3616. "e",
  3617. "Humidifier",
  3618. SwitchbotModel.HUMIDIFIER,
  3619. ),
  3620. ],
  3621. )
  3622. def test_humidifer_passive(test_case: AdvTestCase) -> None:
  3623. """Test humidifier with passive data."""
  3624. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3625. adv_data = generate_advertisement_data(
  3626. manufacturer_data={2409: test_case.manufacturer_data},
  3627. rssi=-97,
  3628. )
  3629. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  3630. assert result == SwitchBotAdvertisement(
  3631. address="aa:bb:cc:dd:ee:ff",
  3632. data={
  3633. "rawAdvData": None,
  3634. "data": test_case.data,
  3635. "isEncrypted": False,
  3636. "model": test_case.model,
  3637. "modelFriendlyName": test_case.modelFriendlyName,
  3638. "modelName": test_case.modelName,
  3639. },
  3640. device=ble_device,
  3641. rssi=-97,
  3642. active=False,
  3643. )
  3644. def test_humidifer_with_empty_data() -> None:
  3645. """Test humidifier with empty data."""
  3646. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  3647. adv_data = generate_advertisement_data(
  3648. manufacturer_data={2409: None},
  3649. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"#\x00\x00\x15\x1c\x00"},
  3650. rssi=-97,
  3651. )
  3652. result = parse_advertisement_data(ble_device, adv_data)
  3653. assert result == SwitchBotAdvertisement(
  3654. address="aa:bb:cc:dd:ee:ff",
  3655. data={
  3656. "rawAdvData": b"#\x00\x00\x15\x1c\x00",
  3657. "data": {},
  3658. "isEncrypted": False,
  3659. "model": "#",
  3660. },
  3661. device=ble_device,
  3662. rssi=-97,
  3663. active=True,
  3664. )
  3665. @pytest.mark.parametrize(
  3666. "test_case",
  3667. [
  3668. AdvTestCase(
  3669. b"\xc0N0\xdd\xb9\xf2\x8a\xc1\x00\x00\x00\x00\x00F\x00\x00",
  3670. b"=\x00\x00\x00",
  3671. {
  3672. 1: {
  3673. "isOn": True,
  3674. "sequence_number": 138,
  3675. "switchMode": True,
  3676. "power": 0.0,
  3677. },
  3678. 2: {
  3679. "isOn": True,
  3680. "sequence_number": 138,
  3681. "switchMode": True,
  3682. "power": 70.0,
  3683. },
  3684. "sequence_number": 138,
  3685. },
  3686. "=",
  3687. "Relay Switch 2PM",
  3688. SwitchbotModel.RELAY_SWITCH_2PM,
  3689. ),
  3690. AdvTestCase(
  3691. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  3692. b";\x00\x00\x00",
  3693. {
  3694. "isOn": True,
  3695. "sequence_number": 71,
  3696. "switchMode": True,
  3697. },
  3698. ";",
  3699. "Relay Switch 1",
  3700. SwitchbotModel.RELAY_SWITCH_1,
  3701. ),
  3702. AdvTestCase(
  3703. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  3704. b"<\x00\x00\x00",
  3705. {
  3706. "isOn": True,
  3707. "sequence_number": 71,
  3708. "switchMode": True,
  3709. "power": 49.0,
  3710. },
  3711. "<",
  3712. "Relay Switch 1PM",
  3713. SwitchbotModel.RELAY_SWITCH_1PM,
  3714. ),
  3715. AdvTestCase(
  3716. b"$X|\x05BN\x0f\x00\x00\x03\x00\x00\x00\x00\x00\x00",
  3717. b">\x00\x00\x00",
  3718. {
  3719. "door_open": True,
  3720. "isOn": False,
  3721. "sequence_number": 15,
  3722. "switchMode": True,
  3723. },
  3724. ">",
  3725. "Garage Door Opener",
  3726. SwitchbotModel.GARAGE_DOOR_OPENER,
  3727. ),
  3728. AdvTestCase(
  3729. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  3730. b"\x00\x00\x00\x00\x10\xd0\xb1",
  3731. {
  3732. "sequence_number": 133,
  3733. "isOn": True,
  3734. "brightness": 30,
  3735. "delay": False,
  3736. "network_state": 2,
  3737. "color_mode": 2,
  3738. "cw": 4753,
  3739. },
  3740. b"\x00\x10\xd0\xb1",
  3741. "Strip Light 3",
  3742. SwitchbotModel.STRIP_LIGHT_3,
  3743. ),
  3744. AdvTestCase(
  3745. b'\xa0\x85\xe3e,\x06P\xaa"\xd4\x00\x00\x00\x00\x00\x00\r\x93\x00',
  3746. b"\x00\x00\x00\x00\x10\xd0\xb0",
  3747. {
  3748. "sequence_number": 80,
  3749. "isOn": True,
  3750. "brightness": 42,
  3751. "delay": False,
  3752. "network_state": 2,
  3753. "color_mode": 2,
  3754. "cw": 3475,
  3755. },
  3756. b"\x00\x10\xd0\xb0",
  3757. "Floor Lamp",
  3758. SwitchbotModel.FLOOR_LAMP,
  3759. ),
  3760. AdvTestCase(
  3761. b"\x90\xe5\xb1h\xda\xaa\n\xb0 \x00",
  3762. b"\x00\x00\x00\x00\x11\x22\xb8",
  3763. {
  3764. "brightness": 48,
  3765. "delay": False,
  3766. "isOn": True,
  3767. "network_state": 2,
  3768. "sequence_number": 10,
  3769. },
  3770. b"\x00\x11\x22\xb8",
  3771. "Candle Warmer Lamp",
  3772. SwitchbotModel.CANDLE_WARMER_LAMP,
  3773. ),
  3774. AdvTestCase(
  3775. b"\xef\xfe\xfb\x9d\x10\xfe\n\x01\x18\xf3$",
  3776. b"q\x00",
  3777. {
  3778. "brightness": 1,
  3779. "color_mode": 1,
  3780. "cw": 6387,
  3781. "isOn": False,
  3782. "sequence_number": 10,
  3783. },
  3784. "q",
  3785. "Ceiling Light",
  3786. SwitchbotModel.CEILING_LIGHT,
  3787. ),
  3788. AdvTestCase(
  3789. b'|,g\xc8\x15&jR"l\x00\x00\x00\x00\x00\x00',
  3790. b"r\x00d",
  3791. {
  3792. "brightness": 82,
  3793. "color_mode": 2,
  3794. "delay": False,
  3795. "isOn": False,
  3796. "sequence_number": 106,
  3797. "network_state": 2,
  3798. },
  3799. "r",
  3800. "Light Strip",
  3801. SwitchbotModel.LIGHT_STRIP,
  3802. ),
  3803. AdvTestCase(
  3804. b"@L\xca\xa7_\x12\x02\x81\x12\x00\x00",
  3805. b"u\x00d",
  3806. {
  3807. "brightness": 1,
  3808. "color_mode": 2,
  3809. "delay": False,
  3810. "isOn": True,
  3811. "loop_index": 0,
  3812. "preset": False,
  3813. "sequence_number": 2,
  3814. "speed": 0,
  3815. },
  3816. "u",
  3817. "Color Bulb",
  3818. SwitchbotModel.COLOR_BULB,
  3819. ),
  3820. AdvTestCase(
  3821. b"\x94\xa9\x90T\x85^?\xa1\x00\x00\x04\xe6\x00\x00\x00\x00",
  3822. b"?\x00\x00\x00",
  3823. {
  3824. "isOn": True,
  3825. "power": 1254.0,
  3826. "sequence_number": 63,
  3827. "switchMode": True,
  3828. },
  3829. "?",
  3830. "Plug Mini (EU)",
  3831. SwitchbotModel.PLUG_MINI_EU,
  3832. ),
  3833. AdvTestCase(
  3834. b'\xdc\x06u\xa6\xfb\xb2y\x9e"\x00\x11\xb8\x00',
  3835. b"\x00\x00\x00\x00\x10\xd0\xb4",
  3836. {
  3837. "sequence_number": 121,
  3838. "isOn": True,
  3839. "brightness": 30,
  3840. "delay": False,
  3841. "network_state": 2,
  3842. "color_mode": 2,
  3843. "cw": 4536,
  3844. },
  3845. b"\x00\x10\xd0\xb4",
  3846. "RGBICWW Floor Lamp",
  3847. SwitchbotModel.RGBICWW_FLOOR_LAMP,
  3848. ),
  3849. AdvTestCase(
  3850. b'(7/L\x94\xb2\x0c\x9e"\x00\x11:\x00',
  3851. b"\x00\x00\x00\x00\x10\xd0\xb3",
  3852. {
  3853. "sequence_number": 12,
  3854. "isOn": True,
  3855. "brightness": 30,
  3856. "delay": False,
  3857. "network_state": 2,
  3858. "color_mode": 2,
  3859. "cw": 4410,
  3860. },
  3861. b"\x00\x10\xd0\xb3",
  3862. "RGBICWW Strip Light",
  3863. SwitchbotModel.RGBICWW_STRIP_LIGHT,
  3864. ),
  3865. AdvTestCase(
  3866. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  3867. b"\x00\x00\x00\x00\x10\xd0\xb7",
  3868. {
  3869. "sequence_number": 133,
  3870. "isOn": True,
  3871. "brightness": 30,
  3872. "delay": False,
  3873. "network_state": 2,
  3874. "color_mode": 2,
  3875. "cw": 0,
  3876. },
  3877. b"\x00\x10\xd0\xb7",
  3878. "Permanent Outdoor Light",
  3879. SwitchbotModel.PERMANENT_OUTDOOR_LIGHT,
  3880. ),
  3881. AdvTestCase(
  3882. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  3883. b"\x00\x00\x00\x00\x10\xd0\xb6",
  3884. {
  3885. "sequence_number": 47,
  3886. "isOn": True,
  3887. "brightness": 11,
  3888. "delay": False,
  3889. "network_state": 2,
  3890. "color_mode": 7,
  3891. },
  3892. b"\x00\x10\xd0\xb6",
  3893. "RGBIC Neon Rope Light",
  3894. SwitchbotModel.RGBIC_NEON_ROPE_LIGHT,
  3895. ),
  3896. AdvTestCase(
  3897. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  3898. b"\x00\x00\x00\x00\x10\xd0\xb5",
  3899. {
  3900. "sequence_number": 47,
  3901. "isOn": True,
  3902. "brightness": 11,
  3903. "delay": False,
  3904. "network_state": 2,
  3905. "color_mode": 7,
  3906. },
  3907. b"\x00\x10\xd0\xb5",
  3908. "RGBIC Neon Wire Rope Light",
  3909. SwitchbotModel.RGBIC_NEON_WIRE_ROPE_LIGHT,
  3910. ),
  3911. AdvTestCase(
  3912. b'(7/L\x94\xb2\x0c\x9e"\x00\x11:\x00\xa0',
  3913. b"\x00\x00\x00\x00\x11\xbb\x10",
  3914. {
  3915. "sequence_number": 12,
  3916. "isOn": True,
  3917. "brightness": 30,
  3918. "delay": False,
  3919. "network_state": 2,
  3920. "color_mode": 2,
  3921. "cw": 4410,
  3922. "main_isOn": True,
  3923. "main_brightness": 32,
  3924. },
  3925. b"\x00\x11\xbb\x10",
  3926. "RGBICWW Ceiling Light",
  3927. SwitchbotModel.RGBICWW_CEILING_LIGHT,
  3928. ),
  3929. AdvTestCase(
  3930. b"\xb0\xe9\xfe\xe4\xbf\xd8\x0b\x01\x11f\x00\x16M\x15",
  3931. b"\x00\x00M\x00\x10\xfb\xa8",
  3932. {
  3933. "battery": 77,
  3934. "mqtt_connected": True,
  3935. "sequence_number": 11,
  3936. "soc_version": "1.1.102",
  3937. "step": 6,
  3938. "work_status": 21,
  3939. },
  3940. b"\x00\x10\xfb\xa8",
  3941. "K11+ Vacuum",
  3942. SwitchbotModel.K11_VACUUM,
  3943. ),
  3944. AdvTestCase(
  3945. b"\xb0\xe9\xfe\x848\x05\x06\x14\x05\x99-\x00\x00\x00\x00\xd4\x00\x0c\x04\x00",
  3946. b"\x00 _\x00\x10\xf3\xd8@",
  3947. {
  3948. "battery": 20,
  3949. "humidity": 45,
  3950. "sequence_number": 6,
  3951. "humidity_alarm": 0,
  3952. "isOn": False,
  3953. "is_light": True,
  3954. "motion_detected": True,
  3955. "temp_alarm": 0,
  3956. "temperature": 25.5,
  3957. "on_keystate": 0,
  3958. "off_keystate": 0,
  3959. "on_keystate_mode": 0,
  3960. "on_keystate_counter": 0,
  3961. "off_keystate_mode": 0,
  3962. "off_keystate_counter": 0,
  3963. },
  3964. b"\x00\x10\xf3\xd8",
  3965. "Climate Panel",
  3966. SwitchbotModel.CLIMATE_PANEL,
  3967. ),
  3968. AdvTestCase(
  3969. b"\xb0\xe9\xfe\x848\x05\x13\x14\t\x99-\x00\x00\x00\x00\xd2\x01\x14\x04\x00",
  3970. b"\x00 _\x00\x10\xf3\xd8@",
  3971. {
  3972. "battery": 20,
  3973. "humidity": 45,
  3974. "sequence_number": 19,
  3975. "humidity_alarm": 0,
  3976. "isOn": False,
  3977. "is_light": False,
  3978. "motion_detected": True,
  3979. "temp_alarm": 0,
  3980. "temperature": 25.9,
  3981. "on_keystate": 0,
  3982. "off_keystate": 0,
  3983. "on_keystate_mode": 0,
  3984. "on_keystate_counter": 0,
  3985. "off_keystate_mode": 0,
  3986. "off_keystate_counter": 0,
  3987. },
  3988. b"\x00\x10\xf3\xd8",
  3989. "Climate Panel",
  3990. SwitchbotModel.CLIMATE_PANEL,
  3991. ),
  3992. AdvTestCase(
  3993. b"\xb0\xe9\xfe\x84\x38\x05\x06\x14\x05\x99\x2d\x00\x00\x25\x62\xd4\x00\x0c\x04\x00",
  3994. b"\x00 _\x00\x10\xf3\xd8@",
  3995. {
  3996. "battery": 20,
  3997. "humidity": 45,
  3998. "sequence_number": 6,
  3999. "humidity_alarm": 0,
  4000. "isOn": False,
  4001. "is_light": True,
  4002. "motion_detected": True,
  4003. "temp_alarm": 0,
  4004. "temperature": 25.5,
  4005. "on_keystate": 37,
  4006. "off_keystate": 98,
  4007. "on_keystate_mode": 1,
  4008. "on_keystate_counter": 5,
  4009. "off_keystate_mode": 3,
  4010. "off_keystate_counter": 2,
  4011. },
  4012. b"\x00\x10\xf3\xd8",
  4013. "Climate Panel",
  4014. SwitchbotModel.CLIMATE_PANEL,
  4015. ),
  4016. AdvTestCase(
  4017. b"\xb0\xe9\xfe\xa2T|6\xe4\x00\x9c\xa3A\x00",
  4018. b"\x00 d\x00\x116@",
  4019. {
  4020. "battery": 100,
  4021. "door_open": False,
  4022. "fault_code": 0,
  4023. "isOn": True,
  4024. "last_mode": "comfort",
  4025. "mode": "manual",
  4026. "sequence_number": 54,
  4027. "need_update_temp": False,
  4028. "restarted": False,
  4029. "target_temperature": 35.0,
  4030. "temperature": 28.0,
  4031. },
  4032. b"\x00\x116@",
  4033. "Smart Thermostat Radiator",
  4034. SwitchbotModel.SMART_THERMOSTAT_RADIATOR,
  4035. ),
  4036. AdvTestCase(
  4037. b"\xb0\xe9\xfe\xc3\x1a!:\x01\x11\x1e\x00\x00d\x03",
  4038. b"\x00\x00d\x00\x10\xe0P",
  4039. {
  4040. "battery": 100,
  4041. "mqtt_connected": False,
  4042. "sequence_number": 58,
  4043. "soc_version": "1.1.030",
  4044. "step": 0,
  4045. "work_status": 3,
  4046. },
  4047. b"\x00\x10\xe0P",
  4048. "S20 Vacuum",
  4049. SwitchbotModel.S20_VACUUM,
  4050. ),
  4051. AdvTestCase(
  4052. b"\xb0\xe9\xfelf\xaa\x06\xcc\x04V\x00\x8c",
  4053. b"\x00 d\x00\x10\xcc\xc8",
  4054. {
  4055. "adaptive_state": True,
  4056. "battery": 100,
  4057. "battery_range": ">=60%",
  4058. "duration": 1110,
  4059. "led_state": True,
  4060. "lightLevel": 12,
  4061. "motion_detected": True,
  4062. "sequence_number": 6,
  4063. "trigger_flag": 0,
  4064. },
  4065. b"\x00\x10\xcc\xc8",
  4066. "Presence Sensor",
  4067. SwitchbotModel.PRESENCE_SENSOR,
  4068. ),
  4069. AdvTestCase(
  4070. b"\xb0\xe9\xfeR\xdd\x84\x06d\x08\x97,\x00\x05",
  4071. b"\x14\x00d",
  4072. {
  4073. "battery": 100,
  4074. "fahrenheit": False,
  4075. "humidity": 44,
  4076. "temp": {"c": 23.8, "f": 74.84},
  4077. "temperature": 23.8,
  4078. },
  4079. b"\x14",
  4080. "Meter Pro",
  4081. SwitchbotModel.METER_PRO,
  4082. ),
  4083. AdvTestCase(
  4084. b"\xb0\xe9\xfe\xe2\xfa8\x157\x03\x08",
  4085. b"\x00\x007\x01\x11>\x10",
  4086. {
  4087. "battery": 55,
  4088. "battery_charging": False,
  4089. "display_mode": 1,
  4090. "display_size": 0,
  4091. "image_index": 3,
  4092. "last_network_status": 0,
  4093. "sequence_number": 21,
  4094. },
  4095. b"\x01\x11>\x10",
  4096. "Art Frame",
  4097. SwitchbotModel.ART_FRAME,
  4098. ),
  4099. AdvTestCase(
  4100. b"\xb0\xe9\xfe\xe5\x04\x1e\xac\xdf\x00\x00\x00\x00\x00\x02",
  4101. b"\x00\x00_\x01\x11\x03x",
  4102. {
  4103. "battery": 95,
  4104. "battery_charging": True,
  4105. "doorbell": False,
  4106. "doorbell_seq": 0,
  4107. "duress_alarm": False,
  4108. "high_temperature": False,
  4109. "lockout_alarm": False,
  4110. "low_temperature": False,
  4111. "pir_triggered_level": 2,
  4112. "sequence_number": 172,
  4113. "tamper_alarm": False,
  4114. },
  4115. b"\x01\x11\x03x",
  4116. "Keypad Vision",
  4117. SwitchbotModel.KEYPAD_VISION,
  4118. ),
  4119. AdvTestCase(
  4120. b"\xb0\xe9\xfe\xde\xb6\x8c+`\x00\x00\x00\x00\x00\x002",
  4121. b"\x00\x00`\x01\x11Q\x98",
  4122. {
  4123. "battery": 96,
  4124. "battery_charging": False,
  4125. "doorbell": False,
  4126. "doorbell_seq": 0,
  4127. "duress_alarm": False,
  4128. "high_temperature": False,
  4129. "lockout_alarm": False,
  4130. "low_temperature": False,
  4131. "radar_triggered_distance": 0,
  4132. "radar_triggered_level": 0,
  4133. "sequence_number": 43,
  4134. "tamper_alarm": False,
  4135. },
  4136. b"\x01\x11Q\x98",
  4137. "Keypad Vision Pro",
  4138. SwitchbotModel.KEYPAD_VISION_PRO,
  4139. ),
  4140. ],
  4141. )
  4142. def test_adv_active(test_case: AdvTestCase) -> None:
  4143. """Test with active data."""
  4144. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4145. adv_data = generate_advertisement_data(
  4146. manufacturer_data={2409: test_case.manufacturer_data},
  4147. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  4148. rssi=-97,
  4149. )
  4150. result = parse_advertisement_data(ble_device, adv_data)
  4151. assert result == SwitchBotAdvertisement(
  4152. address="aa:bb:cc:dd:ee:ff",
  4153. data={
  4154. "rawAdvData": test_case.service_data,
  4155. "data": test_case.data,
  4156. "isEncrypted": False,
  4157. "model": test_case.model,
  4158. "modelFriendlyName": test_case.modelFriendlyName,
  4159. "modelName": test_case.modelName,
  4160. },
  4161. device=ble_device,
  4162. rssi=-97,
  4163. active=True,
  4164. )
  4165. @pytest.mark.parametrize(
  4166. "test_case",
  4167. [
  4168. AdvTestCase(
  4169. b"\xc0N0\xdd\xb9\xf2\x8a\xc1\x00\x00\x00\x00\x00F\x00\x00",
  4170. b"=\x00\x00\x00",
  4171. {
  4172. 1: {
  4173. "isOn": True,
  4174. "sequence_number": 138,
  4175. "switchMode": True,
  4176. "power": 0.0,
  4177. },
  4178. 2: {
  4179. "isOn": True,
  4180. "sequence_number": 138,
  4181. "switchMode": True,
  4182. "power": 70.0,
  4183. },
  4184. "sequence_number": 138,
  4185. },
  4186. "=",
  4187. "Relay Switch 2PM",
  4188. SwitchbotModel.RELAY_SWITCH_2PM,
  4189. ),
  4190. AdvTestCase(
  4191. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  4192. b";\x00\x00\x00",
  4193. {
  4194. "isOn": True,
  4195. "sequence_number": 71,
  4196. "switchMode": True,
  4197. },
  4198. ";",
  4199. "Relay Switch 1",
  4200. SwitchbotModel.RELAY_SWITCH_1,
  4201. ),
  4202. AdvTestCase(
  4203. b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
  4204. b"<\x00\x00\x00",
  4205. {
  4206. "isOn": True,
  4207. "sequence_number": 71,
  4208. "switchMode": True,
  4209. "power": 49.0,
  4210. },
  4211. "<",
  4212. "Relay Switch 1PM",
  4213. SwitchbotModel.RELAY_SWITCH_1PM,
  4214. ),
  4215. AdvTestCase(
  4216. b"$X|\x05BN\x0f\x00\x00\x03\x00\x00\x00\x00\x00\x00",
  4217. b">\x00\x00\x00",
  4218. {
  4219. "door_open": True,
  4220. "isOn": False,
  4221. "sequence_number": 15,
  4222. "switchMode": True,
  4223. },
  4224. ">",
  4225. "Garage Door Opener",
  4226. SwitchbotModel.GARAGE_DOOR_OPENER,
  4227. ),
  4228. AdvTestCase(
  4229. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  4230. None,
  4231. {
  4232. "sequence_number": 133,
  4233. "isOn": True,
  4234. "brightness": 30,
  4235. "delay": False,
  4236. "network_state": 2,
  4237. "color_mode": 2,
  4238. "cw": 4753,
  4239. },
  4240. b"\x00\x10\xd0\xb1",
  4241. "Strip Light 3",
  4242. SwitchbotModel.STRIP_LIGHT_3,
  4243. ),
  4244. AdvTestCase(
  4245. b'\xa0\x85\xe3e,\x06P\xaa"\xd4\x00\x00\x00\x00\x00\x00\r\x93\x00',
  4246. None,
  4247. {
  4248. "sequence_number": 80,
  4249. "isOn": True,
  4250. "brightness": 42,
  4251. "delay": False,
  4252. "network_state": 2,
  4253. "color_mode": 2,
  4254. "cw": 3475,
  4255. },
  4256. b"\x00\x10\xd0\xb0",
  4257. "Floor Lamp",
  4258. SwitchbotModel.FLOOR_LAMP,
  4259. ),
  4260. AdvTestCase(
  4261. b'|,g\xc8\x15&jR"l\x00\x00\x00\x00\x00\x00',
  4262. None,
  4263. {
  4264. "brightness": 82,
  4265. "color_mode": 2,
  4266. "delay": False,
  4267. "isOn": False,
  4268. "sequence_number": 106,
  4269. "network_state": 2,
  4270. },
  4271. "r",
  4272. "Light Strip",
  4273. SwitchbotModel.LIGHT_STRIP,
  4274. ),
  4275. AdvTestCase(
  4276. b"@L\xca\xa7_\x12\x02\x81\x12\x00\x00",
  4277. None,
  4278. {
  4279. "brightness": 1,
  4280. "color_mode": 2,
  4281. "delay": False,
  4282. "isOn": True,
  4283. "loop_index": 0,
  4284. "preset": False,
  4285. "sequence_number": 2,
  4286. "speed": 0,
  4287. },
  4288. "u",
  4289. "Color Bulb",
  4290. SwitchbotModel.COLOR_BULB,
  4291. ),
  4292. AdvTestCase(
  4293. b"\x94\xa9\x90T\x85^?\xa1\x00\x00\x04\xe6\x00\x00\x00\x00",
  4294. None,
  4295. {
  4296. "isOn": True,
  4297. "power": 1254.0,
  4298. "sequence_number": 63,
  4299. "switchMode": True,
  4300. },
  4301. "?",
  4302. "Plug Mini (EU)",
  4303. SwitchbotModel.PLUG_MINI_EU,
  4304. ),
  4305. AdvTestCase(
  4306. b'\xdc\x06u\xa6\xfb\xb2y\x9e"\x00\x11\xb8\x00',
  4307. None,
  4308. {
  4309. "sequence_number": 121,
  4310. "isOn": True,
  4311. "brightness": 30,
  4312. "delay": False,
  4313. "network_state": 2,
  4314. "color_mode": 2,
  4315. "cw": 4536,
  4316. },
  4317. b"\x00\x10\xd0\xb4",
  4318. "RGBICWW Floor Lamp",
  4319. SwitchbotModel.RGBICWW_FLOOR_LAMP,
  4320. ),
  4321. AdvTestCase(
  4322. b"\x90\xe5\xb1h\xda\xaa\n\xb0 \x00",
  4323. None,
  4324. {
  4325. "brightness": 48,
  4326. "delay": False,
  4327. "isOn": True,
  4328. "network_state": 2,
  4329. "sequence_number": 10,
  4330. },
  4331. b"\x00\x11\x22\xb8",
  4332. "Candle Warmer Lamp",
  4333. SwitchbotModel.CANDLE_WARMER_LAMP,
  4334. ),
  4335. AdvTestCase(
  4336. b'(7/L\x94\xb2\x0c\x9e"\x00\x11:\x00',
  4337. None,
  4338. {
  4339. "sequence_number": 12,
  4340. "isOn": True,
  4341. "brightness": 30,
  4342. "delay": False,
  4343. "network_state": 2,
  4344. "color_mode": 2,
  4345. "cw": 4410,
  4346. },
  4347. b"\x00\x10\xd0\xb3",
  4348. "RGBICWW Strip Light",
  4349. SwitchbotModel.RGBICWW_STRIP_LIGHT,
  4350. ),
  4351. AdvTestCase(
  4352. b'\xc0N0\xe0U\x9a\x85\x9e"\xd0\x00\x00\x00\x00\x00\x00\x12\x91\x00',
  4353. None,
  4354. {
  4355. "sequence_number": 133,
  4356. "isOn": True,
  4357. "brightness": 30,
  4358. "delay": False,
  4359. "network_state": 2,
  4360. "color_mode": 2,
  4361. "cw": 0,
  4362. },
  4363. b"\x00\x10\xd0\xb7",
  4364. "Permanent Outdoor Light",
  4365. SwitchbotModel.PERMANENT_OUTDOOR_LIGHT,
  4366. ),
  4367. AdvTestCase(
  4368. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  4369. None,
  4370. {
  4371. "sequence_number": 47,
  4372. "isOn": True,
  4373. "brightness": 11,
  4374. "delay": False,
  4375. "network_state": 2,
  4376. "color_mode": 7,
  4377. },
  4378. b"\x00\x10\xd0\xb6",
  4379. "RGBIC Neon Rope Light",
  4380. SwitchbotModel.RGBIC_NEON_ROPE_LIGHT,
  4381. ),
  4382. AdvTestCase(
  4383. b"@L\xca!pz/\x8b'\x00\x11:\x00",
  4384. None,
  4385. {
  4386. "sequence_number": 47,
  4387. "isOn": True,
  4388. "brightness": 11,
  4389. "delay": False,
  4390. "network_state": 2,
  4391. "color_mode": 7,
  4392. },
  4393. b"\x00\x10\xd0\xb5",
  4394. "RGBIC Neon Wire Rope Light",
  4395. SwitchbotModel.RGBIC_NEON_WIRE_ROPE_LIGHT,
  4396. ),
  4397. AdvTestCase(
  4398. b'(7/L\x94\xb2\x0c\x9e"\x00\x11:\x00\xa0',
  4399. None,
  4400. {
  4401. "sequence_number": 12,
  4402. "isOn": True,
  4403. "brightness": 30,
  4404. "delay": False,
  4405. "network_state": 2,
  4406. "color_mode": 2,
  4407. "cw": 4410,
  4408. "main_isOn": True,
  4409. "main_brightness": 32,
  4410. },
  4411. b"\x00\x11\xbb\x10",
  4412. "RGBICWW Ceiling Light",
  4413. SwitchbotModel.RGBICWW_CEILING_LIGHT,
  4414. ),
  4415. AdvTestCase(
  4416. b"\xb0\xe9\xfe\xe4\xbf\xd8\x0b\x01\x11f\x00\x16M\x15",
  4417. None,
  4418. {
  4419. "battery": 77,
  4420. "mqtt_connected": True,
  4421. "sequence_number": 11,
  4422. "soc_version": "1.1.102",
  4423. "step": 6,
  4424. "work_status": 21,
  4425. },
  4426. b"\x00\x10\xfb\xa8",
  4427. "K11+ Vacuum",
  4428. SwitchbotModel.K11_VACUUM,
  4429. ),
  4430. AdvTestCase(
  4431. b"\xb0\xe9\xfe\x8e\x98Oi_\x06\x9a,\x00\x00\x00\x00\xe4\x00\x08\x04\x00\x01\x00\x00",
  4432. None,
  4433. {
  4434. "battery": 95,
  4435. "humidity": 44,
  4436. "sequence_number": 105,
  4437. "humidity_alarm": 0,
  4438. "isOn": False,
  4439. "is_light": True,
  4440. "motion_detected": True,
  4441. "temp_alarm": 0,
  4442. "temperature": 26.6,
  4443. "on_keystate": 0,
  4444. "off_keystate": 0,
  4445. "on_keystate_mode": 0,
  4446. "on_keystate_counter": 0,
  4447. "off_keystate_mode": 0,
  4448. "off_keystate_counter": 0,
  4449. },
  4450. b"\x00\x10\xf3\xd8",
  4451. "Climate Panel",
  4452. SwitchbotModel.CLIMATE_PANEL,
  4453. ),
  4454. AdvTestCase(
  4455. b"\xb0\xe9\xfe\xa2T|6\xe4\x00\x9c\xa3A\x00",
  4456. None,
  4457. {
  4458. "battery": 100,
  4459. "door_open": False,
  4460. "fault_code": 0,
  4461. "isOn": True,
  4462. "last_mode": "comfort",
  4463. "mode": "manual",
  4464. "sequence_number": 54,
  4465. "need_update_temp": False,
  4466. "restarted": False,
  4467. "target_temperature": 35.0,
  4468. "temperature": 28.0,
  4469. },
  4470. b"\x00\x116@",
  4471. "Smart Thermostat Radiator",
  4472. SwitchbotModel.SMART_THERMOSTAT_RADIATOR,
  4473. ),
  4474. AdvTestCase(
  4475. b"\xb0\xe9\xfe\xc3\x1a!:\x01\x11\x1e\x00\x00d\x03",
  4476. None,
  4477. {
  4478. "battery": 100,
  4479. "mqtt_connected": False,
  4480. "sequence_number": 58,
  4481. "soc_version": "1.1.030",
  4482. "step": 0,
  4483. "work_status": 3,
  4484. },
  4485. b"\x00\x10\xe0P",
  4486. "S20 Vacuum",
  4487. SwitchbotModel.S20_VACUUM,
  4488. ),
  4489. AdvTestCase(
  4490. b"\xb0\xe9\xfelf\xaa\x06\xcc\x04V\x00\x8c",
  4491. None,
  4492. {
  4493. "adaptive_state": True,
  4494. "battery_range": ">=60%",
  4495. "duration": 1110,
  4496. "led_state": True,
  4497. "lightLevel": 12,
  4498. "motion_detected": True,
  4499. "sequence_number": 6,
  4500. "trigger_flag": 0,
  4501. },
  4502. b"\x00\x10\xcc\xc8",
  4503. "Presence Sensor",
  4504. SwitchbotModel.PRESENCE_SENSOR,
  4505. ),
  4506. AdvTestCase(
  4507. b"\xb0\xe9\xfe\xe2\xfa8\x157\x03\x08",
  4508. None,
  4509. {
  4510. "battery": 55,
  4511. "battery_charging": False,
  4512. "display_mode": 1,
  4513. "display_size": 0,
  4514. "image_index": 3,
  4515. "last_network_status": 0,
  4516. "sequence_number": 21,
  4517. },
  4518. b"\x00\x11>\x10",
  4519. "Art Frame",
  4520. SwitchbotModel.ART_FRAME,
  4521. ),
  4522. AdvTestCase(
  4523. b"\xb0\xe9\xfe\xe5\x04\x1e\xac\xdf\x00\x00\x00\x00\x00\x02",
  4524. None,
  4525. {
  4526. "battery": 95,
  4527. "battery_charging": True,
  4528. "doorbell": False,
  4529. "doorbell_seq": 0,
  4530. "duress_alarm": False,
  4531. "high_temperature": False,
  4532. "lockout_alarm": False,
  4533. "low_temperature": False,
  4534. "pir_triggered_level": 2,
  4535. "sequence_number": 172,
  4536. "tamper_alarm": False,
  4537. },
  4538. b"\x00\x11\x03x",
  4539. "Keypad Vision",
  4540. SwitchbotModel.KEYPAD_VISION,
  4541. ),
  4542. AdvTestCase(
  4543. b"\xb0\xe9\xfe\xde\xb6\x8c+`\x00\x00\x00\x00\x00\x002",
  4544. None,
  4545. {
  4546. "battery": 96,
  4547. "battery_charging": False,
  4548. "doorbell": False,
  4549. "doorbell_seq": 0,
  4550. "duress_alarm": False,
  4551. "high_temperature": False,
  4552. "lockout_alarm": False,
  4553. "low_temperature": False,
  4554. "radar_triggered_distance": 0,
  4555. "radar_triggered_level": 0,
  4556. "sequence_number": 43,
  4557. "tamper_alarm": False,
  4558. },
  4559. b"\x00\x11Q\x98",
  4560. "Keypad Vision Pro",
  4561. SwitchbotModel.KEYPAD_VISION_PRO,
  4562. ),
  4563. ],
  4564. )
  4565. def test_adv_passive(test_case: AdvTestCase) -> None:
  4566. """Test with passive data."""
  4567. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4568. adv_data = generate_advertisement_data(
  4569. manufacturer_data={2409: test_case.manufacturer_data},
  4570. rssi=-97,
  4571. )
  4572. result = parse_advertisement_data(ble_device, adv_data, test_case.modelName)
  4573. assert result == SwitchBotAdvertisement(
  4574. address="aa:bb:cc:dd:ee:ff",
  4575. data={
  4576. "rawAdvData": None,
  4577. "data": test_case.data,
  4578. "isEncrypted": False,
  4579. "model": test_case.model,
  4580. "modelFriendlyName": test_case.modelFriendlyName,
  4581. "modelName": test_case.modelName,
  4582. },
  4583. device=ble_device,
  4584. rssi=-97,
  4585. active=False,
  4586. )
  4587. @pytest.mark.parametrize(
  4588. "test_case",
  4589. [
  4590. AdvTestCase(
  4591. None,
  4592. b"=\x00\x00\x00",
  4593. {
  4594. 1: {
  4595. "isOn": True,
  4596. "power": 0.0,
  4597. "sequence_number": 138,
  4598. "switchMode": True,
  4599. },
  4600. 2: {
  4601. "isOn": True,
  4602. "power": 7.0,
  4603. "sequence_number": 138,
  4604. "switchMode": True,
  4605. },
  4606. },
  4607. "=",
  4608. "Relay Switch 2PM",
  4609. SwitchbotModel.RELAY_SWITCH_2PM,
  4610. ),
  4611. AdvTestCase(
  4612. None,
  4613. b";\x00\x00\x00",
  4614. {
  4615. "isOn": True,
  4616. "sequence_number": 71,
  4617. "switchMode": True,
  4618. },
  4619. ";",
  4620. "Relay Switch 1",
  4621. SwitchbotModel.RELAY_SWITCH_1,
  4622. ),
  4623. AdvTestCase(
  4624. None,
  4625. b"<\x00\x00\x00",
  4626. {
  4627. "isOn": True,
  4628. "power": 4.9,
  4629. "sequence_number": 71,
  4630. "switchMode": True,
  4631. },
  4632. "<",
  4633. "Relay Switch 1PM",
  4634. SwitchbotModel.RELAY_SWITCH_1PM,
  4635. ),
  4636. AdvTestCase(
  4637. None,
  4638. b">\x00\x00\x00",
  4639. {
  4640. "door_open": True,
  4641. "isOn": False,
  4642. "sequence_number": 15,
  4643. "switchMode": True,
  4644. },
  4645. ">",
  4646. "Garage Door Opener",
  4647. SwitchbotModel.GARAGE_DOOR_OPENER,
  4648. ),
  4649. AdvTestCase(
  4650. None,
  4651. b"\x00\x00\x00\x00\x10\xd0\xb1",
  4652. {},
  4653. b"\x00\x10\xd0\xb1",
  4654. "Strip Light 3",
  4655. SwitchbotModel.STRIP_LIGHT_3,
  4656. ),
  4657. AdvTestCase(
  4658. None,
  4659. b"\x00\x00\x00\x00\x10\xd0\xb0",
  4660. {},
  4661. b"\x00\x10\xd0\xb0",
  4662. "Floor Lamp",
  4663. SwitchbotModel.FLOOR_LAMP,
  4664. ),
  4665. AdvTestCase(
  4666. None,
  4667. b"\x00\x00\x00\x00\x11\x22\xb8",
  4668. {},
  4669. b"\x00\x11\x22\xb8",
  4670. "Candle Warmer Lamp",
  4671. SwitchbotModel.CANDLE_WARMER_LAMP,
  4672. ),
  4673. AdvTestCase(
  4674. None,
  4675. b"q\x00",
  4676. {},
  4677. "q",
  4678. "Ceiling Light",
  4679. SwitchbotModel.CEILING_LIGHT,
  4680. ),
  4681. AdvTestCase(
  4682. None,
  4683. b"r\x00d",
  4684. {},
  4685. "r",
  4686. "Light Strip",
  4687. SwitchbotModel.LIGHT_STRIP,
  4688. ),
  4689. AdvTestCase(
  4690. None,
  4691. b"u\x00d",
  4692. {},
  4693. "u",
  4694. "Color Bulb",
  4695. SwitchbotModel.COLOR_BULB,
  4696. ),
  4697. AdvTestCase(
  4698. None,
  4699. b"?\x00\x00\x00",
  4700. {
  4701. "isOn": True,
  4702. "power": 1254.0,
  4703. "sequence_number": 63,
  4704. "switchMode": True,
  4705. },
  4706. "?",
  4707. "Plug Mini (EU)",
  4708. SwitchbotModel.PLUG_MINI_EU,
  4709. ),
  4710. AdvTestCase(
  4711. None,
  4712. b"\x00\x00\x00\x00\x10\xd0\xb4",
  4713. {},
  4714. b"\x00\x10\xd0\xb4",
  4715. "RGBICWW Floor Lamp",
  4716. SwitchbotModel.RGBICWW_FLOOR_LAMP,
  4717. ),
  4718. AdvTestCase(
  4719. None,
  4720. b"\x00\x00\x00\x00\x11\xbb\x10",
  4721. {},
  4722. b"\x00\x11\xbb\x10",
  4723. "RGBICWW Ceiling Light",
  4724. SwitchbotModel.RGBICWW_CEILING_LIGHT,
  4725. ),
  4726. AdvTestCase(
  4727. None,
  4728. b"\x00\x00\x00\x00\x10\xd0\xb3",
  4729. {},
  4730. b"\x00\x10\xd0\xb3",
  4731. "RGBICWW Strip Light",
  4732. SwitchbotModel.RGBICWW_STRIP_LIGHT,
  4733. ),
  4734. AdvTestCase(
  4735. None,
  4736. b"\x00\x00\x00\x00\x10\xd0\xb7",
  4737. {},
  4738. b"\x00\x10\xd0\xb7",
  4739. "Permanent Outdoor Light",
  4740. SwitchbotModel.PERMANENT_OUTDOOR_LIGHT,
  4741. ),
  4742. AdvTestCase(
  4743. None,
  4744. b"\x00\x00M\x00\x10\xfb\xa8",
  4745. {},
  4746. b"\x00\x10\xfb\xa8",
  4747. "K11+ Vacuum",
  4748. SwitchbotModel.K11_VACUUM,
  4749. ),
  4750. AdvTestCase(
  4751. None,
  4752. b"\x00 _\x00\x10\xf3\xd8@",
  4753. {},
  4754. b"\x00\x10\xf3\xd8",
  4755. "Climate Panel",
  4756. SwitchbotModel.CLIMATE_PANEL,
  4757. ),
  4758. AdvTestCase(
  4759. None,
  4760. b"\x00 d\x00\x116@",
  4761. {},
  4762. b"\x00\x116@",
  4763. "Smart Thermostat Radiator",
  4764. SwitchbotModel.SMART_THERMOSTAT_RADIATOR,
  4765. ),
  4766. AdvTestCase(
  4767. None,
  4768. b"\x00\x00d\x00\x10\xe0P",
  4769. {},
  4770. b"\x00\x10\xe0P",
  4771. "S20 Vacuum",
  4772. SwitchbotModel.S20_VACUUM,
  4773. ),
  4774. AdvTestCase(
  4775. None,
  4776. b"\x00 d\x00\x10\xcc\xc8",
  4777. {},
  4778. b"\x00\x10\xcc\xc8",
  4779. "Presence Sensor",
  4780. SwitchbotModel.PRESENCE_SENSOR,
  4781. ),
  4782. AdvTestCase(
  4783. None,
  4784. b"\x00\x007\x01\x11>\x10",
  4785. {},
  4786. b"\x01\x11>\x10",
  4787. "Art Frame",
  4788. SwitchbotModel.ART_FRAME,
  4789. ),
  4790. AdvTestCase(
  4791. None,
  4792. b"\x00\x00_\x01\x11\x03x",
  4793. {},
  4794. b"\x01\x11\x03x",
  4795. "Keypad Vision",
  4796. SwitchbotModel.KEYPAD_VISION,
  4797. ),
  4798. AdvTestCase(
  4799. None,
  4800. b"\x00\x00\x00\x00\x10\xd0\xb6",
  4801. {},
  4802. b"\x00\x10\xd0\xb6",
  4803. "RGBIC Neon Rope Light",
  4804. SwitchbotModel.RGBIC_NEON_ROPE_LIGHT,
  4805. ),
  4806. AdvTestCase(
  4807. None,
  4808. b"\x00\x00`\x01\x11Q\x98",
  4809. {},
  4810. b"\x01\x11Q\x98",
  4811. "Keypad Vision Pro",
  4812. SwitchbotModel.KEYPAD_VISION_PRO,
  4813. ),
  4814. ],
  4815. )
  4816. def test_adv_with_empty_data(test_case: AdvTestCase) -> None:
  4817. """Test with empty data."""
  4818. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4819. adv_data = generate_advertisement_data(
  4820. manufacturer_data={2409: None},
  4821. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": test_case.service_data},
  4822. rssi=-97,
  4823. )
  4824. result = parse_advertisement_data(ble_device, adv_data)
  4825. assert result == SwitchBotAdvertisement(
  4826. address="aa:bb:cc:dd:ee:ff",
  4827. data={
  4828. "rawAdvData": test_case.service_data,
  4829. "data": {},
  4830. "isEncrypted": False,
  4831. "model": test_case.model,
  4832. },
  4833. device=ble_device,
  4834. rssi=-97,
  4835. active=True,
  4836. )
  4837. def test_parse_advertisement_with_mac_cache() -> None:
  4838. """Test that populating the MAC cache helps identify unknown passive devices."""
  4839. # Clear the cache to ensure clean test
  4840. _MODEL_TO_MAC_CACHE.clear()
  4841. # Create a passive lock device with manufacturer data only (no service data)
  4842. # This would normally not be identifiable without the cache
  4843. mac_address = "C1:64:B8:7D:06:05"
  4844. ble_device = generate_ble_device(mac_address, "WoLock")
  4845. # Lock passive advertisement with manufacturer data
  4846. # This is real data from a WoLock device in passive mode
  4847. adv_data = generate_advertisement_data(
  4848. manufacturer_data={2409: b"\xc1d\xb8}\x06\x05\x00\x00\x00\x00\x00"},
  4849. service_data={},
  4850. rssi=-70,
  4851. )
  4852. # First attempt: Without cache, parser cannot identify the device model
  4853. result_without_cache = parse_advertisement_data(ble_device, adv_data)
  4854. assert result_without_cache is None, "Should not decode without model hint"
  4855. # Now populate the cache with the device's MAC and model
  4856. populate_model_to_mac_cache(mac_address, SwitchbotModel.LOCK)
  4857. # Second attempt: With cache, parser can now identify and decode the device
  4858. result_with_cache = parse_advertisement_data(ble_device, adv_data)
  4859. assert result_with_cache is not None, "Should decode with MAC cache"
  4860. assert result_with_cache.data["modelName"] == SwitchbotModel.LOCK
  4861. assert result_with_cache.data["modelFriendlyName"] == "Lock"
  4862. assert result_with_cache.active is False # Passive advertisement
  4863. # Clean up
  4864. _MODEL_TO_MAC_CACHE.clear()
  4865. def test_parse_advertisement_with_mac_cache_curtain() -> None:
  4866. """Test MAC cache with a passive curtain device."""
  4867. # Clear the cache
  4868. _MODEL_TO_MAC_CACHE.clear()
  4869. # Create a passive curtain device
  4870. mac_address = "CC:F4:C4:F9:AC:6C"
  4871. ble_device = generate_ble_device(mac_address, None)
  4872. # Curtain passive advertisement with only manufacturer data
  4873. adv_data = generate_advertisement_data(
  4874. manufacturer_data={2409: b"\xccOLG\x00c\x00\x00\x11\x00\x00"},
  4875. service_data={},
  4876. rssi=-85,
  4877. )
  4878. # Without cache, cannot identify
  4879. result_without_cache = parse_advertisement_data(ble_device, adv_data)
  4880. assert result_without_cache is None
  4881. # Populate cache
  4882. populate_model_to_mac_cache(mac_address, SwitchbotModel.CURTAIN)
  4883. # With cache, can identify and parse
  4884. result_with_cache = parse_advertisement_data(ble_device, adv_data)
  4885. assert result_with_cache is not None
  4886. assert result_with_cache.data["modelName"] == SwitchbotModel.CURTAIN
  4887. assert result_with_cache.data["modelFriendlyName"] == "Curtain"
  4888. assert result_with_cache.active is False
  4889. # Clean up
  4890. _MODEL_TO_MAC_CACHE.clear()
  4891. @pytest.mark.parametrize(
  4892. ("manufacturer_data", "service_data", "model"),
  4893. [
  4894. (b"\xff\xff\xff\xff", b"\xff\xff\xff\xff", None),
  4895. (b"\xff\xff\xff\xff", b"\xff\xff\xff\xff", "F"),
  4896. (b"\xff\xff\xff\xff\xff\xff\xff", b"\xff\xff\xff\xff\xff\xff\xff\xff", None),
  4897. (None, None, None),
  4898. ],
  4899. )
  4900. def test_with_invalid_advertisement(manufacturer_data, service_data, model) -> None:
  4901. """Test with invalid advertisement data."""
  4902. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4903. adv_data = generate_advertisement_data(
  4904. manufacturer_data={2409: manufacturer_data},
  4905. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": service_data},
  4906. rssi=-97,
  4907. )
  4908. result = parse_advertisement_data(ble_device, adv_data, model)
  4909. assert result is None
  4910. def test_weather_station_active() -> None:
  4911. """Test Weather Station active advertisement."""
  4912. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4913. adv_data = generate_advertisement_data(
  4914. manufacturer_data={
  4915. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01\x50\x06\x9a\x23\x00\x00\x00\x00\x00"
  4916. },
  4917. service_data={
  4918. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x50\x00\x10\x53\xb0"
  4919. },
  4920. rssi=-67,
  4921. )
  4922. result = parse_advertisement_data(ble_device, adv_data)
  4923. assert result == SwitchBotAdvertisement(
  4924. address="aa:bb:cc:dd:ee:ff",
  4925. data={
  4926. "data": {
  4927. "battery": 80,
  4928. "fahrenheit": False,
  4929. "humidity": 35,
  4930. "temp": {"c": 26.6, "f": 79.88},
  4931. "temperature": 26.6,
  4932. },
  4933. "isEncrypted": False,
  4934. "model": b"\x00\x10\x53\xb0",
  4935. "modelFriendlyName": "Weather Station",
  4936. "modelName": SwitchbotModel.WEATHER_STATION,
  4937. "rawAdvData": b"\x00\x00\x50\x00\x10\x53\xb0",
  4938. },
  4939. device=ble_device,
  4940. rssi=-67,
  4941. active=True,
  4942. )
  4943. def test_weather_station_passive() -> None:
  4944. """Test Weather Station passive advertisement."""
  4945. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4946. adv_data = generate_advertisement_data(
  4947. manufacturer_data={
  4948. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01\x50\x06\x9a\x23\x00\x00\x00\x00\x00"
  4949. },
  4950. rssi=-67,
  4951. )
  4952. result = parse_advertisement_data(
  4953. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  4954. )
  4955. assert result == SwitchBotAdvertisement(
  4956. address="aa:bb:cc:dd:ee:ff",
  4957. data={
  4958. "data": {
  4959. "battery": 80,
  4960. "fahrenheit": False,
  4961. "humidity": 35,
  4962. "temp": {"c": 26.6, "f": 79.88},
  4963. "temperature": 26.6,
  4964. },
  4965. "isEncrypted": False,
  4966. "model": b"\x00\x10\x53\xb0",
  4967. "modelFriendlyName": "Weather Station",
  4968. "modelName": SwitchbotModel.WEATHER_STATION,
  4969. "rawAdvData": None,
  4970. },
  4971. device=ble_device,
  4972. rssi=-67,
  4973. active=False,
  4974. )
  4975. def test_weather_station_service_data_only() -> None:
  4976. """Test Weather Station with service data only (no manufacturer data)."""
  4977. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4978. adv_data = generate_advertisement_data(
  4979. service_data={
  4980. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x50\x06\x9a\x23\x00\x10\x53\xb0"
  4981. },
  4982. rssi=-67,
  4983. )
  4984. result = parse_advertisement_data(
  4985. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  4986. )
  4987. assert result is not None
  4988. assert result.data["data"]["temperature"] == 26.6
  4989. assert result.data["data"]["humidity"] == 35
  4990. assert result.data["data"]["battery"] == 80
  4991. def test_weather_station_empty_data() -> None:
  4992. """Test Weather Station with empty/zero data returns empty dict."""
  4993. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  4994. adv_data = generate_advertisement_data(
  4995. manufacturer_data={
  4996. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01\x00\x00\x80\x00\x00\x00\x00\x00\x00"
  4997. },
  4998. service_data={
  4999. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x00\x00\x10\x53\xb0"
  5000. },
  5001. rssi=-67,
  5002. )
  5003. result = parse_advertisement_data(ble_device, adv_data)
  5004. assert result is not None
  5005. assert result.data["data"] == {}
  5006. def test_weather_station_positive_battery_and_empty_data() -> None:
  5007. """Test Weather Station with positive battery and empty/zero data returns empty dict."""
  5008. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  5009. adv_data = generate_advertisement_data(
  5010. manufacturer_data={
  5011. 2409: b"\xaa\xbb\xcc\xdd\xee\xff\x01P\x00\x80\x00\x00\x00\x00\x00\x00"
  5012. },
  5013. service_data={
  5014. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x00\x00\x00\x10\x53\xb0"
  5015. },
  5016. rssi=-67,
  5017. )
  5018. result = parse_advertisement_data(ble_device, adv_data)
  5019. assert result is not None
  5020. assert result.data["data"] == {}
  5021. def test_weather_station_short_service_data() -> None:
  5022. """Test Weather Station with too-short service data does not raise IndexError."""
  5023. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  5024. adv_data = generate_advertisement_data(
  5025. service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"\x00\x01"},
  5026. rssi=-67,
  5027. )
  5028. result = parse_advertisement_data(
  5029. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  5030. )
  5031. assert result is not None
  5032. assert result.data["data"] == {}
  5033. def test_weather_station_no_data() -> None:
  5034. """Test Weather Station with no usable data."""
  5035. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  5036. adv_data = generate_advertisement_data(
  5037. manufacturer_data={2409: b"\xaa\xbb\xcc\xdd\xee"},
  5038. rssi=-67,
  5039. )
  5040. result = parse_advertisement_data(
  5041. ble_device, adv_data, SwitchbotModel.WEATHER_STATION
  5042. )
  5043. assert result is not None
  5044. assert result.data["data"] == {}
  5045. def test_with_special_manufacturer_data_length() -> None:
  5046. """Test with special manufacturer data length."""
  5047. ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
  5048. adv_data = generate_advertisement_data(
  5049. manufacturer_data={741: b"\xacg\xb2\xcd\xfa\xbe"},
  5050. service_data={
  5051. "0000fd3d-0000-1000-8000-00805f9b34fb": b"\xff\x80\x00\xf9\x80Bc\x00"
  5052. },
  5053. rssi=-97,
  5054. )
  5055. result = parse_advertisement_data(ble_device, adv_data)
  5056. assert result.data["modelName"] == SwitchbotModel.HUMIDIFIER