cmd_client_test.go 212 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734573557365737573857395740574157425743574457455746574757485749575057515752575357545755575657575758575957605761576257635764576557665767576857695770577157725773577457755776577757785779578057815782578357845785578657875788578957905791579257935794579557965797579857995800580158025803580458055806580758085809581058115812581358145815581658175818581958205821582258235824582558265827582858295830583158325833583458355836583758385839584058415842584358445845584658475848584958505851585258535854585558565857585858595860586158625863586458655866586758685869587058715872587358745875587658775878587958805881588258835884588558865887588858895890589158925893589458955896589758985899590059015902590359045905590659075908590959105911591259135914591559165917591859195920592159225923592459255926592759285929593059315932593359345935593659375938593959405941594259435944594559465947594859495950595159525953595459555956595759585959596059615962596359645965596659675968596959705971597259735974597559765977597859795980598159825983598459855986598759885989599059915992599359945995599659975998599960006001600260036004600560066007600860096010601160126013601460156016601760186019602060216022602360246025602660276028602960306031603260336034603560366037603860396040604160426043604460456046604760486049605060516052605360546055605660576058605960606061606260636064606560666067606860696070607160726073607460756076607760786079608060816082608360846085608660876088608960906091609260936094609560966097609860996100610161026103610461056106610761086109611061116112611361146115611661176118611961206121612261236124612561266127612861296130613161326133613461356136613761386139614061416142614361446145614661476148614961506151615261536154615561566157615861596160616161626163616461656166616761686169617061716172617361746175617661776178617961806181618261836184618561866187618861896190619161926193619461956196619761986199620062016202620362046205620662076208620962106211621262136214621562166217621862196220622162226223622462256226622762286229623062316232623362346235623662376238623962406241624262436244624562466247624862496250625162526253625462556256625762586259626062616262626362646265626662676268626962706271627262736274627562766277627862796280628162826283628462856286628762886289629062916292629362946295629662976298629963006301630263036304630563066307630863096310631163126313631463156316631763186319632063216322632363246325632663276328632963306331633263336334633563366337633863396340634163426343634463456346634763486349635063516352635363546355635663576358635963606361636263636364636563666367636863696370637163726373637463756376637763786379638063816382638363846385638663876388638963906391639263936394639563966397639863996400640164026403640464056406640764086409641064116412641364146415641664176418641964206421642264236424642564266427642864296430643164326433643464356436643764386439644064416442644364446445644664476448644964506451645264536454645564566457645864596460646164626463646464656466646764686469647064716472647364746475647664776478647964806481648264836484648564866487648864896490649164926493649464956496649764986499650065016502650365046505650665076508650965106511651265136514651565166517651865196520652165226523652465256526652765286529653065316532653365346535653665376538653965406541654265436544654565466547654865496550655165526553655465556556655765586559656065616562656365646565656665676568656965706571657265736574657565766577657865796580658165826583658465856586658765886589659065916592659365946595659665976598659966006601660266036604660566066607660866096610661166126613661466156616661766186619662066216622662366246625662666276628662966306631663266336634663566366637663866396640664166426643664466456646664766486649665066516652665366546655665666576658665966606661666266636664666566666667666866696670667166726673667466756676667766786679668066816682668366846685668666876688668966906691669266936694669566966697669866996700670167026703670467056706670767086709671067116712671367146715671667176718671967206721672267236724672567266727672867296730673167326733673467356736673767386739674067416742674367446745674667476748674967506751675267536754675567566757675867596760676167626763676467656766676767686769677067716772677367746775677667776778677967806781678267836784678567866787678867896790679167926793679467956796679767986799680068016802680368046805680668076808680968106811681268136814681568166817681868196820682168226823682468256826682768286829683068316832683368346835683668376838683968406841684268436844684568466847684868496850685168526853685468556856685768586859686068616862686368646865686668676868686968706871687268736874687568766877687868796880688168826883688468856886688768886889689068916892689368946895689668976898689969006901690269036904690569066907690869096910691169126913691469156916691769186919692069216922692369246925692669276928692969306931693269336934693569366937693869396940694169426943694469456946694769486949695069516952695369546955695669576958695969606961696269636964696569666967696869696970697169726973697469756976697769786979698069816982698369846985698669876988698969906991699269936994699569966997699869997000700170027003700470057006700770087009701070117012701370147015701670177018701970207021702270237024702570267027702870297030703170327033703470357036703770387039704070417042704370447045704670477048704970507051705270537054705570567057705870597060706170627063706470657066706770687069707070717072707370747075707670777078707970807081708270837084708570867087708870897090709170927093709470957096709770987099710071017102710371047105710671077108710971107111711271137114711571167117711871197120712171227123712471257126712771287129713071317132713371347135713671377138713971407141714271437144714571467147714871497150715171527153715471557156715771587159716071617162716371647165716671677168716971707171717271737174717571767177717871797180718171827183718471857186718771887189719071917192719371947195719671977198719972007201720272037204720572067207720872097210721172127213721472157216721772187219722072217222722372247225722672277228722972307231723272337234723572367237723872397240724172427243
  1. package toc
  2. import (
  3. "context"
  4. "encoding/hex"
  5. "errors"
  6. "io"
  7. "log/slog"
  8. "testing"
  9. "github.com/google/uuid"
  10. "github.com/stretchr/testify/assert"
  11. "github.com/stretchr/testify/mock"
  12. "golang.org/x/sync/errgroup"
  13. "github.com/mk6i/open-oscar-server/config"
  14. "github.com/mk6i/open-oscar-server/state"
  15. "github.com/mk6i/open-oscar-server/wire"
  16. )
  17. func TestOSCARProxy_RecvClientCmd_AddBuddy(t *testing.T) {
  18. cases := []struct {
  19. // name is the unit test name
  20. name string
  21. // me is the TOC user session
  22. me *state.SessionInstance
  23. // givenCmd is the TOC command
  24. givenCmd []byte
  25. // wantMsg is the expected TOC response
  26. wantMsg []string
  27. // mockParams is the list of params sent to mocks that satisfy this
  28. // method's dependencies
  29. mockParams mockParams
  30. }{
  31. {
  32. name: "successfully add buddies",
  33. me: newTestSession("me"),
  34. givenCmd: []byte("toc_add_buddy friend1 friend2 friend3"),
  35. mockParams: mockParams{
  36. buddyParams: buddyParams{
  37. addBuddiesParams: addBuddiesParams{
  38. {
  39. me: state.NewIdentScreenName("me"),
  40. inBody: wire.SNAC_0x03_0x04_BuddyAddBuddies{
  41. Buddies: []struct {
  42. ScreenName string `oscar:"len_prefix=uint8"`
  43. }{
  44. {ScreenName: "friend1"},
  45. {ScreenName: "friend2"},
  46. {ScreenName: "friend3"},
  47. },
  48. },
  49. },
  50. },
  51. },
  52. },
  53. wantMsg: []string{},
  54. },
  55. {
  56. name: "add buddies with empty list",
  57. me: newTestSession("me"),
  58. givenCmd: []byte("toc_add_buddy"),
  59. mockParams: mockParams{
  60. buddyParams: buddyParams{
  61. addBuddiesParams: addBuddiesParams{
  62. {
  63. me: state.NewIdentScreenName("me"),
  64. inBody: wire.SNAC_0x03_0x04_BuddyAddBuddies{},
  65. },
  66. },
  67. },
  68. },
  69. wantMsg: []string{},
  70. },
  71. {
  72. name: "add buddies, receive error from buddy service",
  73. me: newTestSession("me"),
  74. givenCmd: []byte("toc_add_buddy friend1"),
  75. mockParams: mockParams{
  76. buddyParams: buddyParams{
  77. addBuddiesParams: addBuddiesParams{
  78. {
  79. me: state.NewIdentScreenName("me"),
  80. inBody: wire.SNAC_0x03_0x04_BuddyAddBuddies{
  81. Buddies: []struct {
  82. ScreenName string `oscar:"len_prefix=uint8"`
  83. }{
  84. {ScreenName: "friend1"},
  85. },
  86. },
  87. err: io.EOF,
  88. },
  89. },
  90. },
  91. },
  92. wantMsg: []string{cmdInternalSvcErr},
  93. },
  94. }
  95. for _, tc := range cases {
  96. t.Run(tc.name, func(t *testing.T) {
  97. ctx := context.Background()
  98. buddySvc := newMockBuddyService(t)
  99. for _, params := range tc.mockParams.addBuddiesParams {
  100. buddySvc.EXPECT().
  101. AddBuddies(ctx, matchSession(params.me), params.inBody).
  102. Return(params.err)
  103. }
  104. svc := OSCARProxy{
  105. Logger: slog.Default(),
  106. BuddyService: buddySvc,
  107. }
  108. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  109. assert.Equal(t, tc.wantMsg, msg)
  110. })
  111. }
  112. }
  113. func TestOSCARProxy_RecvClientCmd_AddPermit(t *testing.T) {
  114. cases := []struct {
  115. // name is the unit test name
  116. name string
  117. // me is the TOC user session
  118. me *state.SessionInstance
  119. // givenCmd is the TOC command
  120. givenCmd []byte
  121. // wantMsg is the expected TOC response
  122. wantMsg []string
  123. // mockParams is the list of params sent to mocks that satisfy this
  124. // method's dependencies
  125. mockParams mockParams
  126. }{
  127. {
  128. name: "successfully permit buddies",
  129. me: newTestSession("me"),
  130. givenCmd: []byte("toc_add_permit friend1 friend2 friend3"),
  131. mockParams: mockParams{
  132. permitDenyParams: permitDenyParams{
  133. addPermListEntriesParams: addPermListEntriesParams{
  134. {
  135. me: state.NewIdentScreenName("me"),
  136. body: wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{
  137. Users: []struct {
  138. ScreenName string `oscar:"len_prefix=uint8"`
  139. }{
  140. {ScreenName: "friend1"},
  141. {ScreenName: "friend2"},
  142. {ScreenName: "friend3"},
  143. },
  144. },
  145. },
  146. },
  147. },
  148. },
  149. wantMsg: []string{},
  150. },
  151. {
  152. name: "permit buddies, receive error from buddy service",
  153. me: newTestSession("me"),
  154. givenCmd: []byte("toc_add_permit friend1"),
  155. mockParams: mockParams{
  156. permitDenyParams: permitDenyParams{
  157. addPermListEntriesParams: addPermListEntriesParams{
  158. {
  159. me: state.NewIdentScreenName("me"),
  160. body: wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{
  161. Users: []struct {
  162. ScreenName string `oscar:"len_prefix=uint8"`
  163. }{
  164. {ScreenName: "friend1"},
  165. },
  166. },
  167. err: io.EOF,
  168. },
  169. },
  170. },
  171. },
  172. wantMsg: []string{cmdInternalSvcErr},
  173. },
  174. {
  175. name: "permit buddies with empty list",
  176. me: newTestSession("me"),
  177. givenCmd: []byte("toc_add_permit"),
  178. mockParams: mockParams{
  179. permitDenyParams: permitDenyParams{
  180. addPermListEntriesParams: addPermListEntriesParams{
  181. {
  182. me: state.NewIdentScreenName("me"),
  183. body: wire.SNAC_0x09_0x05_PermitDenyAddPermListEntries{},
  184. },
  185. },
  186. },
  187. },
  188. wantMsg: []string{},
  189. },
  190. }
  191. for _, tc := range cases {
  192. t.Run(tc.name, func(t *testing.T) {
  193. ctx := context.Background()
  194. pdSvc := newMockPermitDenyService(t)
  195. for _, params := range tc.mockParams.addPermListEntriesParams {
  196. pdSvc.EXPECT().
  197. AddPermListEntries(ctx, matchSession(params.me), params.body).
  198. Return(params.err)
  199. }
  200. svc := OSCARProxy{
  201. Logger: slog.Default(),
  202. PermitDenyService: pdSvc,
  203. }
  204. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  205. assert.Equal(t, tc.wantMsg, msg)
  206. })
  207. }
  208. }
  209. func TestOSCARProxy_RecvClientCmd_AddDeny(t *testing.T) {
  210. cases := []struct {
  211. // name is the unit test name
  212. name string
  213. // me is the TOC user session
  214. me *state.SessionInstance
  215. // givenCmd is the TOC command
  216. givenCmd []byte
  217. // wantMsg is the expected TOC response
  218. wantMsg []string
  219. // mockParams is the list of params sent to mocks that satisfy this
  220. // method's dependencies
  221. mockParams mockParams
  222. }{
  223. {
  224. name: "successfully deny buddies",
  225. me: newTestSession("me"),
  226. givenCmd: []byte("toc_add_deny friend1 friend2 friend3"),
  227. mockParams: mockParams{
  228. permitDenyParams: permitDenyParams{
  229. addDenyListEntriesParams: addDenyListEntriesParams{
  230. {
  231. me: state.NewIdentScreenName("me"),
  232. body: wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{
  233. Users: []struct {
  234. ScreenName string `oscar:"len_prefix=uint8"`
  235. }{
  236. {ScreenName: "friend1"},
  237. {ScreenName: "friend2"},
  238. {ScreenName: "friend3"},
  239. },
  240. },
  241. },
  242. },
  243. },
  244. },
  245. wantMsg: []string{},
  246. },
  247. {
  248. name: "deny buddies, receive error from buddy service",
  249. me: newTestSession("me"),
  250. givenCmd: []byte("toc_add_deny friend1"),
  251. mockParams: mockParams{
  252. permitDenyParams: permitDenyParams{
  253. addDenyListEntriesParams: addDenyListEntriesParams{
  254. {
  255. me: state.NewIdentScreenName("me"),
  256. body: wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{
  257. Users: []struct {
  258. ScreenName string `oscar:"len_prefix=uint8"`
  259. }{
  260. {ScreenName: "friend1"},
  261. },
  262. },
  263. err: io.EOF,
  264. },
  265. },
  266. },
  267. },
  268. wantMsg: []string{cmdInternalSvcErr},
  269. },
  270. {
  271. name: "deny buddies with empty list",
  272. me: newTestSession("me"),
  273. givenCmd: []byte("toc_add_deny"),
  274. mockParams: mockParams{
  275. permitDenyParams: permitDenyParams{
  276. addDenyListEntriesParams: addDenyListEntriesParams{
  277. {
  278. me: state.NewIdentScreenName("me"),
  279. body: wire.SNAC_0x09_0x07_PermitDenyAddDenyListEntries{},
  280. },
  281. },
  282. },
  283. },
  284. wantMsg: []string{},
  285. },
  286. }
  287. for _, tc := range cases {
  288. t.Run(tc.name, func(t *testing.T) {
  289. ctx := context.Background()
  290. pdSvc := newMockPermitDenyService(t)
  291. for _, params := range tc.mockParams.addDenyListEntriesParams {
  292. pdSvc.EXPECT().
  293. AddDenyListEntries(ctx, matchSession(params.me), params.body).
  294. Return(params.err)
  295. }
  296. svc := OSCARProxy{
  297. Logger: slog.Default(),
  298. PermitDenyService: pdSvc,
  299. }
  300. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  301. assert.Equal(t, tc.wantMsg, msg)
  302. })
  303. }
  304. }
  305. func TestOSCARProxy_RecvClientCmd_FormatNickname(t *testing.T) {
  306. cases := []struct {
  307. // name is the unit test name
  308. name string
  309. // me is the TOC user session
  310. me *state.SessionInstance
  311. // givenCmd is the TOC command
  312. givenCmd []byte
  313. // wantMsg is the expected TOC response
  314. wantMsg []string
  315. // mockParams is the list of params sent to mocks that satisfy this
  316. // method's dependencies
  317. mockParams mockParams
  318. }{
  319. {
  320. name: "successfully change screen name format",
  321. me: newTestSession("myScreenName"),
  322. givenCmd: []byte("toc_format_nickname mYsCrEeNnAmE"),
  323. mockParams: mockParams{
  324. adminParams: adminParams{
  325. infoChangeRequestParams: infoChangeRequestParams{
  326. {
  327. me: state.NewIdentScreenName("myScreenName"),
  328. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  329. TLVRestBlock: wire.TLVRestBlock{
  330. TLVList: wire.TLVList{
  331. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "mYsCrEeNnAmE"),
  332. },
  333. },
  334. },
  335. msg: wire.SNACMessage{
  336. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  337. Permissions: wire.AdminInfoPermissionsReadWrite,
  338. TLVBlock: wire.TLVBlock{
  339. TLVList: wire.TLVList{
  340. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "mYsCrEeNnAmE"),
  341. },
  342. },
  343. },
  344. },
  345. },
  346. },
  347. },
  348. },
  349. wantMsg: []string{"ADMIN_NICK_STATUS:0", "NICK:mYsCrEeNnAmE"},
  350. },
  351. {
  352. name: "format nickname - invalid length",
  353. me: newTestSession("sn"),
  354. givenCmd: []byte("toc_format_nickname sN"),
  355. mockParams: mockParams{
  356. adminParams: adminParams{
  357. infoChangeRequestParams: infoChangeRequestParams{
  358. {
  359. me: state.NewIdentScreenName("sn"),
  360. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  361. TLVRestBlock: wire.TLVRestBlock{
  362. TLVList: wire.TLVList{
  363. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "sN"),
  364. },
  365. },
  366. },
  367. msg: wire.SNACMessage{
  368. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  369. Permissions: wire.AdminInfoPermissionsReadWrite,
  370. TLVBlock: wire.TLVBlock{
  371. TLVList: wire.TLVList{
  372. wire.NewTLVBE(wire.AdminTLVErrorCode, wire.AdminInfoErrorInvalidNickNameLength),
  373. wire.NewTLVBE(wire.AdminTLVUrl, ""),
  374. },
  375. },
  376. },
  377. },
  378. },
  379. },
  380. },
  381. },
  382. wantMsg: []string{"ERROR:911"},
  383. },
  384. {
  385. name: "format nickname - invalid screen name",
  386. me: newTestSession("sn"),
  387. givenCmd: []byte("toc_format_nickname sN"),
  388. mockParams: mockParams{
  389. adminParams: adminParams{
  390. infoChangeRequestParams: infoChangeRequestParams{
  391. {
  392. me: state.NewIdentScreenName("sn"),
  393. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  394. TLVRestBlock: wire.TLVRestBlock{
  395. TLVList: wire.TLVList{
  396. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "sN"),
  397. },
  398. },
  399. },
  400. msg: wire.SNACMessage{
  401. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  402. Permissions: wire.AdminInfoPermissionsReadWrite,
  403. TLVBlock: wire.TLVBlock{
  404. TLVList: wire.TLVList{
  405. wire.NewTLVBE(wire.AdminTLVErrorCode, wire.AdminInfoErrorInvalidNickName),
  406. wire.NewTLVBE(wire.AdminTLVUrl, ""),
  407. },
  408. },
  409. },
  410. },
  411. },
  412. },
  413. },
  414. },
  415. wantMsg: []string{"ERROR:911"},
  416. },
  417. {
  418. name: "format nickname - catch-all error",
  419. me: newTestSession("sn"),
  420. givenCmd: []byte("toc_format_nickname sN"),
  421. mockParams: mockParams{
  422. adminParams: adminParams{
  423. infoChangeRequestParams: infoChangeRequestParams{
  424. {
  425. me: state.NewIdentScreenName("sn"),
  426. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  427. TLVRestBlock: wire.TLVRestBlock{
  428. TLVList: wire.TLVList{
  429. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "sN"),
  430. },
  431. },
  432. },
  433. msg: wire.SNACMessage{
  434. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  435. Permissions: wire.AdminInfoPermissionsReadWrite,
  436. TLVBlock: wire.TLVBlock{
  437. TLVList: wire.TLVList{
  438. wire.NewTLVBE(wire.AdminTLVErrorCode, wire.AdminInfoErrorValidateNickName),
  439. wire.NewTLVBE(wire.AdminTLVUrl, ""),
  440. },
  441. },
  442. },
  443. },
  444. },
  445. },
  446. },
  447. },
  448. wantMsg: []string{"ERROR:913"},
  449. },
  450. {
  451. name: "format nickname - runtime error from admin svc",
  452. me: newTestSession("myScreenName"),
  453. givenCmd: []byte("toc_format_nickname mYsCrEeNnAmE"),
  454. mockParams: mockParams{
  455. adminParams: adminParams{
  456. infoChangeRequestParams: infoChangeRequestParams{
  457. {
  458. me: state.NewIdentScreenName("myScreenName"),
  459. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  460. TLVRestBlock: wire.TLVRestBlock{
  461. TLVList: wire.TLVList{
  462. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "mYsCrEeNnAmE"),
  463. },
  464. },
  465. },
  466. err: io.EOF,
  467. },
  468. },
  469. },
  470. },
  471. wantMsg: []string{cmdInternalSvcErr},
  472. },
  473. {
  474. name: "change password - unexpected response from admin svc",
  475. me: newTestSession("myScreenName"),
  476. givenCmd: []byte("toc_format_nickname mYsCrEeNnAmE"),
  477. mockParams: mockParams{
  478. adminParams: adminParams{
  479. infoChangeRequestParams: infoChangeRequestParams{
  480. {
  481. me: state.NewIdentScreenName("myScreenName"),
  482. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  483. TLVRestBlock: wire.TLVRestBlock{
  484. TLVList: wire.TLVList{
  485. wire.NewTLVBE(wire.AdminTLVScreenNameFormatted, "mYsCrEeNnAmE"),
  486. },
  487. },
  488. },
  489. msg: wire.SNACMessage{
  490. Body: wire.SNACError{},
  491. },
  492. },
  493. },
  494. },
  495. },
  496. wantMsg: []string{cmdInternalSvcErr},
  497. },
  498. {
  499. name: "bad command",
  500. me: newTestSession("me"),
  501. givenCmd: []byte(`toc_format_nickname`),
  502. wantMsg: []string{cmdInternalSvcErr},
  503. },
  504. }
  505. for _, tc := range cases {
  506. t.Run(tc.name, func(t *testing.T) {
  507. ctx := context.Background()
  508. adminSvc := newMockAdminService(t)
  509. for _, params := range tc.mockParams.infoChangeRequestParams {
  510. adminSvc.EXPECT().
  511. InfoChangeRequest(ctx, matchSession(params.me), wire.SNACFrame{}, params.inBody).
  512. Return(params.msg, params.err)
  513. }
  514. svc := OSCARProxy{
  515. Logger: slog.Default(),
  516. AdminService: adminSvc,
  517. }
  518. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  519. assert.Equal(t, tc.wantMsg, msg)
  520. })
  521. }
  522. }
  523. func TestOSCARProxy_RecvClientCmd_ChatAccept(t *testing.T) {
  524. navInfo := wire.SNACMessage{
  525. Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{
  526. TLVRestBlock: wire.TLVRestBlock{
  527. TLVList: wire.TLVList{
  528. wire.NewTLVBE(wire.ChatNavTLVRoomInfo, wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
  529. Cookie: "the-cookie",
  530. TLVBlock: wire.TLVBlock{
  531. TLVList: wire.TLVList{
  532. wire.NewTLVBE(wire.ChatRoomTLVRoomName, "cool room"),
  533. },
  534. },
  535. }),
  536. },
  537. },
  538. },
  539. }
  540. svcReq := wire.SNAC_0x01_0x04_OServiceServiceRequest{
  541. FoodGroup: wire.Chat,
  542. TLVRestBlock: wire.TLVRestBlock{
  543. TLVList: wire.TLVList{
  544. wire.NewTLVBE(0x01, wire.SNAC_0x01_0x04_TLVRoomInfo{
  545. Cookie: "the-cookie",
  546. }),
  547. },
  548. },
  549. }
  550. svcResp := wire.SNACMessage{
  551. Body: wire.SNAC_0x01_0x05_OServiceServiceResponse{
  552. TLVRestBlock: wire.TLVRestBlock{
  553. TLVList: wire.TLVList{
  554. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, "chat-auth-cookie"),
  555. },
  556. },
  557. },
  558. }
  559. cases := []struct {
  560. // name is the unit test name
  561. name string
  562. // me is the TOC user session
  563. me *state.SessionInstance
  564. // givenCmd is the TOC command
  565. givenCmd []byte
  566. // givenChatRegistry is the chat registry passed to the function
  567. givenChatRegistry *ChatRegistry
  568. // wantMsg is the expected TOC response
  569. wantMsg []string
  570. // expectChatSession indicates whether a chat session should be present
  571. // in the chat registry
  572. expectChatSession bool
  573. // checkSession validates the state of the registered chat session (chat ID 0)
  574. checkSession func(*testing.T, *state.SessionInstance)
  575. // mockParams is the list of params sent to mocks that satisfy this
  576. // method's dependencies
  577. mockParams mockParams
  578. }{
  579. {
  580. name: "successfully accept chat",
  581. me: newTestSession("me"),
  582. givenCmd: []byte(`toc_chat_accept 0`),
  583. givenChatRegistry: func() *ChatRegistry {
  584. reg := NewChatRegistry()
  585. reg.Add(wire.ICBMRoomInfo{
  586. Cookie: "the-cookie",
  587. Exchange: 4,
  588. Instance: 0,
  589. })
  590. return reg
  591. }(),
  592. mockParams: mockParams{
  593. chatNavParams: chatNavParams{
  594. requestRoomInfoParams: requestRoomInfoParams{
  595. {
  596. inBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
  597. Cookie: "the-cookie",
  598. Exchange: 4,
  599. InstanceNumber: 0,
  600. },
  601. msg: navInfo,
  602. },
  603. },
  604. },
  605. oServiceParams: oServiceParams{
  606. serviceRequestParams: serviceRequestParams{
  607. {
  608. me: state.NewIdentScreenName("me"),
  609. bodyIn: svcReq,
  610. msg: svcResp,
  611. },
  612. },
  613. clientOnlineParams: clientOnlineParams{
  614. {
  615. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  616. me: state.NewIdentScreenName("me"),
  617. },
  618. },
  619. },
  620. authParams: authParams{
  621. crackCookieParams: crackCookieParams{
  622. {
  623. cookieIn: []byte("chat-auth-cookie"),
  624. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  625. },
  626. },
  627. registerChatSessionParams: registerChatSessionParams{
  628. {
  629. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  630. instance: newTestSession("me"),
  631. },
  632. },
  633. },
  634. },
  635. wantMsg: []string{"CHAT_JOIN:0:cool room"},
  636. expectChatSession: true,
  637. },
  638. {
  639. name: "accept chat - TOC2 with encoded messaging propagates to chat session",
  640. me: newTestSession("me", func(i *state.SessionInstance) { i.SetTOC2(true) }),
  641. givenCmd: []byte(`toc_chat_accept 0`),
  642. givenChatRegistry: func() *ChatRegistry {
  643. reg := NewChatRegistry()
  644. reg.Add(wire.ICBMRoomInfo{
  645. Cookie: "the-cookie",
  646. Exchange: 4,
  647. Instance: 0,
  648. })
  649. return reg
  650. }(),
  651. mockParams: mockParams{
  652. chatNavParams: chatNavParams{
  653. requestRoomInfoParams: requestRoomInfoParams{
  654. {
  655. inBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
  656. Cookie: "the-cookie",
  657. Exchange: 4,
  658. InstanceNumber: 0,
  659. },
  660. msg: navInfo,
  661. },
  662. },
  663. },
  664. oServiceParams: oServiceParams{
  665. serviceRequestParams: serviceRequestParams{
  666. {
  667. me: state.NewIdentScreenName("me"),
  668. bodyIn: svcReq,
  669. msg: svcResp,
  670. },
  671. },
  672. clientOnlineParams: clientOnlineParams{
  673. {
  674. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  675. me: state.NewIdentScreenName("me"),
  676. },
  677. },
  678. },
  679. authParams: authParams{
  680. crackCookieParams: crackCookieParams{
  681. {
  682. cookieIn: []byte("chat-auth-cookie"),
  683. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  684. },
  685. },
  686. registerChatSessionParams: registerChatSessionParams{
  687. {
  688. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  689. instance: newTestSession("me"),
  690. },
  691. },
  692. },
  693. },
  694. wantMsg: []string{"CHAT_JOIN:0:cool room"},
  695. expectChatSession: true,
  696. checkSession: func(t *testing.T, sess *state.SessionInstance) {
  697. assert.NotNil(t, sess, "chat session should be registered")
  698. assert.True(t, sess.IsTOC2(), "chat session should have TOC2 set")
  699. assert.True(t, sess.SupportsTOC2MsgEnc(), "chat session should have TOC2 msg enc set")
  700. },
  701. },
  702. {
  703. name: "accept chat, receive error from client online",
  704. me: newTestSession("me"),
  705. givenCmd: []byte(`toc_chat_accept 0`),
  706. givenChatRegistry: func() *ChatRegistry {
  707. reg := NewChatRegistry()
  708. reg.Add(wire.ICBMRoomInfo{
  709. Cookie: "the-cookie",
  710. Exchange: 4,
  711. Instance: 0,
  712. })
  713. return reg
  714. }(),
  715. mockParams: mockParams{
  716. chatNavParams: chatNavParams{
  717. requestRoomInfoParams: requestRoomInfoParams{
  718. {
  719. inBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
  720. Cookie: "the-cookie",
  721. Exchange: 4,
  722. InstanceNumber: 0,
  723. },
  724. msg: navInfo,
  725. },
  726. },
  727. },
  728. oServiceParams: oServiceParams{
  729. serviceRequestParams: serviceRequestParams{
  730. {
  731. me: state.NewIdentScreenName("me"),
  732. bodyIn: svcReq,
  733. msg: svcResp,
  734. },
  735. },
  736. clientOnlineParams: clientOnlineParams{
  737. {
  738. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  739. me: state.NewIdentScreenName("me"),
  740. err: io.EOF,
  741. },
  742. },
  743. },
  744. authParams: authParams{
  745. crackCookieParams: crackCookieParams{
  746. {
  747. cookieIn: []byte("chat-auth-cookie"),
  748. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  749. },
  750. },
  751. registerChatSessionParams: registerChatSessionParams{
  752. {
  753. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  754. instance: newTestSession("me"),
  755. },
  756. },
  757. },
  758. },
  759. wantMsg: []string{cmdInternalSvcErr},
  760. expectChatSession: false,
  761. },
  762. {
  763. name: "accept chat, receive error from register chat session",
  764. me: newTestSession("me"),
  765. givenCmd: []byte(`toc_chat_accept 0`),
  766. givenChatRegistry: func() *ChatRegistry {
  767. reg := NewChatRegistry()
  768. reg.Add(wire.ICBMRoomInfo{
  769. Cookie: "the-cookie",
  770. Exchange: 4,
  771. Instance: 0,
  772. })
  773. return reg
  774. }(),
  775. mockParams: mockParams{
  776. chatNavParams: chatNavParams{
  777. requestRoomInfoParams: requestRoomInfoParams{
  778. {
  779. inBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
  780. Cookie: "the-cookie",
  781. Exchange: 4,
  782. InstanceNumber: 0,
  783. },
  784. msg: navInfo,
  785. },
  786. },
  787. },
  788. oServiceParams: oServiceParams{
  789. serviceRequestParams: serviceRequestParams{
  790. {
  791. me: state.NewIdentScreenName("me"),
  792. bodyIn: svcReq,
  793. msg: svcResp,
  794. },
  795. },
  796. },
  797. authParams: authParams{
  798. crackCookieParams: crackCookieParams{
  799. {
  800. cookieIn: []byte("chat-auth-cookie"),
  801. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  802. },
  803. },
  804. registerChatSessionParams: registerChatSessionParams{
  805. {
  806. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  807. instance: newTestSession("me"),
  808. err: io.EOF,
  809. },
  810. },
  811. },
  812. },
  813. wantMsg: []string{cmdInternalSvcErr},
  814. expectChatSession: false,
  815. },
  816. {
  817. name: "accept chat, receive error from BOS oservice svc",
  818. me: newTestSession("me"),
  819. givenCmd: []byte(`toc_chat_accept 0`),
  820. givenChatRegistry: func() *ChatRegistry {
  821. reg := NewChatRegistry()
  822. reg.Add(wire.ICBMRoomInfo{
  823. Cookie: "the-cookie",
  824. Exchange: 4,
  825. Instance: 0,
  826. })
  827. return reg
  828. }(),
  829. mockParams: mockParams{
  830. chatNavParams: chatNavParams{
  831. requestRoomInfoParams: requestRoomInfoParams{
  832. {
  833. inBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
  834. Cookie: "the-cookie",
  835. Exchange: 4,
  836. InstanceNumber: 0,
  837. },
  838. msg: navInfo,
  839. },
  840. },
  841. },
  842. oServiceParams: oServiceParams{
  843. serviceRequestParams: serviceRequestParams{
  844. {
  845. me: state.NewIdentScreenName("me"),
  846. bodyIn: svcReq,
  847. err: io.EOF,
  848. },
  849. },
  850. },
  851. },
  852. wantMsg: []string{cmdInternalSvcErr},
  853. expectChatSession: false,
  854. },
  855. {
  856. name: "accept chat, receive error from chat nav svc",
  857. me: newTestSession("me"),
  858. givenCmd: []byte(`toc_chat_accept 0`),
  859. givenChatRegistry: func() *ChatRegistry {
  860. reg := NewChatRegistry()
  861. reg.Add(wire.ICBMRoomInfo{
  862. Cookie: "the-cookie",
  863. Exchange: 4,
  864. Instance: 0,
  865. })
  866. return reg
  867. }(),
  868. mockParams: mockParams{
  869. chatNavParams: chatNavParams{
  870. requestRoomInfoParams: requestRoomInfoParams{
  871. {
  872. inBody: wire.SNAC_0x0D_0x04_ChatNavRequestRoomInfo{
  873. Cookie: "the-cookie",
  874. Exchange: 4,
  875. InstanceNumber: 0,
  876. },
  877. err: io.EOF,
  878. },
  879. },
  880. },
  881. },
  882. wantMsg: []string{cmdInternalSvcErr},
  883. expectChatSession: false,
  884. },
  885. {
  886. name: "bad command",
  887. me: newTestSession("me"),
  888. givenCmd: []byte(`toc_chat_accept`),
  889. givenChatRegistry: NewChatRegistry(),
  890. wantMsg: []string{cmdInternalSvcErr},
  891. expectChatSession: false,
  892. },
  893. {
  894. name: "bad exchange number",
  895. me: newTestSession("me"),
  896. givenCmd: []byte(`toc_chat_accept four`),
  897. givenChatRegistry: NewChatRegistry(),
  898. wantMsg: []string{cmdInternalSvcErr},
  899. expectChatSession: false,
  900. },
  901. }
  902. for _, tc := range cases {
  903. t.Run(tc.name, func(t *testing.T) {
  904. ctx, cancel := context.WithCancel(context.Background())
  905. cancel()
  906. chatNavSvc := newMockChatNavService(t)
  907. for _, params := range tc.mockParams.requestRoomInfoParams {
  908. chatNavSvc.EXPECT().
  909. RequestRoomInfo(ctx, wire.SNACFrame{}, params.inBody).
  910. Return(params.msg, params.err)
  911. }
  912. oServiceSvc := newMockOServiceService(t)
  913. for _, params := range tc.mockParams.oServiceParams.serviceRequestParams {
  914. oServiceSvc.EXPECT().
  915. ServiceRequest(ctx, wire.BOS, matchSession(params.me), wire.SNACFrame{}, params.bodyIn, config.Listener{}).
  916. Return(params.msg, params.err)
  917. }
  918. for _, params := range tc.mockParams.oServiceParams.clientOnlineParams {
  919. oServiceSvc.EXPECT().
  920. ClientOnline(ctx, wire.Chat, params.body, matchSession(params.me)).
  921. Return(params.err)
  922. }
  923. authSvc := newMockAuthService(t)
  924. for _, params := range tc.mockParams.authParams.registerChatSessionParams {
  925. authSvc.EXPECT().
  926. RegisterChatSession(ctx, params.authCookie).
  927. Return(params.instance, params.err)
  928. }
  929. for _, params := range tc.mockParams.authParams.crackCookieParams {
  930. authSvc.EXPECT().
  931. CrackCookie(params.cookieIn).
  932. Return(params.cookieOut, params.err)
  933. }
  934. svc := OSCARProxy{
  935. AuthService: authSvc,
  936. ChatNavService: chatNavSvc,
  937. Logger: slog.Default(),
  938. OServiceService: oServiceSvc,
  939. }
  940. g := &errgroup.Group{}
  941. tc.me.CloseInstance()
  942. msg := svc.RecvClientCmd(ctx, tc.me, tc.givenChatRegistry, tc.givenCmd, nil, g.Go)
  943. assert.NoError(t, g.Wait())
  944. assert.Equal(t, tc.wantMsg, msg)
  945. assert.Equal(t, tc.expectChatSession, len(tc.givenChatRegistry.Sessions()) == 1)
  946. if tc.checkSession != nil {
  947. tc.checkSession(t, tc.givenChatRegistry.RetrieveSess(0))
  948. }
  949. })
  950. }
  951. }
  952. func TestOSCARProxy_RecvClientCmd_ChatInvite(t *testing.T) {
  953. cases := []struct {
  954. // name is the unit test name
  955. name string
  956. // me is the TOC user session
  957. me *state.SessionInstance
  958. // givenCmd is the TOC command
  959. givenCmd []byte
  960. // givenChatRegistry is the chat registry passed to the function
  961. givenChatRegistry *ChatRegistry
  962. // wantMsg is the expected TOC response
  963. wantMsg []string
  964. // mockParams is the list of params sent to mocks that satisfy this
  965. // method's dependencies
  966. mockParams mockParams
  967. }{
  968. {
  969. name: "successfully send chat invitation",
  970. me: newTestSession("me"),
  971. givenCmd: []byte(`toc_chat_invite 0 "join my chat! :\)" friend1`),
  972. givenChatRegistry: func() *ChatRegistry {
  973. reg := NewChatRegistry()
  974. reg.Add(wire.ICBMRoomInfo{
  975. Exchange: 4,
  976. Cookie: "the-cookie",
  977. Instance: 0,
  978. })
  979. return reg
  980. }(),
  981. mockParams: mockParams{
  982. icbmParams: icbmParams{
  983. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  984. {
  985. sender: state.NewIdentScreenName("me"),
  986. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  987. ChannelID: wire.ICBMChannelRendezvous,
  988. ScreenName: "friend1",
  989. TLVRestBlock: wire.TLVRestBlock{
  990. TLVList: wire.TLVList{
  991. wire.NewTLVBE(0x05, wire.ICBMCh2Fragment{
  992. Type: 0,
  993. Capability: wire.CapChat,
  994. TLVRestBlock: wire.TLVRestBlock{
  995. TLVList: wire.TLVList{
  996. wire.NewTLVBE(10, uint16(1)),
  997. wire.NewTLVBE(12, "join my chat! :)"),
  998. wire.NewTLVBE(13, "us-ascii"),
  999. wire.NewTLVBE(14, "en"),
  1000. wire.NewTLVBE(10001, wire.ICBMRoomInfo{
  1001. Exchange: 4,
  1002. Cookie: "the-cookie",
  1003. Instance: 0,
  1004. }),
  1005. },
  1006. },
  1007. }),
  1008. },
  1009. },
  1010. },
  1011. },
  1012. },
  1013. },
  1014. },
  1015. wantMsg: []string{},
  1016. },
  1017. {
  1018. name: "send chat invitation, receive error from ICBM svc",
  1019. me: newTestSession("me"),
  1020. givenCmd: []byte(`toc_chat_invite 0 "join my chat!" friend1`),
  1021. givenChatRegistry: func() *ChatRegistry {
  1022. reg := NewChatRegistry()
  1023. reg.Add(wire.ICBMRoomInfo{
  1024. Exchange: 4,
  1025. Cookie: "the-cookie",
  1026. Instance: 0,
  1027. })
  1028. return reg
  1029. }(),
  1030. mockParams: mockParams{
  1031. icbmParams: icbmParams{
  1032. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  1033. {
  1034. sender: state.NewIdentScreenName("me"),
  1035. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  1036. ChannelID: wire.ICBMChannelRendezvous,
  1037. ScreenName: "friend1",
  1038. TLVRestBlock: wire.TLVRestBlock{
  1039. TLVList: wire.TLVList{
  1040. wire.NewTLVBE(0x05, wire.ICBMCh2Fragment{
  1041. Type: 0,
  1042. Capability: wire.CapChat,
  1043. TLVRestBlock: wire.TLVRestBlock{
  1044. TLVList: wire.TLVList{
  1045. wire.NewTLVBE(10, uint16(1)),
  1046. wire.NewTLVBE(12, "join my chat!"),
  1047. wire.NewTLVBE(13, "us-ascii"),
  1048. wire.NewTLVBE(14, "en"),
  1049. wire.NewTLVBE(10001, wire.ICBMRoomInfo{
  1050. Exchange: 4,
  1051. Cookie: "the-cookie",
  1052. Instance: 0,
  1053. }),
  1054. },
  1055. },
  1056. }),
  1057. },
  1058. },
  1059. },
  1060. err: io.EOF,
  1061. },
  1062. },
  1063. },
  1064. },
  1065. wantMsg: []string{cmdInternalSvcErr},
  1066. },
  1067. {
  1068. name: "send chat invitation to non-existent room",
  1069. me: newTestSession("me"),
  1070. givenCmd: []byte(`toc_chat_invite 0 "join my chat!" friend1`),
  1071. givenChatRegistry: NewChatRegistry(),
  1072. wantMsg: []string{cmdInternalSvcErr},
  1073. },
  1074. {
  1075. name: "bad chat room ID",
  1076. givenCmd: []byte(`toc_chat_invite zero "join my chat!" friend1`),
  1077. wantMsg: []string{cmdInternalSvcErr},
  1078. },
  1079. {
  1080. name: "bad command",
  1081. me: newTestSession("me"),
  1082. givenCmd: []byte(`toc_chat_invite`),
  1083. wantMsg: []string{cmdInternalSvcErr},
  1084. },
  1085. }
  1086. for _, tc := range cases {
  1087. t.Run(tc.name, func(t *testing.T) {
  1088. ctx := context.Background()
  1089. icbmSvc := newMockICBMService(t)
  1090. for _, params := range tc.mockParams.channelMsgToHostParamsICBM {
  1091. icbmSvc.EXPECT().
  1092. ChannelMsgToHost(ctx, matchSession(params.sender), wire.SNACFrame{}, params.inBody).
  1093. Return(nil, params.err)
  1094. }
  1095. svc := OSCARProxy{
  1096. Logger: slog.Default(),
  1097. ICBMService: icbmSvc,
  1098. }
  1099. msg := svc.RecvClientCmd(ctx, tc.me, tc.givenChatRegistry, tc.givenCmd, nil, nil)
  1100. assert.Equal(t, tc.wantMsg, msg)
  1101. })
  1102. }
  1103. }
  1104. func TestOSCARProxy_RecvClientCmd_ChatJoin(t *testing.T) {
  1105. roomInfo := wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
  1106. Exchange: 4,
  1107. Cookie: "create",
  1108. TLVBlock: wire.TLVBlock{
  1109. TLVList: wire.TLVList{
  1110. wire.NewTLVBE(wire.ChatRoomTLVRoomName, "cool room :)"),
  1111. },
  1112. },
  1113. }
  1114. navInfo := wire.SNACMessage{
  1115. Body: wire.SNAC_0x0D_0x09_ChatNavNavInfo{
  1116. TLVRestBlock: wire.TLVRestBlock{
  1117. TLVList: wire.TLVList{
  1118. wire.NewTLVBE(wire.ChatNavTLVRoomInfo, wire.SNAC_0x0E_0x02_ChatRoomInfoUpdate{
  1119. Cookie: "the-cookie",
  1120. }),
  1121. },
  1122. },
  1123. },
  1124. }
  1125. svcReq := wire.SNAC_0x01_0x04_OServiceServiceRequest{
  1126. FoodGroup: wire.Chat,
  1127. TLVRestBlock: wire.TLVRestBlock{
  1128. TLVList: wire.TLVList{
  1129. wire.NewTLVBE(0x01, wire.SNAC_0x01_0x04_TLVRoomInfo{
  1130. Cookie: "the-cookie",
  1131. }),
  1132. },
  1133. },
  1134. }
  1135. svcResp := wire.SNACMessage{
  1136. Body: wire.SNAC_0x01_0x05_OServiceServiceResponse{
  1137. TLVRestBlock: wire.TLVRestBlock{
  1138. TLVList: wire.TLVList{
  1139. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, "chat-auth-cookie"),
  1140. },
  1141. },
  1142. },
  1143. }
  1144. cases := []struct {
  1145. // name is the unit test name
  1146. name string
  1147. // me is the TOC user session
  1148. me *state.SessionInstance
  1149. // givenCmd is the TOC command
  1150. givenCmd []byte
  1151. // givenChatRegistry is the chat registry passed to the function
  1152. givenChatRegistry *ChatRegistry
  1153. // wantMsg is the expected TOC response
  1154. wantMsg []string
  1155. // expectChatSession indicates whether a chat session should be present
  1156. // in the chat registry
  1157. expectChatSession bool
  1158. // checkSession validates the state of the registered chat session (chat ID 0)
  1159. checkSession func(*testing.T, *state.SessionInstance)
  1160. // mockParams is the list of params sent to mocks that satisfy this
  1161. // method's dependencies
  1162. mockParams mockParams
  1163. }{
  1164. {
  1165. name: "successfully join chat",
  1166. me: newTestSession("me"),
  1167. givenCmd: []byte(`toc_chat_join 4 "cool room :\)"`),
  1168. givenChatRegistry: NewChatRegistry(),
  1169. mockParams: mockParams{
  1170. chatNavParams: chatNavParams{
  1171. createRoomParams: createRoomParams{
  1172. {
  1173. me: state.NewIdentScreenName("me"),
  1174. inBody: roomInfo,
  1175. msg: navInfo,
  1176. },
  1177. },
  1178. },
  1179. oServiceParams: oServiceParams{
  1180. serviceRequestParams: serviceRequestParams{
  1181. {
  1182. me: state.NewIdentScreenName("me"),
  1183. bodyIn: svcReq,
  1184. msg: svcResp,
  1185. },
  1186. },
  1187. clientOnlineParams: clientOnlineParams{
  1188. {
  1189. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  1190. me: state.NewIdentScreenName("me"),
  1191. },
  1192. },
  1193. },
  1194. authParams: authParams{
  1195. crackCookieParams: crackCookieParams{
  1196. {
  1197. cookieIn: []byte("chat-auth-cookie"),
  1198. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1199. },
  1200. },
  1201. registerChatSessionParams: registerChatSessionParams{
  1202. {
  1203. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1204. instance: newTestSession("me"),
  1205. },
  1206. },
  1207. },
  1208. },
  1209. wantMsg: []string{"CHAT_JOIN:0:cool room :)"},
  1210. expectChatSession: true,
  1211. },
  1212. {
  1213. name: "successfully join chat - TOC2 with encoded messaging propagates to chat session",
  1214. me: newTestSession("me", func(i *state.SessionInstance) { i.SetTOC2(true) }),
  1215. givenCmd: []byte(`toc_chat_join 4 "cool room :\)"`),
  1216. givenChatRegistry: NewChatRegistry(),
  1217. mockParams: mockParams{
  1218. chatNavParams: chatNavParams{
  1219. createRoomParams: createRoomParams{
  1220. {
  1221. me: state.NewIdentScreenName("me"),
  1222. inBody: roomInfo,
  1223. msg: navInfo,
  1224. },
  1225. },
  1226. },
  1227. oServiceParams: oServiceParams{
  1228. serviceRequestParams: serviceRequestParams{
  1229. {
  1230. me: state.NewIdentScreenName("me"),
  1231. bodyIn: svcReq,
  1232. msg: svcResp,
  1233. },
  1234. },
  1235. clientOnlineParams: clientOnlineParams{
  1236. {
  1237. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  1238. me: state.NewIdentScreenName("me"),
  1239. },
  1240. },
  1241. },
  1242. authParams: authParams{
  1243. crackCookieParams: crackCookieParams{
  1244. {
  1245. cookieIn: []byte("chat-auth-cookie"),
  1246. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1247. },
  1248. },
  1249. registerChatSessionParams: registerChatSessionParams{
  1250. {
  1251. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1252. instance: newTestSession("me"),
  1253. },
  1254. },
  1255. },
  1256. },
  1257. wantMsg: []string{"CHAT_JOIN:0:cool room :)"},
  1258. expectChatSession: true,
  1259. checkSession: func(t *testing.T, sess *state.SessionInstance) {
  1260. assert.NotNil(t, sess, "chat session should be registered")
  1261. assert.True(t, sess.IsTOC2(), "chat session should have TOC2 set")
  1262. assert.True(t, sess.SupportsTOC2MsgEnc(), "chat session should have TOC2 msg enc set")
  1263. },
  1264. },
  1265. {
  1266. name: "accept chat, receive error from client online",
  1267. me: newTestSession("me"),
  1268. givenCmd: []byte(`toc_chat_join 4 "cool room :\)"`),
  1269. givenChatRegistry: NewChatRegistry(),
  1270. mockParams: mockParams{
  1271. chatNavParams: chatNavParams{
  1272. createRoomParams: createRoomParams{
  1273. {
  1274. me: state.NewIdentScreenName("me"),
  1275. inBody: roomInfo,
  1276. msg: navInfo,
  1277. },
  1278. },
  1279. },
  1280. oServiceParams: oServiceParams{
  1281. serviceRequestParams: serviceRequestParams{
  1282. {
  1283. me: state.NewIdentScreenName("me"),
  1284. bodyIn: svcReq,
  1285. msg: svcResp,
  1286. },
  1287. },
  1288. clientOnlineParams: clientOnlineParams{
  1289. {
  1290. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  1291. me: state.NewIdentScreenName("me"),
  1292. err: io.EOF,
  1293. },
  1294. },
  1295. },
  1296. authParams: authParams{
  1297. crackCookieParams: crackCookieParams{
  1298. {
  1299. cookieIn: []byte("chat-auth-cookie"),
  1300. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1301. },
  1302. },
  1303. registerChatSessionParams: registerChatSessionParams{
  1304. {
  1305. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1306. instance: newTestSession("me"),
  1307. },
  1308. },
  1309. },
  1310. },
  1311. wantMsg: []string{cmdInternalSvcErr},
  1312. expectChatSession: false,
  1313. },
  1314. {
  1315. name: "join chat, receive error from register chat session",
  1316. me: newTestSession("me"),
  1317. givenCmd: []byte(`toc_chat_join 4 "cool room :\)"`),
  1318. givenChatRegistry: NewChatRegistry(),
  1319. mockParams: mockParams{
  1320. chatNavParams: chatNavParams{
  1321. createRoomParams: createRoomParams{
  1322. {
  1323. me: state.NewIdentScreenName("me"),
  1324. inBody: roomInfo,
  1325. msg: navInfo,
  1326. },
  1327. },
  1328. },
  1329. oServiceParams: oServiceParams{
  1330. serviceRequestParams: serviceRequestParams{
  1331. {
  1332. me: state.NewIdentScreenName("me"),
  1333. bodyIn: svcReq,
  1334. msg: svcResp,
  1335. },
  1336. },
  1337. },
  1338. authParams: authParams{
  1339. crackCookieParams: crackCookieParams{
  1340. {
  1341. cookieIn: []byte("chat-auth-cookie"),
  1342. cookieOut: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1343. },
  1344. },
  1345. registerChatSessionParams: registerChatSessionParams{
  1346. {
  1347. authCookie: state.ServerCookie{ChatCookie: "chat-auth-cookie"},
  1348. instance: newTestSession("me"),
  1349. err: io.EOF,
  1350. },
  1351. },
  1352. },
  1353. },
  1354. wantMsg: []string{cmdInternalSvcErr},
  1355. expectChatSession: false,
  1356. },
  1357. {
  1358. name: "join chat, receive error from service request",
  1359. me: newTestSession("me"),
  1360. givenCmd: []byte(`toc_chat_join 4 "cool room :\)"`),
  1361. givenChatRegistry: NewChatRegistry(),
  1362. mockParams: mockParams{
  1363. chatNavParams: chatNavParams{
  1364. createRoomParams: createRoomParams{
  1365. {
  1366. me: state.NewIdentScreenName("me"),
  1367. inBody: roomInfo,
  1368. msg: navInfo,
  1369. },
  1370. },
  1371. },
  1372. oServiceParams: oServiceParams{
  1373. serviceRequestParams: serviceRequestParams{
  1374. {
  1375. me: state.NewIdentScreenName("me"),
  1376. bodyIn: svcReq,
  1377. err: io.EOF,
  1378. },
  1379. },
  1380. },
  1381. },
  1382. wantMsg: []string{cmdInternalSvcErr},
  1383. expectChatSession: false,
  1384. },
  1385. {
  1386. name: "join chat, receive error from chat nav svc",
  1387. me: newTestSession("me"),
  1388. givenCmd: []byte(`toc_chat_join 4 "cool room :\)"`),
  1389. givenChatRegistry: NewChatRegistry(),
  1390. mockParams: mockParams{
  1391. chatNavParams: chatNavParams{
  1392. createRoomParams: createRoomParams{
  1393. {
  1394. me: state.NewIdentScreenName("me"),
  1395. inBody: roomInfo,
  1396. err: io.EOF,
  1397. },
  1398. },
  1399. },
  1400. },
  1401. wantMsg: []string{cmdInternalSvcErr},
  1402. expectChatSession: false,
  1403. },
  1404. {
  1405. name: "bad command",
  1406. me: newTestSession("me"),
  1407. givenCmd: []byte(`toc_chat_join`),
  1408. givenChatRegistry: NewChatRegistry(),
  1409. wantMsg: []string{cmdInternalSvcErr},
  1410. expectChatSession: false,
  1411. },
  1412. {
  1413. name: "bad exchange number",
  1414. me: newTestSession("me"),
  1415. givenCmd: []byte(`toc_chat_join four "cool room :\)"`),
  1416. givenChatRegistry: NewChatRegistry(),
  1417. wantMsg: []string{cmdInternalSvcErr},
  1418. expectChatSession: false,
  1419. },
  1420. }
  1421. for _, tc := range cases {
  1422. t.Run(tc.name, func(t *testing.T) {
  1423. ctx, cancel := context.WithCancel(context.Background())
  1424. cancel()
  1425. chatNavSvc := newMockChatNavService(t)
  1426. for _, params := range tc.mockParams.createRoomParams {
  1427. chatNavSvc.EXPECT().
  1428. CreateRoom(ctx, matchSession(params.me), wire.SNACFrame{}, params.inBody).
  1429. Return(params.msg, params.err)
  1430. }
  1431. bosOServiceSvc := newMockOServiceService(t)
  1432. for _, params := range tc.mockParams.oServiceParams.serviceRequestParams {
  1433. bosOServiceSvc.EXPECT().
  1434. ServiceRequest(ctx, wire.BOS, matchSession(params.me), wire.SNACFrame{}, params.bodyIn, config.Listener{}).
  1435. Return(params.msg, params.err)
  1436. }
  1437. for _, params := range tc.mockParams.oServiceParams.clientOnlineParams {
  1438. bosOServiceSvc.EXPECT().
  1439. ClientOnline(ctx, wire.Chat, params.body, matchSession(params.me)).
  1440. Return(params.err)
  1441. }
  1442. authSvc := newMockAuthService(t)
  1443. for _, params := range tc.mockParams.authParams.registerChatSessionParams {
  1444. authSvc.EXPECT().
  1445. RegisterChatSession(ctx, params.authCookie).
  1446. Return(params.instance, params.err)
  1447. }
  1448. for _, params := range tc.mockParams.authParams.crackCookieParams {
  1449. authSvc.EXPECT().
  1450. CrackCookie(params.cookieIn).
  1451. Return(params.cookieOut, params.err)
  1452. }
  1453. svc := OSCARProxy{
  1454. AuthService: authSvc,
  1455. ChatNavService: chatNavSvc,
  1456. Logger: slog.Default(),
  1457. OServiceService: bosOServiceSvc,
  1458. }
  1459. g := &errgroup.Group{}
  1460. tc.me.CloseInstance()
  1461. msg := svc.RecvClientCmd(ctx, tc.me, tc.givenChatRegistry, tc.givenCmd, nil, g.Go)
  1462. assert.NoError(t, g.Wait())
  1463. assert.Equal(t, tc.wantMsg, msg)
  1464. assert.Equal(t, tc.expectChatSession, len(tc.givenChatRegistry.Sessions()) == 1)
  1465. if tc.checkSession != nil {
  1466. tc.checkSession(t, tc.givenChatRegistry.RetrieveSess(0))
  1467. }
  1468. })
  1469. }
  1470. }
  1471. func TestOSCARProxy_RecvClientCmd_ChatLeave(t *testing.T) {
  1472. cases := []struct {
  1473. // name is the unit test name
  1474. name string
  1475. // me is the TOC user session
  1476. me *state.SessionInstance
  1477. // givenCmd is the TOC command
  1478. givenCmd []byte
  1479. // givenChatRegistry is the chat registry passed to the function
  1480. givenChatRegistry *ChatRegistry
  1481. // wantMsg is the expected TOC response
  1482. wantMsg []string
  1483. // mockParams is the list of params sent to mocks that satisfy this
  1484. // method's dependencies
  1485. mockParams mockParams
  1486. }{
  1487. {
  1488. name: "successfully leave chat",
  1489. me: newTestSession("me"),
  1490. givenCmd: []byte(`toc_chat_leave 0`),
  1491. givenChatRegistry: func() *ChatRegistry {
  1492. reg := NewChatRegistry()
  1493. reg.RegisterSess(0, newTestSession("me"))
  1494. return reg
  1495. }(),
  1496. mockParams: mockParams{
  1497. authParams: authParams{
  1498. signoutChatParams: signoutChatParams{
  1499. {
  1500. me: state.NewIdentScreenName("me"),
  1501. },
  1502. },
  1503. },
  1504. },
  1505. wantMsg: []string{"CHAT_LEFT:0"},
  1506. },
  1507. {
  1508. name: "chat room ID with invalid format",
  1509. givenCmd: []byte(`toc_chat_leave zero`),
  1510. wantMsg: []string{cmdInternalSvcErr},
  1511. },
  1512. {
  1513. name: "missing chat session",
  1514. givenCmd: []byte(`toc_chat_leave 0`),
  1515. givenChatRegistry: NewChatRegistry(),
  1516. wantMsg: []string{cmdInternalSvcErr},
  1517. },
  1518. {
  1519. name: "bad command",
  1520. me: newTestSession("me"),
  1521. givenCmd: []byte(`toc_chat_leave`),
  1522. wantMsg: []string{cmdInternalSvcErr},
  1523. },
  1524. }
  1525. for _, tc := range cases {
  1526. t.Run(tc.name, func(t *testing.T) {
  1527. ctx := context.Background()
  1528. authSvc := newMockAuthService(t)
  1529. for _, params := range tc.mockParams.signoutChatParams {
  1530. authSvc.EXPECT().SignoutChat(ctx, matchSession(params.me))
  1531. }
  1532. svc := OSCARProxy{
  1533. Logger: slog.Default(),
  1534. AuthService: authSvc,
  1535. }
  1536. msg := svc.RecvClientCmd(ctx, nil, tc.givenChatRegistry, tc.givenCmd, nil, nil)
  1537. assert.Equal(t, tc.wantMsg, msg)
  1538. })
  1539. }
  1540. }
  1541. func TestOSCARProxy_RecvClientCmd_ChatSend(t *testing.T) {
  1542. cases := []struct {
  1543. // name is the unit test name
  1544. name string
  1545. // me is the TOC user session
  1546. me *state.SessionInstance
  1547. // givenCmd is the TOC command
  1548. givenCmd []byte
  1549. // givenChatRegistry is the chat registry passed to the function
  1550. givenChatRegistry *ChatRegistry
  1551. // wantMsg is the expected TOC response
  1552. wantMsg []string
  1553. // mockParams is the list of params sent to mocks that satisfy this
  1554. // method's dependencies
  1555. mockParams mockParams
  1556. }{
  1557. {
  1558. name: "successfully send chat message",
  1559. me: newTestSession("me"),
  1560. givenCmd: []byte(`toc_chat_send 0 "Hello world! :\)"`),
  1561. givenChatRegistry: func() *ChatRegistry {
  1562. reg := NewChatRegistry()
  1563. reg.RegisterSess(0, newTestSession("me"))
  1564. return reg
  1565. }(),
  1566. mockParams: mockParams{
  1567. chatParams: chatParams{
  1568. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1569. {
  1570. sender: state.NewIdentScreenName("me"),
  1571. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1572. Channel: wire.ICBMChannelMIME,
  1573. TLVRestBlock: wire.TLVRestBlock{
  1574. TLVList: wire.TLVList{
  1575. wire.NewTLVBE(wire.ChatTLVEnableReflectionFlag, uint8(1)),
  1576. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1577. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1578. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1579. TLVList: wire.TLVList{
  1580. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world! :)"),
  1581. },
  1582. }),
  1583. },
  1584. },
  1585. },
  1586. result: &wire.SNACMessage{
  1587. Body: wire.SNAC_0x0E_0x06_ChatChannelMsgToClient{
  1588. Channel: wire.ICBMChannelMIME,
  1589. TLVRestBlock: wire.TLVRestBlock{
  1590. TLVList: wire.TLVList{
  1591. wire.NewTLVBE(wire.ChatTLVSenderInformation,
  1592. newTestSession("me").Session().TLVUserInfo()),
  1593. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1594. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1595. TLVList: wire.TLVList{
  1596. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world! :)"),
  1597. },
  1598. }),
  1599. },
  1600. },
  1601. },
  1602. },
  1603. },
  1604. },
  1605. },
  1606. },
  1607. wantMsg: []string{"CHAT_IN:0:me:F:Hello world! :)"},
  1608. },
  1609. {
  1610. name: "successfully send chat message - TOC2 encoded returns CHAT_IN_ENC",
  1611. me: newTestSession("me"),
  1612. givenCmd: []byte(`toc_chat_send 0 "Hello world! :\)"`),
  1613. givenChatRegistry: func() *ChatRegistry {
  1614. reg := NewChatRegistry()
  1615. reg.RegisterSess(0, newTestSession("me", func(i *state.SessionInstance) { i.SetTOC2(true) }))
  1616. return reg
  1617. }(),
  1618. mockParams: mockParams{
  1619. chatParams: chatParams{
  1620. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1621. {
  1622. sender: state.NewIdentScreenName("me"),
  1623. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1624. Channel: wire.ICBMChannelMIME,
  1625. TLVRestBlock: wire.TLVRestBlock{
  1626. TLVList: wire.TLVList{
  1627. wire.NewTLVBE(wire.ChatTLVEnableReflectionFlag, uint8(1)),
  1628. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1629. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1630. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1631. TLVList: wire.TLVList{
  1632. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world! :)"),
  1633. },
  1634. }),
  1635. },
  1636. },
  1637. },
  1638. result: &wire.SNACMessage{
  1639. Body: wire.SNAC_0x0E_0x06_ChatChannelMsgToClient{
  1640. Channel: wire.ICBMChannelMIME,
  1641. TLVRestBlock: wire.TLVRestBlock{
  1642. TLVList: wire.TLVList{
  1643. wire.NewTLVBE(wire.ChatTLVSenderInformation,
  1644. newTestSession("me").Session().TLVUserInfo()),
  1645. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1646. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1647. TLVList: wire.TLVList{
  1648. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world! :)"),
  1649. },
  1650. }),
  1651. },
  1652. },
  1653. },
  1654. },
  1655. },
  1656. },
  1657. },
  1658. },
  1659. wantMsg: []string{"CHAT_IN_ENC:0:me:F:A:en:Hello world! :)"},
  1660. },
  1661. {
  1662. name: "send chat message, receive error from chat svc",
  1663. me: newTestSession("me"),
  1664. givenCmd: []byte(`toc_chat_send 0 "Hello world!"`),
  1665. givenChatRegistry: func() *ChatRegistry {
  1666. reg := NewChatRegistry()
  1667. reg.RegisterSess(0, newTestSession("me"))
  1668. return reg
  1669. }(),
  1670. mockParams: mockParams{
  1671. chatParams: chatParams{
  1672. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1673. {
  1674. sender: state.NewIdentScreenName("me"),
  1675. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1676. Channel: wire.ICBMChannelMIME,
  1677. TLVRestBlock: wire.TLVRestBlock{
  1678. TLVList: wire.TLVList{
  1679. wire.NewTLVBE(wire.ChatTLVEnableReflectionFlag, uint8(1)),
  1680. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1681. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1682. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1683. TLVList: wire.TLVList{
  1684. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world!"),
  1685. },
  1686. }),
  1687. },
  1688. },
  1689. },
  1690. err: io.EOF,
  1691. },
  1692. },
  1693. },
  1694. },
  1695. wantMsg: []string{cmdInternalSvcErr},
  1696. },
  1697. {
  1698. name: "send chat message, receive nil response from chat svc",
  1699. me: newTestSession("me"),
  1700. givenCmd: []byte(`toc_chat_send 0 "Hello world!"`),
  1701. givenChatRegistry: func() *ChatRegistry {
  1702. reg := NewChatRegistry()
  1703. reg.RegisterSess(0, newTestSession("me"))
  1704. return reg
  1705. }(),
  1706. mockParams: mockParams{
  1707. chatParams: chatParams{
  1708. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1709. {
  1710. sender: state.NewIdentScreenName("me"),
  1711. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1712. Channel: wire.ICBMChannelMIME,
  1713. TLVRestBlock: wire.TLVRestBlock{
  1714. TLVList: wire.TLVList{
  1715. wire.NewTLVBE(wire.ChatTLVEnableReflectionFlag, uint8(1)),
  1716. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1717. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1718. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1719. TLVList: wire.TLVList{
  1720. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world!"),
  1721. },
  1722. }),
  1723. },
  1724. },
  1725. },
  1726. result: nil,
  1727. },
  1728. },
  1729. },
  1730. },
  1731. wantMsg: []string{cmdInternalSvcErr},
  1732. },
  1733. {
  1734. name: "send chat message, receive unexpected response from chat svc",
  1735. me: newTestSession("me"),
  1736. givenCmd: []byte(`toc_chat_send 0 "Hello world!"`),
  1737. givenChatRegistry: func() *ChatRegistry {
  1738. reg := NewChatRegistry()
  1739. reg.RegisterSess(0, newTestSession("me"))
  1740. return reg
  1741. }(),
  1742. mockParams: mockParams{
  1743. chatParams: chatParams{
  1744. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1745. {
  1746. sender: state.NewIdentScreenName("me"),
  1747. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1748. Channel: wire.ICBMChannelMIME,
  1749. TLVRestBlock: wire.TLVRestBlock{
  1750. TLVList: wire.TLVList{
  1751. wire.NewTLVBE(wire.ChatTLVEnableReflectionFlag, uint8(1)),
  1752. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1753. wire.NewTLVBE(wire.ChatTLVPublicWhisperFlag, []byte{}),
  1754. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1755. TLVList: wire.TLVList{
  1756. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world!"),
  1757. },
  1758. }),
  1759. },
  1760. },
  1761. },
  1762. result: &wire.SNACMessage{
  1763. Body: wire.SNACError{},
  1764. },
  1765. },
  1766. },
  1767. },
  1768. },
  1769. wantMsg: []string{cmdInternalSvcErr},
  1770. },
  1771. {
  1772. name: "chat room ID with invalid format",
  1773. givenCmd: []byte(`toc_chat_send zero "Hello world!"`),
  1774. wantMsg: []string{cmdInternalSvcErr},
  1775. },
  1776. {
  1777. name: "missing chat session",
  1778. givenCmd: []byte(`toc_chat_send 0 "Hello world!"`),
  1779. givenChatRegistry: NewChatRegistry(),
  1780. wantMsg: []string{cmdInternalSvcErr},
  1781. },
  1782. {
  1783. name: "bad command",
  1784. me: newTestSession("me"),
  1785. givenCmd: []byte(`toc_chat_send`),
  1786. wantMsg: []string{cmdInternalSvcErr},
  1787. },
  1788. }
  1789. for _, tc := range cases {
  1790. t.Run(tc.name, func(t *testing.T) {
  1791. ctx := context.Background()
  1792. chatSvc := newMockChatService(t)
  1793. for _, params := range tc.mockParams.channelMsgToHostParamsChat {
  1794. chatSvc.EXPECT().
  1795. ChannelMsgToHost(ctx, matchSession(params.sender), wire.SNACFrame{}, params.inBody).
  1796. Return(params.result, params.err)
  1797. }
  1798. svc := OSCARProxy{
  1799. Logger: slog.Default(),
  1800. ChatService: chatSvc,
  1801. }
  1802. msg := svc.RecvClientCmd(ctx, nil, tc.givenChatRegistry, tc.givenCmd, nil, nil)
  1803. assert.Equal(t, tc.wantMsg, msg)
  1804. })
  1805. }
  1806. }
  1807. func TestOSCARProxy_RecvClientCmd_ChatWhisper(t *testing.T) {
  1808. cases := []struct {
  1809. // name is the unit test name
  1810. name string
  1811. // me is the TOC user session
  1812. me *state.SessionInstance
  1813. // givenCmd is the TOC command
  1814. givenCmd []byte
  1815. // givenChatRegistry is the chat registry passed to the function
  1816. givenChatRegistry *ChatRegistry
  1817. // wantMsg is the expected TOC response
  1818. wantMsg []string
  1819. // mockParams is the list of params sent to mocks that satisfy this
  1820. // method's dependencies
  1821. mockParams mockParams
  1822. }{
  1823. {
  1824. name: "successfully send chat whisper",
  1825. me: newTestSession("me"),
  1826. givenCmd: []byte(`toc_chat_whisper 0 them "Hello world! :\)"`),
  1827. givenChatRegistry: func() *ChatRegistry {
  1828. reg := NewChatRegistry()
  1829. reg.RegisterSess(0, newTestSession("me"))
  1830. return reg
  1831. }(),
  1832. mockParams: mockParams{
  1833. chatParams: chatParams{
  1834. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1835. {
  1836. sender: state.NewIdentScreenName("me"),
  1837. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1838. Channel: wire.ICBMChannelMIME,
  1839. TLVRestBlock: wire.TLVRestBlock{
  1840. TLVList: wire.TLVList{
  1841. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1842. wire.NewTLVBE(wire.ChatTLVWhisperToUser, "them"),
  1843. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1844. TLVList: wire.TLVList{
  1845. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world! :)"),
  1846. },
  1847. }),
  1848. },
  1849. },
  1850. },
  1851. result: nil,
  1852. },
  1853. },
  1854. },
  1855. },
  1856. wantMsg: []string{},
  1857. },
  1858. {
  1859. name: "send chat whisper, receive error from chat svc",
  1860. me: newTestSession("me"),
  1861. givenCmd: []byte(`toc_chat_whisper 0 them "Hello world!"`),
  1862. givenChatRegistry: func() *ChatRegistry {
  1863. reg := NewChatRegistry()
  1864. reg.RegisterSess(0, newTestSession("me"))
  1865. return reg
  1866. }(),
  1867. mockParams: mockParams{
  1868. chatParams: chatParams{
  1869. channelMsgToHostParamsChat: channelMsgToHostParamsChat{
  1870. {
  1871. sender: state.NewIdentScreenName("me"),
  1872. inBody: wire.SNAC_0x0E_0x05_ChatChannelMsgToHost{
  1873. Channel: wire.ICBMChannelMIME,
  1874. TLVRestBlock: wire.TLVRestBlock{
  1875. TLVList: wire.TLVList{
  1876. wire.NewTLVBE(wire.ChatTLVSenderInformation, newTestSession("me").Session().TLVUserInfo()),
  1877. wire.NewTLVBE(wire.ChatTLVWhisperToUser, "them"),
  1878. wire.NewTLVBE(wire.ChatTLVMessageInfo, wire.TLVRestBlock{
  1879. TLVList: wire.TLVList{
  1880. wire.NewTLVBE(wire.ChatTLVMessageInfoText, "Hello world!"),
  1881. },
  1882. }),
  1883. },
  1884. },
  1885. },
  1886. err: io.EOF,
  1887. },
  1888. },
  1889. },
  1890. },
  1891. wantMsg: []string{cmdInternalSvcErr},
  1892. },
  1893. {
  1894. name: "chat room ID with invalid format",
  1895. givenCmd: []byte(`toc_chat_whisper zero them "Hello world!"`),
  1896. wantMsg: []string{cmdInternalSvcErr},
  1897. },
  1898. {
  1899. name: "missing chat session",
  1900. givenCmd: []byte(`toc_chat_whisper 0 them "Hello world!"`),
  1901. givenChatRegistry: NewChatRegistry(),
  1902. wantMsg: []string{cmdInternalSvcErr},
  1903. },
  1904. {
  1905. name: "bad command",
  1906. me: newTestSession("me"),
  1907. givenCmd: []byte(`toc_chat_whisper`),
  1908. wantMsg: []string{cmdInternalSvcErr},
  1909. },
  1910. }
  1911. for _, tc := range cases {
  1912. t.Run(tc.name, func(t *testing.T) {
  1913. ctx := context.Background()
  1914. chatSvc := newMockChatService(t)
  1915. for _, params := range tc.mockParams.channelMsgToHostParamsChat {
  1916. chatSvc.EXPECT().
  1917. ChannelMsgToHost(ctx, matchSession(params.sender), wire.SNACFrame{}, params.inBody).
  1918. Return(params.result, params.err)
  1919. }
  1920. svc := OSCARProxy{
  1921. Logger: slog.Default(),
  1922. ChatService: chatSvc,
  1923. }
  1924. msg := svc.RecvClientCmd(ctx, nil, tc.givenChatRegistry, tc.givenCmd, nil, nil)
  1925. assert.Equal(t, tc.wantMsg, msg)
  1926. })
  1927. }
  1928. }
  1929. func TestOSCARProxy_RecvClientCmd_Evil(t *testing.T) {
  1930. cases := []struct {
  1931. // name is the unit test name
  1932. name string
  1933. // me is the TOC user session
  1934. me *state.SessionInstance
  1935. // givenCmd is the TOC command
  1936. givenCmd []byte
  1937. // wantMsg is the expected TOC response
  1938. wantMsg []string
  1939. // mockParams is the list of params sent to mocks that satisfy this
  1940. // method's dependencies
  1941. mockParams mockParams
  1942. }{
  1943. {
  1944. name: "successfully warn normally",
  1945. me: newTestSession("me"),
  1946. givenCmd: []byte(`toc_evil them norm`),
  1947. mockParams: mockParams{
  1948. icbmParams: icbmParams{
  1949. evilRequestParams: evilRequestParams{
  1950. {
  1951. me: state.NewIdentScreenName("me"),
  1952. inBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
  1953. SendAs: 0,
  1954. ScreenName: "them",
  1955. },
  1956. msg: wire.SNACMessage{
  1957. Body: wire.SNAC_0x04_0x09_ICBMEvilReply{},
  1958. },
  1959. },
  1960. },
  1961. },
  1962. },
  1963. wantMsg: []string{},
  1964. },
  1965. {
  1966. name: "successfully warn anonymously",
  1967. me: newTestSession("me"),
  1968. givenCmd: []byte(`toc_evil them anon`),
  1969. mockParams: mockParams{
  1970. icbmParams: icbmParams{
  1971. evilRequestParams: evilRequestParams{
  1972. {
  1973. me: state.NewIdentScreenName("me"),
  1974. inBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
  1975. SendAs: 1,
  1976. ScreenName: "them",
  1977. },
  1978. msg: wire.SNACMessage{
  1979. Body: wire.SNAC_0x04_0x09_ICBMEvilReply{},
  1980. },
  1981. },
  1982. },
  1983. },
  1984. },
  1985. wantMsg: []string{},
  1986. },
  1987. {
  1988. name: "warn, receive error from ICBM service",
  1989. me: newTestSession("me"),
  1990. givenCmd: []byte(`toc_evil them anon`),
  1991. mockParams: mockParams{
  1992. icbmParams: icbmParams{
  1993. evilRequestParams: evilRequestParams{
  1994. {
  1995. me: state.NewIdentScreenName("me"),
  1996. inBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
  1997. SendAs: 1,
  1998. ScreenName: "them",
  1999. },
  2000. err: io.EOF,
  2001. },
  2002. },
  2003. },
  2004. },
  2005. wantMsg: []string{cmdInternalSvcErr},
  2006. },
  2007. {
  2008. name: "warn, receive snac err",
  2009. me: newTestSession("me"),
  2010. givenCmd: []byte(`toc_evil them anon`),
  2011. mockParams: mockParams{
  2012. icbmParams: icbmParams{
  2013. evilRequestParams: evilRequestParams{
  2014. {
  2015. me: state.NewIdentScreenName("me"),
  2016. inBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
  2017. SendAs: 1,
  2018. ScreenName: "them",
  2019. },
  2020. msg: wire.SNACMessage{
  2021. Body: wire.SNACError{},
  2022. },
  2023. },
  2024. },
  2025. },
  2026. },
  2027. wantMsg: []string{"ERROR:902"},
  2028. },
  2029. {
  2030. name: "warn, ICBM svc returns unexpected snac type",
  2031. me: newTestSession("me"),
  2032. givenCmd: []byte(`toc_evil them anon`),
  2033. mockParams: mockParams{
  2034. icbmParams: icbmParams{
  2035. evilRequestParams: evilRequestParams{
  2036. {
  2037. me: state.NewIdentScreenName("me"),
  2038. inBody: wire.SNAC_0x04_0x08_ICBMEvilRequest{
  2039. SendAs: 1,
  2040. ScreenName: "them",
  2041. },
  2042. msg: wire.SNACMessage{
  2043. Body: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{},
  2044. },
  2045. },
  2046. },
  2047. },
  2048. },
  2049. wantMsg: []string{cmdInternalSvcErr},
  2050. },
  2051. {
  2052. name: "warn with incorrect type",
  2053. me: newTestSession("me"),
  2054. givenCmd: []byte(`toc_evil them blah`),
  2055. wantMsg: []string{cmdInternalSvcErr},
  2056. },
  2057. {
  2058. name: "bad command",
  2059. me: newTestSession("me"),
  2060. givenCmd: []byte(`toc_evil`),
  2061. wantMsg: []string{cmdInternalSvcErr},
  2062. },
  2063. }
  2064. for _, tc := range cases {
  2065. t.Run(tc.name, func(t *testing.T) {
  2066. ctx := context.Background()
  2067. icbmSvc := newMockICBMService(t)
  2068. for _, params := range tc.mockParams.evilRequestParams {
  2069. icbmSvc.EXPECT().
  2070. EvilRequest(ctx, matchSession(params.me), wire.SNACFrame{}, params.inBody).
  2071. Return(params.msg, params.err)
  2072. }
  2073. svc := OSCARProxy{
  2074. Logger: slog.Default(),
  2075. ICBMService: icbmSvc,
  2076. }
  2077. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2078. assert.Equal(t, tc.wantMsg, msg)
  2079. })
  2080. }
  2081. }
  2082. func TestOSCARProxy_RecvClientCmd_ClientEvent(t *testing.T) {
  2083. cases := []struct {
  2084. name string
  2085. me *state.SessionInstance
  2086. givenCmd []byte
  2087. wantMsg []string
  2088. mockParams mockParams
  2089. }{
  2090. {
  2091. name: "send typing status 0 (no activity)",
  2092. me: newTestSession("me"),
  2093. givenCmd: []byte("toc2_client_event friend 0"),
  2094. mockParams: mockParams{
  2095. icbmParams: icbmParams{
  2096. clientEventParams: clientEventParams{
  2097. {
  2098. sender: state.NewIdentScreenName("me"),
  2099. inBody: wire.SNAC_0x04_0x14_ICBMClientEvent{
  2100. Cookie: 0,
  2101. ChannelID: wire.ICBMChannelIM,
  2102. ScreenName: "friend",
  2103. Event: 0,
  2104. },
  2105. },
  2106. },
  2107. },
  2108. },
  2109. wantMsg: []string{},
  2110. },
  2111. {
  2112. name: "send typing status 1 (typing paused)",
  2113. me: newTestSession("me"),
  2114. givenCmd: []byte("toc2_client_event buddy 1"),
  2115. mockParams: mockParams{
  2116. icbmParams: icbmParams{
  2117. clientEventParams: clientEventParams{
  2118. {
  2119. sender: state.NewIdentScreenName("me"),
  2120. inBody: wire.SNAC_0x04_0x14_ICBMClientEvent{
  2121. Cookie: 0,
  2122. ChannelID: wire.ICBMChannelIM,
  2123. ScreenName: "buddy",
  2124. Event: 1,
  2125. },
  2126. },
  2127. },
  2128. },
  2129. },
  2130. wantMsg: []string{},
  2131. },
  2132. {
  2133. name: "send typing status 2 (currently typing)",
  2134. me: newTestSession("me"),
  2135. givenCmd: []byte("toc2_client_event chatter 2"),
  2136. mockParams: mockParams{
  2137. icbmParams: icbmParams{
  2138. clientEventParams: clientEventParams{
  2139. {
  2140. sender: state.NewIdentScreenName("me"),
  2141. inBody: wire.SNAC_0x04_0x14_ICBMClientEvent{
  2142. Cookie: 0,
  2143. ChannelID: wire.ICBMChannelIM,
  2144. ScreenName: "chatter",
  2145. Event: 2,
  2146. },
  2147. },
  2148. },
  2149. },
  2150. },
  2151. wantMsg: []string{},
  2152. },
  2153. {
  2154. name: "ICBM ClientEvent returns error",
  2155. me: newTestSession("me"),
  2156. givenCmd: []byte("toc2_client_event friend 2"),
  2157. mockParams: mockParams{
  2158. icbmParams: icbmParams{
  2159. clientEventParams: clientEventParams{
  2160. {
  2161. sender: state.NewIdentScreenName("me"),
  2162. inBody: wire.SNAC_0x04_0x14_ICBMClientEvent{
  2163. Cookie: 0,
  2164. ChannelID: wire.ICBMChannelIM,
  2165. ScreenName: "friend",
  2166. Event: 2,
  2167. },
  2168. err: io.EOF,
  2169. },
  2170. },
  2171. },
  2172. },
  2173. wantMsg: []string{cmdInternalSvcErr},
  2174. },
  2175. {
  2176. name: "missing args",
  2177. me: newTestSession("me"),
  2178. givenCmd: []byte("toc2_client_event"),
  2179. wantMsg: []string{cmdInternalSvcErr},
  2180. },
  2181. {
  2182. name: "missing typing status",
  2183. me: newTestSession("me"),
  2184. givenCmd: []byte("toc2_client_event friend"),
  2185. wantMsg: []string{cmdInternalSvcErr},
  2186. },
  2187. {
  2188. name: "invalid typing status (must be 0, 1, or 2)",
  2189. me: newTestSession("me"),
  2190. givenCmd: []byte("toc2_client_event friend 3"),
  2191. wantMsg: []string{cmdInternalSvcErr},
  2192. },
  2193. {
  2194. name: "invalid typing status non-numeric",
  2195. me: newTestSession("me"),
  2196. givenCmd: []byte("toc2_client_event friend typing"),
  2197. wantMsg: []string{cmdInternalSvcErr},
  2198. },
  2199. }
  2200. for _, tc := range cases {
  2201. t.Run(tc.name, func(t *testing.T) {
  2202. ctx := context.Background()
  2203. icbmSvc := newMockICBMService(t)
  2204. for _, params := range tc.mockParams.clientEventParams {
  2205. icbmSvc.EXPECT().
  2206. ClientEvent(ctx, matchSession(params.sender), wire.SNACFrame{}, params.inBody).
  2207. Return(params.err)
  2208. }
  2209. svc := OSCARProxy{
  2210. Logger: slog.Default(),
  2211. ICBMService: icbmSvc,
  2212. }
  2213. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2214. assert.Equal(t, tc.wantMsg, msg)
  2215. })
  2216. }
  2217. }
  2218. func TestOSCARProxy_RecvClientCmd_ChangePassword(t *testing.T) {
  2219. cases := []struct {
  2220. // name is the unit test name
  2221. name string
  2222. // me is the TOC user session
  2223. me *state.SessionInstance
  2224. // givenCmd is the TOC command
  2225. givenCmd []byte
  2226. // wantMsg is the expected TOC response
  2227. wantMsg []string
  2228. // mockParams is the list of params sent to mocks that satisfy this
  2229. // method's dependencies
  2230. mockParams mockParams
  2231. }{
  2232. {
  2233. name: "successfully change password",
  2234. me: newTestSession("me"),
  2235. givenCmd: []byte("toc_change_passwd oldpa\\$\\$ newpa\\$\\$"),
  2236. mockParams: mockParams{
  2237. adminParams: adminParams{
  2238. infoChangeRequestParams: infoChangeRequestParams{
  2239. {
  2240. me: state.NewIdentScreenName("me"),
  2241. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  2242. TLVRestBlock: wire.TLVRestBlock{
  2243. TLVList: wire.TLVList{
  2244. wire.NewTLVBE(wire.AdminTLVOldPassword, "oldpa$$"),
  2245. wire.NewTLVBE(wire.AdminTLVNewPassword, "newpa$$"),
  2246. },
  2247. },
  2248. },
  2249. msg: wire.SNACMessage{
  2250. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  2251. Permissions: wire.AdminInfoPermissionsReadWrite,
  2252. TLVBlock: wire.TLVBlock{
  2253. TLVList: wire.TLVList{
  2254. wire.NewTLVBE(wire.AdminTLVNewPassword, []byte{}),
  2255. },
  2256. },
  2257. },
  2258. },
  2259. },
  2260. },
  2261. },
  2262. },
  2263. wantMsg: []string{"ADMIN_PASSWD_STATUS:0"},
  2264. },
  2265. {
  2266. name: "change password - invalid password length",
  2267. me: newTestSession("me"),
  2268. givenCmd: []byte("toc_change_passwd oldpass np"),
  2269. mockParams: mockParams{
  2270. adminParams: adminParams{
  2271. infoChangeRequestParams: infoChangeRequestParams{
  2272. {
  2273. me: state.NewIdentScreenName("me"),
  2274. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  2275. TLVRestBlock: wire.TLVRestBlock{
  2276. TLVList: wire.TLVList{
  2277. wire.NewTLVBE(wire.AdminTLVOldPassword, "oldpass"),
  2278. wire.NewTLVBE(wire.AdminTLVNewPassword, "np"),
  2279. },
  2280. },
  2281. },
  2282. msg: wire.SNACMessage{
  2283. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  2284. Permissions: wire.AdminInfoPermissionsReadWrite,
  2285. TLVBlock: wire.TLVBlock{
  2286. TLVList: wire.TLVList{
  2287. wire.NewTLVBE(wire.AdminTLVNewPassword, []byte{}),
  2288. wire.NewTLVBE(wire.AdminTLVErrorCode, wire.AdminInfoErrorInvalidPasswordLength),
  2289. },
  2290. },
  2291. },
  2292. },
  2293. },
  2294. },
  2295. },
  2296. },
  2297. wantMsg: []string{"ERROR:911"},
  2298. },
  2299. {
  2300. name: "change password - incorrect password",
  2301. me: newTestSession("me"),
  2302. givenCmd: []byte("toc_change_passwd oldpass baddpass"),
  2303. mockParams: mockParams{
  2304. adminParams: adminParams{
  2305. infoChangeRequestParams: infoChangeRequestParams{
  2306. {
  2307. me: state.NewIdentScreenName("me"),
  2308. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  2309. TLVRestBlock: wire.TLVRestBlock{
  2310. TLVList: wire.TLVList{
  2311. wire.NewTLVBE(wire.AdminTLVOldPassword, "oldpass"),
  2312. wire.NewTLVBE(wire.AdminTLVNewPassword, "baddpass"),
  2313. },
  2314. },
  2315. },
  2316. msg: wire.SNACMessage{
  2317. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  2318. Permissions: wire.AdminInfoPermissionsReadWrite,
  2319. TLVBlock: wire.TLVBlock{
  2320. TLVList: wire.TLVList{
  2321. wire.NewTLVBE(wire.AdminTLVNewPassword, []byte{}),
  2322. wire.NewTLVBE(wire.AdminTLVErrorCode, wire.AdminInfoErrorValidatePassword),
  2323. },
  2324. },
  2325. },
  2326. },
  2327. },
  2328. },
  2329. },
  2330. },
  2331. wantMsg: []string{"ERROR:980"},
  2332. },
  2333. {
  2334. name: "change password - catch-all error response",
  2335. me: newTestSession("me"),
  2336. givenCmd: []byte("toc_change_passwd oldpass baddpass"),
  2337. mockParams: mockParams{
  2338. adminParams: adminParams{
  2339. infoChangeRequestParams: infoChangeRequestParams{
  2340. {
  2341. me: state.NewIdentScreenName("me"),
  2342. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  2343. TLVRestBlock: wire.TLVRestBlock{
  2344. TLVList: wire.TLVList{
  2345. wire.NewTLVBE(wire.AdminTLVOldPassword, "oldpass"),
  2346. wire.NewTLVBE(wire.AdminTLVNewPassword, "baddpass"),
  2347. },
  2348. },
  2349. },
  2350. msg: wire.SNACMessage{
  2351. Body: wire.SNAC_0x07_0x05_AdminChangeReply{
  2352. Permissions: wire.AdminInfoPermissionsReadWrite,
  2353. TLVBlock: wire.TLVBlock{
  2354. TLVList: wire.TLVList{
  2355. wire.NewTLVBE(wire.AdminTLVNewPassword, []byte{}),
  2356. wire.NewTLVBE(wire.AdminTLVErrorCode, wire.AdminInfoErrorAllOtherErrors),
  2357. },
  2358. },
  2359. },
  2360. },
  2361. },
  2362. },
  2363. },
  2364. },
  2365. wantMsg: []string{"ERROR:913"},
  2366. },
  2367. {
  2368. name: "change password - runtime error from admin svc",
  2369. me: newTestSession("me"),
  2370. givenCmd: []byte("toc_change_passwd oldpass baddpass"),
  2371. mockParams: mockParams{
  2372. adminParams: adminParams{
  2373. infoChangeRequestParams: infoChangeRequestParams{
  2374. {
  2375. me: state.NewIdentScreenName("me"),
  2376. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  2377. TLVRestBlock: wire.TLVRestBlock{
  2378. TLVList: wire.TLVList{
  2379. wire.NewTLVBE(wire.AdminTLVOldPassword, "oldpass"),
  2380. wire.NewTLVBE(wire.AdminTLVNewPassword, "baddpass"),
  2381. },
  2382. },
  2383. },
  2384. err: io.EOF,
  2385. },
  2386. },
  2387. },
  2388. },
  2389. wantMsg: []string{cmdInternalSvcErr},
  2390. },
  2391. {
  2392. name: "change password - unexpected response from admin svc",
  2393. me: newTestSession("me"),
  2394. givenCmd: []byte("toc_change_passwd oldpass baddpass"),
  2395. mockParams: mockParams{
  2396. adminParams: adminParams{
  2397. infoChangeRequestParams: infoChangeRequestParams{
  2398. {
  2399. me: state.NewIdentScreenName("me"),
  2400. inBody: wire.SNAC_0x07_0x04_AdminInfoChangeRequest{
  2401. TLVRestBlock: wire.TLVRestBlock{
  2402. TLVList: wire.TLVList{
  2403. wire.NewTLVBE(wire.AdminTLVOldPassword, "oldpass"),
  2404. wire.NewTLVBE(wire.AdminTLVNewPassword, "baddpass"),
  2405. },
  2406. },
  2407. },
  2408. msg: wire.SNACMessage{
  2409. Body: wire.SNACError{},
  2410. },
  2411. },
  2412. },
  2413. },
  2414. },
  2415. wantMsg: []string{cmdInternalSvcErr},
  2416. },
  2417. {
  2418. name: "bad command",
  2419. me: newTestSession("me"),
  2420. givenCmd: []byte(`toc_change_passwd`),
  2421. wantMsg: []string{cmdInternalSvcErr},
  2422. },
  2423. }
  2424. for _, tc := range cases {
  2425. t.Run(tc.name, func(t *testing.T) {
  2426. ctx := context.Background()
  2427. adminSvc := newMockAdminService(t)
  2428. for _, params := range tc.mockParams.infoChangeRequestParams {
  2429. adminSvc.EXPECT().
  2430. InfoChangeRequest(ctx, matchSession(params.me), wire.SNACFrame{}, params.inBody).
  2431. Return(params.msg, params.err)
  2432. }
  2433. svc := OSCARProxy{
  2434. Logger: slog.Default(),
  2435. AdminService: adminSvc,
  2436. }
  2437. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2438. assert.Equal(t, tc.wantMsg, msg)
  2439. })
  2440. }
  2441. }
  2442. func TestOSCARProxy_RecvClientCmd_GetDirSearchURL(t *testing.T) {
  2443. cases := []struct {
  2444. // name is the unit test name
  2445. name string
  2446. // me is the TOC user session
  2447. me *state.SessionInstance
  2448. // givenCmd is the TOC command
  2449. givenCmd []byte
  2450. // wantMsg is the expected TOC response
  2451. wantMsg []string
  2452. // mockParams is the list of params sent to mocks that satisfy this
  2453. // method's dependencies
  2454. mockParams mockParams
  2455. }{
  2456. {
  2457. name: "successfully request user info",
  2458. me: newTestSession("me"),
  2459. givenCmd: []byte(`toc_dir_search "first \[name\]":"middle name":"last name":"maiden name":"city":"state":"country":"email"`),
  2460. mockParams: mockParams{
  2461. cookieBakerParams: cookieBakerParams{
  2462. issueParams: issueParams{
  2463. {
  2464. data: []byte("me"),
  2465. returnData: []byte("monster"),
  2466. },
  2467. },
  2468. },
  2469. },
  2470. wantMsg: []string{"GOTO_URL:search results:dir_search?city=city&cookie=6d6f6e73746572&country=country&email=email&first_name=first+%5Bname%5D&last_name=last+name&maiden_name=maiden+name&middle_name=middle+name&state=state"},
  2471. },
  2472. {
  2473. name: "successfully request user info by keywords",
  2474. me: newTestSession("me"),
  2475. givenCmd: []byte(`toc_dir_search ::::::::::"searchkw"`),
  2476. mockParams: mockParams{
  2477. cookieBakerParams: cookieBakerParams{
  2478. issueParams: issueParams{
  2479. {
  2480. data: []byte("me"),
  2481. returnData: []byte("monster"),
  2482. },
  2483. },
  2484. },
  2485. },
  2486. wantMsg: []string{"GOTO_URL:search results:dir_search?cookie=6d6f6e73746572&keyword=searchkw"},
  2487. },
  2488. {
  2489. name: "request user info with too many params",
  2490. me: newTestSession("me"),
  2491. givenCmd: []byte(`toc_dir_search ::::::::::::::::::::"searchkw"`),
  2492. wantMsg: []string{cmdInternalSvcErr},
  2493. },
  2494. {
  2495. name: "request user info, get cookie issue error",
  2496. me: newTestSession("me"),
  2497. givenCmd: []byte(`toc_dir_search them`),
  2498. mockParams: mockParams{
  2499. cookieBakerParams: cookieBakerParams{
  2500. issueParams: issueParams{
  2501. {
  2502. data: []byte("me"),
  2503. returnErr: io.EOF,
  2504. },
  2505. },
  2506. },
  2507. },
  2508. wantMsg: []string{cmdInternalSvcErr},
  2509. },
  2510. {
  2511. name: "bad command",
  2512. me: newTestSession("me"),
  2513. givenCmd: []byte(`toc_dir_search`),
  2514. wantMsg: []string{cmdInternalSvcErr},
  2515. },
  2516. }
  2517. for _, tc := range cases {
  2518. t.Run(tc.name, func(t *testing.T) {
  2519. ctx := context.Background()
  2520. cookieBaker := newMockCookieBaker(t)
  2521. for _, params := range tc.mockParams.issueParams {
  2522. cookieBaker.EXPECT().
  2523. Issue(params.data).
  2524. Return(params.returnData, params.returnErr)
  2525. }
  2526. svc := OSCARProxy{
  2527. Logger: slog.Default(),
  2528. CookieBaker: cookieBaker,
  2529. SNACRateLimits: wire.DefaultSNACRateLimits(),
  2530. }
  2531. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2532. assert.Equal(t, tc.wantMsg, msg)
  2533. })
  2534. }
  2535. }
  2536. func TestOSCARProxy_RecvClientCmd_GetDirURL(t *testing.T) {
  2537. cases := []struct {
  2538. // name is the unit test name
  2539. name string
  2540. // me is the TOC user session
  2541. me *state.SessionInstance
  2542. // givenCmd is the TOC command
  2543. givenCmd []byte
  2544. // wantMsg is the expected TOC response
  2545. wantMsg []string
  2546. // mockParams is the list of params sent to mocks that satisfy this
  2547. // method's dependencies
  2548. mockParams mockParams
  2549. }{
  2550. {
  2551. name: "successfully request user dir info",
  2552. me: newTestSession("me"),
  2553. givenCmd: []byte(`toc_get_dir them`),
  2554. mockParams: mockParams{
  2555. cookieBakerParams: cookieBakerParams{
  2556. issueParams: issueParams{
  2557. {
  2558. data: []byte("me"),
  2559. returnData: []byte("monster"),
  2560. },
  2561. },
  2562. },
  2563. },
  2564. wantMsg: []string{"GOTO_URL:directory info:dir_info?cookie=6d6f6e73746572&user=them"},
  2565. },
  2566. {
  2567. name: "request user info, get cookie issue error",
  2568. me: newTestSession("me"),
  2569. givenCmd: []byte(`toc_get_dir them`),
  2570. mockParams: mockParams{
  2571. cookieBakerParams: cookieBakerParams{
  2572. issueParams: issueParams{
  2573. {
  2574. data: []byte("me"),
  2575. returnErr: io.EOF,
  2576. },
  2577. },
  2578. },
  2579. },
  2580. wantMsg: []string{cmdInternalSvcErr},
  2581. },
  2582. {
  2583. name: "bad command",
  2584. me: newTestSession("me"),
  2585. givenCmd: []byte(`toc_get_dir`),
  2586. wantMsg: []string{cmdInternalSvcErr},
  2587. },
  2588. }
  2589. for _, tc := range cases {
  2590. t.Run(tc.name, func(t *testing.T) {
  2591. ctx := context.Background()
  2592. cookieBaker := newMockCookieBaker(t)
  2593. for _, params := range tc.mockParams.issueParams {
  2594. cookieBaker.EXPECT().
  2595. Issue(params.data).
  2596. Return(params.returnData, params.returnErr)
  2597. }
  2598. svc := OSCARProxy{
  2599. Logger: slog.Default(),
  2600. CookieBaker: cookieBaker,
  2601. }
  2602. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2603. assert.Equal(t, tc.wantMsg, msg)
  2604. })
  2605. }
  2606. }
  2607. func TestOSCARProxy_RecvClientCmd_GetInfoURL(t *testing.T) {
  2608. cases := []struct {
  2609. // name is the unit test name
  2610. name string
  2611. // me is the TOC user session
  2612. me *state.SessionInstance
  2613. // givenCmd is the TOC command
  2614. givenCmd []byte
  2615. // wantMsg is the expected TOC response
  2616. wantMsg []string
  2617. // mockParams is the list of params sent to mocks that satisfy this
  2618. // method's dependencies
  2619. mockParams mockParams
  2620. }{
  2621. {
  2622. name: "successfully request user info",
  2623. me: newTestSession("me"),
  2624. givenCmd: []byte(`toc_get_info them`),
  2625. mockParams: mockParams{
  2626. cookieBakerParams: cookieBakerParams{
  2627. issueParams: issueParams{
  2628. {
  2629. data: []byte("me"),
  2630. returnData: []byte("monster"),
  2631. },
  2632. },
  2633. },
  2634. },
  2635. wantMsg: []string{"GOTO_URL:profile:info?cookie=6d6f6e73746572&from=me&user=them"},
  2636. },
  2637. {
  2638. name: "request user info, get cookie issue error",
  2639. me: newTestSession("me"),
  2640. givenCmd: []byte(`toc_get_info them`),
  2641. mockParams: mockParams{
  2642. cookieBakerParams: cookieBakerParams{
  2643. issueParams: issueParams{
  2644. {
  2645. data: []byte("me"),
  2646. returnErr: io.EOF,
  2647. },
  2648. },
  2649. },
  2650. },
  2651. wantMsg: []string{cmdInternalSvcErr},
  2652. },
  2653. {
  2654. name: "bad command",
  2655. me: newTestSession("me"),
  2656. givenCmd: []byte(`toc_get_info`),
  2657. wantMsg: []string{cmdInternalSvcErr},
  2658. },
  2659. }
  2660. for _, tc := range cases {
  2661. t.Run(tc.name, func(t *testing.T) {
  2662. ctx := context.Background()
  2663. cookieBaker := newMockCookieBaker(t)
  2664. for _, params := range tc.mockParams.issueParams {
  2665. cookieBaker.EXPECT().
  2666. Issue(params.data).
  2667. Return(params.returnData, params.returnErr)
  2668. }
  2669. svc := OSCARProxy{
  2670. Logger: slog.Default(),
  2671. CookieBaker: cookieBaker,
  2672. }
  2673. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2674. assert.Equal(t, tc.wantMsg, msg)
  2675. })
  2676. }
  2677. }
  2678. func TestOSCARProxy_RecvClientCmd_GetStatus(t *testing.T) {
  2679. cases := []struct {
  2680. // name is the unit test name
  2681. name string
  2682. // me is the TOC user session
  2683. me *state.SessionInstance
  2684. // givenCmd is the TOC command
  2685. givenCmd []byte
  2686. // wantMsg is the expected TOC response
  2687. wantMsg []string
  2688. // mockParams is the list of params sent to mocks that satisfy this
  2689. // method's dependencies
  2690. mockParams mockParams
  2691. }{
  2692. {
  2693. name: "successfully request status",
  2694. me: newTestSession("me"),
  2695. givenCmd: []byte("toc_get_status them"),
  2696. mockParams: mockParams{
  2697. locateParams: locateParams{
  2698. userInfoQueryParams: userInfoQueryParams{
  2699. {
  2700. me: state.NewIdentScreenName("me"),
  2701. inBody: wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  2702. ScreenName: "them",
  2703. },
  2704. msg: wire.SNACMessage{
  2705. Body: wire.SNAC_0x02_0x06_LocateUserInfoReply{
  2706. TLVUserInfo: wire.TLVUserInfo{
  2707. ScreenName: "them",
  2708. WarningLevel: 0,
  2709. TLVBlock: wire.TLVBlock{
  2710. TLVList: wire.TLVList{
  2711. wire.NewTLVBE(wire.OServiceUserInfoSignonTOD, uint32(1234)),
  2712. wire.NewTLVBE(wire.OServiceUserInfoIdleTime, uint16(5678)),
  2713. wire.NewTLVBE(wire.OServiceUserInfoUserFlags, wire.OServiceUserFlagOSCARFree),
  2714. },
  2715. },
  2716. },
  2717. },
  2718. },
  2719. },
  2720. },
  2721. },
  2722. },
  2723. wantMsg: []string{"UPDATE_BUDDY:them:T:0:1234:5678: O "},
  2724. },
  2725. {
  2726. name: "request status, receive err from locate svc",
  2727. me: newTestSession("me"),
  2728. givenCmd: []byte("toc_get_status them"),
  2729. mockParams: mockParams{
  2730. locateParams: locateParams{
  2731. userInfoQueryParams: userInfoQueryParams{
  2732. {
  2733. me: state.NewIdentScreenName("me"),
  2734. inBody: wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  2735. ScreenName: "them",
  2736. },
  2737. err: io.EOF,
  2738. },
  2739. },
  2740. },
  2741. },
  2742. wantMsg: []string{cmdInternalSvcErr},
  2743. },
  2744. {
  2745. name: "request status, user not online",
  2746. me: newTestSession("me"),
  2747. givenCmd: []byte("toc_get_status them"),
  2748. mockParams: mockParams{
  2749. locateParams: locateParams{
  2750. userInfoQueryParams: userInfoQueryParams{
  2751. {
  2752. me: state.NewIdentScreenName("me"),
  2753. inBody: wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  2754. ScreenName: "them",
  2755. },
  2756. msg: wire.SNACMessage{
  2757. Body: wire.SNACError{
  2758. Code: wire.ErrorCodeNotLoggedOn,
  2759. },
  2760. },
  2761. },
  2762. },
  2763. },
  2764. },
  2765. wantMsg: []string{"ERROR:901:them"},
  2766. },
  2767. {
  2768. name: "request status, receive unexpected error code",
  2769. me: newTestSession("me"),
  2770. givenCmd: []byte("toc_get_status them"),
  2771. mockParams: mockParams{
  2772. locateParams: locateParams{
  2773. userInfoQueryParams: userInfoQueryParams{
  2774. {
  2775. me: state.NewIdentScreenName("me"),
  2776. inBody: wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  2777. ScreenName: "them",
  2778. },
  2779. msg: wire.SNACMessage{
  2780. Body: wire.SNACError{
  2781. Code: wire.ErrorCodeInvalidSnac,
  2782. },
  2783. },
  2784. },
  2785. },
  2786. },
  2787. },
  2788. wantMsg: []string{cmdInternalSvcErr},
  2789. },
  2790. {
  2791. name: "request status, unexpected response from locate svc",
  2792. me: newTestSession("me"),
  2793. givenCmd: []byte("toc_get_status them"),
  2794. mockParams: mockParams{
  2795. locateParams: locateParams{
  2796. userInfoQueryParams: userInfoQueryParams{
  2797. {
  2798. me: state.NewIdentScreenName("me"),
  2799. inBody: wire.SNAC_0x02_0x05_LocateUserInfoQuery{
  2800. ScreenName: "them",
  2801. },
  2802. msg: wire.SNACMessage{
  2803. Body: wire.SNAC_0x0E_0x04_ChatUsersLeft{},
  2804. },
  2805. },
  2806. },
  2807. },
  2808. },
  2809. wantMsg: []string{cmdInternalSvcErr},
  2810. },
  2811. {
  2812. name: "bad command",
  2813. me: newTestSession("me"),
  2814. givenCmd: []byte(`toc_get_status`),
  2815. wantMsg: []string{cmdInternalSvcErr},
  2816. },
  2817. }
  2818. for _, tc := range cases {
  2819. t.Run(tc.name, func(t *testing.T) {
  2820. ctx := context.Background()
  2821. locateSvc := newMockLocateService(t)
  2822. for _, params := range tc.mockParams.userInfoQueryParams {
  2823. locateSvc.EXPECT().
  2824. UserInfoQuery(mock.Anything, matchSession(params.me), wire.SNACFrame{}, params.inBody).
  2825. Return(params.msg, params.err)
  2826. }
  2827. svc := OSCARProxy{
  2828. Logger: slog.Default(),
  2829. LocateService: locateSvc,
  2830. }
  2831. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2832. assert.Equal(t, tc.wantMsg, msg)
  2833. })
  2834. }
  2835. }
  2836. func TestOSCARProxy_RecvClientCmd_InitDone(t *testing.T) {
  2837. cases := []struct {
  2838. // name is the unit test name
  2839. name string
  2840. // me is the TOC user session
  2841. me *state.SessionInstance
  2842. // givenCmd is the TOC command
  2843. givenCmd []byte
  2844. // wantMsg is the expected TOC response
  2845. wantMsg []string
  2846. // mockParams is the list of params sent to mocks that satisfy this
  2847. // method's dependencies
  2848. mockParams mockParams
  2849. }{
  2850. {
  2851. name: "successfully initialize connection",
  2852. me: newTestSession("me"),
  2853. givenCmd: []byte(`toc_init_done`),
  2854. mockParams: mockParams{
  2855. oServiceParams: oServiceParams{
  2856. clientOnlineParams: clientOnlineParams{
  2857. {
  2858. me: state.NewIdentScreenName("me"),
  2859. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  2860. service: wire.BOS,
  2861. },
  2862. },
  2863. },
  2864. feedBagParams: feedBagParams{
  2865. useFeedbagParams: useFeedbagParams{
  2866. {
  2867. me: state.NewIdentScreenName("me"),
  2868. },
  2869. },
  2870. },
  2871. },
  2872. wantMsg: []string{},
  2873. },
  2874. {
  2875. name: "initialize connection, receive err from BOS oservice svc",
  2876. me: newTestSession("me"),
  2877. givenCmd: []byte(`toc_init_done`),
  2878. mockParams: mockParams{
  2879. oServiceParams: oServiceParams{
  2880. clientOnlineParams: clientOnlineParams{
  2881. {
  2882. me: state.NewIdentScreenName("me"),
  2883. body: wire.SNAC_0x01_0x02_OServiceClientOnline{},
  2884. service: wire.BOS,
  2885. err: io.EOF,
  2886. },
  2887. },
  2888. },
  2889. feedBagParams: feedBagParams{
  2890. useFeedbagParams: useFeedbagParams{
  2891. {
  2892. me: state.NewIdentScreenName("me"),
  2893. },
  2894. },
  2895. },
  2896. },
  2897. wantMsg: []string{cmdInternalSvcErr},
  2898. },
  2899. }
  2900. for _, tc := range cases {
  2901. t.Run(tc.name, func(t *testing.T) {
  2902. ctx := context.Background()
  2903. oSvc := newMockOServiceService(t)
  2904. for _, params := range tc.mockParams.oServiceParams.clientOnlineParams {
  2905. oSvc.EXPECT().
  2906. ClientOnline(ctx, params.service, params.body, matchSession(params.me)).
  2907. Return(params.err)
  2908. }
  2909. fbMgr := newMockFeedbagManager(t)
  2910. for _, params := range tc.mockParams.feedBagParams.useFeedbagParams {
  2911. fbMgr.EXPECT().
  2912. UseFeedbag(ctx, params.me).
  2913. Return(params.err)
  2914. }
  2915. svc := OSCARProxy{
  2916. Logger: slog.Default(),
  2917. OServiceService: oSvc,
  2918. FeedbagManager: fbMgr,
  2919. }
  2920. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  2921. assert.Equal(t, tc.wantMsg, msg)
  2922. })
  2923. }
  2924. }
  2925. func TestOSCARProxy_RecvClientCmd_RemoveBuddy(t *testing.T) {
  2926. cases := []struct {
  2927. // name is the unit test name
  2928. name string
  2929. // me is the TOC user session
  2930. me *state.SessionInstance
  2931. // givenCmd is the TOC command
  2932. givenCmd []byte
  2933. // wantMsg is the expected TOC response
  2934. wantMsg []string
  2935. // mockParams is the list of params sent to mocks that satisfy this
  2936. // method's dependencies
  2937. mockParams mockParams
  2938. }{
  2939. {
  2940. name: "successfully remove buddies",
  2941. me: newTestSession("me"),
  2942. givenCmd: []byte("toc_remove_buddy friend1 friend2 friend3"),
  2943. mockParams: mockParams{
  2944. buddyParams: buddyParams{
  2945. delBuddiesParams: delBuddiesParams{
  2946. {
  2947. me: state.NewIdentScreenName("me"),
  2948. inBody: wire.SNAC_0x03_0x05_BuddyDelBuddies{
  2949. Buddies: []struct {
  2950. ScreenName string `oscar:"len_prefix=uint8"`
  2951. }{
  2952. {ScreenName: "friend1"},
  2953. {ScreenName: "friend2"},
  2954. {ScreenName: "friend3"},
  2955. },
  2956. },
  2957. },
  2958. },
  2959. },
  2960. },
  2961. wantMsg: []string{},
  2962. },
  2963. {
  2964. name: "remove buddies with empty list",
  2965. me: newTestSession("me"),
  2966. givenCmd: []byte("toc_remove_buddy"),
  2967. mockParams: mockParams{
  2968. buddyParams: buddyParams{
  2969. delBuddiesParams: delBuddiesParams{
  2970. {
  2971. me: state.NewIdentScreenName("me"),
  2972. inBody: wire.SNAC_0x03_0x05_BuddyDelBuddies{},
  2973. },
  2974. },
  2975. },
  2976. },
  2977. wantMsg: []string{},
  2978. },
  2979. {
  2980. name: "remove buddies, receive error from buddy service",
  2981. me: newTestSession("me"),
  2982. givenCmd: []byte("toc_remove_buddy friend1"),
  2983. mockParams: mockParams{
  2984. buddyParams: buddyParams{
  2985. delBuddiesParams: delBuddiesParams{
  2986. {
  2987. me: state.NewIdentScreenName("me"),
  2988. inBody: wire.SNAC_0x03_0x05_BuddyDelBuddies{
  2989. Buddies: []struct {
  2990. ScreenName string `oscar:"len_prefix=uint8"`
  2991. }{
  2992. {ScreenName: "friend1"},
  2993. },
  2994. },
  2995. err: io.EOF,
  2996. },
  2997. },
  2998. },
  2999. },
  3000. wantMsg: []string{cmdInternalSvcErr},
  3001. },
  3002. }
  3003. for _, tc := range cases {
  3004. t.Run(tc.name, func(t *testing.T) {
  3005. ctx := context.Background()
  3006. buddySvc := newMockBuddyService(t)
  3007. for _, params := range tc.mockParams.delBuddiesParams {
  3008. buddySvc.EXPECT().
  3009. DelBuddies(ctx, matchSession(params.me), params.inBody).
  3010. Return(params.err)
  3011. }
  3012. svc := OSCARProxy{
  3013. Logger: slog.Default(),
  3014. BuddyService: buddySvc,
  3015. }
  3016. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  3017. assert.Equal(t, tc.wantMsg, msg)
  3018. })
  3019. }
  3020. }
  3021. func TestOSCARProxy_RecvClientCmd_NewBuddies(t *testing.T) {
  3022. cases := []struct {
  3023. name string
  3024. me *state.SessionInstance
  3025. givenCmd []byte
  3026. wantMsg []string
  3027. mockParams mockParams
  3028. }{
  3029. {
  3030. name: "add new users to existing group",
  3031. me: newTestSession("me"),
  3032. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:mike\nb:mk6i\ng:Family\nb:alice\nb:bob\n}"),
  3033. wantMsg: []string{"NEW_BUDDY_REPLY2:mike:added", "NEW_BUDDY_REPLY2:mk6i:added", "NEW_BUDDY_REPLY2:alice:added", "NEW_BUDDY_REPLY2:bob:added"},
  3034. mockParams: mockParams{
  3035. feedBagParams: feedBagParams{
  3036. feedbagParams: feedbagParams{
  3037. {
  3038. screenName: state.NewIdentScreenName("me"),
  3039. results: []wire.FeedbagItem{
  3040. {
  3041. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3042. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724, 21827, 29709})}},
  3043. },
  3044. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3045. {Name: "Co-Workers", GroupID: 21827, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3046. {Name: "Family", GroupID: 29709, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3047. },
  3048. err: nil,
  3049. },
  3050. },
  3051. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3052. {
  3053. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3054. items: []wire.FeedbagItem{
  3055. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 17724, Name: "mike", TLVLBlock: wire.TLVLBlock{}},
  3056. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 17724, Name: "mk6i", TLVLBlock: wire.TLVLBlock{}},
  3057. },
  3058. msg: nil, err: nil,
  3059. },
  3060. {
  3061. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3062. items: []wire.FeedbagItem{
  3063. {
  3064. Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3065. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 2})}},
  3066. },
  3067. },
  3068. msg: nil, err: nil,
  3069. },
  3070. {
  3071. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3072. items: []wire.FeedbagItem{
  3073. {ItemID: 3, ClassID: wire.FeedbagClassIdBuddy, GroupID: 29709, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  3074. {ItemID: 4, ClassID: wire.FeedbagClassIdBuddy, GroupID: 29709, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  3075. },
  3076. msg: nil, err: nil,
  3077. },
  3078. {
  3079. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3080. items: []wire.FeedbagItem{
  3081. {
  3082. Name: "Family", GroupID: 29709, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3083. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{3, 4})}},
  3084. },
  3085. },
  3086. msg: nil, err: nil,
  3087. },
  3088. },
  3089. },
  3090. },
  3091. },
  3092. {
  3093. name: "empty feedbag: add new group and buddies",
  3094. me: newTestSession("me"),
  3095. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:mike\nb:mk6i\n}"),
  3096. wantMsg: []string{"NEW_BUDDY_REPLY2:mike:added", "NEW_BUDDY_REPLY2:mk6i:added"},
  3097. mockParams: mockParams{
  3098. feedBagParams: feedBagParams{
  3099. feedbagParams: feedbagParams{
  3100. {
  3101. screenName: state.NewIdentScreenName("me"),
  3102. results: []wire.FeedbagItem{},
  3103. err: nil,
  3104. },
  3105. },
  3106. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3107. // First: insert the two buddy items (deterministic randIntn gives GroupID 1, ItemIDs 2 and 3)
  3108. {
  3109. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3110. items: []wire.FeedbagItem{
  3111. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, Name: "mike", TLVLBlock: wire.TLVLBlock{}},
  3112. {ItemID: 3, ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, Name: "mk6i", TLVLBlock: wire.TLVLBlock{}},
  3113. },
  3114. msg: nil, err: nil,
  3115. },
  3116. // Second: insert the new Buddies group with Order listing the buddy item IDs
  3117. {
  3118. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3119. items: []wire.FeedbagItem{
  3120. {
  3121. Name: "Buddies", GroupID: 1, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3122. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{2, 3})}},
  3123. },
  3124. },
  3125. msg: nil, err: nil,
  3126. },
  3127. // Third: insert the root group so its Order lists the new group ID
  3128. {
  3129. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3130. items: []wire.FeedbagItem{
  3131. {
  3132. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3133. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}},
  3134. },
  3135. },
  3136. msg: nil, err: nil,
  3137. },
  3138. },
  3139. },
  3140. },
  3141. },
  3142. {
  3143. name: "add new group and buddies",
  3144. me: newTestSession("me"),
  3145. givenCmd: []byte("toc2_new_buddies {g:Co-Workers\nb:carol\nb:dan\ng:Family\nb:alice\nb:bob\n}"),
  3146. wantMsg: []string{"NEW_BUDDY_REPLY2:carol:added", "NEW_BUDDY_REPLY2:dan:added", "NEW_BUDDY_REPLY2:alice:added", "NEW_BUDDY_REPLY2:bob:added"},
  3147. mockParams: mockParams{
  3148. feedBagParams: feedBagParams{
  3149. feedbagParams: feedbagParams{
  3150. {
  3151. screenName: state.NewIdentScreenName("me"),
  3152. results: []wire.FeedbagItem{
  3153. {
  3154. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3155. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3156. },
  3157. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3158. },
  3159. err: nil,
  3160. },
  3161. },
  3162. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3163. {
  3164. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3165. items: []wire.FeedbagItem{
  3166. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, Name: "carol", TLVLBlock: wire.TLVLBlock{}},
  3167. {ItemID: 3, ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, Name: "dan", TLVLBlock: wire.TLVLBlock{}},
  3168. },
  3169. msg: nil, err: nil,
  3170. },
  3171. {
  3172. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3173. items: []wire.FeedbagItem{
  3174. {
  3175. Name: "Co-Workers", GroupID: 1, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3176. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{2, 3})}},
  3177. },
  3178. },
  3179. msg: nil, err: nil,
  3180. },
  3181. {
  3182. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3183. items: []wire.FeedbagItem{
  3184. {ItemID: 5, ClassID: wire.FeedbagClassIdBuddy, GroupID: 4, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  3185. {ItemID: 6, ClassID: wire.FeedbagClassIdBuddy, GroupID: 4, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  3186. },
  3187. msg: nil, err: nil,
  3188. },
  3189. {
  3190. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3191. items: []wire.FeedbagItem{
  3192. {
  3193. Name: "Family", GroupID: 4, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3194. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{5, 6})}},
  3195. },
  3196. },
  3197. msg: nil, err: nil,
  3198. },
  3199. // Root updated once at end
  3200. {
  3201. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3202. items: []wire.FeedbagItem{
  3203. {
  3204. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3205. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724, 1, 4})}},
  3206. },
  3207. },
  3208. msg: nil, err: nil,
  3209. },
  3210. },
  3211. },
  3212. },
  3213. },
  3214. {
  3215. name: "empty config returns error",
  3216. me: newTestSession("me"),
  3217. givenCmd: []byte("toc2_new_buddies "),
  3218. wantMsg: []string{cmdInternalSvcErr},
  3219. mockParams: mockParams{},
  3220. },
  3221. {
  3222. name: "whitespace-only config returns error",
  3223. me: newTestSession("me"),
  3224. givenCmd: []byte("toc2_new_buddies "),
  3225. wantMsg: []string{cmdInternalSvcErr},
  3226. mockParams: mockParams{},
  3227. },
  3228. {
  3229. name: "buddy without group returns parse error",
  3230. me: newTestSession("me"),
  3231. givenCmd: []byte("toc2_new_buddies {b:alice\n}"),
  3232. wantMsg: []string{cmdInternalSvcErr},
  3233. mockParams: mockParams{},
  3234. },
  3235. {
  3236. name: "empty group name returns parse error",
  3237. me: newTestSession("me"),
  3238. givenCmd: []byte("toc2_new_buddies {g:\nb:alice\n}"),
  3239. wantMsg: []string{cmdInternalSvcErr},
  3240. mockParams: mockParams{},
  3241. },
  3242. {
  3243. name: "empty buddy name returns parse error",
  3244. me: newTestSession("me"),
  3245. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:\n}"),
  3246. wantMsg: []string{cmdInternalSvcErr},
  3247. mockParams: mockParams{},
  3248. },
  3249. {
  3250. name: "group with no buddies does not call UpsertItem",
  3251. me: newTestSession("me"),
  3252. givenCmd: []byte("toc2_new_buddies {g:EmptyGroup\n}"),
  3253. wantMsg: nil,
  3254. mockParams: mockParams{
  3255. feedBagParams: feedBagParams{
  3256. feedbagParams: feedbagParams{
  3257. {
  3258. screenName: state.NewIdentScreenName("me"),
  3259. results: []wire.FeedbagItem{
  3260. {
  3261. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3262. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3263. },
  3264. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3265. },
  3266. err: nil,
  3267. },
  3268. },
  3269. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  3270. },
  3271. },
  3272. },
  3273. {
  3274. name: "FeedbagManager.Feedbag error returns internal error",
  3275. me: newTestSession("me"),
  3276. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:alice\n}"),
  3277. wantMsg: []string{cmdInternalSvcErr},
  3278. mockParams: mockParams{
  3279. feedBagParams: feedBagParams{
  3280. feedbagParams: feedbagParams{
  3281. {
  3282. screenName: state.NewIdentScreenName("me"),
  3283. results: nil,
  3284. err: errors.New("EOF"),
  3285. },
  3286. },
  3287. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  3288. },
  3289. },
  3290. },
  3291. {
  3292. name: "FeedbagService.UpsertItem error on first insert returns internal error",
  3293. me: newTestSession("me"),
  3294. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:alice\n}"),
  3295. wantMsg: []string{cmdInternalSvcErr},
  3296. mockParams: mockParams{
  3297. feedBagParams: feedBagParams{
  3298. feedbagParams: feedbagParams{
  3299. {
  3300. screenName: state.NewIdentScreenName("me"),
  3301. results: []wire.FeedbagItem{
  3302. {
  3303. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3304. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{100})}},
  3305. },
  3306. {
  3307. Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3308. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{})}},
  3309. },
  3310. },
  3311. err: nil,
  3312. },
  3313. },
  3314. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3315. {
  3316. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3317. items: []wire.FeedbagItem{
  3318. {
  3319. Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3320. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}},
  3321. },
  3322. },
  3323. msg: nil, err: nil,
  3324. },
  3325. {
  3326. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3327. items: []wire.FeedbagItem{
  3328. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  3329. },
  3330. msg: nil, err: errors.New("EOF"),
  3331. },
  3332. },
  3333. },
  3334. },
  3335. },
  3336. {
  3337. name: "buddy already in group skips and adds only new buddy",
  3338. me: newTestSession("me"),
  3339. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:mike\nb:mk6i\n}"),
  3340. wantMsg: []string{"NEW_BUDDY_REPLY2:mk6i:added"},
  3341. mockParams: mockParams{
  3342. feedBagParams: feedBagParams{
  3343. feedbagParams: feedbagParams{
  3344. {
  3345. screenName: state.NewIdentScreenName("me"),
  3346. results: []wire.FeedbagItem{
  3347. {
  3348. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3349. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3350. },
  3351. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3352. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}}},
  3353. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 17724, Name: "mike", TLVLBlock: wire.TLVLBlock{}},
  3354. },
  3355. err: nil,
  3356. },
  3357. },
  3358. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3359. {
  3360. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3361. items: []wire.FeedbagItem{
  3362. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 17724, Name: "mk6i", TLVLBlock: wire.TLVLBlock{}},
  3363. },
  3364. msg: nil, err: nil,
  3365. },
  3366. {
  3367. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3368. items: []wire.FeedbagItem{
  3369. {
  3370. Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3371. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 2})}},
  3372. },
  3373. },
  3374. msg: nil, err: nil,
  3375. },
  3376. },
  3377. },
  3378. },
  3379. },
  3380. {
  3381. name: "all buddies already in group returns no replies and no UpsertItem",
  3382. me: newTestSession("me"),
  3383. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:mike\nb:mk6i\n}"),
  3384. wantMsg: nil,
  3385. mockParams: mockParams{
  3386. feedBagParams: feedBagParams{
  3387. feedbagParams: feedbagParams{
  3388. {
  3389. screenName: state.NewIdentScreenName("me"),
  3390. results: []wire.FeedbagItem{
  3391. {
  3392. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3393. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3394. },
  3395. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3396. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 2})}}},
  3397. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 17724, Name: "mike", TLVLBlock: wire.TLVLBlock{}},
  3398. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 17724, Name: "mk6i", TLVLBlock: wire.TLVLBlock{}},
  3399. },
  3400. err: nil,
  3401. },
  3402. },
  3403. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  3404. },
  3405. },
  3406. },
  3407. {
  3408. name: "single buddy in new group",
  3409. me: newTestSession("me"),
  3410. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:alice\n}"),
  3411. wantMsg: []string{"NEW_BUDDY_REPLY2:alice:added"},
  3412. mockParams: mockParams{
  3413. feedBagParams: feedBagParams{
  3414. feedbagParams: feedbagParams{
  3415. {
  3416. screenName: state.NewIdentScreenName("me"),
  3417. results: []wire.FeedbagItem{},
  3418. err: nil,
  3419. },
  3420. },
  3421. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3422. {
  3423. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3424. items: []wire.FeedbagItem{
  3425. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 1, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  3426. },
  3427. msg: nil, err: nil,
  3428. },
  3429. {
  3430. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3431. items: []wire.FeedbagItem{
  3432. {
  3433. Name: "Buddies", GroupID: 1, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3434. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{2})}},
  3435. },
  3436. },
  3437. msg: nil, err: nil,
  3438. },
  3439. {
  3440. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3441. items: []wire.FeedbagItem{
  3442. {
  3443. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3444. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}},
  3445. },
  3446. },
  3447. msg: nil, err: nil,
  3448. },
  3449. },
  3450. },
  3451. },
  3452. },
  3453. {
  3454. name: "buddy with alias and note",
  3455. me: newTestSession("me"),
  3456. givenCmd: []byte("toc2_new_buddies {g:Buddies\nb:bob:Bob Smith:::::Friend from work\n}"),
  3457. wantMsg: []string{"NEW_BUDDY_REPLY2:bob:added"},
  3458. mockParams: mockParams{
  3459. feedBagParams: feedBagParams{
  3460. feedbagParams: feedbagParams{
  3461. {
  3462. screenName: state.NewIdentScreenName("me"),
  3463. results: []wire.FeedbagItem{
  3464. {
  3465. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3466. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{100})}},
  3467. },
  3468. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3469. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{})}}},
  3470. },
  3471. err: nil,
  3472. },
  3473. },
  3474. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3475. {
  3476. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3477. items: []wire.FeedbagItem{
  3478. {
  3479. ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "bob",
  3480. TLVLBlock: wire.TLVLBlock{
  3481. TLVList: wire.TLVList{
  3482. wire.NewTLVBE(wire.FeedbagAttributesAlias, "Bob Smith"),
  3483. wire.NewTLVBE(wire.FeedbagAttributesNote, "Friend from work"),
  3484. },
  3485. },
  3486. },
  3487. },
  3488. msg: nil, err: nil,
  3489. },
  3490. {
  3491. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  3492. items: []wire.FeedbagItem{
  3493. {
  3494. Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3495. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}},
  3496. },
  3497. },
  3498. msg: nil, err: nil,
  3499. },
  3500. },
  3501. },
  3502. },
  3503. },
  3504. }
  3505. for _, tc := range cases {
  3506. t.Run(tc.name, func(t *testing.T) {
  3507. ctx := context.Background()
  3508. fbMgr := newMockFeedbagManager(t)
  3509. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  3510. fbMgr.EXPECT().
  3511. Feedbag(ctx, params.screenName).
  3512. Return(params.results, params.err)
  3513. }
  3514. fbSvc := newMockFeedbagService(t)
  3515. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  3516. fbSvc.EXPECT().
  3517. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  3518. Return(params.msg, params.err)
  3519. }
  3520. // Deterministic item ID: increment by 1 per call so IDs are unique and predictable per test.
  3521. var randCall int
  3522. randIntn := func(n int) int {
  3523. randCall++
  3524. return randCall
  3525. }
  3526. svc := OSCARProxy{
  3527. Logger: slog.Default(),
  3528. FeedbagManager: fbMgr,
  3529. FeedbagService: fbSvc,
  3530. RandIntn: randIntn,
  3531. }
  3532. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  3533. assert.Equal(t, tc.wantMsg, msg)
  3534. })
  3535. }
  3536. }
  3537. func TestOSCARProxy_RecvClientCmd_NewGroup(t *testing.T) {
  3538. cases := []struct {
  3539. name string
  3540. me *state.SessionInstance
  3541. givenCmd []byte
  3542. wantMsg []string
  3543. mockParams mockParams
  3544. }{
  3545. {
  3546. name: "add new group when feedbag has root and existing groups",
  3547. me: newTestSession("me"),
  3548. givenCmd: []byte("toc2_new_group Family"),
  3549. wantMsg: []string{},
  3550. mockParams: mockParams{
  3551. feedBagParams: feedBagParams{
  3552. feedbagParams: feedbagParams{
  3553. {
  3554. screenName: state.NewIdentScreenName("me"),
  3555. results: []wire.FeedbagItem{
  3556. {
  3557. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3558. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3559. },
  3560. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3561. },
  3562. err: nil,
  3563. },
  3564. },
  3565. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3566. {
  3567. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3568. items: []wire.FeedbagItem{
  3569. {
  3570. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3571. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724, 1})}},
  3572. },
  3573. {ClassID: wire.FeedbagClassIdGroup, GroupID: 1, Name: "Family"},
  3574. },
  3575. msg: nil, err: nil,
  3576. },
  3577. },
  3578. },
  3579. },
  3580. },
  3581. {
  3582. name: "add new group when feedbag is empty creates root and group",
  3583. me: newTestSession("me"),
  3584. givenCmd: []byte("toc2_new_group Buddies"),
  3585. wantMsg: []string{},
  3586. mockParams: mockParams{
  3587. feedBagParams: feedBagParams{
  3588. feedbagParams: feedbagParams{
  3589. {
  3590. screenName: state.NewIdentScreenName("me"),
  3591. results: []wire.FeedbagItem{},
  3592. err: nil,
  3593. },
  3594. },
  3595. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3596. {
  3597. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3598. items: []wire.FeedbagItem{
  3599. {
  3600. ClassID: wire.FeedbagClassIdGroup,
  3601. GroupID: 0,
  3602. Name: "",
  3603. TLVLBlock: wire.TLVLBlock{
  3604. TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})},
  3605. },
  3606. },
  3607. {ClassID: wire.FeedbagClassIdGroup, GroupID: 1, Name: "Buddies"},
  3608. },
  3609. msg: nil, err: nil,
  3610. },
  3611. },
  3612. },
  3613. },
  3614. },
  3615. {
  3616. name: "group already exists returns success idempotently",
  3617. me: newTestSession("me"),
  3618. givenCmd: []byte("toc2_new_group Buddies"),
  3619. wantMsg: []string{},
  3620. mockParams: mockParams{
  3621. feedBagParams: feedBagParams{
  3622. feedbagParams: feedbagParams{
  3623. {
  3624. screenName: state.NewIdentScreenName("me"),
  3625. results: []wire.FeedbagItem{
  3626. {
  3627. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3628. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3629. },
  3630. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3631. },
  3632. err: nil,
  3633. },
  3634. },
  3635. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  3636. },
  3637. },
  3638. },
  3639. {
  3640. name: "empty group name returns error",
  3641. me: newTestSession("me"),
  3642. givenCmd: []byte("toc2_new_group"),
  3643. wantMsg: []string{cmdInternalSvcErr},
  3644. mockParams: mockParams{
  3645. feedBagParams: feedBagParams{
  3646. feedbagParams: feedbagParams{},
  3647. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  3648. },
  3649. },
  3650. },
  3651. {
  3652. name: "FeedbagManager.Feedbag error returns error",
  3653. me: newTestSession("me"),
  3654. givenCmd: []byte("toc2_new_group Family"),
  3655. wantMsg: []string{cmdInternalSvcErr},
  3656. mockParams: mockParams{
  3657. feedBagParams: feedBagParams{
  3658. feedbagParams: feedbagParams{
  3659. {
  3660. screenName: state.NewIdentScreenName("me"),
  3661. results: nil,
  3662. err: assert.AnError,
  3663. },
  3664. },
  3665. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  3666. },
  3667. },
  3668. },
  3669. {
  3670. name: "FeedbagService.UpsertItem error returns error",
  3671. me: newTestSession("me"),
  3672. givenCmd: []byte("toc2_new_group Family"),
  3673. wantMsg: []string{cmdInternalSvcErr},
  3674. mockParams: mockParams{
  3675. feedBagParams: feedBagParams{
  3676. feedbagParams: feedbagParams{
  3677. {
  3678. screenName: state.NewIdentScreenName("me"),
  3679. results: []wire.FeedbagItem{
  3680. {
  3681. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3682. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3683. },
  3684. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3685. },
  3686. err: nil,
  3687. },
  3688. },
  3689. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3690. {
  3691. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3692. items: []wire.FeedbagItem{
  3693. {
  3694. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3695. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724, 1})}},
  3696. },
  3697. {ClassID: wire.FeedbagClassIdGroup, GroupID: 1, Name: "Family"},
  3698. },
  3699. msg: nil, err: assert.AnError,
  3700. },
  3701. },
  3702. },
  3703. },
  3704. },
  3705. }
  3706. for _, tc := range cases {
  3707. t.Run(tc.name, func(t *testing.T) {
  3708. ctx := context.Background()
  3709. fbMgr := newMockFeedbagManager(t)
  3710. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  3711. fbMgr.EXPECT().
  3712. Feedbag(ctx, params.screenName).
  3713. Return(params.results, params.err)
  3714. }
  3715. fbSvc := newMockFeedbagService(t)
  3716. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  3717. fbSvc.EXPECT().
  3718. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  3719. Return(params.msg, params.err)
  3720. }
  3721. var randCall int
  3722. randIntn := func(n int) int {
  3723. randCall++
  3724. return randCall
  3725. }
  3726. svc := OSCARProxy{
  3727. Logger: slog.Default(),
  3728. FeedbagManager: fbMgr,
  3729. FeedbagService: fbSvc,
  3730. RandIntn: randIntn,
  3731. }
  3732. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  3733. assert.Equal(t, tc.wantMsg, msg)
  3734. })
  3735. }
  3736. }
  3737. func TestOSCARProxy_RecvClientCmd_DelGroup(t *testing.T) {
  3738. cases := []struct {
  3739. name string
  3740. me *state.SessionInstance
  3741. givenCmd []byte
  3742. wantMsg []string
  3743. mockParams mockParams
  3744. }{
  3745. {
  3746. name: "delete existing group",
  3747. me: newTestSession("me"),
  3748. givenCmd: []byte("toc2_del_group Family"),
  3749. wantMsg: []string{},
  3750. mockParams: mockParams{
  3751. feedBagParams: feedBagParams{
  3752. feedbagParams: feedbagParams{
  3753. {
  3754. screenName: state.NewIdentScreenName("me"),
  3755. results: []wire.FeedbagItem{
  3756. {
  3757. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3758. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724, 29709})}},
  3759. },
  3760. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3761. {Name: "Family", GroupID: 29709, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3762. },
  3763. err: nil,
  3764. },
  3765. },
  3766. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  3767. {
  3768. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  3769. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  3770. Items: []wire.FeedbagItem{
  3771. {Name: "Family", GroupID: 29709, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3772. },
  3773. },
  3774. msg: nil, err: nil,
  3775. },
  3776. },
  3777. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3778. {
  3779. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3780. items: []wire.FeedbagItem{
  3781. {
  3782. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3783. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3784. },
  3785. },
  3786. msg: nil, err: nil,
  3787. },
  3788. },
  3789. },
  3790. },
  3791. },
  3792. {
  3793. name: "empty group name returns error",
  3794. me: newTestSession("me"),
  3795. givenCmd: []byte("toc2_del_group"),
  3796. wantMsg: []string{cmdInternalSvcErr},
  3797. mockParams: mockParams{
  3798. feedBagParams: feedBagParams{
  3799. feedbagParams: feedbagParams{},
  3800. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  3801. },
  3802. },
  3803. },
  3804. {
  3805. name: "group not found returns error",
  3806. me: newTestSession("me"),
  3807. givenCmd: []byte("toc2_del_group Nonexistent"),
  3808. wantMsg: []string{cmdInternalSvcErr},
  3809. mockParams: mockParams{
  3810. feedBagParams: feedBagParams{
  3811. feedbagParams: feedbagParams{
  3812. {
  3813. screenName: state.NewIdentScreenName("me"),
  3814. results: []wire.FeedbagItem{
  3815. {
  3816. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3817. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724})}},
  3818. },
  3819. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3820. },
  3821. err: nil,
  3822. },
  3823. },
  3824. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  3825. },
  3826. },
  3827. },
  3828. {
  3829. name: "FeedbagManager.Feedbag error returns error",
  3830. me: newTestSession("me"),
  3831. givenCmd: []byte("toc2_del_group Family"),
  3832. wantMsg: []string{cmdInternalSvcErr},
  3833. mockParams: mockParams{
  3834. feedBagParams: feedBagParams{
  3835. feedbagParams: feedbagParams{
  3836. {
  3837. screenName: state.NewIdentScreenName("me"),
  3838. results: nil,
  3839. err: assert.AnError,
  3840. },
  3841. },
  3842. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  3843. },
  3844. },
  3845. },
  3846. {
  3847. name: "FeedbagService.DeleteItem error returns error",
  3848. me: newTestSession("me"),
  3849. givenCmd: []byte("toc2_del_group Family"),
  3850. wantMsg: []string{cmdInternalSvcErr},
  3851. mockParams: mockParams{
  3852. feedBagParams: feedBagParams{
  3853. feedbagParams: feedbagParams{
  3854. {
  3855. screenName: state.NewIdentScreenName("me"),
  3856. results: []wire.FeedbagItem{
  3857. {
  3858. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  3859. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{17724, 29709})}},
  3860. },
  3861. {Name: "Buddies", GroupID: 17724, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3862. {Name: "Family", GroupID: 29709, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3863. },
  3864. err: nil,
  3865. },
  3866. },
  3867. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  3868. {
  3869. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  3870. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  3871. Items: []wire.FeedbagItem{
  3872. {Name: "Family", GroupID: 29709, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  3873. },
  3874. },
  3875. msg: nil, err: assert.AnError,
  3876. },
  3877. },
  3878. },
  3879. },
  3880. },
  3881. }
  3882. for _, tc := range cases {
  3883. t.Run(tc.name, func(t *testing.T) {
  3884. ctx := context.Background()
  3885. fbMgr := newMockFeedbagManager(t)
  3886. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  3887. fbMgr.EXPECT().
  3888. Feedbag(ctx, params.screenName).
  3889. Return(params.results, params.err)
  3890. }
  3891. fbSvc := newMockFeedbagService(t)
  3892. for _, params := range tc.mockParams.feedBagParams.feedbagServiceDeleteItemParams {
  3893. fbSvc.EXPECT().
  3894. DeleteItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.inBody).
  3895. Return(params.msg, params.err)
  3896. }
  3897. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  3898. fbSvc.EXPECT().
  3899. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  3900. Return(params.msg, params.err)
  3901. }
  3902. svc := OSCARProxy{
  3903. Logger: slog.Default(),
  3904. FeedbagManager: fbMgr,
  3905. FeedbagService: fbSvc,
  3906. }
  3907. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  3908. assert.Equal(t, tc.wantMsg, msg)
  3909. })
  3910. }
  3911. }
  3912. func TestOSCARProxy_RecvClientCmd_SetPDMode(t *testing.T) {
  3913. cases := []struct {
  3914. name string
  3915. me *state.SessionInstance
  3916. givenCmd []byte
  3917. wantMsg []string
  3918. mockParams mockParams
  3919. }{
  3920. {
  3921. name: "empty feedbag creates new Pdinfo with mode 4",
  3922. me: newTestSession("me"),
  3923. givenCmd: []byte("toc2_set_pdmode 4"),
  3924. wantMsg: []string{},
  3925. mockParams: mockParams{
  3926. feedBagParams: feedBagParams{
  3927. feedbagParams: feedbagParams{
  3928. {
  3929. screenName: state.NewIdentScreenName("me"),
  3930. results: []wire.FeedbagItem{},
  3931. err: nil,
  3932. },
  3933. },
  3934. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3935. {
  3936. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3937. items: []wire.FeedbagItem{
  3938. {
  3939. ClassID: wire.FeedbagClassIdPdinfo,
  3940. GroupID: 0,
  3941. ItemID: 1,
  3942. TLVLBlock: wire.TLVLBlock{
  3943. TLVList: wire.TLVList{
  3944. wire.NewTLVBE(wire.FeedbagAttributesPdMode, uint8(4)),
  3945. },
  3946. },
  3947. },
  3948. },
  3949. msg: nil, err: nil,
  3950. },
  3951. },
  3952. },
  3953. },
  3954. },
  3955. {
  3956. name: "pdinfo already exists but does not have FeedbagAttributesPdMode",
  3957. me: newTestSession("me"),
  3958. givenCmd: []byte("toc2_set_pdmode 2"),
  3959. wantMsg: []string{},
  3960. mockParams: mockParams{
  3961. feedBagParams: feedBagParams{
  3962. feedbagParams: feedbagParams{
  3963. {
  3964. screenName: state.NewIdentScreenName("me"),
  3965. results: []wire.FeedbagItem{
  3966. {
  3967. ClassID: wire.FeedbagClassIdPdinfo,
  3968. GroupID: 0,
  3969. ItemID: 99,
  3970. TLVLBlock: wire.TLVLBlock{
  3971. TLVList: wire.TLVList{},
  3972. },
  3973. },
  3974. },
  3975. err: nil,
  3976. },
  3977. },
  3978. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  3979. {
  3980. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  3981. items: []wire.FeedbagItem{
  3982. {
  3983. ClassID: wire.FeedbagClassIdPdinfo,
  3984. GroupID: 0,
  3985. ItemID: 99,
  3986. TLVLBlock: wire.TLVLBlock{
  3987. TLVList: wire.TLVList{
  3988. wire.NewTLVBE(wire.FeedbagAttributesPdMode, uint8(2)),
  3989. },
  3990. },
  3991. },
  3992. },
  3993. msg: nil, err: nil,
  3994. },
  3995. },
  3996. },
  3997. },
  3998. },
  3999. {
  4000. name: "pdinfo already exists and already has FeedbagAttributesPdMode with same mode",
  4001. me: newTestSession("me"),
  4002. givenCmd: []byte("toc2_set_pdmode 3"),
  4003. wantMsg: []string{},
  4004. mockParams: mockParams{
  4005. feedBagParams: feedBagParams{
  4006. feedbagParams: feedbagParams{
  4007. {
  4008. screenName: state.NewIdentScreenName("me"),
  4009. results: []wire.FeedbagItem{
  4010. {
  4011. ClassID: wire.FeedbagClassIdPdinfo,
  4012. GroupID: 0,
  4013. ItemID: 99,
  4014. TLVLBlock: wire.TLVLBlock{
  4015. TLVList: wire.TLVList{
  4016. wire.NewTLVBE(wire.FeedbagAttributesPdMode, uint8(3)),
  4017. },
  4018. },
  4019. },
  4020. },
  4021. err: nil,
  4022. },
  4023. },
  4024. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4025. },
  4026. },
  4027. },
  4028. {
  4029. name: "pdinfo already exists and already has FeedbagAttributesPdMode with different mode",
  4030. me: newTestSession("me"),
  4031. givenCmd: []byte("toc2_set_pdmode 3"),
  4032. wantMsg: []string{},
  4033. mockParams: mockParams{
  4034. feedBagParams: feedBagParams{
  4035. feedbagParams: feedbagParams{
  4036. {
  4037. screenName: state.NewIdentScreenName("me"),
  4038. results: []wire.FeedbagItem{
  4039. {
  4040. ClassID: wire.FeedbagClassIdPdinfo,
  4041. GroupID: 0,
  4042. ItemID: 99,
  4043. TLVLBlock: wire.TLVLBlock{
  4044. TLVList: wire.TLVList{
  4045. wire.NewTLVBE(wire.FeedbagAttributesPdMode, uint8(1)),
  4046. },
  4047. },
  4048. },
  4049. },
  4050. err: nil,
  4051. },
  4052. },
  4053. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4054. {
  4055. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4056. items: []wire.FeedbagItem{
  4057. {
  4058. ClassID: wire.FeedbagClassIdPdinfo,
  4059. GroupID: 0,
  4060. ItemID: 99,
  4061. TLVLBlock: wire.TLVLBlock{
  4062. TLVList: wire.TLVList{
  4063. wire.NewTLVBE(wire.FeedbagAttributesPdMode, uint8(3)),
  4064. },
  4065. },
  4066. },
  4067. },
  4068. msg: nil, err: nil,
  4069. },
  4070. },
  4071. },
  4072. },
  4073. },
  4074. }
  4075. for _, tc := range cases {
  4076. t.Run(tc.name, func(t *testing.T) {
  4077. ctx := context.Background()
  4078. fbMgr := newMockFeedbagManager(t)
  4079. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  4080. fbMgr.EXPECT().
  4081. Feedbag(ctx, params.screenName).
  4082. Return(params.results, params.err)
  4083. }
  4084. fbSvc := newMockFeedbagService(t)
  4085. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  4086. fbSvc.EXPECT().
  4087. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  4088. Return(params.msg, params.err)
  4089. }
  4090. var randCall int
  4091. randIntn := func(n int) int {
  4092. randCall++
  4093. return randCall
  4094. }
  4095. svc := OSCARProxy{
  4096. Logger: slog.Default(),
  4097. FeedbagManager: fbMgr,
  4098. FeedbagService: fbSvc,
  4099. RandIntn: randIntn,
  4100. }
  4101. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  4102. assert.Equal(t, tc.wantMsg, msg)
  4103. })
  4104. }
  4105. }
  4106. func TestOSCARProxy_RecvClientCmd_AddPermit2(t *testing.T) {
  4107. cases := []struct {
  4108. name string
  4109. me *state.SessionInstance
  4110. givenCmd []byte
  4111. wantMsg []string
  4112. mockParams mockParams
  4113. }{
  4114. {
  4115. name: "no screennames provided returns internal error",
  4116. me: newTestSession("me"),
  4117. givenCmd: []byte("toc2_add_permit"),
  4118. wantMsg: []string{cmdInternalSvcErr},
  4119. mockParams: mockParams{},
  4120. },
  4121. {
  4122. name: "FeedbagManager.Feedbag error returns internal error",
  4123. me: newTestSession("me"),
  4124. givenCmd: []byte("toc2_add_permit alice"),
  4125. wantMsg: []string{cmdInternalSvcErr},
  4126. mockParams: mockParams{
  4127. feedBagParams: feedBagParams{
  4128. feedbagParams: feedbagParams{
  4129. {
  4130. screenName: state.NewIdentScreenName("me"),
  4131. results: nil,
  4132. err: errors.New("EOF"),
  4133. },
  4134. },
  4135. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4136. },
  4137. },
  4138. },
  4139. {
  4140. name: "FeedbagService.UpsertItem error returns internal error",
  4141. me: newTestSession("me"),
  4142. givenCmd: []byte("toc2_add_permit alice"),
  4143. wantMsg: []string{cmdInternalSvcErr},
  4144. mockParams: mockParams{
  4145. feedBagParams: feedBagParams{
  4146. feedbagParams: feedbagParams{
  4147. {
  4148. screenName: state.NewIdentScreenName("me"),
  4149. results: []wire.FeedbagItem{},
  4150. err: nil,
  4151. },
  4152. },
  4153. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4154. {
  4155. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4156. items: []wire.FeedbagItem{
  4157. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4158. },
  4159. msg: nil, err: errors.New("EOF"),
  4160. },
  4161. },
  4162. },
  4163. },
  4164. },
  4165. {
  4166. name: "add permits to empty feedbag",
  4167. me: newTestSession("me"),
  4168. givenCmd: []byte("toc2_add_permit alice bob"),
  4169. wantMsg: []string{},
  4170. mockParams: mockParams{
  4171. feedBagParams: feedBagParams{
  4172. feedbagParams: feedbagParams{
  4173. {
  4174. screenName: state.NewIdentScreenName("me"),
  4175. results: []wire.FeedbagItem{},
  4176. err: nil,
  4177. },
  4178. },
  4179. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4180. {
  4181. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4182. items: []wire.FeedbagItem{
  4183. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4184. {ItemID: 2, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4185. },
  4186. msg: nil, err: nil,
  4187. },
  4188. },
  4189. },
  4190. },
  4191. },
  4192. {
  4193. name: "skip existing permit, add only new",
  4194. me: newTestSession("me"),
  4195. givenCmd: []byte("toc2_add_permit alice bob"),
  4196. wantMsg: []string{},
  4197. mockParams: mockParams{
  4198. feedBagParams: feedBagParams{
  4199. feedbagParams: feedbagParams{
  4200. {
  4201. screenName: state.NewIdentScreenName("me"),
  4202. results: []wire.FeedbagItem{
  4203. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4204. },
  4205. err: nil,
  4206. },
  4207. },
  4208. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4209. {
  4210. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4211. items: []wire.FeedbagItem{
  4212. {ItemID: 2, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4213. },
  4214. msg: nil, err: nil,
  4215. },
  4216. },
  4217. },
  4218. },
  4219. },
  4220. {
  4221. name: "no-op when permit already in feedbag",
  4222. me: newTestSession("me"),
  4223. givenCmd: []byte("toc2_add_permit alice"),
  4224. wantMsg: []string{},
  4225. mockParams: mockParams{
  4226. feedBagParams: feedBagParams{
  4227. feedbagParams: feedbagParams{
  4228. {
  4229. screenName: state.NewIdentScreenName("me"),
  4230. results: []wire.FeedbagItem{
  4231. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4232. },
  4233. err: nil,
  4234. },
  4235. },
  4236. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4237. },
  4238. },
  4239. },
  4240. }
  4241. for _, tc := range cases {
  4242. t.Run(tc.name, func(t *testing.T) {
  4243. ctx := context.Background()
  4244. fbMgr := newMockFeedbagManager(t)
  4245. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  4246. fbMgr.EXPECT().
  4247. Feedbag(ctx, params.screenName).
  4248. Return(params.results, params.err)
  4249. }
  4250. fbSvc := newMockFeedbagService(t)
  4251. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  4252. fbSvc.EXPECT().
  4253. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  4254. Return(params.msg, params.err)
  4255. }
  4256. var randCall int
  4257. randIntn := func(n int) int {
  4258. randCall++
  4259. return randCall
  4260. }
  4261. svc := OSCARProxy{
  4262. Logger: slog.Default(),
  4263. FeedbagManager: fbMgr,
  4264. FeedbagService: fbSvc,
  4265. RandIntn: randIntn,
  4266. }
  4267. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  4268. assert.Equal(t, tc.wantMsg, msg)
  4269. })
  4270. }
  4271. }
  4272. func TestOSCARProxy_RecvClientCmd_AddDeny2(t *testing.T) {
  4273. cases := []struct {
  4274. name string
  4275. me *state.SessionInstance
  4276. givenCmd []byte
  4277. wantMsg []string
  4278. mockParams mockParams
  4279. }{
  4280. {
  4281. name: "no screennames provided returns internal error",
  4282. me: newTestSession("me"),
  4283. givenCmd: []byte("toc2_add_deny"),
  4284. wantMsg: []string{cmdInternalSvcErr},
  4285. mockParams: mockParams{},
  4286. },
  4287. {
  4288. name: "FeedbagManager.Feedbag error returns internal error",
  4289. me: newTestSession("me"),
  4290. givenCmd: []byte("toc2_add_deny alice"),
  4291. wantMsg: []string{cmdInternalSvcErr},
  4292. mockParams: mockParams{
  4293. feedBagParams: feedBagParams{
  4294. feedbagParams: feedbagParams{
  4295. {
  4296. screenName: state.NewIdentScreenName("me"),
  4297. results: nil,
  4298. err: errors.New("EOF"),
  4299. },
  4300. },
  4301. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4302. },
  4303. },
  4304. },
  4305. {
  4306. name: "FeedbagService.UpsertItem error returns internal error",
  4307. me: newTestSession("me"),
  4308. givenCmd: []byte("toc2_add_deny alice"),
  4309. wantMsg: []string{cmdInternalSvcErr},
  4310. mockParams: mockParams{
  4311. feedBagParams: feedBagParams{
  4312. feedbagParams: feedbagParams{
  4313. {
  4314. screenName: state.NewIdentScreenName("me"),
  4315. results: []wire.FeedbagItem{},
  4316. err: nil,
  4317. },
  4318. },
  4319. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4320. {
  4321. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4322. items: []wire.FeedbagItem{
  4323. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4324. },
  4325. msg: nil, err: errors.New("EOF"),
  4326. },
  4327. },
  4328. },
  4329. },
  4330. },
  4331. {
  4332. name: "add deny to empty feedbag",
  4333. me: newTestSession("me"),
  4334. givenCmd: []byte("toc2_add_deny alice"),
  4335. wantMsg: []string{},
  4336. mockParams: mockParams{
  4337. feedBagParams: feedBagParams{
  4338. feedbagParams: feedbagParams{
  4339. {
  4340. screenName: state.NewIdentScreenName("me"),
  4341. results: []wire.FeedbagItem{},
  4342. err: nil,
  4343. },
  4344. },
  4345. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4346. {
  4347. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4348. items: []wire.FeedbagItem{
  4349. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4350. },
  4351. msg: nil, err: nil,
  4352. },
  4353. },
  4354. },
  4355. },
  4356. },
  4357. {
  4358. name: "skip existing deny, add only new",
  4359. me: newTestSession("me"),
  4360. givenCmd: []byte("toc2_add_deny alice bob"),
  4361. wantMsg: []string{},
  4362. mockParams: mockParams{
  4363. feedBagParams: feedBagParams{
  4364. feedbagParams: feedbagParams{
  4365. {
  4366. screenName: state.NewIdentScreenName("me"),
  4367. results: []wire.FeedbagItem{
  4368. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4369. },
  4370. err: nil,
  4371. },
  4372. },
  4373. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4374. {
  4375. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagInsertItem},
  4376. items: []wire.FeedbagItem{
  4377. {ItemID: 2, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4378. },
  4379. msg: nil, err: nil,
  4380. },
  4381. },
  4382. },
  4383. },
  4384. },
  4385. {
  4386. name: "no-op when deny already in feedbag",
  4387. me: newTestSession("me"),
  4388. givenCmd: []byte("toc2_add_deny alice"),
  4389. wantMsg: []string{},
  4390. mockParams: mockParams{
  4391. feedBagParams: feedBagParams{
  4392. feedbagParams: feedbagParams{
  4393. {
  4394. screenName: state.NewIdentScreenName("me"),
  4395. results: []wire.FeedbagItem{
  4396. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4397. },
  4398. err: nil,
  4399. },
  4400. },
  4401. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4402. },
  4403. },
  4404. },
  4405. }
  4406. for _, tc := range cases {
  4407. t.Run(tc.name, func(t *testing.T) {
  4408. ctx := context.Background()
  4409. fbMgr := newMockFeedbagManager(t)
  4410. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  4411. fbMgr.EXPECT().
  4412. Feedbag(ctx, params.screenName).
  4413. Return(params.results, params.err)
  4414. }
  4415. fbSvc := newMockFeedbagService(t)
  4416. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  4417. fbSvc.EXPECT().
  4418. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  4419. Return(params.msg, params.err)
  4420. }
  4421. var randCall int
  4422. randIntn := func(n int) int {
  4423. randCall++
  4424. return randCall
  4425. }
  4426. svc := OSCARProxy{
  4427. Logger: slog.Default(),
  4428. FeedbagManager: fbMgr,
  4429. FeedbagService: fbSvc,
  4430. RandIntn: randIntn,
  4431. }
  4432. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  4433. assert.Equal(t, tc.wantMsg, msg)
  4434. })
  4435. }
  4436. }
  4437. func TestOSCARProxy_RecvClientCmd_RemoveBuddy2(t *testing.T) {
  4438. cases := []struct {
  4439. name string
  4440. me *state.SessionInstance
  4441. givenCmd []byte
  4442. wantMsg []string
  4443. mockParams mockParams
  4444. }{
  4445. {
  4446. name: "removes buddy from group order TLV and upserts group",
  4447. me: newTestSession("me"),
  4448. givenCmd: []byte("toc2_remove_buddy friend2 Buddies"),
  4449. wantMsg: []string{},
  4450. mockParams: mockParams{
  4451. feedBagParams: feedBagParams{
  4452. feedbagParams: feedbagParams{
  4453. {
  4454. screenName: state.NewIdentScreenName("me"),
  4455. results: []wire.FeedbagItem{
  4456. {
  4457. Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  4458. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 2, 3})}},
  4459. },
  4460. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "friend1", TLVLBlock: wire.TLVLBlock{}},
  4461. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "friend2", TLVLBlock: wire.TLVLBlock{}},
  4462. {ItemID: 3, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "friend3", TLVLBlock: wire.TLVLBlock{}},
  4463. },
  4464. err: nil,
  4465. },
  4466. },
  4467. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4468. {
  4469. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4470. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4471. Items: []wire.FeedbagItem{
  4472. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "friend2", TLVLBlock: wire.TLVLBlock{}},
  4473. },
  4474. },
  4475. msg: nil, err: nil,
  4476. },
  4477. },
  4478. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{
  4479. {
  4480. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagUpdateItem},
  4481. items: []wire.FeedbagItem{
  4482. {
  4483. Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  4484. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 3})}},
  4485. },
  4486. },
  4487. msg: nil, err: nil,
  4488. },
  4489. },
  4490. },
  4491. },
  4492. },
  4493. {
  4494. name: "parseArgs error returns internal error",
  4495. me: newTestSession("me"),
  4496. givenCmd: []byte("toc2_remove_buddy \"unclosed"),
  4497. wantMsg: []string{cmdInternalSvcErr},
  4498. mockParams: mockParams{},
  4499. },
  4500. {
  4501. name: "missing params returns internal error",
  4502. me: newTestSession("me"),
  4503. givenCmd: []byte("toc2_remove_buddy friend1"),
  4504. wantMsg: []string{cmdInternalSvcErr},
  4505. mockParams: mockParams{},
  4506. },
  4507. {
  4508. name: "group not found returns internal error",
  4509. me: newTestSession("me"),
  4510. givenCmd: []byte("toc2_remove_buddy friend1 NoSuchGroup"),
  4511. wantMsg: []string{cmdInternalSvcErr},
  4512. mockParams: mockParams{
  4513. feedBagParams: feedBagParams{
  4514. feedbagParams: feedbagParams{
  4515. {
  4516. screenName: state.NewIdentScreenName("me"),
  4517. results: []wire.FeedbagItem{
  4518. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup},
  4519. },
  4520. err: nil,
  4521. },
  4522. },
  4523. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4524. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4525. },
  4526. },
  4527. },
  4528. {
  4529. name: "FeedbagManager.Feedbag error returns internal error",
  4530. me: newTestSession("me"),
  4531. givenCmd: []byte("toc2_remove_buddy friend1 Buddies"),
  4532. wantMsg: []string{cmdInternalSvcErr},
  4533. mockParams: mockParams{
  4534. feedBagParams: feedBagParams{
  4535. feedbagParams: feedbagParams{
  4536. {
  4537. screenName: state.NewIdentScreenName("me"),
  4538. results: nil,
  4539. err: errors.New("EOF"),
  4540. },
  4541. },
  4542. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4543. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4544. },
  4545. },
  4546. },
  4547. {
  4548. name: "no-op when buddy not in group",
  4549. me: newTestSession("me"),
  4550. givenCmd: []byte("toc2_remove_buddy stranger Buddies"),
  4551. wantMsg: []string{},
  4552. mockParams: mockParams{
  4553. feedBagParams: feedBagParams{
  4554. feedbagParams: feedbagParams{
  4555. {
  4556. screenName: state.NewIdentScreenName("me"),
  4557. results: []wire.FeedbagItem{
  4558. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  4559. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}}},
  4560. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "friend1", TLVLBlock: wire.TLVLBlock{}},
  4561. },
  4562. err: nil,
  4563. },
  4564. },
  4565. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4566. feedbagServiceUpsertItemParams: feedbagServiceUpsertItemParams{},
  4567. },
  4568. },
  4569. },
  4570. }
  4571. for _, tc := range cases {
  4572. t.Run(tc.name, func(t *testing.T) {
  4573. ctx := context.Background()
  4574. fbMgr := newMockFeedbagManager(t)
  4575. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  4576. fbMgr.EXPECT().
  4577. Feedbag(ctx, params.screenName).
  4578. Return(params.results, params.err)
  4579. }
  4580. fbSvc := newMockFeedbagService(t)
  4581. for _, params := range tc.mockParams.feedBagParams.feedbagServiceDeleteItemParams {
  4582. fbSvc.EXPECT().
  4583. DeleteItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.inBody).
  4584. Return(params.msg, params.err)
  4585. }
  4586. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUpsertItemParams {
  4587. fbSvc.EXPECT().
  4588. UpsertItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.items).
  4589. Return(params.msg, params.err)
  4590. }
  4591. svc := OSCARProxy{
  4592. Logger: slog.Default(),
  4593. FeedbagManager: fbMgr,
  4594. FeedbagService: fbSvc,
  4595. }
  4596. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  4597. assert.Equal(t, tc.wantMsg, msg)
  4598. })
  4599. }
  4600. }
  4601. func TestOSCARProxy_RecvClientCmd_RemovePermit2(t *testing.T) {
  4602. cases := []struct {
  4603. name string
  4604. me *state.SessionInstance
  4605. givenCmd []byte
  4606. wantMsg []string
  4607. mockParams mockParams
  4608. }{
  4609. {
  4610. name: "no screennames provided returns internal error",
  4611. me: newTestSession("me"),
  4612. givenCmd: []byte("toc2_remove_permit"),
  4613. wantMsg: []string{cmdInternalSvcErr},
  4614. mockParams: mockParams{},
  4615. },
  4616. {
  4617. name: "FeedbagManager.Feedbag error returns internal error",
  4618. me: newTestSession("me"),
  4619. givenCmd: []byte("toc2_remove_permit alice"),
  4620. wantMsg: []string{cmdInternalSvcErr},
  4621. mockParams: mockParams{
  4622. feedBagParams: feedBagParams{
  4623. feedbagParams: feedbagParams{
  4624. {
  4625. screenName: state.NewIdentScreenName("me"),
  4626. results: nil,
  4627. err: errors.New("EOF"),
  4628. },
  4629. },
  4630. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4631. },
  4632. },
  4633. },
  4634. {
  4635. name: "FeedbagService.DeleteItem error returns internal error",
  4636. me: newTestSession("me"),
  4637. givenCmd: []byte("toc2_remove_permit alice"),
  4638. wantMsg: []string{cmdInternalSvcErr},
  4639. mockParams: mockParams{
  4640. feedBagParams: feedBagParams{
  4641. feedbagParams: feedbagParams{
  4642. {
  4643. screenName: state.NewIdentScreenName("me"),
  4644. results: []wire.FeedbagItem{
  4645. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4646. },
  4647. err: nil,
  4648. },
  4649. },
  4650. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4651. {
  4652. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4653. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4654. Items: []wire.FeedbagItem{
  4655. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4656. },
  4657. },
  4658. msg: nil, err: errors.New("EOF"),
  4659. },
  4660. },
  4661. },
  4662. },
  4663. },
  4664. {
  4665. name: "remove permits from feedbag",
  4666. me: newTestSession("me"),
  4667. givenCmd: []byte("toc2_remove_permit alice bob"),
  4668. wantMsg: []string{},
  4669. mockParams: mockParams{
  4670. feedBagParams: feedBagParams{
  4671. feedbagParams: feedbagParams{
  4672. {
  4673. screenName: state.NewIdentScreenName("me"),
  4674. results: []wire.FeedbagItem{
  4675. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4676. {ItemID: 2, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4677. },
  4678. err: nil,
  4679. },
  4680. },
  4681. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4682. {
  4683. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4684. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4685. Items: []wire.FeedbagItem{
  4686. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4687. {ItemID: 2, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4688. },
  4689. },
  4690. msg: nil, err: nil,
  4691. },
  4692. },
  4693. },
  4694. },
  4695. },
  4696. {
  4697. name: "remove only existing permits",
  4698. me: newTestSession("me"),
  4699. givenCmd: []byte("toc2_remove_permit alice bob"),
  4700. wantMsg: []string{},
  4701. mockParams: mockParams{
  4702. feedBagParams: feedBagParams{
  4703. feedbagParams: feedbagParams{
  4704. {
  4705. screenName: state.NewIdentScreenName("me"),
  4706. results: []wire.FeedbagItem{
  4707. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4708. },
  4709. err: nil,
  4710. },
  4711. },
  4712. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4713. {
  4714. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4715. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4716. Items: []wire.FeedbagItem{
  4717. {ItemID: 1, ClassID: wire.FeedbagClassIDPermit, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4718. },
  4719. },
  4720. msg: nil, err: nil,
  4721. },
  4722. },
  4723. },
  4724. },
  4725. },
  4726. {
  4727. name: "no-op when no permits in feedbag",
  4728. me: newTestSession("me"),
  4729. givenCmd: []byte("toc2_remove_permit alice"),
  4730. wantMsg: []string{},
  4731. mockParams: mockParams{
  4732. feedBagParams: feedBagParams{
  4733. feedbagParams: feedbagParams{
  4734. {
  4735. screenName: state.NewIdentScreenName("me"),
  4736. results: []wire.FeedbagItem{},
  4737. err: nil,
  4738. },
  4739. },
  4740. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4741. },
  4742. },
  4743. },
  4744. }
  4745. for _, tc := range cases {
  4746. t.Run(tc.name, func(t *testing.T) {
  4747. ctx := context.Background()
  4748. fbMgr := newMockFeedbagManager(t)
  4749. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  4750. fbMgr.EXPECT().
  4751. Feedbag(ctx, params.screenName).
  4752. Return(params.results, params.err)
  4753. }
  4754. fbSvc := newMockFeedbagService(t)
  4755. for _, params := range tc.mockParams.feedBagParams.feedbagServiceDeleteItemParams {
  4756. fbSvc.EXPECT().
  4757. DeleteItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.inBody).
  4758. Return(params.msg, params.err)
  4759. }
  4760. svc := OSCARProxy{
  4761. Logger: slog.Default(),
  4762. FeedbagManager: fbMgr,
  4763. FeedbagService: fbSvc,
  4764. }
  4765. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  4766. assert.Equal(t, tc.wantMsg, msg)
  4767. })
  4768. }
  4769. }
  4770. func TestOSCARProxy_RecvClientCmd_RemoveDeny2(t *testing.T) {
  4771. cases := []struct {
  4772. name string
  4773. me *state.SessionInstance
  4774. givenCmd []byte
  4775. wantMsg []string
  4776. mockParams mockParams
  4777. }{
  4778. {
  4779. name: "no screennames provided returns internal error",
  4780. me: newTestSession("me"),
  4781. givenCmd: []byte("toc2_remove_deny"),
  4782. wantMsg: []string{cmdInternalSvcErr},
  4783. mockParams: mockParams{},
  4784. },
  4785. {
  4786. name: "FeedbagManager.Feedbag error returns internal error",
  4787. me: newTestSession("me"),
  4788. givenCmd: []byte("toc2_remove_deny alice"),
  4789. wantMsg: []string{cmdInternalSvcErr},
  4790. mockParams: mockParams{
  4791. feedBagParams: feedBagParams{
  4792. feedbagParams: feedbagParams{
  4793. {
  4794. screenName: state.NewIdentScreenName("me"),
  4795. results: nil,
  4796. err: errors.New("EOF"),
  4797. },
  4798. },
  4799. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4800. },
  4801. },
  4802. },
  4803. {
  4804. name: "FeedbagService.DeleteItem error returns internal error",
  4805. me: newTestSession("me"),
  4806. givenCmd: []byte("toc2_remove_deny alice"),
  4807. wantMsg: []string{cmdInternalSvcErr},
  4808. mockParams: mockParams{
  4809. feedBagParams: feedBagParams{
  4810. feedbagParams: feedbagParams{
  4811. {
  4812. screenName: state.NewIdentScreenName("me"),
  4813. results: []wire.FeedbagItem{
  4814. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4815. },
  4816. err: nil,
  4817. },
  4818. },
  4819. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4820. {
  4821. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4822. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4823. Items: []wire.FeedbagItem{
  4824. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4825. },
  4826. },
  4827. msg: nil, err: errors.New("EOF"),
  4828. },
  4829. },
  4830. },
  4831. },
  4832. },
  4833. {
  4834. name: "remove denies from feedbag",
  4835. me: newTestSession("me"),
  4836. givenCmd: []byte("toc2_remove_deny alice bob"),
  4837. wantMsg: []string{},
  4838. mockParams: mockParams{
  4839. feedBagParams: feedBagParams{
  4840. feedbagParams: feedbagParams{
  4841. {
  4842. screenName: state.NewIdentScreenName("me"),
  4843. results: []wire.FeedbagItem{
  4844. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4845. {ItemID: 2, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4846. },
  4847. err: nil,
  4848. },
  4849. },
  4850. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4851. {
  4852. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4853. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4854. Items: []wire.FeedbagItem{
  4855. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4856. {ItemID: 2, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "bob", TLVLBlock: wire.TLVLBlock{}},
  4857. },
  4858. },
  4859. msg: nil, err: nil,
  4860. },
  4861. },
  4862. },
  4863. },
  4864. },
  4865. {
  4866. name: "remove only existing denies",
  4867. me: newTestSession("me"),
  4868. givenCmd: []byte("toc2_remove_deny alice bob"),
  4869. wantMsg: []string{},
  4870. mockParams: mockParams{
  4871. feedBagParams: feedBagParams{
  4872. feedbagParams: feedbagParams{
  4873. {
  4874. screenName: state.NewIdentScreenName("me"),
  4875. results: []wire.FeedbagItem{
  4876. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4877. },
  4878. err: nil,
  4879. },
  4880. },
  4881. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{
  4882. {
  4883. frame: wire.SNACFrame{FoodGroup: wire.Feedbag, SubGroup: wire.FeedbagDeleteItem},
  4884. inBody: wire.SNAC_0x13_0x0A_FeedbagDeleteItem{
  4885. Items: []wire.FeedbagItem{
  4886. {ItemID: 1, ClassID: wire.FeedbagClassIDDeny, GroupID: 0, Name: "alice", TLVLBlock: wire.TLVLBlock{}},
  4887. },
  4888. },
  4889. msg: nil, err: nil,
  4890. },
  4891. },
  4892. },
  4893. },
  4894. },
  4895. {
  4896. name: "no-op when no denies in feedbag",
  4897. me: newTestSession("me"),
  4898. givenCmd: []byte("toc2_remove_deny alice"),
  4899. wantMsg: []string{},
  4900. mockParams: mockParams{
  4901. feedBagParams: feedBagParams{
  4902. feedbagParams: feedbagParams{
  4903. {
  4904. screenName: state.NewIdentScreenName("me"),
  4905. results: []wire.FeedbagItem{},
  4906. err: nil,
  4907. },
  4908. },
  4909. feedbagServiceDeleteItemParams: feedbagServiceDeleteItemParams{},
  4910. },
  4911. },
  4912. },
  4913. }
  4914. for _, tc := range cases {
  4915. t.Run(tc.name, func(t *testing.T) {
  4916. ctx := context.Background()
  4917. fbMgr := newMockFeedbagManager(t)
  4918. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  4919. fbMgr.EXPECT().
  4920. Feedbag(ctx, params.screenName).
  4921. Return(params.results, params.err)
  4922. }
  4923. fbSvc := newMockFeedbagService(t)
  4924. for _, params := range tc.mockParams.feedBagParams.feedbagServiceDeleteItemParams {
  4925. fbSvc.EXPECT().
  4926. DeleteItem(ctx, matchSession(tc.me.IdentScreenName()), params.frame, params.inBody).
  4927. Return(params.msg, params.err)
  4928. }
  4929. svc := OSCARProxy{
  4930. Logger: slog.Default(),
  4931. FeedbagManager: fbMgr,
  4932. FeedbagService: fbSvc,
  4933. }
  4934. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  4935. assert.Equal(t, tc.wantMsg, msg)
  4936. })
  4937. }
  4938. }
  4939. func TestOSCARProxy_RecvClientCmd_RvousAccept(t *testing.T) {
  4940. cases := []struct {
  4941. // name is the unit test name
  4942. name string
  4943. // me is the TOC user session
  4944. me *state.SessionInstance
  4945. // givenCmd is the TOC command
  4946. givenCmd []byte
  4947. // wantMsg is the expected TOC response
  4948. wantMsg []string
  4949. // mockParams is the list of params sent to mocks that satisfy this
  4950. // method's dependencies
  4951. mockParams mockParams
  4952. }{
  4953. {
  4954. name: "successfully send rendezvous request",
  4955. me: newTestSession("me"),
  4956. givenCmd: []byte(`toc_rvous_accept them aGFoYWhhaGE= 09461343-4C7F-11D1-8222-444553540000`),
  4957. mockParams: mockParams{
  4958. icbmParams: icbmParams{
  4959. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  4960. {
  4961. sender: state.NewIdentScreenName("me"),
  4962. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  4963. ChannelID: wire.ICBMChannelRendezvous,
  4964. ScreenName: "them",
  4965. TLVRestBlock: wire.TLVRestBlock{
  4966. TLVList: wire.TLVList{
  4967. wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
  4968. Type: wire.ICBMRdvMessageAccept,
  4969. Cookie: [8]byte{'h', 'a', 'h', 'a', 'h', 'a', 'h', 'a'},
  4970. Capability: wire.CapFileTransfer,
  4971. }),
  4972. },
  4973. },
  4974. },
  4975. },
  4976. },
  4977. },
  4978. },
  4979. wantMsg: []string{},
  4980. },
  4981. {
  4982. name: "send rendezvous request, receive error from ICBM service",
  4983. me: newTestSession("me"),
  4984. givenCmd: []byte(`toc_rvous_accept them aGFoYWhhaGE= 09461343-4C7F-11D1-8222-444553540000`),
  4985. mockParams: mockParams{
  4986. icbmParams: icbmParams{
  4987. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  4988. {
  4989. sender: state.NewIdentScreenName("me"),
  4990. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  4991. ChannelID: wire.ICBMChannelRendezvous,
  4992. ScreenName: "them",
  4993. TLVRestBlock: wire.TLVRestBlock{
  4994. TLVList: wire.TLVList{
  4995. wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
  4996. Type: wire.ICBMRdvMessageAccept,
  4997. Cookie: [8]byte{'h', 'a', 'h', 'a', 'h', 'a', 'h', 'a'},
  4998. Capability: wire.CapFileTransfer,
  4999. }),
  5000. },
  5001. },
  5002. },
  5003. err: io.EOF,
  5004. },
  5005. },
  5006. },
  5007. },
  5008. wantMsg: []string{cmdInternalSvcErr},
  5009. },
  5010. {
  5011. name: "bad command",
  5012. me: newTestSession("me"),
  5013. givenCmd: []byte(`toc_rvous_accept`),
  5014. wantMsg: []string{cmdInternalSvcErr},
  5015. },
  5016. }
  5017. for _, tc := range cases {
  5018. t.Run(tc.name, func(t *testing.T) {
  5019. ctx := context.Background()
  5020. icbmSvc := newMockICBMService(t)
  5021. for _, params := range tc.mockParams.channelMsgToHostParamsICBM {
  5022. icbmSvc.EXPECT().
  5023. ChannelMsgToHost(ctx, matchSession(params.sender), params.inFrame, params.inBody).
  5024. Return(params.result, params.err)
  5025. }
  5026. svc := OSCARProxy{
  5027. Logger: slog.Default(),
  5028. ICBMService: icbmSvc,
  5029. }
  5030. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5031. assert.Equal(t, tc.wantMsg, msg)
  5032. })
  5033. }
  5034. }
  5035. func TestOSCARProxy_RecvClientCmd_RvousCancel(t *testing.T) {
  5036. cases := []struct {
  5037. // name is the unit test name
  5038. name string
  5039. // me is the TOC user session
  5040. me *state.SessionInstance
  5041. // givenCmd is the TOC command
  5042. givenCmd []byte
  5043. // wantMsg is the expected TOC response
  5044. wantMsg []string
  5045. // mockParams is the list of params sent to mocks that satisfy this
  5046. // method's dependencies
  5047. mockParams mockParams
  5048. }{
  5049. {
  5050. name: "successfully send rendezvous cancellation",
  5051. me: newTestSession("me"),
  5052. givenCmd: []byte(`toc_rvous_cancel them aGFoYWhhaGE= 09461343-4C7F-11D1-8222-444553540000`),
  5053. mockParams: mockParams{
  5054. icbmParams: icbmParams{
  5055. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5056. {
  5057. sender: state.NewIdentScreenName("me"),
  5058. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5059. ChannelID: wire.ICBMChannelRendezvous,
  5060. ScreenName: "them",
  5061. TLVRestBlock: wire.TLVRestBlock{
  5062. TLVList: wire.TLVList{
  5063. wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
  5064. Type: wire.ICBMRdvMessageCancel,
  5065. Cookie: [8]byte{'h', 'a', 'h', 'a', 'h', 'a', 'h', 'a'},
  5066. Capability: wire.CapFileTransfer,
  5067. TLVRestBlock: wire.TLVRestBlock{
  5068. TLVList: wire.TLVList{
  5069. wire.NewTLVBE(wire.ICBMRdvTLVTagsCancelReason, wire.ICBMRdvCancelReasonsUserCancel),
  5070. },
  5071. },
  5072. }),
  5073. },
  5074. },
  5075. },
  5076. },
  5077. },
  5078. },
  5079. },
  5080. wantMsg: []string{},
  5081. },
  5082. {
  5083. name: "send rendezvous cancellation, receive error from ICBM service",
  5084. me: newTestSession("me"),
  5085. givenCmd: []byte(`toc_rvous_cancel them aGFoYWhhaGE= 09461343-4C7F-11D1-8222-444553540000`),
  5086. mockParams: mockParams{
  5087. icbmParams: icbmParams{
  5088. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5089. {
  5090. sender: state.NewIdentScreenName("me"),
  5091. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5092. ChannelID: wire.ICBMChannelRendezvous,
  5093. ScreenName: "them",
  5094. TLVRestBlock: wire.TLVRestBlock{
  5095. TLVList: wire.TLVList{
  5096. wire.NewTLVBE(wire.ICBMTLVData, wire.ICBMCh2Fragment{
  5097. Type: wire.ICBMRdvMessageCancel,
  5098. Cookie: [8]byte{'h', 'a', 'h', 'a', 'h', 'a', 'h', 'a'},
  5099. Capability: wire.CapFileTransfer,
  5100. TLVRestBlock: wire.TLVRestBlock{
  5101. TLVList: wire.TLVList{
  5102. wire.NewTLVBE(wire.ICBMRdvTLVTagsCancelReason, wire.ICBMRdvCancelReasonsUserCancel),
  5103. },
  5104. },
  5105. }),
  5106. },
  5107. },
  5108. },
  5109. err: io.EOF,
  5110. },
  5111. },
  5112. },
  5113. },
  5114. wantMsg: []string{cmdInternalSvcErr},
  5115. },
  5116. {
  5117. name: "bad command",
  5118. me: newTestSession("me"),
  5119. givenCmd: []byte(`toc_rvous_cancel`),
  5120. wantMsg: []string{cmdInternalSvcErr},
  5121. },
  5122. }
  5123. for _, tc := range cases {
  5124. t.Run(tc.name, func(t *testing.T) {
  5125. ctx := context.Background()
  5126. icbmSvc := newMockICBMService(t)
  5127. for _, params := range tc.mockParams.channelMsgToHostParamsICBM {
  5128. icbmSvc.EXPECT().
  5129. ChannelMsgToHost(ctx, matchSession(params.sender), params.inFrame, params.inBody).
  5130. Return(params.result, params.err)
  5131. }
  5132. svc := OSCARProxy{
  5133. Logger: slog.Default(),
  5134. ICBMService: icbmSvc,
  5135. }
  5136. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5137. assert.Equal(t, tc.wantMsg, msg)
  5138. })
  5139. }
  5140. }
  5141. func TestOSCARProxy_RecvClientCmd_SendIM(t *testing.T) {
  5142. cases := []struct {
  5143. // name is the unit test name
  5144. name string
  5145. // me is the TOC user session
  5146. me *state.SessionInstance
  5147. // givenCmd is the TOC command
  5148. givenCmd []byte
  5149. // wantMsg is the expected TOC response
  5150. wantMsg []string
  5151. // mockParams is the list of params sent to mocks that satisfy this
  5152. // method's dependencies
  5153. mockParams mockParams
  5154. }{
  5155. {
  5156. name: "successfully send instant message",
  5157. me: newTestSession("me"),
  5158. givenCmd: []byte(`toc_send_im chattingChuck "hello world! :\)"`),
  5159. mockParams: mockParams{
  5160. icbmParams: icbmParams{
  5161. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5162. {
  5163. sender: state.NewIdentScreenName("me"),
  5164. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5165. ChannelID: wire.ICBMChannelIM,
  5166. ScreenName: "chattingChuck",
  5167. TLVRestBlock: wire.TLVRestBlock{
  5168. TLVList: wire.TLVList{
  5169. wire.NewTLVBE(wire.ICBMTLVAOLIMData, []wire.ICBMCh1Fragment{
  5170. {
  5171. ID: 5,
  5172. Version: 1,
  5173. Payload: []byte{1, 1, 2},
  5174. },
  5175. {
  5176. ID: 1,
  5177. Version: 1,
  5178. Payload: []byte{
  5179. 0x00, 0x00,
  5180. 0x00, 0x00,
  5181. 'h', 'e', 'l', 'l', 'o', ' ', 'w', 'o', 'r', 'l', 'd', '!', ' ', ':', ')',
  5182. },
  5183. },
  5184. }),
  5185. },
  5186. },
  5187. },
  5188. },
  5189. },
  5190. },
  5191. },
  5192. wantMsg: []string{},
  5193. },
  5194. {
  5195. name: "successfully auto-reply send instant message",
  5196. me: newTestSession("me"),
  5197. givenCmd: []byte(`toc_send_im chattingChuck "hello world!" auto`),
  5198. mockParams: mockParams{
  5199. icbmParams: icbmParams{
  5200. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5201. {
  5202. sender: state.NewIdentScreenName("me"),
  5203. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5204. ChannelID: wire.ICBMChannelIM,
  5205. ScreenName: "chattingChuck",
  5206. TLVRestBlock: wire.TLVRestBlock{
  5207. TLVList: wire.TLVList{
  5208. wire.NewTLVBE(wire.ICBMTLVAOLIMData, []wire.ICBMCh1Fragment{
  5209. {
  5210. ID: 5,
  5211. Version: 1,
  5212. Payload: []byte{1, 1, 2},
  5213. },
  5214. {
  5215. ID: 1,
  5216. Version: 1,
  5217. Payload: []byte{
  5218. 0x00, 0x00,
  5219. 0x00, 0x00,
  5220. 'h', 'e', 'l', 'l', 'o', ' ', 'w', 'o', 'r', 'l', 'd', '!',
  5221. },
  5222. },
  5223. }),
  5224. wire.NewTLVBE(wire.ICBMTLVAutoResponse, []byte{}),
  5225. },
  5226. },
  5227. },
  5228. },
  5229. },
  5230. },
  5231. },
  5232. wantMsg: []string{},
  5233. },
  5234. {
  5235. name: "send instant message, receive error from ICBM service",
  5236. me: newTestSession("me"),
  5237. givenCmd: []byte(`toc_send_im chattingChuck "hello world!"`),
  5238. mockParams: mockParams{
  5239. icbmParams: icbmParams{
  5240. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5241. {
  5242. sender: state.NewIdentScreenName("me"),
  5243. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5244. ChannelID: wire.ICBMChannelIM,
  5245. ScreenName: "chattingChuck",
  5246. TLVRestBlock: wire.TLVRestBlock{
  5247. TLVList: wire.TLVList{
  5248. wire.NewTLVBE(wire.ICBMTLVAOLIMData, []wire.ICBMCh1Fragment{
  5249. {
  5250. ID: 5,
  5251. Version: 1,
  5252. Payload: []byte{1, 1, 2},
  5253. },
  5254. {
  5255. ID: 1,
  5256. Version: 1,
  5257. Payload: []byte{
  5258. 0x00, 0x00,
  5259. 0x00, 0x00,
  5260. 'h', 'e', 'l', 'l', 'o', ' ', 'w', 'o', 'r', 'l', 'd', '!',
  5261. },
  5262. },
  5263. }),
  5264. },
  5265. },
  5266. },
  5267. err: io.EOF,
  5268. },
  5269. },
  5270. },
  5271. },
  5272. wantMsg: []string{cmdInternalSvcErr},
  5273. },
  5274. {
  5275. name: "bad command",
  5276. me: newTestSession("me"),
  5277. givenCmd: []byte(`toc_send_im`),
  5278. wantMsg: []string{cmdInternalSvcErr},
  5279. },
  5280. }
  5281. for _, tc := range cases {
  5282. t.Run(tc.name, func(t *testing.T) {
  5283. ctx := context.Background()
  5284. icbmSvc := newMockICBMService(t)
  5285. for _, params := range tc.mockParams.channelMsgToHostParamsICBM {
  5286. icbmSvc.EXPECT().
  5287. ChannelMsgToHost(ctx, matchSession(params.sender), params.inFrame, params.inBody).
  5288. Return(params.result, params.err)
  5289. }
  5290. svc := OSCARProxy{
  5291. Logger: slog.Default(),
  5292. ICBMService: icbmSvc,
  5293. }
  5294. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5295. assert.Equal(t, tc.wantMsg, msg)
  5296. })
  5297. }
  5298. }
  5299. func TestOSCARProxy_RecvClientCmd_SendIMEnc(t *testing.T) {
  5300. cases := []struct {
  5301. name string
  5302. me *state.SessionInstance
  5303. givenCmd []byte
  5304. wantMsg []string
  5305. mockParams mockParams
  5306. }{
  5307. {
  5308. name: "successfully send encoded instant message",
  5309. me: newTestSession("me"),
  5310. givenCmd: []byte(`toc2_send_im_enc chattingChuck "F" utf-8 en "hello"`),
  5311. mockParams: mockParams{
  5312. icbmParams: icbmParams{
  5313. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5314. {
  5315. sender: state.NewIdentScreenName("me"),
  5316. inFrame: wire.SNACFrame{},
  5317. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5318. ChannelID: wire.ICBMChannelIM,
  5319. ScreenName: "chattingChuck",
  5320. TLVRestBlock: wire.TLVRestBlock{
  5321. TLVList: wire.TLVList{
  5322. wire.NewTLVBE(wire.ICBMTLVAOLIMData, []wire.ICBMCh1Fragment{
  5323. {
  5324. ID: 5,
  5325. Version: 1,
  5326. Payload: []byte{1, 1, 2},
  5327. },
  5328. {
  5329. ID: 1,
  5330. Version: 1,
  5331. Payload: []byte{
  5332. 0x00, 0x00,
  5333. 0x00, 0x00,
  5334. 'h', 'e', 'l', 'l', 'o',
  5335. },
  5336. },
  5337. }),
  5338. },
  5339. },
  5340. },
  5341. },
  5342. },
  5343. },
  5344. },
  5345. wantMsg: []string{},
  5346. },
  5347. {
  5348. name: "successfully send encoded instant message with auto",
  5349. me: newTestSession("me"),
  5350. givenCmd: []byte(`toc2_send_im_enc chattingChuck "F" utf-8 en "hello" auto`),
  5351. mockParams: mockParams{
  5352. icbmParams: icbmParams{
  5353. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5354. {
  5355. sender: state.NewIdentScreenName("me"),
  5356. inFrame: wire.SNACFrame{},
  5357. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5358. ChannelID: wire.ICBMChannelIM,
  5359. ScreenName: "chattingChuck",
  5360. TLVRestBlock: wire.TLVRestBlock{
  5361. TLVList: wire.TLVList{
  5362. wire.NewTLVBE(wire.ICBMTLVAOLIMData, []wire.ICBMCh1Fragment{
  5363. {
  5364. ID: 5,
  5365. Version: 1,
  5366. Payload: []byte{1, 1, 2},
  5367. },
  5368. {
  5369. ID: 1,
  5370. Version: 1,
  5371. Payload: []byte{
  5372. 0x00, 0x00,
  5373. 0x00, 0x00,
  5374. 'h', 'e', 'l', 'l', 'o',
  5375. },
  5376. },
  5377. }),
  5378. wire.NewTLVBE(wire.ICBMTLVAutoResponse, []byte{}),
  5379. },
  5380. },
  5381. },
  5382. },
  5383. },
  5384. },
  5385. },
  5386. wantMsg: []string{},
  5387. },
  5388. {
  5389. name: "send encoded instant message, receive error from ICBM service",
  5390. me: newTestSession("me"),
  5391. givenCmd: []byte(`toc2_send_im_enc chattingChuck "F" utf-8 en "hello"`),
  5392. mockParams: mockParams{
  5393. icbmParams: icbmParams{
  5394. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{
  5395. {
  5396. sender: state.NewIdentScreenName("me"),
  5397. inFrame: wire.SNACFrame{},
  5398. inBody: wire.SNAC_0x04_0x06_ICBMChannelMsgToHost{
  5399. ChannelID: wire.ICBMChannelIM,
  5400. ScreenName: "chattingChuck",
  5401. TLVRestBlock: wire.TLVRestBlock{
  5402. TLVList: wire.TLVList{
  5403. wire.NewTLVBE(wire.ICBMTLVAOLIMData, []wire.ICBMCh1Fragment{
  5404. {
  5405. ID: 5,
  5406. Version: 1,
  5407. Payload: []byte{1, 1, 2},
  5408. },
  5409. {
  5410. ID: 1,
  5411. Version: 1,
  5412. Payload: []byte{
  5413. 0x00, 0x00,
  5414. 0x00, 0x00,
  5415. 'h', 'e', 'l', 'l', 'o',
  5416. },
  5417. },
  5418. }),
  5419. },
  5420. },
  5421. },
  5422. err: io.EOF,
  5423. },
  5424. },
  5425. },
  5426. },
  5427. wantMsg: []string{cmdInternalSvcErr},
  5428. },
  5429. {
  5430. name: "invalid args too few",
  5431. me: newTestSession("me"),
  5432. givenCmd: []byte(`toc2_send_im_enc`),
  5433. mockParams: mockParams{
  5434. icbmParams: icbmParams{
  5435. channelMsgToHostParamsICBM: channelMsgToHostParamsICBM{},
  5436. },
  5437. },
  5438. wantMsg: []string{cmdInternalSvcErr},
  5439. },
  5440. }
  5441. for _, tc := range cases {
  5442. t.Run(tc.name, func(t *testing.T) {
  5443. ctx := context.Background()
  5444. icbmSvc := newMockICBMService(t)
  5445. for _, params := range tc.mockParams.channelMsgToHostParamsICBM {
  5446. icbmSvc.EXPECT().
  5447. ChannelMsgToHost(ctx, matchSession(params.sender), params.inFrame, params.inBody).
  5448. Return(params.result, params.err)
  5449. }
  5450. svc := OSCARProxy{
  5451. Logger: slog.Default(),
  5452. ICBMService: icbmSvc,
  5453. }
  5454. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5455. assert.Equal(t, tc.wantMsg, msg)
  5456. })
  5457. }
  5458. }
  5459. func TestOSCARProxy_RecvClientCmd_SetAway(t *testing.T) {
  5460. cases := []struct {
  5461. // name is the unit test name
  5462. name string
  5463. // me is the TOC user session
  5464. me *state.SessionInstance
  5465. // givenCmd is the TOC command
  5466. givenCmd []byte
  5467. // wantMsg is the expected TOC response
  5468. wantMsg []string
  5469. // mockParams is the list of params sent to mocks that satisfy this
  5470. // method's dependencies
  5471. mockParams mockParams
  5472. }{
  5473. {
  5474. name: "successfully set away with message",
  5475. me: newTestSession("me"),
  5476. givenCmd: []byte(`toc_set_away "I'm away from my computer right now. :\)"`),
  5477. mockParams: mockParams{
  5478. locateParams: locateParams{
  5479. setInfoParams: setInfoParams{
  5480. {
  5481. me: state.NewIdentScreenName("me"),
  5482. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5483. TLVRestBlock: wire.TLVRestBlock{
  5484. TLVList: wire.TLVList{
  5485. wire.NewTLVBE(wire.LocateTLVTagsInfoUnavailableData, "I'm away from my computer right now. :)"),
  5486. },
  5487. },
  5488. },
  5489. },
  5490. },
  5491. },
  5492. },
  5493. wantMsg: []string{},
  5494. },
  5495. {
  5496. name: "successfully set away without message",
  5497. me: newTestSession("me"),
  5498. givenCmd: []byte(`toc_set_away`),
  5499. mockParams: mockParams{
  5500. locateParams: locateParams{
  5501. setInfoParams: setInfoParams{
  5502. {
  5503. me: state.NewIdentScreenName("me"),
  5504. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5505. TLVRestBlock: wire.TLVRestBlock{
  5506. TLVList: wire.TLVList{
  5507. wire.NewTLVBE(wire.LocateTLVTagsInfoUnavailableData, ""),
  5508. },
  5509. },
  5510. },
  5511. },
  5512. },
  5513. },
  5514. },
  5515. wantMsg: []string{},
  5516. },
  5517. {
  5518. name: "set away message, receive error from locate service",
  5519. me: newTestSession("me"),
  5520. givenCmd: []byte(`toc_set_away "I'm away from my computer right now."`),
  5521. mockParams: mockParams{
  5522. locateParams: locateParams{
  5523. setInfoParams: setInfoParams{
  5524. {
  5525. me: state.NewIdentScreenName("me"),
  5526. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5527. TLVRestBlock: wire.TLVRestBlock{
  5528. TLVList: wire.TLVList{
  5529. wire.NewTLVBE(wire.LocateTLVTagsInfoUnavailableData, "I'm away from my computer right now."),
  5530. },
  5531. },
  5532. },
  5533. err: io.EOF,
  5534. },
  5535. },
  5536. },
  5537. },
  5538. wantMsg: []string{cmdInternalSvcErr},
  5539. },
  5540. }
  5541. for _, tc := range cases {
  5542. t.Run(tc.name, func(t *testing.T) {
  5543. ctx := context.Background()
  5544. locateSvc := newMockLocateService(t)
  5545. for _, params := range tc.mockParams.setInfoParams {
  5546. locateSvc.EXPECT().
  5547. SetInfo(ctx, matchSession(params.me), params.inBody).
  5548. Return(params.err)
  5549. }
  5550. svc := OSCARProxy{
  5551. Logger: slog.Default(),
  5552. LocateService: locateSvc,
  5553. }
  5554. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5555. assert.Equal(t, tc.wantMsg, msg)
  5556. })
  5557. }
  5558. }
  5559. func TestOSCARProxy_RecvClientCmd_SetCaps(t *testing.T) {
  5560. cases := []struct {
  5561. // name is the unit test name
  5562. name string
  5563. // me is the TOC user session
  5564. me *state.SessionInstance
  5565. // givenCmd is the TOC command
  5566. givenCmd []byte
  5567. // wantMsg is the expected TOC response
  5568. wantMsg []string
  5569. // mockParams is the list of params sent to mocks that satisfy this
  5570. // method's dependencies
  5571. mockParams mockParams
  5572. }{
  5573. {
  5574. name: "successfully set capabilities",
  5575. me: newTestSession("me"),
  5576. givenCmd: []byte(`toc_set_caps 09460000-4C7F-11D1-8222-444553540000 09460001-4C7F-11D1-8222-444553540000`),
  5577. mockParams: mockParams{
  5578. locateParams: locateParams{
  5579. setInfoParams: setInfoParams{
  5580. {
  5581. me: state.NewIdentScreenName("me"),
  5582. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5583. TLVRestBlock: wire.TLVRestBlock{
  5584. TLVList: wire.TLVList{
  5585. wire.NewTLVBE(wire.LocateTLVTagsInfoCapabilities, []uuid.UUID{
  5586. uuid.MustParse("09460000-4C7F-11D1-8222-444553540000"),
  5587. uuid.MustParse("09460001-4C7F-11D1-8222-444553540000"),
  5588. wire.CapChat,
  5589. }),
  5590. },
  5591. },
  5592. },
  5593. },
  5594. },
  5595. },
  5596. },
  5597. wantMsg: []string{},
  5598. },
  5599. {
  5600. name: "set capabilities with empty list",
  5601. me: newTestSession("me"),
  5602. givenCmd: []byte(`toc_set_caps`),
  5603. mockParams: mockParams{
  5604. locateParams: locateParams{
  5605. setInfoParams: setInfoParams{
  5606. {
  5607. me: state.NewIdentScreenName("me"),
  5608. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5609. TLVRestBlock: wire.TLVRestBlock{
  5610. TLVList: wire.TLVList{
  5611. wire.NewTLVBE(wire.LocateTLVTagsInfoCapabilities, []uuid.UUID{
  5612. wire.CapChat,
  5613. }),
  5614. },
  5615. },
  5616. },
  5617. },
  5618. },
  5619. },
  5620. },
  5621. wantMsg: []string{},
  5622. },
  5623. {
  5624. name: "set capability, receive error from locate service",
  5625. me: newTestSession("me"),
  5626. givenCmd: []byte(`toc_set_caps 09460000-4C7F-11D1-8222-444553540000`),
  5627. mockParams: mockParams{
  5628. locateParams: locateParams{
  5629. setInfoParams: setInfoParams{
  5630. {
  5631. me: state.NewIdentScreenName("me"),
  5632. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5633. TLVRestBlock: wire.TLVRestBlock{
  5634. TLVList: wire.TLVList{
  5635. wire.NewTLVBE(wire.LocateTLVTagsInfoCapabilities, []uuid.UUID{
  5636. uuid.MustParse("09460000-4C7F-11D1-8222-444553540000"),
  5637. wire.CapChat,
  5638. }),
  5639. },
  5640. },
  5641. },
  5642. err: io.EOF,
  5643. },
  5644. },
  5645. },
  5646. },
  5647. wantMsg: []string{cmdInternalSvcErr},
  5648. },
  5649. {
  5650. name: "set malformed capability UUID is skipped",
  5651. me: newTestSession("me"),
  5652. givenCmd: []byte(`toc_set_caps 09460000-`),
  5653. mockParams: mockParams{
  5654. locateParams: locateParams{
  5655. setInfoParams: setInfoParams{
  5656. {
  5657. me: state.NewIdentScreenName("me"),
  5658. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5659. TLVRestBlock: wire.TLVRestBlock{
  5660. TLVList: wire.TLVList{
  5661. wire.NewTLVBE(wire.LocateTLVTagsInfoCapabilities, []uuid.UUID{
  5662. wire.CapChat,
  5663. }),
  5664. },
  5665. },
  5666. },
  5667. },
  5668. },
  5669. },
  5670. },
  5671. wantMsg: []string{},
  5672. },
  5673. {
  5674. name: "set capabilities with comma-separated list (TameClone format)",
  5675. me: newTestSession("me"),
  5676. givenCmd: []byte(`toc_set_caps 748F2420-6287-11D1-8222-444553540000,1348,134B,1341,1343,1FF,1345,1346,1347,`),
  5677. mockParams: mockParams{
  5678. locateParams: locateParams{
  5679. setInfoParams: setInfoParams{
  5680. {
  5681. me: state.NewIdentScreenName("me"),
  5682. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  5683. TLVRestBlock: wire.TLVRestBlock{
  5684. TLVList: wire.TLVList{
  5685. wire.NewTLVBE(wire.LocateTLVTagsInfoCapabilities, []uuid.UUID{
  5686. wire.CapChat, // 748F... from client
  5687. wire.CapFileSharing, // 1348
  5688. wire.CapBuddyListTransfer, // 134B
  5689. wire.CapVoiceChat, // 1341
  5690. wire.CapFileTransfer, // 1343
  5691. wire.CapSmartCaps, // 1FF
  5692. wire.CapDirectICBM, // 1345
  5693. wire.CapAvatarService, // 1346
  5694. wire.CapStocksAddins, // 1347
  5695. wire.CapChat, // auto-appended
  5696. }),
  5697. },
  5698. },
  5699. },
  5700. },
  5701. },
  5702. },
  5703. },
  5704. wantMsg: []string{},
  5705. },
  5706. }
  5707. for _, tc := range cases {
  5708. t.Run(tc.name, func(t *testing.T) {
  5709. ctx := context.Background()
  5710. locateSvc := newMockLocateService(t)
  5711. for _, params := range tc.mockParams.setInfoParams {
  5712. locateSvc.EXPECT().
  5713. SetInfo(ctx, matchSession(params.me), params.inBody).
  5714. Return(params.err)
  5715. }
  5716. svc := OSCARProxy{
  5717. Logger: slog.Default(),
  5718. LocateService: locateSvc,
  5719. }
  5720. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5721. assert.Equal(t, tc.wantMsg, msg)
  5722. })
  5723. }
  5724. }
  5725. func TestOSCARProxy_RecvClientCmd_SetConfig(t *testing.T) {
  5726. cases := []struct {
  5727. // name is the unit test name
  5728. name string
  5729. // me is the TOC user session
  5730. me *state.SessionInstance
  5731. // givenCmd is the TOC command
  5732. givenCmd []byte
  5733. // wantMsg is the expected TOC response
  5734. wantMsg []string
  5735. // mockParams is the list of params sent to mocks that satisfy this
  5736. // method's dependencies
  5737. mockParams mockParams
  5738. }{
  5739. {
  5740. name: "successfully set permit all config (unquoted)",
  5741. me: newTestSession("me"),
  5742. givenCmd: []byte("toc_set_config {m 1\ng Buddies\nb friend1\nb friend2\n}\n"),
  5743. mockParams: mockParams{
  5744. tocConfigParams: tocConfigParams{
  5745. setTOCConfigParams: setTOCConfigParams{
  5746. {
  5747. user: state.NewIdentScreenName("me"),
  5748. config: "{m 1\ng Buddies\nb friend1\nb friend2\n}\n",
  5749. },
  5750. },
  5751. },
  5752. },
  5753. wantMsg: []string{},
  5754. },
  5755. {
  5756. name: "successfully set permit all config (double-quoted)",
  5757. me: newTestSession("me"),
  5758. givenCmd: []byte("toc_set_config \"{m 1\ng Buddies\nb friend1\nb friend2\n}\n\""),
  5759. mockParams: mockParams{
  5760. tocConfigParams: tocConfigParams{
  5761. setTOCConfigParams: setTOCConfigParams{
  5762. {
  5763. user: state.NewIdentScreenName("me"),
  5764. config: "{m 1\ng Buddies\nb friend1\nb friend2\n}\n",
  5765. },
  5766. },
  5767. },
  5768. },
  5769. wantMsg: []string{},
  5770. },
  5771. {
  5772. name: "successfully set permit all config (single-quoted)",
  5773. me: newTestSession("me"),
  5774. givenCmd: []byte("toc_set_config '{m 1\ng Buddies\nb friend1\nb friend2\n}\n'"),
  5775. mockParams: mockParams{
  5776. tocConfigParams: tocConfigParams{
  5777. setTOCConfigParams: setTOCConfigParams{
  5778. {
  5779. user: state.NewIdentScreenName("me"),
  5780. config: "{m 1\ng Buddies\nb friend1\nb friend2\n}\n",
  5781. },
  5782. },
  5783. },
  5784. },
  5785. wantMsg: []string{},
  5786. },
  5787. {
  5788. name: "successfully set permit all config (double-quoted with spaces)",
  5789. me: newTestSession("me"),
  5790. givenCmd: []byte("toc_set_config \" {m 1\ng Buddies\nb friend1\nb friend2\n}\n \""),
  5791. mockParams: mockParams{
  5792. tocConfigParams: tocConfigParams{
  5793. setTOCConfigParams: setTOCConfigParams{
  5794. {
  5795. user: state.NewIdentScreenName("me"),
  5796. config: "{m 1\ng Buddies\nb friend1\nb friend2\n}\n",
  5797. },
  5798. },
  5799. },
  5800. },
  5801. wantMsg: []string{},
  5802. },
  5803. {
  5804. name: "set config, receive error from toc config store",
  5805. me: newTestSession("me"),
  5806. givenCmd: []byte("toc_set_config {m 1\ng Buddies\nb friend1\nb friend2\n}\n"),
  5807. mockParams: mockParams{
  5808. tocConfigParams: tocConfigParams{
  5809. setTOCConfigParams: setTOCConfigParams{
  5810. {
  5811. user: state.NewIdentScreenName("me"),
  5812. config: "{m 1\ng Buddies\nb friend1\nb friend2\n}\n",
  5813. err: io.EOF,
  5814. },
  5815. },
  5816. },
  5817. },
  5818. wantMsg: []string{cmdInternalSvcErr},
  5819. },
  5820. {
  5821. name: "bad command",
  5822. me: newTestSession("me"),
  5823. givenCmd: []byte(`toc_set_config`),
  5824. wantMsg: []string{cmdInternalSvcErr},
  5825. },
  5826. }
  5827. for _, tc := range cases {
  5828. t.Run(tc.name, func(t *testing.T) {
  5829. ctx := context.Background()
  5830. pdSvc := newMockPermitDenyService(t)
  5831. for _, params := range tc.mockParams.addDenyListEntriesParams {
  5832. pdSvc.EXPECT().
  5833. AddDenyListEntries(ctx, matchSession(params.me), params.body).
  5834. Return(params.err)
  5835. }
  5836. for _, params := range tc.mockParams.addPermListEntriesParams {
  5837. pdSvc.EXPECT().
  5838. AddPermListEntries(ctx, matchSession(params.me), params.body).
  5839. Return(params.err)
  5840. }
  5841. buddySvc := newMockBuddyService(t)
  5842. for _, params := range tc.mockParams.addBuddiesParams {
  5843. buddySvc.EXPECT().
  5844. AddBuddies(ctx, matchSession(params.me), params.inBody).
  5845. Return(params.err)
  5846. }
  5847. tocConfigSvc := newMockTOCConfigStore(t)
  5848. for _, params := range tc.mockParams.setTOCConfigParams {
  5849. tocConfigSvc.EXPECT().
  5850. SetTOCConfig(matchContext(), params.user, params.config).
  5851. Return(params.err)
  5852. }
  5853. svc := OSCARProxy{
  5854. BuddyService: buddySvc,
  5855. Logger: slog.Default(),
  5856. PermitDenyService: pdSvc,
  5857. TOCConfigStore: tocConfigSvc,
  5858. }
  5859. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  5860. assert.Equal(t, tc.wantMsg, msg)
  5861. })
  5862. }
  5863. }
  5864. func TestOSCARProxy_RecvClientCmd_SetDir(t *testing.T) {
  5865. cases := []struct {
  5866. // name is the unit test name
  5867. name string
  5868. // me is the TOC user session
  5869. me *state.SessionInstance
  5870. // givenCmd is the TOC command
  5871. givenCmd []byte
  5872. // wantMsg is the expected TOC response
  5873. wantMsg []string
  5874. // mockParams is the list of params sent to mocks that satisfy this
  5875. // method's dependencies
  5876. mockParams mockParams
  5877. }{
  5878. {
  5879. name: "successfully set directory info with quoted fields",
  5880. me: newTestSession("me"),
  5881. givenCmd: []byte(`toc_set_dir "first name\$":"middle name":"last name":"maiden name":"city":"state":"country":"email":"allow web searches"`),
  5882. mockParams: mockParams{
  5883. locateParams: locateParams{
  5884. setDirInfoParams: setDirInfoParams{
  5885. {
  5886. me: state.NewIdentScreenName("me"),
  5887. inBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
  5888. TLVRestBlock: wire.TLVRestBlock{
  5889. TLVList: wire.TLVList{
  5890. wire.NewTLVBE(wire.ODirTLVFirstName, "first name$"),
  5891. wire.NewTLVBE(wire.ODirTLVMiddleName, "middle name"),
  5892. wire.NewTLVBE(wire.ODirTLVLastName, "last name"),
  5893. wire.NewTLVBE(wire.ODirTLVMaidenName, "maiden name"),
  5894. wire.NewTLVBE(wire.ODirTLVCountry, "country"),
  5895. wire.NewTLVBE(wire.ODirTLVState, "state"),
  5896. wire.NewTLVBE(wire.ODirTLVCity, "city"),
  5897. },
  5898. },
  5899. },
  5900. },
  5901. },
  5902. },
  5903. },
  5904. wantMsg: []string{},
  5905. },
  5906. {
  5907. name: "successfully set directory info with some blank fields",
  5908. me: newTestSession("me"),
  5909. givenCmd: []byte(`toc_set_dir "first name"::"last name"::"city":"state":"country":"email":"allow web searches"`),
  5910. mockParams: mockParams{
  5911. locateParams: locateParams{
  5912. setDirInfoParams: setDirInfoParams{
  5913. {
  5914. me: state.NewIdentScreenName("me"),
  5915. inBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
  5916. TLVRestBlock: wire.TLVRestBlock{
  5917. TLVList: wire.TLVList{
  5918. wire.NewTLVBE(wire.ODirTLVFirstName, "first name"),
  5919. wire.NewTLVBE(wire.ODirTLVMiddleName, ""),
  5920. wire.NewTLVBE(wire.ODirTLVLastName, "last name"),
  5921. wire.NewTLVBE(wire.ODirTLVMaidenName, ""),
  5922. wire.NewTLVBE(wire.ODirTLVCountry, "country"),
  5923. wire.NewTLVBE(wire.ODirTLVState, "state"),
  5924. wire.NewTLVBE(wire.ODirTLVCity, "city"),
  5925. },
  5926. },
  5927. },
  5928. },
  5929. },
  5930. },
  5931. },
  5932. wantMsg: []string{},
  5933. },
  5934. {
  5935. name: "successfully set directory info with last two fields absent",
  5936. me: newTestSession("me"),
  5937. givenCmd: []byte(`toc_set_dir "first name"::"last name"::"city":"state":"country"`),
  5938. mockParams: mockParams{
  5939. locateParams: locateParams{
  5940. setDirInfoParams: setDirInfoParams{
  5941. {
  5942. me: state.NewIdentScreenName("me"),
  5943. inBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
  5944. TLVRestBlock: wire.TLVRestBlock{
  5945. TLVList: wire.TLVList{
  5946. wire.NewTLVBE(wire.ODirTLVFirstName, "first name"),
  5947. wire.NewTLVBE(wire.ODirTLVMiddleName, ""),
  5948. wire.NewTLVBE(wire.ODirTLVLastName, "last name"),
  5949. wire.NewTLVBE(wire.ODirTLVMaidenName, ""),
  5950. wire.NewTLVBE(wire.ODirTLVCountry, "country"),
  5951. wire.NewTLVBE(wire.ODirTLVState, "state"),
  5952. wire.NewTLVBE(wire.ODirTLVCity, "city"),
  5953. },
  5954. },
  5955. },
  5956. },
  5957. },
  5958. },
  5959. },
  5960. wantMsg: []string{},
  5961. },
  5962. {
  5963. name: "set directory info, receive error from locate svc",
  5964. me: newTestSession("me"),
  5965. givenCmd: []byte(`toc_set_dir "first name":"middle name":"last name":"maiden name":"city":"state":"country":"email":"allow web searches"`),
  5966. mockParams: mockParams{
  5967. locateParams: locateParams{
  5968. setDirInfoParams: setDirInfoParams{
  5969. {
  5970. me: state.NewIdentScreenName("me"),
  5971. inBody: wire.SNAC_0x02_0x09_LocateSetDirInfo{
  5972. TLVRestBlock: wire.TLVRestBlock{
  5973. TLVList: wire.TLVList{
  5974. wire.NewTLVBE(wire.ODirTLVFirstName, "first name"),
  5975. wire.NewTLVBE(wire.ODirTLVMiddleName, "middle name"),
  5976. wire.NewTLVBE(wire.ODirTLVLastName, "last name"),
  5977. wire.NewTLVBE(wire.ODirTLVMaidenName, "maiden name"),
  5978. wire.NewTLVBE(wire.ODirTLVCountry, "country"),
  5979. wire.NewTLVBE(wire.ODirTLVState, "state"),
  5980. wire.NewTLVBE(wire.ODirTLVCity, "city"),
  5981. },
  5982. },
  5983. },
  5984. err: io.EOF,
  5985. },
  5986. },
  5987. },
  5988. },
  5989. wantMsg: []string{cmdInternalSvcErr},
  5990. },
  5991. {
  5992. name: "set directory with too many fields present",
  5993. me: newTestSession("me"),
  5994. givenCmd: []byte(`toc_set_dir "first name"::"last name"::"city":"state":"country":"email":"allow web searches":"extra":"extra"`),
  5995. wantMsg: []string{cmdInternalSvcErr},
  5996. },
  5997. {
  5998. name: "bad command",
  5999. me: newTestSession("me"),
  6000. givenCmd: []byte(`toc_set_dir`),
  6001. wantMsg: []string{cmdInternalSvcErr},
  6002. },
  6003. }
  6004. for _, tc := range cases {
  6005. t.Run(tc.name, func(t *testing.T) {
  6006. ctx := context.Background()
  6007. locateSvc := newMockLocateService(t)
  6008. for _, params := range tc.mockParams.setDirInfoParams {
  6009. locateSvc.EXPECT().
  6010. SetDirInfo(ctx, matchSession(params.me), wire.SNACFrame{}, params.inBody).
  6011. Return(params.msg, params.err)
  6012. }
  6013. svc := OSCARProxy{
  6014. Logger: slog.Default(),
  6015. LocateService: locateSvc,
  6016. }
  6017. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  6018. assert.Equal(t, tc.wantMsg, msg)
  6019. })
  6020. }
  6021. }
  6022. func TestOSCARProxy_RecvClientCmd_SetIdle(t *testing.T) {
  6023. cases := []struct {
  6024. // name is the unit test name
  6025. name string
  6026. // me is the TOC user session
  6027. me *state.SessionInstance
  6028. // givenCmd is the TOC command
  6029. givenCmd []byte
  6030. // wantMsg is the expected TOC response
  6031. wantMsg []string
  6032. // mockParams is the list of params sent to mocks that satisfy this
  6033. // method's dependencies
  6034. mockParams mockParams
  6035. }{
  6036. {
  6037. name: "successfully set idle status",
  6038. me: newTestSession("me"),
  6039. givenCmd: []byte(`toc_set_idle 10`),
  6040. mockParams: mockParams{
  6041. oServiceParams: oServiceParams{
  6042. idleNotificationParams: idleNotificationParams{
  6043. {
  6044. me: state.NewIdentScreenName("me"),
  6045. bodyIn: wire.SNAC_0x01_0x11_OServiceIdleNotification{
  6046. IdleTime: uint32(10),
  6047. },
  6048. },
  6049. },
  6050. },
  6051. },
  6052. wantMsg: []string{},
  6053. },
  6054. {
  6055. name: "set idle status, receive err from BOS oservice svc",
  6056. me: newTestSession("me"),
  6057. givenCmd: []byte(`toc_set_idle 10`),
  6058. mockParams: mockParams{
  6059. oServiceParams: oServiceParams{
  6060. idleNotificationParams: idleNotificationParams{
  6061. {
  6062. me: state.NewIdentScreenName("me"),
  6063. bodyIn: wire.SNAC_0x01_0x11_OServiceIdleNotification{
  6064. IdleTime: uint32(10),
  6065. },
  6066. err: io.EOF,
  6067. },
  6068. },
  6069. },
  6070. },
  6071. wantMsg: []string{cmdInternalSvcErr},
  6072. },
  6073. {
  6074. name: "bad secs param",
  6075. givenCmd: []byte(`toc_set_idle zero`),
  6076. wantMsg: []string{cmdInternalSvcErr},
  6077. },
  6078. {
  6079. name: "bad command",
  6080. me: newTestSession("me"),
  6081. givenCmd: []byte(`toc_set_idle`),
  6082. wantMsg: []string{cmdInternalSvcErr},
  6083. },
  6084. }
  6085. for _, tc := range cases {
  6086. t.Run(tc.name, func(t *testing.T) {
  6087. ctx := context.Background()
  6088. oServiceSvc := newMockOServiceService(t)
  6089. for _, params := range tc.mockParams.oServiceParams.idleNotificationParams {
  6090. oServiceSvc.EXPECT().
  6091. IdleNotification(ctx, matchSession(params.me), params.bodyIn).
  6092. Return(params.err)
  6093. }
  6094. svc := OSCARProxy{
  6095. Logger: slog.Default(),
  6096. OServiceService: oServiceSvc,
  6097. }
  6098. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  6099. assert.Equal(t, tc.wantMsg, msg)
  6100. })
  6101. }
  6102. }
  6103. func TestOSCARProxy_RecvClientCmd_SetInfo(t *testing.T) {
  6104. cases := []struct {
  6105. // name is the unit test name
  6106. name string
  6107. // me is the TOC user session
  6108. me *state.SessionInstance
  6109. // givenCmd is the TOC command
  6110. givenCmd []byte
  6111. // wantMsg is the expected TOC response
  6112. wantMsg []string
  6113. // mockParams is the list of params sent to mocks that satisfy this
  6114. // method's dependencies
  6115. mockParams mockParams
  6116. }{
  6117. {
  6118. name: "successfully set profile",
  6119. me: newTestSession("me"),
  6120. givenCmd: []byte(`toc_set_info "my profile! :\)"`),
  6121. mockParams: mockParams{
  6122. locateParams: locateParams{
  6123. setInfoParams: setInfoParams{
  6124. {
  6125. me: state.NewIdentScreenName("me"),
  6126. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  6127. TLVRestBlock: wire.TLVRestBlock{
  6128. TLVList: wire.TLVList{
  6129. wire.NewTLVBE(wire.LocateTLVTagsInfoSigData, "my profile! :)"),
  6130. },
  6131. },
  6132. },
  6133. },
  6134. },
  6135. },
  6136. },
  6137. wantMsg: []string{},
  6138. },
  6139. {
  6140. name: "set profile, receive error from locate svc",
  6141. me: newTestSession("me"),
  6142. givenCmd: []byte(`toc_set_info "my profile!"`),
  6143. mockParams: mockParams{
  6144. locateParams: locateParams{
  6145. setInfoParams: setInfoParams{
  6146. {
  6147. me: state.NewIdentScreenName("me"),
  6148. inBody: wire.SNAC_0x02_0x04_LocateSetInfo{
  6149. TLVRestBlock: wire.TLVRestBlock{
  6150. TLVList: wire.TLVList{
  6151. wire.NewTLVBE(wire.LocateTLVTagsInfoSigData, "my profile!"),
  6152. },
  6153. },
  6154. },
  6155. err: io.EOF,
  6156. },
  6157. },
  6158. },
  6159. },
  6160. wantMsg: []string{cmdInternalSvcErr},
  6161. },
  6162. {
  6163. name: "bad command",
  6164. me: newTestSession("me"),
  6165. givenCmd: []byte(`toc_set_info`),
  6166. wantMsg: []string{cmdInternalSvcErr},
  6167. },
  6168. }
  6169. for _, tc := range cases {
  6170. t.Run(tc.name, func(t *testing.T) {
  6171. ctx := context.Background()
  6172. locateSvc := newMockLocateService(t)
  6173. for _, params := range tc.mockParams.setInfoParams {
  6174. locateSvc.EXPECT().
  6175. SetInfo(ctx, matchSession(params.me), params.inBody).
  6176. Return(params.err)
  6177. }
  6178. svc := OSCARProxy{
  6179. Logger: slog.Default(),
  6180. LocateService: locateSvc,
  6181. }
  6182. msg := svc.RecvClientCmd(ctx, tc.me, nil, tc.givenCmd, nil, nil)
  6183. assert.Equal(t, tc.wantMsg, msg)
  6184. })
  6185. }
  6186. }
  6187. func TestOSCARProxy_Signon(t *testing.T) {
  6188. roastedPass := wire.RoastTOCPassword([]byte("thepass"))
  6189. cases := []struct {
  6190. // name is the unit test name
  6191. name string
  6192. // checkSession validates the session returned from Signon when non-nil; leave nil for error cases
  6193. checkSession func(*testing.T, *state.SessionInstance)
  6194. // givenCmd is the TOC command
  6195. givenCmd []byte
  6196. // wantMsg is the expected TOC response
  6197. wantMsg []string
  6198. // mockParams is the list of params sent to mocks that satisfy this
  6199. // method's dependencies
  6200. mockParams mockParams
  6201. }{
  6202. {
  6203. name: "successfully login TOC1",
  6204. checkSession: func(t *testing.T, s *state.SessionInstance) {
  6205. assert.Equal(t, state.NewIdentScreenName("me"), s.IdentScreenName())
  6206. assert.Equal(t, [][16]byte{wire.CapChat}, s.Session().Caps())
  6207. assert.False(t, s.IsTOC2())
  6208. },
  6209. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6210. mockParams: mockParams{
  6211. authParams: authParams{
  6212. flapLoginParams: flapLoginParams{
  6213. {
  6214. frame: wire.FLAPSignonFrame{
  6215. TLVRestBlock: wire.TLVRestBlock{
  6216. TLVList: wire.TLVList{
  6217. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6218. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6219. },
  6220. },
  6221. },
  6222. tlv: wire.TLVRestBlock{
  6223. TLVList: wire.TLVList{
  6224. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6225. },
  6226. },
  6227. },
  6228. },
  6229. crackCookieParams: crackCookieParams{
  6230. {
  6231. cookieIn: []byte("thecookie"),
  6232. cookieOut: state.ServerCookie{Service: wire.BOS},
  6233. },
  6234. },
  6235. registerBOSSessionParams: registerBOSSessionParams{
  6236. {
  6237. authCookie: state.ServerCookie{Service: wire.BOS},
  6238. instance: newTestSession("me"),
  6239. },
  6240. },
  6241. },
  6242. buddyListRegistryParams: buddyListRegistryParams{
  6243. registerBuddyListParams: registerBuddyListParams{
  6244. {
  6245. user: state.NewIdentScreenName("me"),
  6246. },
  6247. },
  6248. },
  6249. tocConfigParams: tocConfigParams{
  6250. userParams: userParams{
  6251. {
  6252. screenName: state.NewIdentScreenName("me"),
  6253. returnedUser: &state.User{
  6254. TOCConfig: "my-toc-config",
  6255. },
  6256. },
  6257. },
  6258. },
  6259. },
  6260. wantMsg: []string{"SIGN_ON:TOC1.0", "CONFIG:my-toc-config", "NICK:me"},
  6261. },
  6262. {
  6263. name: "successfully login TOC2",
  6264. checkSession: func(t *testing.T, s *state.SessionInstance) {
  6265. assert.Equal(t, state.NewIdentScreenName("me"), s.IdentScreenName())
  6266. assert.Equal(t, [][16]byte{wire.CapChat}, s.Session().Caps())
  6267. assert.True(t, s.IsTOC2())
  6268. },
  6269. givenCmd: []byte(`toc2_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6270. mockParams: mockParams{
  6271. authParams: authParams{
  6272. flapLoginParams: flapLoginParams{
  6273. {
  6274. frame: wire.FLAPSignonFrame{
  6275. TLVRestBlock: wire.TLVRestBlock{
  6276. TLVList: wire.TLVList{
  6277. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6278. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6279. wire.NewTLVBE(wire.LoginTLVTagsMultiConnFlags, wire.MultiConnFlagsRecentClient),
  6280. },
  6281. },
  6282. },
  6283. tlv: wire.TLVRestBlock{
  6284. TLVList: wire.TLVList{
  6285. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6286. },
  6287. },
  6288. },
  6289. },
  6290. crackCookieParams: crackCookieParams{
  6291. {
  6292. cookieIn: []byte("thecookie"),
  6293. cookieOut: state.ServerCookie{Service: wire.BOS},
  6294. },
  6295. },
  6296. registerBOSSessionParams: registerBOSSessionParams{
  6297. {
  6298. authCookie: state.ServerCookie{Service: wire.BOS},
  6299. instance: newTestSession("me"),
  6300. },
  6301. },
  6302. },
  6303. buddyListRegistryParams: buddyListRegistryParams{
  6304. registerBuddyListParams: registerBuddyListParams{
  6305. {
  6306. user: state.NewIdentScreenName("me"),
  6307. },
  6308. },
  6309. },
  6310. feedBagParams: feedBagParams{
  6311. feedbagServiceUseParams: feedbagServiceUseParams{{err: nil}},
  6312. feedbagParams: feedbagParams{
  6313. {
  6314. screenName: state.NewIdentScreenName("me"),
  6315. results: nil,
  6316. err: nil,
  6317. },
  6318. },
  6319. },
  6320. },
  6321. wantMsg: []string{"SIGN_ON:TOC2.0", "NICK:me", "CONFIG2:done:\n"},
  6322. },
  6323. {
  6324. name: "successfully login toc2_login (TOC2 with encoded messaging)",
  6325. checkSession: func(t *testing.T, s *state.SessionInstance) {
  6326. assert.Equal(t, state.NewIdentScreenName("me"), s.IdentScreenName())
  6327. assert.Equal(t, [][16]byte{wire.CapChat}, s.Session().Caps())
  6328. assert.True(t, s.IsTOC2())
  6329. assert.True(t, s.SupportsTOC2MsgEnc())
  6330. },
  6331. givenCmd: []byte(`toc2_login "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6332. mockParams: mockParams{
  6333. authParams: authParams{
  6334. flapLoginParams: flapLoginParams{
  6335. {
  6336. frame: wire.FLAPSignonFrame{
  6337. TLVRestBlock: wire.TLVRestBlock{
  6338. TLVList: wire.TLVList{
  6339. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6340. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6341. wire.NewTLVBE(wire.LoginTLVTagsMultiConnFlags, wire.MultiConnFlagsRecentClient),
  6342. },
  6343. },
  6344. },
  6345. tlv: wire.TLVRestBlock{
  6346. TLVList: wire.TLVList{
  6347. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6348. },
  6349. },
  6350. },
  6351. },
  6352. crackCookieParams: crackCookieParams{
  6353. {
  6354. cookieIn: []byte("thecookie"),
  6355. cookieOut: state.ServerCookie{Service: wire.BOS},
  6356. },
  6357. },
  6358. registerBOSSessionParams: registerBOSSessionParams{
  6359. {
  6360. authCookie: state.ServerCookie{Service: wire.BOS},
  6361. instance: newTestSession("me"),
  6362. },
  6363. },
  6364. },
  6365. buddyListRegistryParams: buddyListRegistryParams{
  6366. registerBuddyListParams: registerBuddyListParams{
  6367. {
  6368. user: state.NewIdentScreenName("me"),
  6369. },
  6370. },
  6371. },
  6372. feedBagParams: feedBagParams{
  6373. feedbagServiceUseParams: feedbagServiceUseParams{{err: nil}},
  6374. feedbagParams: feedbagParams{
  6375. {
  6376. screenName: state.NewIdentScreenName("me"),
  6377. results: nil,
  6378. err: nil,
  6379. },
  6380. },
  6381. },
  6382. },
  6383. wantMsg: []string{"SIGN_ON:TOC2.0", "NICK:me", "CONFIG2:done:\n"},
  6384. },
  6385. {
  6386. name: "login TOC2, receive error from FeedbagService.Use",
  6387. // Signon returns (sess, err) on feedbag errors, so session is non-nil; use no-op check
  6388. checkSession: func(*testing.T, *state.SessionInstance) {},
  6389. givenCmd: []byte(`toc2_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6390. mockParams: mockParams{
  6391. authParams: authParams{
  6392. flapLoginParams: flapLoginParams{
  6393. {
  6394. frame: wire.FLAPSignonFrame{
  6395. TLVRestBlock: wire.TLVRestBlock{
  6396. TLVList: wire.TLVList{
  6397. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6398. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6399. wire.NewTLVBE(wire.LoginTLVTagsMultiConnFlags, wire.MultiConnFlagsRecentClient),
  6400. },
  6401. },
  6402. },
  6403. tlv: wire.TLVRestBlock{
  6404. TLVList: wire.TLVList{
  6405. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6406. },
  6407. },
  6408. },
  6409. },
  6410. crackCookieParams: crackCookieParams{
  6411. {
  6412. cookieIn: []byte("thecookie"),
  6413. cookieOut: state.ServerCookie{Service: wire.BOS},
  6414. },
  6415. },
  6416. registerBOSSessionParams: registerBOSSessionParams{
  6417. {
  6418. authCookie: state.ServerCookie{Service: wire.BOS},
  6419. instance: newTestSession("me"),
  6420. },
  6421. },
  6422. },
  6423. buddyListRegistryParams: buddyListRegistryParams{
  6424. registerBuddyListParams: registerBuddyListParams{
  6425. {user: state.NewIdentScreenName("me")},
  6426. },
  6427. },
  6428. feedBagParams: feedBagParams{
  6429. feedbagServiceUseParams: feedbagServiceUseParams{{err: io.EOF}},
  6430. },
  6431. },
  6432. wantMsg: []string{cmdInternalSvcErr},
  6433. },
  6434. {
  6435. name: "login TOC2, receive error from FeedbagManager.Feedbag",
  6436. // Signon returns (sess, err) on feedbag errors, so session is non-nil; use no-op check
  6437. checkSession: func(*testing.T, *state.SessionInstance) {},
  6438. givenCmd: []byte(`toc2_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6439. mockParams: mockParams{
  6440. authParams: authParams{
  6441. flapLoginParams: flapLoginParams{
  6442. {
  6443. frame: wire.FLAPSignonFrame{
  6444. TLVRestBlock: wire.TLVRestBlock{
  6445. TLVList: wire.TLVList{
  6446. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6447. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6448. wire.NewTLVBE(wire.LoginTLVTagsMultiConnFlags, wire.MultiConnFlagsRecentClient),
  6449. },
  6450. },
  6451. },
  6452. tlv: wire.TLVRestBlock{
  6453. TLVList: wire.TLVList{
  6454. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6455. },
  6456. },
  6457. },
  6458. },
  6459. crackCookieParams: crackCookieParams{
  6460. {
  6461. cookieIn: []byte("thecookie"),
  6462. cookieOut: state.ServerCookie{Service: wire.BOS},
  6463. },
  6464. },
  6465. registerBOSSessionParams: registerBOSSessionParams{
  6466. {
  6467. authCookie: state.ServerCookie{Service: wire.BOS},
  6468. instance: newTestSession("me"),
  6469. },
  6470. },
  6471. },
  6472. buddyListRegistryParams: buddyListRegistryParams{
  6473. registerBuddyListParams: registerBuddyListParams{
  6474. {user: state.NewIdentScreenName("me")},
  6475. },
  6476. },
  6477. feedBagParams: feedBagParams{
  6478. feedbagServiceUseParams: feedbagServiceUseParams{{err: nil}},
  6479. feedbagParams: feedbagParams{
  6480. {
  6481. screenName: state.NewIdentScreenName("me"),
  6482. results: nil,
  6483. err: io.EOF,
  6484. },
  6485. },
  6486. },
  6487. },
  6488. wantMsg: []string{cmdInternalSvcErr},
  6489. },
  6490. {
  6491. name: "login, receive error from auth svc FLAP login",
  6492. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6493. mockParams: mockParams{
  6494. authParams: authParams{
  6495. flapLoginParams: flapLoginParams{
  6496. {
  6497. frame: wire.FLAPSignonFrame{
  6498. TLVRestBlock: wire.TLVRestBlock{
  6499. TLVList: wire.TLVList{
  6500. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6501. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6502. },
  6503. },
  6504. },
  6505. err: io.EOF,
  6506. },
  6507. },
  6508. },
  6509. },
  6510. wantMsg: []string{cmdInternalSvcErr},
  6511. },
  6512. {
  6513. name: "login, receive error from auth svc registration",
  6514. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6515. mockParams: mockParams{
  6516. authParams: authParams{
  6517. flapLoginParams: flapLoginParams{
  6518. {
  6519. frame: wire.FLAPSignonFrame{
  6520. TLVRestBlock: wire.TLVRestBlock{
  6521. TLVList: wire.TLVList{
  6522. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6523. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6524. },
  6525. },
  6526. },
  6527. tlv: wire.TLVRestBlock{
  6528. TLVList: wire.TLVList{
  6529. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6530. },
  6531. },
  6532. },
  6533. },
  6534. crackCookieParams: crackCookieParams{
  6535. {
  6536. cookieIn: []byte("thecookie"),
  6537. cookieOut: state.ServerCookie{Service: wire.BOS},
  6538. },
  6539. },
  6540. registerBOSSessionParams: registerBOSSessionParams{
  6541. {
  6542. authCookie: state.ServerCookie{Service: wire.BOS},
  6543. err: io.EOF,
  6544. },
  6545. },
  6546. },
  6547. },
  6548. wantMsg: []string{cmdInternalSvcErr},
  6549. },
  6550. {
  6551. name: "login, receive error from buddy list registry",
  6552. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6553. mockParams: mockParams{
  6554. authParams: authParams{
  6555. flapLoginParams: flapLoginParams{
  6556. {
  6557. frame: wire.FLAPSignonFrame{
  6558. TLVRestBlock: wire.TLVRestBlock{
  6559. TLVList: wire.TLVList{
  6560. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6561. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6562. },
  6563. },
  6564. },
  6565. tlv: wire.TLVRestBlock{
  6566. TLVList: wire.TLVList{
  6567. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6568. },
  6569. },
  6570. },
  6571. },
  6572. crackCookieParams: crackCookieParams{
  6573. {
  6574. cookieIn: []byte("thecookie"),
  6575. cookieOut: state.ServerCookie{Service: wire.BOS},
  6576. },
  6577. },
  6578. registerBOSSessionParams: registerBOSSessionParams{
  6579. {
  6580. authCookie: state.ServerCookie{Service: wire.BOS},
  6581. instance: newTestSession("me"),
  6582. },
  6583. },
  6584. },
  6585. buddyListRegistryParams: buddyListRegistryParams{
  6586. registerBuddyListParams: registerBuddyListParams{
  6587. {
  6588. user: state.NewIdentScreenName("me"),
  6589. err: io.EOF,
  6590. },
  6591. },
  6592. },
  6593. },
  6594. wantMsg: []string{cmdInternalSvcErr},
  6595. },
  6596. {
  6597. name: "login, receive error from TOC config store",
  6598. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6599. mockParams: mockParams{
  6600. authParams: authParams{
  6601. flapLoginParams: flapLoginParams{
  6602. {
  6603. frame: wire.FLAPSignonFrame{
  6604. TLVRestBlock: wire.TLVRestBlock{
  6605. TLVList: wire.TLVList{
  6606. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6607. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6608. },
  6609. },
  6610. },
  6611. tlv: wire.TLVRestBlock{
  6612. TLVList: wire.TLVList{
  6613. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6614. },
  6615. },
  6616. },
  6617. },
  6618. crackCookieParams: crackCookieParams{
  6619. {
  6620. cookieIn: []byte("thecookie"),
  6621. cookieOut: state.ServerCookie{Service: wire.BOS},
  6622. },
  6623. },
  6624. registerBOSSessionParams: registerBOSSessionParams{
  6625. {
  6626. authCookie: state.ServerCookie{Service: wire.BOS},
  6627. instance: newTestSession("me"),
  6628. },
  6629. },
  6630. },
  6631. buddyListRegistryParams: buddyListRegistryParams{
  6632. registerBuddyListParams: registerBuddyListParams{
  6633. {
  6634. user: state.NewIdentScreenName("me"),
  6635. },
  6636. },
  6637. },
  6638. tocConfigParams: tocConfigParams{
  6639. userParams: userParams{
  6640. {
  6641. screenName: state.NewIdentScreenName("me"),
  6642. err: io.EOF,
  6643. },
  6644. },
  6645. },
  6646. },
  6647. wantMsg: []string{cmdInternalSvcErr},
  6648. },
  6649. {
  6650. name: "login, user not found after login",
  6651. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6652. mockParams: mockParams{
  6653. authParams: authParams{
  6654. flapLoginParams: flapLoginParams{
  6655. {
  6656. frame: wire.FLAPSignonFrame{
  6657. TLVRestBlock: wire.TLVRestBlock{
  6658. TLVList: wire.TLVList{
  6659. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6660. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6661. },
  6662. },
  6663. },
  6664. tlv: wire.TLVRestBlock{
  6665. TLVList: wire.TLVList{
  6666. wire.NewTLVBE(wire.OServiceTLVTagsLoginCookie, []byte("thecookie")),
  6667. },
  6668. },
  6669. },
  6670. },
  6671. crackCookieParams: crackCookieParams{
  6672. {
  6673. cookieIn: []byte("thecookie"),
  6674. cookieOut: state.ServerCookie{Service: wire.BOS},
  6675. },
  6676. },
  6677. registerBOSSessionParams: registerBOSSessionParams{
  6678. {
  6679. authCookie: state.ServerCookie{Service: wire.BOS},
  6680. instance: newTestSession("me"),
  6681. },
  6682. },
  6683. },
  6684. buddyListRegistryParams: buddyListRegistryParams{
  6685. registerBuddyListParams: registerBuddyListParams{
  6686. {
  6687. user: state.NewIdentScreenName("me"),
  6688. },
  6689. },
  6690. },
  6691. tocConfigParams: tocConfigParams{
  6692. userParams: userParams{
  6693. {
  6694. screenName: state.NewIdentScreenName("me"),
  6695. returnedUser: nil,
  6696. },
  6697. },
  6698. },
  6699. },
  6700. wantMsg: []string{cmdInternalSvcErr},
  6701. },
  6702. {
  6703. name: "login with bad credentials",
  6704. givenCmd: []byte(`toc_signon "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6705. mockParams: mockParams{
  6706. authParams: authParams{
  6707. flapLoginParams: flapLoginParams{
  6708. {
  6709. frame: wire.FLAPSignonFrame{
  6710. TLVRestBlock: wire.TLVRestBlock{
  6711. TLVList: wire.TLVList{
  6712. wire.NewTLVBE(wire.LoginTLVTagsScreenName, "me"),
  6713. wire.NewTLVBE(wire.LoginTLVTagsRoastedTOCPassword, roastedPass),
  6714. },
  6715. },
  6716. },
  6717. tlv: wire.TLVRestBlock{
  6718. TLVList: wire.TLVList{
  6719. wire.NewTLVBE(wire.LoginTLVTagsErrorSubcode, wire.LoginErrInvalidUsernameOrPassword),
  6720. },
  6721. },
  6722. },
  6723. },
  6724. },
  6725. },
  6726. wantMsg: []string{"ERROR:980"},
  6727. },
  6728. {
  6729. name: "bad command",
  6730. givenCmd: []byte(`toc_bad "" "" me "0x` + hex.EncodeToString(roastedPass) + `"`),
  6731. wantMsg: []string{cmdInternalSvcErr},
  6732. },
  6733. }
  6734. for _, tc := range cases {
  6735. t.Run(tc.name, func(t *testing.T) {
  6736. ctx := context.Background()
  6737. authSvc := newMockAuthService(t)
  6738. for _, params := range tc.mockParams.flapLoginParams {
  6739. authSvc.EXPECT().
  6740. FLAPLogin(matchContext(), params.frame, "").
  6741. Return(params.tlv, params.err)
  6742. }
  6743. for _, params := range tc.mockParams.crackCookieParams {
  6744. authSvc.EXPECT().
  6745. CrackCookie(params.cookieIn).
  6746. Return(params.cookieOut, params.err)
  6747. }
  6748. for _, params := range tc.mockParams.registerBOSSessionParams {
  6749. authSvc.EXPECT().
  6750. RegisterBOSSession(matchContext(), params.authCookie).
  6751. Return(params.instance, params.err)
  6752. }
  6753. buddyRegistry := newMockBuddyListRegistry(t)
  6754. for _, params := range tc.mockParams.registerBuddyListParams {
  6755. buddyRegistry.EXPECT().
  6756. RegisterBuddyList(matchContext(), params.user).
  6757. Return(params.err)
  6758. }
  6759. tocCfg := newMockTOCConfigStore(t)
  6760. for _, params := range tc.mockParams.userParams {
  6761. tocCfg.EXPECT().
  6762. User(matchContext(), params.screenName).
  6763. Return(params.returnedUser, params.err)
  6764. }
  6765. fbSvc := newMockFeedbagService(t)
  6766. for _, params := range tc.mockParams.feedBagParams.feedbagServiceUseParams {
  6767. fbSvc.EXPECT().
  6768. Use(matchContext(), mock.Anything).
  6769. Return(params.err)
  6770. }
  6771. fbMgr := newMockFeedbagManager(t)
  6772. for _, params := range tc.mockParams.feedBagParams.feedbagParams {
  6773. fbMgr.EXPECT().
  6774. Feedbag(matchContext(), params.screenName).
  6775. Return(params.results, params.err)
  6776. }
  6777. buddySvc := newMockBuddyService(t)
  6778. svc := OSCARProxy{
  6779. AuthService: authSvc,
  6780. BuddyListRegistry: buddyRegistry,
  6781. BuddyService: buddySvc,
  6782. Logger: slog.Default(),
  6783. TOCConfigStore: tocCfg,
  6784. FeedbagService: fbSvc,
  6785. FeedbagManager: fbMgr,
  6786. }
  6787. sess, msg := svc.Signon(ctx, tc.givenCmd,
  6788. func(ctx context.Context, instance *state.SessionInstance) error { return nil },
  6789. func(ctx context.Context, instance *state.SessionInstance) {},
  6790. NewChatRegistry(),
  6791. )
  6792. assert.Equal(t, tc.wantMsg, msg)
  6793. if tc.checkSession != nil {
  6794. tc.checkSession(t, sess)
  6795. } else {
  6796. assert.Nil(t, sess)
  6797. }
  6798. })
  6799. }
  6800. }
  6801. func TestOSCARProxy_RecvClientCmd_UnknownCmd(t *testing.T) {
  6802. ctx := context.Background()
  6803. svc := OSCARProxy{
  6804. Logger: slog.Default(),
  6805. }
  6806. cmd := []byte("toc_unknown_cmd")
  6807. msg := svc.RecvClientCmd(ctx, nil, nil, cmd, nil, nil)
  6808. assert.Equal(t, cmdInternalSvcErr, msg[0])
  6809. }
  6810. func Test_parseArgs(t *testing.T) {
  6811. type testCase struct {
  6812. name string
  6813. givenPayload string
  6814. givenArgs []*string
  6815. wantVarArgs []string
  6816. wantArgs []string
  6817. wantErrMsg string
  6818. }
  6819. tests := []testCase{
  6820. {
  6821. name: "no positional args or varargs",
  6822. givenPayload: ``,
  6823. givenArgs: nil,
  6824. wantVarArgs: []string{},
  6825. },
  6826. {
  6827. name: "positional args with varargs",
  6828. givenPayload: `1234 "Join me!" user1 user2 user3`,
  6829. givenArgs: []*string{new(string), new(string)},
  6830. wantVarArgs: []string{"user1", "user2", "user3"},
  6831. wantArgs: []string{"1234", "Join me!"},
  6832. },
  6833. {
  6834. name: "nil positional argument placeholders should get skipped",
  6835. givenPayload: `1234 "Join me!" user1 user2 user3`,
  6836. givenArgs: []*string{nil, nil}, // still 2 placeholders, both nil
  6837. wantVarArgs: []string{"user1", "user2", "user3"},
  6838. wantArgs: []string{"", ""},
  6839. },
  6840. {
  6841. name: "positional args with no varargs",
  6842. givenPayload: `1234 "Join me!"`,
  6843. givenArgs: []*string{new(string), new(string)}, // roomID + msg
  6844. wantVarArgs: []string{},
  6845. wantArgs: []string{"1234", "Join me!"},
  6846. },
  6847. {
  6848. name: "varargs only",
  6849. givenPayload: `user1 user2 user3`,
  6850. givenArgs: nil,
  6851. wantVarArgs: []string{"user1", "user2", "user3"},
  6852. },
  6853. {
  6854. name: "too many positional arg placeholders",
  6855. givenPayload: `toc_chat_invite`,
  6856. givenArgs: []*string{new(string), new(string)},
  6857. wantVarArgs: []string{},
  6858. wantErrMsg: "command contains fewer arguments than expected",
  6859. },
  6860. {
  6861. name: "CSV parser error",
  6862. givenPayload: ``,
  6863. givenArgs: []*string{nil},
  6864. wantVarArgs: []string{},
  6865. wantErrMsg: "CSV reader error",
  6866. },
  6867. }
  6868. for _, tt := range tests {
  6869. t.Run(tt.name, func(t *testing.T) {
  6870. varArgs, err := parseArgs([]byte(tt.givenPayload), tt.givenArgs...)
  6871. if tt.wantErrMsg != "" {
  6872. assert.ErrorContains(t, err, tt.wantErrMsg)
  6873. return
  6874. }
  6875. assert.NoError(t, err)
  6876. // verify the placeholder pointers got populated
  6877. for i, want := range tt.wantArgs {
  6878. if want == "" {
  6879. assert.Nil(t, tt.givenArgs[i])
  6880. } else {
  6881. got := *tt.givenArgs[i]
  6882. assert.Equal(t, want, got)
  6883. }
  6884. }
  6885. // verify we have the same varargs
  6886. assert.Equal(t, tt.wantVarArgs, varArgs)
  6887. assert.Equal(t, len(tt.wantArgs), len(tt.givenArgs))
  6888. })
  6889. }
  6890. }
  6891. func TestBuildToc2Config(t *testing.T) {
  6892. tests := []struct {
  6893. name string
  6894. fb []wire.FeedbagItem
  6895. want []string // CONFIG2 command string(s), each starts with "CONFIG2:"
  6896. wantErr string
  6897. }{
  6898. {
  6899. name: "root group missing order attribute returns error",
  6900. fb: []wire.FeedbagItem{
  6901. {Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup, TLVLBlock: wire.TLVLBlock{}},
  6902. },
  6903. wantErr: "root group missing order attribute",
  6904. },
  6905. {
  6906. name: "empty buddylist yields only done",
  6907. fb: []wire.FeedbagItem{
  6908. {
  6909. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6910. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{})}},
  6911. },
  6912. },
  6913. want: []string{"CONFIG2:done:\n"},
  6914. },
  6915. {
  6916. name: "single group with buddies",
  6917. fb: []wire.FeedbagItem{
  6918. {
  6919. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6920. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{100})}},
  6921. },
  6922. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6923. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1, 2})}}},
  6924. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "user1", TLVLBlock: wire.TLVLBlock{}},
  6925. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "user2", TLVLBlock: wire.TLVLBlock{}},
  6926. },
  6927. want: []string{"CONFIG2:g:Buddies\nb:user1\nb:user2\ndone:\n"},
  6928. },
  6929. {
  6930. name: "blocked user (d:) and private user (p:)",
  6931. fb: []wire.FeedbagItem{
  6932. {
  6933. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6934. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{})}},
  6935. },
  6936. {ClassID: wire.FeedbagClassIDDeny, Name: "blockeduser"},
  6937. {ClassID: wire.FeedbagClassIDPermit, Name: "allowuser"},
  6938. },
  6939. want: []string{"CONFIG2:d:blockeduser\np:allowuser\ndone:\n"},
  6940. },
  6941. {
  6942. name: "privacy level (m:) from pdinfo",
  6943. fb: []wire.FeedbagItem{
  6944. {
  6945. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6946. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{})}},
  6947. },
  6948. {
  6949. ClassID: wire.FeedbagClassIdPdinfo,
  6950. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesPdMode, uint8(3))}},
  6951. },
  6952. },
  6953. want: []string{"CONFIG2:m:3\ndone:\n"},
  6954. },
  6955. {
  6956. name: "buddy with alias",
  6957. fb: []wire.FeedbagItem{
  6958. {
  6959. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6960. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{100})}},
  6961. },
  6962. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6963. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}}},
  6964. {
  6965. ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "bob",
  6966. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesAlias, "Bob Smith")}},
  6967. },
  6968. },
  6969. want: []string{"CONFIG2:g:Buddies\nb:bob:Bob Smith\ndone:\n"},
  6970. },
  6971. {
  6972. name: "buddy with note (comment) uses five colons before note",
  6973. fb: []wire.FeedbagItem{
  6974. {
  6975. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6976. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{100})}},
  6977. },
  6978. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6979. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}}},
  6980. {
  6981. ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "alice",
  6982. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesNote, "Friend from work")}},
  6983. },
  6984. },
  6985. want: []string{"CONFIG2:g:Buddies\nb:alice:::::Friend from work\ndone:\n"},
  6986. },
  6987. {
  6988. name: "multiple groups in root order",
  6989. fb: []wire.FeedbagItem{
  6990. {
  6991. Name: "", GroupID: 0, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6992. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{100, 200})}},
  6993. },
  6994. {Name: "Buddies", GroupID: 100, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6995. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{1})}}},
  6996. {Name: "Family", GroupID: 200, ItemID: 0, ClassID: wire.FeedbagClassIdGroup,
  6997. TLVLBlock: wire.TLVLBlock{TLVList: wire.TLVList{wire.NewTLVBE(wire.FeedbagAttributesOrder, []uint16{2})}}},
  6998. {ItemID: 1, ClassID: wire.FeedbagClassIdBuddy, GroupID: 100, Name: "friend1", TLVLBlock: wire.TLVLBlock{}},
  6999. {ItemID: 2, ClassID: wire.FeedbagClassIdBuddy, GroupID: 200, Name: "mom", TLVLBlock: wire.TLVLBlock{}},
  7000. },
  7001. want: []string{"CONFIG2:g:Buddies\nb:friend1\ng:Family\nb:mom\ndone:\n"},
  7002. },
  7003. }
  7004. for _, tt := range tests {
  7005. t.Run(tt.name, func(t *testing.T) {
  7006. got, err := buildToc2Config(tt.fb)
  7007. if tt.wantErr != "" {
  7008. assert.ErrorContains(t, err, tt.wantErr)
  7009. assert.Empty(t, got)
  7010. return
  7011. }
  7012. assert.NoError(t, err)
  7013. assert.Equal(t, tt.want, got)
  7014. })
  7015. }
  7016. }
  7017. // TestUnescape tests the unescape function
  7018. func TestUnescape(t *testing.T) {
  7019. tests := []struct {
  7020. name string
  7021. input string
  7022. expected string
  7023. }{
  7024. {"No Escapes", "Hello World", "Hello World"},
  7025. {"Escaped Brace", "Hello \\{World\\}", "Hello {World}"},
  7026. {"Escaped Parentheses", "Test\\(123\\)", "Test(123)"},
  7027. {"Escaped Brackets", "\\[List\\]", "[List]"},
  7028. {"Escaped Dollar", "Price: \\$100", "Price: $100"},
  7029. {"Escaped Quote", "She said \\\"Hello\\\"", "She said \"Hello\""},
  7030. {"Multiple Escapes", "One\\, Two\\, Three", "One, Two, Three"},
  7031. {"Consecutive Escapes", "\\\\\\$100", "\\$100"},
  7032. {"Only Escape Character", "\\", ""},
  7033. {"Empty Input", "", ""},
  7034. }
  7035. for _, tt := range tests {
  7036. t.Run(tt.name, func(t *testing.T) {
  7037. result := unescape(tt.input)
  7038. if result != tt.expected {
  7039. t.Errorf("unescape(%q) = %q; want %q", tt.input, result, tt.expected)
  7040. }
  7041. })
  7042. }
  7043. }