test_adv_parser.py 174 KB

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