| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626 |
- package oscar
- import (
- "bytes"
- "context"
- "log/slog"
- "math"
- "testing"
- "github.com/mk6i/retro-aim-server/config"
- "github.com/mk6i/retro-aim-server/server/oscar/middleware"
- "github.com/mk6i/retro-aim-server/state"
- "github.com/mk6i/retro-aim-server/wire"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/mock"
- )
- func TestHandler_AdminConfirmRequest(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x07_0x06_AdminConfirmRequest
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x07_0x06_AdminConfirmRequest{},
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x07_0x06_AdminConfirmRequest{},
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x07_0x06_AdminConfirmRequest{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Admin,
- SubGroup: wire.AdminAcctConfirmRequest,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Admin,
- SubGroup: wire.AdminAcctConfirmReply,
- },
- Body: wire.SNAC_0x07_0x07_AdminConfirmReply{
- Status: wire.AdminAcctConfirmStatusEmailSent,
- },
- }
- svc := newMockAdminService(t)
- svc.EXPECT().
- ConfirmRequest(mock.Anything, mock.Anything, input.Frame).
- Return(output, tt.serviceError)
- h := Handler{
- AdminService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_AdminInfoQuery_RegistrationStatus(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x07_0x02_AdminInfoQuery
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x07_0x02_AdminInfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVRegistrationStatus, uint16(0x00)),
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x07_0x02_AdminInfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVRegistrationStatus, uint16(0x00)),
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x07_0x02_AdminInfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVRegistrationStatus, uint16(0x00)),
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Admin,
- SubGroup: wire.AdminInfoQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Admin,
- SubGroup: wire.AdminInfoReply,
- },
- Body: wire.SNAC_0x07_0x03_AdminInfoReply{
- Permissions: wire.AdminInfoPermissionsReadWrite,
- TLVBlock: wire.TLVBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVRegistrationStatus, wire.AdminInfoRegStatusFullDisclosure),
- },
- },
- },
- }
- svc := newMockAdminService(t)
- svc.EXPECT().
- InfoQuery(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- AdminService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_AdminInfoChangeRequest_ScreenNameFormatted(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x07_0x04_AdminInfoChangeRequest
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "Chatting Chuck"),
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "Chatting Chuck"),
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "Chatting Chuck"),
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Admin,
- SubGroup: wire.AdminInfoChangeRequest,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Admin,
- SubGroup: wire.AdminInfoChangeReply,
- },
- Body: wire.SNAC_0x07_0x05_AdminChangeReply{
- Permissions: wire.AdminInfoPermissionsReadWrite,
- TLVBlock: wire.TLVBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "Chatting Chuck"),
- },
- },
- },
- }
- svc := newMockAdminService(t)
- svc.EXPECT().
- InfoChangeRequest(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- AdminService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_AlertNotifyCapabilities(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNACFrame
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNACFrame{},
- },
- {
- name: "empty body",
- inputBody: wire.SNACFrame{},
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Alert,
- SubGroup: wire.AlertNotifyCapabilities,
- },
- Body: tt.inputBody,
- }
- h := Handler{
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_AlertNotifyDisplayCapabilities(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNACFrame
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNACFrame{},
- },
- {
- name: "empty body",
- inputBody: wire.SNACFrame{},
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Alert,
- SubGroup: wire.AlertNotifyDisplayCapabilities,
- },
- Body: tt.inputBody,
- }
- h := Handler{
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BARTDownloadQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x10_0x04_BARTDownloadQuery
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x10_0x04_BARTDownloadQuery{},
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x10_0x04_BARTDownloadQuery{},
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x10_0x04_BARTDownloadQuery{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.BART,
- SubGroup: wire.BARTDownloadQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.BART,
- SubGroup: wire.BARTDownloadReply,
- },
- Body: wire.SNAC_0x10_0x05_BARTDownloadReply{
- ScreenName: "the-screen-name",
- },
- }
- svc := newMockBARTService(t)
- svc.EXPECT().
- RetrieveItem(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- BARTService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BARTDownload2Query(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x10_0x06_BARTDownload2Query
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x10_0x06_BARTDownload2Query{},
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x10_0x06_BARTDownload2Query{},
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x10_0x06_BARTDownload2Query{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.BART,
- SubGroup: wire.BARTDownload2Query,
- },
- Body: tt.inputBody,
- }
- output := []wire.SNACMessage{
- {
- Frame: wire.SNACFrame{
- FoodGroup: wire.BART,
- SubGroup: wire.BARTDownload2Reply,
- },
- Body: wire.SNAC_0x10_0x07_BARTDownload2Reply{
- ScreenName: "the-screen-name",
- },
- },
- }
- svc := newMockBARTService(t)
- svc.EXPECT().
- RetrieveItemV2(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- BARTService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- for _, msg := range output {
- responseWriter.EXPECT().
- SendSNAC(msg.Frame, msg.Body).
- Return(tt.responseError)
- }
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BARTUploadQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x10_0x02_BARTUploadQuery
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x10_0x02_BARTUploadQuery{
- Type: 1,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x10_0x02_BARTUploadQuery{
- Type: 1,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x10_0x02_BARTUploadQuery{
- Type: 1,
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.BART,
- SubGroup: wire.BARTUploadQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.BART,
- SubGroup: wire.BARTUploadReply,
- },
- Body: wire.SNAC_0x10_0x03_BARTUploadReply{
- Code: wire.BARTReplyCodesSuccess,
- },
- }
- svc := newMockBARTService(t)
- svc.EXPECT().
- UpsertItem(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- BARTService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BuddyRightsQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x03_0x02_BuddyRightsQuery
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x03_0x02_BuddyRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(1000)),
- },
- },
- },
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x03_0x02_BuddyRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(1000)),
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Buddy,
- SubGroup: wire.BuddyRightsQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Buddy,
- SubGroup: wire.BuddyRightsReply,
- },
- Body: wire.SNAC_0x03_0x03_BuddyRightsReply{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(1000)),
- },
- },
- },
- }
- svc := newMockBuddyService(t)
- svc.EXPECT().
- RightsQuery(mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- BuddyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BuddyAddBuddies(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x03_0x04_BuddyAddBuddies
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x03_0x04_BuddyAddBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x03_0x04_BuddyAddBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Buddy,
- SubGroup: wire.BuddyAddBuddies,
- },
- Body: tt.inputBody,
- }
- svc := newMockBuddyService(t)
- svc.EXPECT().
- AddBuddies(mock.Anything, mock.Anything, input.Body).
- Return(tt.serviceError)
- h := Handler{
- BuddyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BuddyDelBuddies(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x03_0x05_BuddyDelBuddies
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x03_0x05_BuddyDelBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x03_0x05_BuddyDelBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Buddy,
- SubGroup: wire.BuddyDelBuddies,
- },
- Body: tt.inputBody,
- }
- svc := newMockBuddyService(t)
- svc.EXPECT().
- DelBuddies(mock.Anything, mock.Anything, input.Body).
- Return(tt.serviceError)
- h := Handler{
- BuddyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BuddyAddTempBuddies(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x03_0x0F_BuddyAddTempBuddies
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x03_0x0F_BuddyAddTempBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x03_0x0F_BuddyAddTempBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Buddy,
- SubGroup: wire.BuddyAddTempBuddies,
- },
- Body: tt.inputBody,
- }
- svc := newMockBuddyService(t)
- svc.EXPECT().
- AddTempBuddies(mock.Anything, mock.Anything, input.Body).
- Return(tt.serviceError)
- h := Handler{
- BuddyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_BuddyDelTempBuddies(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x03_0x10_BuddyDelTempBuddies
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x03_0x10_BuddyDelTempBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x03_0x10_BuddyDelTempBuddies{
- Buddies: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "user1",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Buddy,
- SubGroup: wire.BuddyDelTempBuddies,
- },
- Body: tt.inputBody,
- }
- svc := newMockBuddyService(t)
- svc.EXPECT().
- DelTempBuddies(mock.Anything, mock.Anything, input.Body).
- Return(tt.serviceError)
- h := Handler{
- BuddyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ChatNavCreateRoom(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
- Exchange: 1,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
- Exchange: 1,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
- Exchange: 1,
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavCreateRoom,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavNavInfo,
- },
- Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{},
- }
- sess := state.NewSession()
- svc := newMockChatNavService(t)
- svc.EXPECT().
- CreateRoom(mock.Anything, sess, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- ChatNavService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ChatNavCreateRoom_ReadErr(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavCreateRoom,
- },
- Body: wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
- Exchange: 1,
- },
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavNavInfo,
- },
- Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{},
- }
- sess := state.NewSession()
- svc := newMockChatNavService(t)
- svc.EXPECT().
- CreateRoom(mock.Anything, sess, input.Frame, input.Body).
- Return(output, nil)
- h := Handler{
- ChatNavService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- ss := newMockResponseWriter(t)
- ss.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(nil)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, ss, config.Listener{}))
- }
- func TestHandler_ChatNavRequestChatRights(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "response writer error",
- inputBody: struct{}{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavRequestChatRights,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavNavInfo,
- },
- Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{},
- }
- svc := newMockChatNavService(t)
- svc.EXPECT().
- RequestChatRights(mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- ChatNavService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ChatNavRequestRoomInfo(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
- Exchange: 1,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
- Exchange: 1,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
- Exchange: 1,
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavRequestRoomInfo,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavNavInfo,
- },
- Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{},
- }
- svc := newMockChatNavService(t)
- svc.EXPECT().
- RequestRoomInfo(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- ChatNavService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ChatNavRequestExchangeInfo(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0D_0x03_ChatNavRequestExchangeInfo
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x0D_0x03_ChatNavRequestExchangeInfo{
- Exchange: 4,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x0D_0x03_ChatNavRequestExchangeInfo{
- Exchange: 4,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x0D_0x03_ChatNavRequestExchangeInfo{
- Exchange: 4,
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavRequestExchangeInfo,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ChatNav,
- SubGroup: wire.ChatNavNavInfo,
- },
- Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{},
- }
- svc := newMockChatNavService(t)
- svc.EXPECT().
- ExchangeInfo(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- ChatNavService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ChatChannelMsgToHost(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0E_0x05_ChatChannelMsgToHost
- serviceResponse *wire.SNACMessage
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success with reflected response",
- inputBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: 4,
- },
- serviceResponse: &wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Chat,
- SubGroup: wire.ChatChannelMsgToClient,
- },
- Body: wire.SNAC_0x0E_0x06_ChatChannelMsgToClient{
- Channel: 4,
- },
- },
- },
- {
- name: "service error with reflected response",
- inputBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: 4,
- },
- serviceResponse: &wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Chat,
- SubGroup: wire.ChatChannelMsgToClient,
- },
- Body: wire.SNAC_0x0E_0x06_ChatChannelMsgToClient{
- Channel: 4,
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error with reflected response",
- inputBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: 4,
- },
- serviceResponse: &wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Chat,
- SubGroup: wire.ChatChannelMsgToClient,
- },
- Body: wire.SNAC_0x0E_0x06_ChatChannelMsgToClient{
- Channel: 4,
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "success without reflected response",
- inputBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: 4,
- },
- serviceResponse: nil, // nil response means no reflection back to caller
- },
- {
- name: "service error without reflected response",
- inputBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
- Channel: 4,
- },
- serviceResponse: nil,
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Chat,
- SubGroup: wire.ChatChannelMsgToHost,
- },
- Body: tt.inputBody,
- }
- svc := newMockChatService(t)
- svc.EXPECT().
- ChannelMsgToHost(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(tt.serviceResponse, tt.serviceError)
- h := Handler{
- ChatService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil && tt.serviceResponse != nil {
- responseWriter.EXPECT().
- SendSNAC(tt.serviceResponse.Frame, tt.serviceResponse.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagDeleteItem(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x0A_FeedbagDeleteItem
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagDeleteItem,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagStatus,
- },
- Body: wire.SNAC_0x13_0x0E_FeedbagStatus{
- Results: []uint16{1234},
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- DeleteItem(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagEndCluster(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagEndCluster,
- },
- Body: tt.inputBody,
- }
- svc := newMockFeedbagService(t)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagInsertItem(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x08_FeedbagInsertItem
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x08_FeedbagInsertItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x13_0x08_FeedbagInsertItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x13_0x08_FeedbagInsertItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagInsertItem,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagStatus,
- },
- Body: wire.SNAC_0x13_0x0E_FeedbagStatus{
- Results: []uint16{1234},
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- UpsertItem(mock.Anything, mock.Anything, input.Frame, tt.inputBody.Items).
- Return(output, tt.serviceError)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x02_FeedbagRightsQuery
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x02_FeedbagRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x13_0x02_FeedbagRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x13_0x02_FeedbagRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagReply,
- },
- Body: wire.SNAC_0x13_0x06_FeedbagReply{
- Version: 4,
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- Query(mock.Anything, mock.Anything, input.Frame).
- Return(output, tt.serviceError)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagQueryIfModified(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x05_FeedbagQueryIfModified
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x05_FeedbagQueryIfModified{
- LastUpdate: 1234,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x13_0x05_FeedbagQueryIfModified{
- LastUpdate: 1234,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x13_0x05_FeedbagQueryIfModified{
- LastUpdate: 1234,
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagQueryIfModified,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagReply,
- },
- Body: wire.SNAC_0x13_0x06_FeedbagReply{
- LastUpdate: 1234,
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- QueryIfModified(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagRightsQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x02_FeedbagRightsQuery
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x02_FeedbagRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x13_0x02_FeedbagRightsQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagRightsQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagRightsReply,
- },
- Body: wire.SNAC_0x13_0x03_FeedbagRightsReply{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- RightsQuery(mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagStartCluster(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x11_FeedbagStartCluster
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x11_FeedbagStartCluster{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagStartCluster,
- },
- Body: tt.inputBody,
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- StartCluster(mock.Anything, input.Frame, input.Body)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagUpdateItem(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x13_0x09_FeedbagUpdateItem
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x13_0x09_FeedbagUpdateItem{
- Items: []wire.FeedbagItem{
- {
- Name: "my-item",
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagUpdateItem,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagStatus,
- },
- Body: wire.SNAC_0x13_0x0E_FeedbagStatus{
- Results: []uint16{1234},
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- UpsertItem(mock.Anything, mock.Anything, input.Frame, tt.inputBody.Items).
- Return(output, tt.serviceError)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagUse(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "service error",
- inputBody: struct{}{},
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagUse,
- },
- Body: tt.inputBody,
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- Use(mock.Anything, mock.Anything).
- Return(tt.serviceError)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_FeedbagRespondAuthorizeToHost(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Feedbag,
- SubGroup: wire.FeedbagRespondAuthorizeToHost,
- },
- Body: wire.SNAC_0x13_0x1A_FeedbagRespondAuthorizeToHost{
- ScreenName: "theScreenName",
- },
- }
- svc := newMockFeedbagService(t)
- svc.EXPECT().
- RespondAuthorizeToHost(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(nil)
- h := Handler{
- FeedbagService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{}))
- }
- func TestHandler_ICBMAddParameters(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x04_0x02_ICBMAddParameters
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x04_0x02_ICBMAddParameters{
- Channel: 1,
- },
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMAddParameters,
- },
- Body: tt.inputBody,
- }
- svc := newMockICBMService(t)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICBMChannelMsgToHost(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x04_0x06_ICBMChannelMsgToHost
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ScreenName: "recipient-screen-name",
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ScreenName: "recipient-screen-name",
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
- ScreenName: "recipient-screen-name",
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMChannelMsgToHost,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMHostAck,
- },
- Body: wire.SNAC_0x04_0x0C_ICBMHostAck{
- ChannelID: 4,
- },
- }
- svc := newMockICBMService(t)
- svc.EXPECT().
- ChannelMsgToHost(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(&output, tt.serviceError)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICBMClientErr(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x04_0x0B_ICBMClientErr
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x04_0x0B_ICBMClientErr{
- Code: 4,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x04_0x0B_ICBMClientErr{
- Code: 4,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMClientErr,
- },
- Body: tt.inputBody,
- }
- svc := newMockICBMService(t)
- svc.EXPECT().
- ClientErr(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(tt.serviceError)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICBMClientEvent(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x04_0x14_ICBMClientEvent
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x04_0x14_ICBMClientEvent{
- ScreenName: "recipient-screen-name",
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x04_0x14_ICBMClientEvent{
- ScreenName: "recipient-screen-name",
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMClientEvent,
- },
- Body: tt.inputBody,
- }
- svc := newMockICBMService(t)
- svc.EXPECT().
- ClientEvent(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(tt.serviceError)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICBMEvilRequest(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x04_0x08_ICBMEvilRequest
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
- ScreenName: "recipient-screen-name",
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
- ScreenName: "recipient-screen-name",
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
- ScreenName: "recipient-screen-name",
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMEvilRequest,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMEvilReply,
- },
- Body: wire.SNAC_0x04_0x09_ICBMEvilReply{
- EvilDeltaApplied: 100,
- },
- }
- svc := newMockICBMService(t)
- svc.EXPECT().
- EvilRequest(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICBMOfflineRetrieve(t *testing.T) {
- tests := []struct {
- name string
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- },
- {
- name: "service error",
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMOfflineRetrieve,
- },
- Body: wire.SNAC_0x04_0x0A_ICBMOfflineRetrieve{},
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMOfflineRetrieveReply,
- },
- Body: wire.SNAC_0x04_0x17_ICBMOfflineRetrieveReply{},
- }
- svc := newMockICBMService(t)
- svc.EXPECT().
- OfflineRetrieve(mock.Anything, mock.Anything, input.Frame).
- Return(output, tt.serviceError)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICBMParameterQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "response writer error",
- inputBody: struct{}{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMParameterQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ICBM,
- SubGroup: wire.ICBMParameterReply,
- },
- Body: wire.SNAC_0x04_0x05_ICBMParameterReply{
- MaxSlots: 100,
- },
- }
- svc := newMockICBMService(t)
- svc.EXPECT().
- ParameterQuery(mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- ICBMService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ICQDBQuery(t *testing.T) {
- type ICQMetaRequest struct {
- wire.ICQMetadata
- ReqSubType uint16
- MetaRequest any
- }
- type reqParams struct {
- ctx context.Context
- sess *state.Session
- inFrame wire.SNACFrame
- inBody wire.SNAC_0x15_0x02_BQuery
- rw ResponseWriter
- seq uint16
- wantErr error
- }
- type mockParam struct {
- req any
- wantErr error
- }
- type allMockParams struct {
- deleteMsgReq *mockParam
- findByDetails *mockParam
- findByEmail *mockParam
- findByEmail3 *mockParam
- findByInterests *mockParam
- findByUIN *mockParam
- findByUIN2 *mockParam
- findByWhitePages2 *mockParam
- fullUserInfo *mockParam
- offlineMsgReq *mockParam
- setAffiliations *mockParam
- setBasicInfo *mockParam
- setEmails *mockParam
- setInterests *mockParam
- setMoreInfo *mockParam
- setPermissions *mockParam
- setUserNotes *mockParam
- setWorkInfo *mockParam
- shortUserInfo *mockParam
- xmlReqData *mockParam
- }
- tests := []struct {
- name string
- reqParams reqParams
- allMockParams allMockParams
- }{
- {
- name: "MetaReqFullInfo - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqFullInfo,
- MetaRequest: wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{
- UIN: 123456789,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- fullUserInfo: &mockParam{
- req: wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{
- UIN: 123456789,
- },
- },
- },
- },
- {
- name: "MetaReqShortInfo - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqShortInfo,
- MetaRequest: wire.ICQ_0x07D0_0x04BA_DBQueryMetaReqShortInfo{
- UIN: 123456789,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- shortUserInfo: &mockParam{
- req: wire.ICQ_0x07D0_0x04BA_DBQueryMetaReqShortInfo{
- UIN: 123456789,
- },
- },
- },
- },
- {
- name: "MetaReqFullInfo2 - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqFullInfo2,
- MetaRequest: wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{
- UIN: 123456789,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- fullUserInfo: &mockParam{
- req: wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{
- UIN: 123456789,
- },
- },
- },
- },
- {
- name: "MetaReqXMLReq - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqXMLReq,
- MetaRequest: wire.ICQ_0x07D0_0x0898_DBQueryMetaReqXMLReq{
- XMLRequest: "<xml></xml>",
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- xmlReqData: &mockParam{
- req: wire.ICQ_0x07D0_0x0898_DBQueryMetaReqXMLReq{
- XMLRequest: "<xml></xml>",
- },
- },
- },
- },
- {
- name: "MetaReqSetPermissions - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetPermissions,
- MetaRequest: wire.ICQ_0x07D0_0x0424_DBQueryMetaReqSetPermissions{
- Authorization: 1,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setPermissions: &mockParam{
- req: wire.ICQ_0x07D0_0x0424_DBQueryMetaReqSetPermissions{
- Authorization: 1,
- },
- },
- },
- },
- {
- name: "MetaReqSearchByUIN - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchByUIN,
- MetaRequest: wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{
- UIN: 123456789,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByUIN: &mockParam{
- req: wire.ICQ_0x07D0_0x051F_DBQueryMetaReqSearchByUIN{
- UIN: 123456789,
- },
- },
- },
- },
- {
- name: "MetaReqSearchByUIN2 - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchByUIN2,
- MetaRequest: wire.ICQ_0x07D0_0x0569_DBQueryMetaReqSearchByUIN2{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(1)),
- },
- },
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByUIN2: &mockParam{
- req: wire.ICQ_0x07D0_0x0569_DBQueryMetaReqSearchByUIN2{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(1)),
- },
- },
- },
- },
- },
- },
- {
- name: "MetaReqSearchByEmail - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchByEmail,
- MetaRequest: wire.ICQ_0x07D0_0x0529_DBQueryMetaReqSearchByEmail{
- Email: "test@aol.com",
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByEmail: &mockParam{
- req: wire.ICQ_0x07D0_0x0529_DBQueryMetaReqSearchByEmail{
- Email: "test@aol.com",
- },
- },
- },
- },
- {
- name: "MetaReqSearchByEmail3 - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchByEmail3,
- MetaRequest: wire.ICQ_0x07D0_0x0573_DBQueryMetaReqSearchByEmail3{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(1)),
- },
- },
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByEmail3: &mockParam{
- req: wire.ICQ_0x07D0_0x0573_DBQueryMetaReqSearchByEmail3{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(1)),
- },
- },
- },
- },
- },
- },
- {
- name: "MetaReqSearchByDetails - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchByDetails,
- MetaRequest: wire.ICQ_0x07D0_0x0515_DBQueryMetaReqSearchByDetails{
- FirstName: "john",
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByDetails: &mockParam{
- req: wire.ICQ_0x07D0_0x0515_DBQueryMetaReqSearchByDetails{
- FirstName: "john",
- },
- },
- },
- },
- {
- name: "MetaReqSearchWhitePages - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchWhitePages,
- MetaRequest: wire.ICQ_0x07D0_0x0533_DBQueryMetaReqSearchWhitePages{
- InterestsCode: 1,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByInterests: &mockParam{
- req: wire.ICQ_0x07D0_0x0533_DBQueryMetaReqSearchWhitePages{
- InterestsCode: 1,
- },
- },
- },
- },
- {
- name: "MetaReqSearchWhitePages2 - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchWhitePages2,
- MetaRequest: wire.ICQ_0x07D0_0x055F_DBQueryMetaReqSearchWhitePages2{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(1)),
- },
- },
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- findByWhitePages2: &mockParam{
- req: wire.ICQ_0x07D0_0x055F_DBQueryMetaReqSearchWhitePages2{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(1)),
- },
- },
- },
- },
- },
- },
- {
- name: "MetaReqSetBasicInfo - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetBasicInfo,
- MetaRequest: wire.ICQ_0x07D0_0x03EA_DBQueryMetaReqSetBasicInfo{
- FirstName: "john",
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setBasicInfo: &mockParam{
- req: wire.ICQ_0x07D0_0x03EA_DBQueryMetaReqSetBasicInfo{
- FirstName: "john",
- },
- },
- },
- },
- {
- name: "MetaReqSetWorkInfo - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetWorkInfo,
- MetaRequest: wire.ICQ_0x07D0_0x03F3_DBQueryMetaReqSetWorkInfo{
- ZIP: "11111",
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setWorkInfo: &mockParam{
- req: wire.ICQ_0x07D0_0x03F3_DBQueryMetaReqSetWorkInfo{
- ZIP: "11111",
- },
- },
- },
- },
- {
- name: "MetaReqSetMoreInfo - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetMoreInfo,
- MetaRequest: wire.ICQ_0x07D0_0x03FD_DBQueryMetaReqSetMoreInfo{
- Age: 100,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setMoreInfo: &mockParam{
- req: wire.ICQ_0x07D0_0x03FD_DBQueryMetaReqSetMoreInfo{
- Age: 100,
- },
- },
- },
- },
- {
- name: "MetaReqSetNotes - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetNotes,
- MetaRequest: wire.ICQ_0x07D0_0x0406_DBQueryMetaReqSetNotes{
- Notes: "my note",
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setUserNotes: &mockParam{
- req: wire.ICQ_0x07D0_0x0406_DBQueryMetaReqSetNotes{
- Notes: "my note",
- },
- },
- },
- },
- {
- name: "MetaReqSetEmails - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetEmails,
- MetaRequest: wire.ICQ_0x07D0_0x040B_DBQueryMetaReqSetEmails{
- Emails: []struct {
- Publish uint8
- Email string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Email: "test@aol.com",
- },
- },
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setEmails: &mockParam{
- req: wire.ICQ_0x07D0_0x040B_DBQueryMetaReqSetEmails{
- Emails: []struct {
- Publish uint8
- Email string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Email: "test@aol.com",
- },
- },
- },
- },
- },
- },
- {
- name: "MetaReqSetInterests - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetInterests,
- MetaRequest: wire.ICQ_0x07D0_0x0410_DBQueryMetaReqSetInterests{
- Interests: []struct {
- Code uint16
- Keyword string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Keyword: "an_interest",
- },
- },
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setInterests: &mockParam{
- req: wire.ICQ_0x07D0_0x0410_DBQueryMetaReqSetInterests{
- Interests: []struct {
- Code uint16
- Keyword string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Keyword: "an_interest",
- },
- },
- },
- },
- },
- },
- {
- name: "MetaReqSetAffiliations - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSetAffiliations,
- MetaRequest: wire.ICQ_0x07D0_0x041A_DBQueryMetaReqSetAffiliations{
- PastAffiliations: []struct {
- Code uint16
- Keyword string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Keyword: "a_past_affiliation",
- },
- },
- Affiliations: []struct {
- Code uint16
- Keyword string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Keyword: "an_affiliation",
- },
- },
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- setAffiliations: &mockParam{
- req: wire.ICQ_0x07D0_0x041A_DBQueryMetaReqSetAffiliations{
- PastAffiliations: []struct {
- Code uint16
- Keyword string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Keyword: "a_past_affiliation",
- },
- },
- Affiliations: []struct {
- Code uint16
- Keyword string `oscar:"len_prefix=uint16,nullterm"`
- }{
- {
- Keyword: "an_affiliation",
- },
- },
- },
- },
- },
- },
- {
- name: "MetaReqStat - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqStat0a8c,
- MetaRequest: struct{}{},
- },
- }),
- },
- },
- },
- seq: 1,
- },
- },
- {
- name: "unknown metadata request subtype",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: 0xCA8E,
- MetaRequest: struct{}{},
- },
- }),
- },
- },
- },
- seq: 1,
- wantErr: errUnknownICQMetaReqSubType,
- },
- },
- {
- name: "OfflineMsgReq - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: wire.ICQMetadataWithSubType{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryOfflineMsgReq,
- Seq: 1,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- offlineMsgReq: &mockParam{},
- },
- },
- {
- name: "DeleteMsgReq - happy path",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: wire.ICQMetadataWithSubType{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryDeleteMsgReq,
- Seq: 1,
- },
- },
- }),
- },
- },
- },
- seq: 1,
- },
- allMockParams: allMockParams{
- deleteMsgReq: &mockParam{},
- },
- },
- {
- name: "unknown request type",
- reqParams: reqParams{
- sess: &state.Session{},
- inBody: wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: 0x13B4,
- Seq: 1,
- },
- ReqSubType: 0xCA8E,
- MetaRequest: struct{}{},
- },
- }),
- },
- },
- },
- seq: 1,
- wantErr: errUnknownICQMetaReqType,
- },
- }, // todo: add to a separate test
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- icqService := newMockICQService(t)
- switch {
- case tt.allMockParams.fullUserInfo != nil:
- icqService.EXPECT().
- FullUserInfo(mock.Anything, tt.reqParams.sess, tt.allMockParams.fullUserInfo.req, tt.reqParams.seq).
- Return(tt.allMockParams.fullUserInfo.wantErr)
- case tt.allMockParams.shortUserInfo != nil:
- icqService.EXPECT().
- ShortUserInfo(mock.Anything, tt.reqParams.sess, tt.allMockParams.shortUserInfo.req, tt.reqParams.seq).
- Return(tt.allMockParams.shortUserInfo.wantErr)
- case tt.allMockParams.xmlReqData != nil:
- icqService.EXPECT().
- XMLReqData(mock.Anything, tt.reqParams.sess, tt.allMockParams.xmlReqData.req, tt.reqParams.seq).
- Return(tt.allMockParams.xmlReqData.wantErr)
- case tt.allMockParams.setPermissions != nil:
- icqService.EXPECT().
- SetPermissions(mock.Anything, tt.reqParams.sess, tt.allMockParams.setPermissions.req, tt.reqParams.seq).
- Return(tt.allMockParams.setPermissions.wantErr)
- case tt.allMockParams.findByUIN != nil:
- icqService.EXPECT().
- FindByUIN(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByUIN.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByUIN.wantErr)
- case tt.allMockParams.findByUIN2 != nil:
- icqService.EXPECT().
- FindByUIN2(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByUIN2.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByUIN2.wantErr)
- case tt.allMockParams.findByEmail != nil:
- icqService.EXPECT().
- FindByICQEmail(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByEmail.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByEmail.wantErr)
- case tt.allMockParams.findByEmail3 != nil:
- icqService.EXPECT().
- FindByEmail3(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByEmail3.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByEmail3.wantErr)
- case tt.allMockParams.findByDetails != nil:
- icqService.EXPECT().
- FindByICQName(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByDetails.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByDetails.wantErr)
- case tt.allMockParams.findByInterests != nil:
- icqService.EXPECT().
- FindByICQInterests(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByInterests.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByInterests.wantErr)
- case tt.allMockParams.findByWhitePages2 != nil:
- icqService.EXPECT().
- FindByWhitePages2(mock.Anything, tt.reqParams.sess, tt.allMockParams.findByWhitePages2.req, tt.reqParams.seq).
- Return(tt.allMockParams.findByWhitePages2.wantErr)
- case tt.allMockParams.setBasicInfo != nil:
- icqService.EXPECT().
- SetBasicInfo(mock.Anything, tt.reqParams.sess, tt.allMockParams.setBasicInfo.req, tt.reqParams.seq).
- Return(tt.allMockParams.setBasicInfo.wantErr)
- case tt.allMockParams.setWorkInfo != nil:
- icqService.EXPECT().
- SetWorkInfo(mock.Anything, tt.reqParams.sess, tt.allMockParams.setWorkInfo.req, tt.reqParams.seq).
- Return(tt.allMockParams.setWorkInfo.wantErr)
- case tt.allMockParams.setMoreInfo != nil:
- icqService.EXPECT().
- SetMoreInfo(mock.Anything, tt.reqParams.sess, tt.allMockParams.setMoreInfo.req, tt.reqParams.seq).
- Return(tt.allMockParams.setMoreInfo.wantErr)
- case tt.allMockParams.setUserNotes != nil:
- icqService.EXPECT().
- SetUserNotes(mock.Anything, tt.reqParams.sess, tt.allMockParams.setUserNotes.req, tt.reqParams.seq).
- Return(tt.allMockParams.setUserNotes.wantErr)
- case tt.allMockParams.setEmails != nil:
- icqService.EXPECT().
- SetEmails(mock.Anything, tt.reqParams.sess, tt.allMockParams.setEmails.req, tt.reqParams.seq).
- Return(tt.allMockParams.setEmails.wantErr)
- case tt.allMockParams.setInterests != nil:
- icqService.EXPECT().
- SetInterests(mock.Anything, tt.reqParams.sess, tt.allMockParams.setInterests.req, tt.reqParams.seq).
- Return(tt.allMockParams.setInterests.wantErr)
- case tt.allMockParams.setAffiliations != nil:
- icqService.EXPECT().
- SetAffiliations(mock.Anything, tt.reqParams.sess, tt.allMockParams.setAffiliations.req, tt.reqParams.seq).
- Return(tt.allMockParams.setAffiliations.wantErr)
- case tt.allMockParams.offlineMsgReq != nil:
- icqService.EXPECT().
- OfflineMsgReq(mock.Anything, tt.reqParams.sess, tt.reqParams.seq).
- Return(tt.allMockParams.offlineMsgReq.wantErr)
- case tt.allMockParams.deleteMsgReq != nil:
- icqService.EXPECT().
- DeleteMsgReq(mock.Anything, tt.reqParams.sess, tt.reqParams.seq).
- Return(tt.allMockParams.deleteMsgReq.wantErr)
- }
- h := Handler{
- ICQService: icqService,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(tt.reqParams.inBody, buf))
- frame := wire.SNACFrame{
- FoodGroup: wire.ICQ,
- SubGroup: wire.ICQDBQuery,
- }
- err := h.Handle(context.TODO(), wire.BOS, tt.reqParams.sess, frame, buf, nil, config.Listener{})
- assert.ErrorIs(t, err, tt.reqParams.wantErr)
- })
- }
- }
- // Test workaround for QIP 2005 bug where TLV length is incorrect.
- func TestHandler_ICQDBQuery_QIP2005UINSearchBug(t *testing.T) {
- icqService := newMockICQService(t)
- type ICQMetaRequest struct {
- wire.ICQMetadata
- ReqSubType uint16
- MetaRequest any
- }
- expect := wire.ICQ_0x07D0_0x0569_DBQueryMetaReqSearchByUIN2{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsUIN, uint32(100009)),
- },
- },
- }
- icqService.EXPECT().
- FindByUIN2(mock.Anything, &state.Session{}, expect, uint16(1)).
- Return(nil)
- h := Handler{
- ICQService: icqService,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- inBody := wire.SNAC_0x15_0x02_BQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsMetadata, wire.ICQMessageReplyEnvelope{
- Message: ICQMetaRequest{
- ICQMetadata: wire.ICQMetadata{
- ReqType: wire.ICQDBQueryMetaReq,
- Seq: 1,
- },
- ReqSubType: wire.ICQDBQueryMetaReqSearchByUIN2,
- MetaRequest: wire.ICQ_0x07D0_0x0569_DBQueryMetaReqSearchByUIN2{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(wire.ICQTLVTagsUIN, uint32(100009)),
- },
- },
- },
- },
- }),
- },
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(inBody, buf))
- b := buf.Bytes()
- b[18] = 6 // incorrectly set TLV length to 6 (should be 4)
- err := h.ICQDBQuery(nil, &state.Session{}, wire.SNACFrame{}, buf, nil)
- assert.NoError(t, err)
- }
- func TestHandler_ODirInfoQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0F_0x02_InfoQuery
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x0F_0x02_InfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(2)),
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x0F_0x02_InfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(2)),
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x0F_0x02_InfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(2)),
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ODir,
- SubGroup: wire.ODirInfoQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ODir,
- SubGroup: wire.ODirInfoReply,
- },
- Body: wire.SNAC_0x0F_0x03_InfoReply{
- Status: 5, // OK has results/not found
- },
- }
- svc := newMockODirService(t)
- svc.EXPECT().
- InfoQuery(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- ODirService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- ss := newMockResponseWriter(t)
- if tt.serviceError == nil {
- ss.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, ss, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_ODirKeywordListQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0F_0x02_InfoQuery
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x0F_0x02_InfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(2)),
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x0F_0x02_InfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(2)),
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x0F_0x02_InfoQuery{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(1, uint16(2)),
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ODir,
- SubGroup: wire.ODirKeywordListQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.ODir,
- SubGroup: wire.ODirKeywordListReply,
- },
- Body: wire.SNAC_0x0F_0x04_KeywordListReply{
- Status: 0x01,
- },
- }
- svc := newMockODirService(t)
- svc.EXPECT().
- KeywordListQuery(mock.Anything, input.Frame).
- Return(output, tt.serviceError)
- h := Handler{
- ODirService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- ss := newMockResponseWriter(t)
- if tt.serviceError == nil {
- ss.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, ss, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceClientOnline(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x02_OServiceClientOnline
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x02_OServiceClientOnline{
- GroupVersions: []struct {
- FoodGroup uint16
- Version uint16
- ToolID uint16
- ToolVersion uint16
- }{
- {
- FoodGroup: 10,
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x01_0x02_OServiceClientOnline{
- GroupVersions: []struct {
- FoodGroup uint16
- Version uint16
- ToolID uint16
- ToolVersion uint16
- }{
- {
- FoodGroup: 10,
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceClientOnline,
- },
- Body: tt.inputBody,
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- ClientOnline(mock.Anything, wire.BOS, input.Body, mock.Anything).
- Return(tt.serviceError)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceServiceRequest(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x04_OServiceServiceRequest
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x04_OServiceServiceRequest{
- FoodGroup: wire.Chat,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x01_0x04_OServiceServiceRequest{
- FoodGroup: wire.Chat,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x01_0x04_OServiceServiceRequest{
- FoodGroup: wire.Chat,
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceServiceRequest,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceServiceResponse,
- },
- Body: wire.SNAC_0x01_0x05_OServiceServiceResponse{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(1000)),
- },
- },
- },
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- ServiceRequest(mock.Anything, wire.BOS, mock.Anything, input.Frame, input.Body, config.Listener{BOSAdvertisedHostPlain: "127.0.0.1:1234"}).
- Return(output, tt.serviceError)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{BOSAdvertisedHostPlain: "127.0.0.1:1234"})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceIdleNotification(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x11_OServiceIdleNotification
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x11_OServiceIdleNotification{
- IdleTime: 10,
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x01_0x11_OServiceIdleNotification{
- IdleTime: 10,
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceIdleNotification,
- },
- Body: tt.inputBody,
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- IdleNotification(mock.Anything, mock.Anything, input.Body).
- Return(tt.serviceError)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceClientVersions(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x17_OServiceClientVersions
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x17_OServiceClientVersions{
- Versions: []uint16{
- 10,
- },
- },
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x01_0x17_OServiceClientVersions{
- Versions: []uint16{
- 10,
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceClientVersions,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceHostVersions,
- },
- Body: wire.SNAC_0x01_0x18_OServiceHostVersions{
- Versions: []uint16{
- 10,
- },
- },
- }
- sess := state.NewSession()
- svc := newMockOServiceService(t)
- svc.EXPECT().
- ClientVersions(mock.Anything, sess, input.Frame, input.Body).
- Return(output)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.Background(), wire.BOS, sess, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceRateParamsQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "response writer error",
- inputBody: struct{}{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceRateParamsQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceRateParamsReply,
- },
- Body: wire.SNAC_0x01_0x07_OServiceRateParamsReply{
- RateGroups: []struct {
- ID uint16
- Pairs []struct {
- FoodGroup uint16
- SubGroup uint16
- } `oscar:"count_prefix=uint16"`
- }{
- {
- ID: 1,
- },
- },
- },
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- RateParamsQuery(mock.Anything, mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceRateParamsSubAdd(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x08_OServiceRateParamsSubAdd{
- ClassIDs: []uint16{1, 2, 3, 4},
- },
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceRateParamsSubAdd,
- },
- Body: tt.inputBody,
- }
- sess := state.NewSession()
- svc := newMockOServiceService(t)
- svc.EXPECT().
- RateParamsSubAdd(mock.Anything, sess, input.Body)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.Background(), wire.BOS, sess, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceSetUserInfoFields(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, []byte{1, 2, 3, 4}),
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, []byte{1, 2, 3, 4}),
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x01_0x1E_OServiceSetUserInfoFields{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, []byte{1, 2, 3, 4}),
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceSetUserInfoFields,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceUserInfoUpdate,
- },
- Body: wire.SNAC_0x01_0x0F_OServiceUserInfoUpdate{
- UserInfo: []wire.TLVUserInfo{
- {ScreenName: "screen-name"},
- {ScreenName: "screen-name"},
- },
- },
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- SetUserInfoFields(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceUserInfoQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "response writer error",
- inputBody: struct{}{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceUserInfoQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceUserInfoUpdate,
- },
- Body: wire.SNAC_0x01_0x0F_OServiceUserInfoUpdate{
- UserInfo: []wire.TLVUserInfo{
- {ScreenName: "screen-name"},
- {ScreenName: "screen-name"},
- },
- },
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- UserInfoQuery(mock.Anything, mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceNoop(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceNoop,
- },
- Body: tt.inputBody,
- }
- h := Handler{
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_OServiceServiceSetPrivacyFlags(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x01_0x14_OServiceSetPrivacyFlags{
- PrivacyFlags: wire.OServicePrivacyFlagMember,
- },
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.OService,
- SubGroup: wire.OServiceSetPrivacyFlags,
- },
- Body: tt.inputBody,
- }
- svc := newMockOServiceService(t)
- svc.EXPECT().
- SetPrivacyFlags(mock.Anything, input.Body)
- h := Handler{
- OServiceService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_PermitDenyRightsQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "response writer error",
- inputBody: struct{}{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenyRightsQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenyRightsReply,
- },
- Body: wire.SNAC_0x09_0x03_PermitDenyRightsReply{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(1000)),
- },
- },
- },
- }
- svc := newMockPermitDenyService(t)
- svc.EXPECT().
- RightsQuery(mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- PermitDenyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_PermitDenyAddDenyListEntries(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- sess := state.NewSession()
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenyAddDenyListEntries,
- },
- Body: tt.inputBody,
- }
- svc := newMockPermitDenyService(t)
- svc.EXPECT().
- AddDenyListEntries(mock.Anything, sess, input.Body).
- Return(tt.serviceError)
- h := Handler{
- PermitDenyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_PermitDenyDelDenyListEntries(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x09_0x08_PermitDenyDelDenyListEntries
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x09_0x08_PermitDenyDelDenyListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x09_0x08_PermitDenyDelDenyListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- sess := state.NewSession()
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenyDelDenyListEntries,
- },
- Body: tt.inputBody,
- }
- svc := newMockPermitDenyService(t)
- svc.EXPECT().
- DelDenyListEntries(mock.Anything, sess, input.Body).
- Return(tt.serviceError)
- h := Handler{
- PermitDenyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_PermitDenyAddPermListEntries(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- sess := state.NewSession()
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenyAddPermListEntries,
- },
- Body: tt.inputBody,
- }
- svc := newMockPermitDenyService(t)
- svc.EXPECT().
- AddPermListEntries(mock.Anything, sess, input.Body).
- Return(tt.serviceError)
- h := Handler{
- PermitDenyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_PermitDenyDelPermListEntries(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x09_0x06_PermitDenyDelPermListEntries
- serviceError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x09_0x06_PermitDenyDelPermListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x09_0x06_PermitDenyDelPermListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- {
- ScreenName: "friend2",
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- sess := state.NewSession()
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenyDelPermListEntries,
- },
- Body: tt.inputBody,
- }
- svc := newMockPermitDenyService(t)
- svc.EXPECT().
- DelPermListEntries(mock.Anything, sess, input.Body).
- Return(tt.serviceError)
- h := Handler{
- PermitDenyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_PermitDenySetGroupPermitMask(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x09_0x06_PermitDenyDelPermListEntries
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x09_0x06_PermitDenyDelPermListEntries{
- Users: []struct {
- ScreenName string `oscar:"len_prefix=uint8"`
- }{
- {
- ScreenName: "friend1",
- },
- },
- },
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- sess := state.NewSession()
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.PermitDeny,
- SubGroup: wire.PermitDenySetGroupPermitMask,
- },
- Body: tt.inputBody,
- }
- svc := newMockPermitDenyService(t)
- h := Handler{
- PermitDenyService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, sess, input.Frame, buf, nil, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestUserLookupHandler_FindByEmail(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x0A_0x02_UserLookupFindByEmail
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x0A_0x02_UserLookupFindByEmail{
- Email: []byte("haha@aol.com"),
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x0A_0x02_UserLookupFindByEmail{
- Email: []byte("haha@aol.com"),
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x0A_0x02_UserLookupFindByEmail{
- Email: []byte("haha@aol.com"),
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.UserLookup,
- SubGroup: wire.UserLookupFindByEmail,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.UserLookup,
- SubGroup: wire.UserLookupFindReply,
- },
- Body: wire.SNAC_0x0A_0x03_UserLookupFindReply{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(0x02)),
- },
- },
- },
- }
- svc := newMockUserLookupService(t)
- svc.EXPECT().
- FindByEmail(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- UserLookupService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- ss := newMockResponseWriter(t)
- if tt.serviceError == nil {
- ss.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, ss, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_LocateGetDirInfo(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x02_0x0B_LocateGetDirInfo
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x02_0x0B_LocateGetDirInfo{
- ScreenName: "screen-name",
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x02_0x0B_LocateGetDirInfo{
- ScreenName: "screen-name",
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x02_0x0B_LocateGetDirInfo{
- ScreenName: "screen-name",
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateGetDirInfo,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateGetDirReply,
- },
- Body: wire.SNAC_0x02_0x0C_LocateGetDirReply{
- Status: 1,
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- DirInfo(mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_LocateRightsQuery(t *testing.T) {
- tests := []struct {
- name string
- inputBody struct{}
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: struct{}{},
- },
- {
- name: "response writer error",
- inputBody: struct{}{},
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateRightsQuery,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateRightsReply,
- },
- Body: wire.SNAC_0x02_0x03_LocateRightsReply{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- wire.NewTLVBE(0x01, uint16(1000)),
- },
- },
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- RightsQuery(mock.Anything, input.Frame).
- Return(output)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_LocateSetDirInfo(t *testing.T) {
- tests := []struct {
- name string
- inputBody wire.SNAC_0x02_0x09_LocateSetDirInfo
- serviceError error
- responseError error
- expectedError error
- }{
- {
- name: "success",
- inputBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- },
- {
- name: "service error",
- inputBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- serviceError: assert.AnError,
- expectedError: assert.AnError,
- },
- {
- name: "response writer error",
- inputBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- responseError: assert.AnError,
- expectedError: assert.AnError,
- },
- }
- for _, tt := range tests {
- t.Run(tt.name, func(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateSetDirInfo,
- },
- Body: tt.inputBody,
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateSetDirReply,
- },
- Body: wire.SNAC_0x02_0x0A_LocateSetDirReply{
- Result: 1,
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- SetDirInfo(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, tt.serviceError)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- if tt.serviceError == nil {
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(tt.responseError)
- }
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- err := h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{})
- if tt.expectedError != nil {
- assert.ErrorIs(t, err, tt.expectedError)
- } else {
- assert.NoError(t, err)
- }
- })
- }
- }
- func TestHandler_LocateSetInfo(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateSetInfo,
- },
- Body: wire.SNAC_0x02_0x04_LocateSetInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- SetInfo(mock.Anything, mock.Anything, input.Body).
- Return(nil)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{}))
- }
- func TestHandler_LocateSetKeywordInfo(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateSetKeywordInfo,
- },
- Body: wire.SNAC_0x02_0x0F_LocateSetKeywordInfo{
- TLVRestBlock: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateSetKeywordReply,
- },
- Body: wire.SNAC_0x02_0x10_LocateSetKeywordReply{
- Unknown: 1,
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- SetKeywordInfo(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, nil)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(nil)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{}))
- }
- func TestHandler_LocateUserInfoQuery(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateUserInfoQuery,
- },
- Body: wire.SNAC_0x02_0x05_LocateUserInfoQuery{
- Type: 1,
- },
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateUserInfoReply,
- },
- Body: wire.SNAC_0x02_0x06_LocateUserInfoReply{
- TLVUserInfo: wire.TLVUserInfo{
- ScreenName: "screen-name",
- },
- LocateInfo: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- UserInfoQuery(mock.Anything, mock.Anything, input.Frame, input.Body).
- Return(output, nil)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(nil)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{}))
- }
- func TestHandler_LocateUserInfoQuery2(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateUserInfoQuery2,
- },
- Body: wire.SNAC_0x02_0x15_LocateUserInfoQuery2{
- Type2: 1,
- },
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Locate,
- SubGroup: wire.LocateUserInfoReply,
- },
- Body: wire.SNAC_0x02_0x06_LocateUserInfoReply{
- TLVUserInfo: wire.TLVUserInfo{
- ScreenName: "screen-name",
- },
- LocateInfo: wire.TLVRestBlock{
- TLVList: wire.TLVList{
- {
- Tag: 0x01,
- Value: []byte{1, 2, 3, 4},
- },
- },
- },
- },
- }
- svc := newMockLocateService(t)
- svc.EXPECT().
- UserInfoQuery(mock.Anything, mock.Anything, input.Frame, wire.SNAC_0x02_0x05_LocateUserInfoQuery{Type: 1}).
- Return(output, nil)
- h := Handler{
- LocateService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- responseWriter := newMockResponseWriter(t)
- responseWriter.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(nil)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, responseWriter, config.Listener{}))
- }
- func TestHandler_StatsReportEvents(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Stats,
- SubGroup: wire.StatsReportEvents,
- },
- Body: wire.SNAC_0x0B_0x03_StatsReportEvents{},
- }
- output := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Stats,
- SubGroup: wire.StatsReportAck,
- },
- Body: wire.SNAC_0x0B_0x04_StatsReportAck{},
- }
- svc := newMockStatsService(t)
- svc.EXPECT().
- ReportEvents(mock.Anything, input.Frame, input.Body).
- Return(output)
- h := Handler{
- StatsService: svc,
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- ss := newMockResponseWriter(t)
- ss.EXPECT().
- SendSNAC(output.Frame, output.Body).
- Return(nil)
- buf := &bytes.Buffer{}
- assert.NoError(t, wire.MarshalBE(input.Body, buf))
- assert.NoError(t, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, buf, ss, config.Listener{}))
- }
- func TestHandler_RouteNotFound(t *testing.T) {
- input := wire.SNACMessage{
- Frame: wire.SNACFrame{
- FoodGroup: wire.Stats,
- SubGroup: math.MaxUint16,
- },
- Body: wire.SNAC_0x0B_0x03_StatsReportEvents{},
- }
- h := Handler{
- RouteLogger: middleware.RouteLogger{
- Logger: slog.Default(),
- },
- }
- assert.ErrorIs(t, ErrRouteNotFound, h.Handle(context.TODO(), wire.BOS, nil, input.Frame, nil, nil, config.Listener{}))
- }
|