4
0

cmd_client_test.go 213 KB

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