1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003400440054006400740084009401040114012401340144015401640174018401940204021402240234024402540264027402840294030403140324033403440354036403740384039404040414042404340444045404640474048404940504051405240534054405540564057405840594060406140624063406440654066406740684069407040714072407340744075407640774078407940804081408240834084408540864087408840894090409140924093409440954096409740984099410041014102410341044105410641074108410941104111411241134114411541164117411841194120412141224123412441254126412741284129413041314132413341344135413641374138413941404141414241434144414541464147414841494150415141524153415441554156415741584159416041614162416341644165416641674168416941704171417241734174417541764177417841794180418141824183418441854186418741884189419041914192419341944195419641974198419942004201420242034204420542064207420842094210421142124213421442154216421742184219422042214222422342244225422642274228422942304231423242334234423542364237423842394240424142424243424442454246424742484249425042514252425342544255425642574258425942604261426242634264426542664267426842694270427142724273427442754276427742784279428042814282428342844285428642874288428942904291429242934294429542964297429842994300430143024303430443054306430743084309431043114312431343144315431643174318431943204321432243234324432543264327432843294330433143324333433443354336433743384339434043414342434343444345434643474348434943504351435243534354435543564357435843594360436143624363436443654366436743684369437043714372437343744375437643774378437943804381438243834384438543864387438843894390439143924393439443954396439743984399440044014402440344044405440644074408440944104411441244134414441544164417441844194420442144224423442444254426442744284429443044314432443344344435443644374438443944404441444244434444444544464447444844494450445144524453445444554456445744584459446044614462446344644465446644674468446944704471447244734474447544764477447844794480448144824483448444854486448744884489449044914492449344944495449644974498449945004501450245034504450545064507450845094510451145124513451445154516451745184519452045214522452345244525452645274528452945304531453245334534453545364537453845394540454145424543454445454546454745484549455045514552455345544555455645574558455945604561456245634564456545664567456845694570457145724573457445754576457745784579458045814582458345844585458645874588458945904591459245934594459545964597459845994600460146024603460446054606460746084609461046114612461346144615461646174618461946204621462246234624462546264627462846294630463146324633463446354636463746384639464046414642464346444645464646474648464946504651465246534654465546564657465846594660466146624663466446654666466746684669467046714672467346744675467646774678467946804681468246834684468546864687468846894690469146924693469446954696469746984699470047014702470347044705470647074708470947104711471247134714471547164717471847194720472147224723472447254726472747284729473047314732473347344735473647374738473947404741474247434744474547464747474847494750475147524753475447554756475747584759476047614762476347644765476647674768476947704771477247734774477547764777477847794780478147824783478447854786478747884789479047914792479347944795479647974798479948004801480248034804480548064807480848094810481148124813481448154816481748184819482048214822482348244825482648274828482948304831483248334834483548364837483848394840484148424843484448454846484748484849485048514852485348544855485648574858485948604861486248634864486548664867486848694870487148724873487448754876487748784879488048814882488348844885488648874888488948904891489248934894489548964897489848994900490149024903490449054906490749084909491049114912491349144915491649174918491949204921492249234924492549264927492849294930493149324933493449354936493749384939494049414942494349444945494649474948494949504951495249534954495549564957495849594960496149624963496449654966496749684969497049714972497349744975497649774978497949804981498249834984498549864987498849894990499149924993499449954996499749984999500050015002500350045005500650075008500950105011501250135014501550165017501850195020502150225023502450255026502750285029503050315032503350345035503650375038503950405041504250435044504550465047504850495050505150525053505450555056505750585059506050615062506350645065506650675068506950705071507250735074507550765077507850795080508150825083508450855086508750885089509050915092509350945095509650975098509951005101510251035104510551065107510851095110511151125113511451155116511751185119512051215122512351245125512651275128512951305131513251335134513551365137513851395140514151425143514451455146514751485149515051515152515351545155515651575158515951605161516251635164516551665167516851695170517151725173517451755176517751785179518051815182518351845185518651875188518951905191519251935194519551965197519851995200520152025203520452055206520752085209521052115212521352145215521652175218521952205221522252235224522552265227522852295230523152325233523452355236523752385239524052415242524352445245524652475248524952505251525252535254525552565257525852595260526152625263526452655266526752685269527052715272527352745275527652775278527952805281528252835284528552865287528852895290529152925293529452955296529752985299530053015302530353045305530653075308530953105311531253135314531553165317531853195320532153225323532453255326532753285329533053315332533353345335533653375338533953405341534253435344534553465347534853495350535153525353535453555356535753585359536053615362536353645365536653675368536953705371537253735374537553765377537853795380538153825383538453855386538753885389539053915392539353945395539653975398539954005401540254035404540554065407540854095410541154125413541454155416541754185419542054215422542354245425542654275428542954305431543254335434543554365437543854395440544154425443544454455446544754485449545054515452545354545455545654575458545954605461546254635464546554665467546854695470547154725473547454755476547754785479548054815482548354845485548654875488548954905491549254935494549554965497549854995500550155025503550455055506550755085509551055115512551355145515551655175518551955205521552255235524552555265527552855295530553155325533553455355536553755385539554055415542554355445545554655475548554955505551555255535554555555565557555855595560556155625563556455655566556755685569557055715572557355745575557655775578557955805581558255835584558555865587558855895590559155925593559455955596559755985599560056015602560356045605560656075608560956105611561256135614561556165617561856195620562156225623562456255626562756285629563056315632563356345635563656375638563956405641564256435644564556465647564856495650565156525653565456555656565756585659566056615662566356645665566656675668566956705671567256735674567556765677567856795680568156825683568456855686568756885689569056915692569356945695569656975698569957005701570257035704570557065707570857095710571157125713571457155716571757185719572057215722572357245725572657275728572957305731573257335734573557365737573857395740574157425743574457455746574757485749575057515752575357545755575657575758575957605761576257635764576557665767576857695770577157725773577457755776577757785779578057815782578357845785578657875788578957905791579257935794579557965797579857995800580158025803580458055806580758085809581058115812581358145815581658175818581958205821582258235824582558265827582858295830583158325833583458355836583758385839584058415842584358445845584658475848584958505851585258535854585558565857585858595860586158625863586458655866586758685869587058715872587358745875587658775878587958805881588258835884588558865887588858895890589158925893589458955896589758985899590059015902590359045905590659075908590959105911591259135914591559165917591859195920592159225923592459255926592759285929593059315932593359345935593659375938593959405941594259435944594559465947594859495950595159525953595459555956595759585959596059615962596359645965596659675968596959705971597259735974597559765977597859795980598159825983598459855986598759885989599059915992599359945995599659975998599960006001600260036004600560066007600860096010601160126013601460156016601760186019602060216022602360246025602660276028602960306031603260336034603560366037603860396040604160426043604460456046604760486049605060516052605360546055605660576058605960606061606260636064606560666067606860696070607160726073607460756076607760786079608060816082608360846085608660876088608960906091609260936094609560966097609860996100610161026103610461056106610761086109611061116112611361146115611661176118611961206121612261236124612561266127612861296130613161326133613461356136613761386139614061416142614361446145614661476148614961506151615261536154615561566157615861596160616161626163616461656166616761686169617061716172617361746175617661776178617961806181618261836184618561866187618861896190619161926193619461956196619761986199620062016202620362046205620662076208620962106211621262136214621562166217621862196220622162226223622462256226622762286229623062316232623362346235623662376238623962406241624262436244624562466247624862496250625162526253625462556256625762586259626062616262626362646265626662676268626962706271627262736274627562766277627862796280628162826283628462856286628762886289629062916292629362946295629662976298629963006301630263036304 |
- //! Program state processor tests
- use {
- mollusk_svm::{result::Check, Mollusk},
- serial_test::serial,
- solana_sdk::{
- account::{create_account_for_test, Account as SolanaAccount, ReadableAccount},
- account_info::{AccountInfo, IntoAccountInfo},
- entrypoint::ProgramResult,
- instruction::Instruction,
- program_error::ProgramError,
- program_option::COption,
- program_pack::Pack,
- pubkey::Pubkey,
- rent::Rent,
- sysvar::rent,
- },
- spl_token::{
- error::TokenError,
- instruction::{
- amount_to_ui_amount, approve, approve_checked, burn, burn_checked, close_account,
- freeze_account, get_account_data_size, initialize_account, initialize_account2,
- initialize_account3, initialize_immutable_owner, initialize_mint, initialize_mint2,
- initialize_multisig, initialize_multisig2, mint_to, mint_to_checked, revoke,
- set_authority, sync_native, thaw_account, transfer, transfer_checked,
- ui_amount_to_amount, AuthorityType, MAX_SIGNERS,
- },
- state::{Account, AccountState, Mint, Multisig},
- },
- std::collections::HashMap,
- };
- fn do_process_instruction(
- instruction: Instruction,
- mut accounts: Vec<&mut SolanaAccount>,
- checks: &[Check],
- ) -> ProgramResult {
- // Prepare accounts for mollusk.
- let instruction_accounts: Vec<(Pubkey, SolanaAccount)> = instruction
- .accounts
- .iter()
- .zip(&accounts)
- .map(|(account_meta, account)| (account_meta.pubkey, (*account).clone()))
- .collect();
- let mollusk = Mollusk::new(&spl_token::ID, "spl_token");
- let result =
- mollusk.process_and_validate_instruction(&instruction, &instruction_accounts, checks);
- // Update accounts after the instruction is processed.
- for (original, (_, updated)) in accounts
- .iter_mut()
- .zip(result.resulting_accounts.into_iter())
- {
- original.data = updated.data().to_vec();
- original.lamports = updated.lamports();
- original.owner = *updated.owner();
- }
- result
- .raw_result
- .map_err(|e| ProgramError::try_from(e).unwrap())
- }
- fn do_process_instruction_dups(
- instruction: Instruction,
- account_infos: Vec<AccountInfo>,
- checks: &[Check],
- ) -> ProgramResult {
- let mut cached_accounts = HashMap::new();
- let mut dedup_accounts = Vec::new();
- // Dedup accounts for mollusk.
- account_infos.iter().for_each(|account_info| {
- if !cached_accounts.contains_key(account_info.key) {
- let account = SolanaAccount {
- lamports: account_info.lamports(),
- data: account_info.try_borrow_data().unwrap().to_vec(),
- owner: *account_info.owner,
- executable: account_info.executable,
- rent_epoch: account_info.rent_epoch,
- };
- dedup_accounts.push((*account_info.key, account));
- cached_accounts.insert(account_info.key, account_info);
- }
- });
- let mollusk = Mollusk::new(&spl_token::ID, "spl_token");
- let result = mollusk.process_and_validate_instruction(&instruction, &dedup_accounts, checks);
- // Update accounts after the instruction is processed.
- result
- .resulting_accounts
- .into_iter()
- .for_each(|(pubkey, account)| {
- let account = account.clone();
- let account_info = cached_accounts.get(&pubkey).unwrap();
- if account.data.is_empty() {
- // When the account is closed, the tests expect the data to
- // be zeroed.
- account_info.try_borrow_mut_data().unwrap().fill(0);
- } else {
- account_info
- .try_borrow_mut_data()
- .unwrap()
- .copy_from_slice(account.data());
- }
- **account_info.try_borrow_mut_lamports().unwrap() = account.lamports();
- account_info.assign(account.owner());
- });
- result
- .raw_result
- .map_err(|e| ProgramError::try_from(e).unwrap())
- }
- fn rent_sysvar() -> SolanaAccount {
- create_account_for_test(&Rent::default())
- }
- fn mint_minimum_balance() -> u64 {
- Rent::default().minimum_balance(Mint::get_packed_len())
- }
- fn account_minimum_balance() -> u64 {
- Rent::default().minimum_balance(Account::get_packed_len())
- }
- fn multisig_minimum_balance() -> u64 {
- Rent::default().minimum_balance(Multisig::get_packed_len())
- }
- #[test]
- fn test_initialize_mint() {
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account = SolanaAccount::new(42, Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // mint is not rent exempt
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::err(TokenError::NotRentExempt.into())],
- )
- );
- mint_account.lamports = mint_minimum_balance();
- // create new mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create twice
- assert_eq!(
- Err(TokenError::AlreadyInUse.into()),
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2,).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::err(TokenError::AlreadyInUse.into())],
- )
- );
- // create another mint that can freeze
- do_process_instruction(
- initialize_mint(&program_id, &mint2_key, &owner_key, Some(&owner_key), 2).unwrap(),
- vec![&mut mint2_account, &mut rent_sysvar],
- &[
- Check::success(),
- // freeze authority is set
- Check::account(&mint2_key)
- .data_slice(46, &[1, 0, 0, 0])
- .build(),
- // freeze authority matches owner
- Check::account(&mint2_key)
- .data_slice(50, owner_key.as_ref())
- .build(),
- ],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint2_account.data).unwrap();
- assert_eq!(mint.freeze_authority, COption::Some(owner_key));
- }
- #[test]
- fn test_initialize_mint2() {
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account = SolanaAccount::new(42, Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- // mint is not rent exempt
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_mint2(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account],
- &[Check::err(TokenError::NotRentExempt.into())],
- )
- );
- mint_account.lamports = mint_minimum_balance();
- // create new mint
- do_process_instruction(
- initialize_mint2(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account],
- &[Check::success()],
- )
- .unwrap();
- // create twice
- assert_eq!(
- Err(TokenError::AlreadyInUse.into()),
- do_process_instruction(
- initialize_mint2(&program_id, &mint_key, &owner_key, None, 2,).unwrap(),
- vec![&mut mint_account],
- &[Check::err(TokenError::AlreadyInUse.into())],
- )
- );
- // create another mint that can freeze
- do_process_instruction(
- initialize_mint2(&program_id, &mint2_key, &owner_key, Some(&owner_key), 2).unwrap(),
- vec![&mut mint2_account],
- &[
- Check::success(),
- // freeze authority is set
- Check::account(&mint2_key)
- .data_slice(46, &[1, 0, 0, 0])
- .build(),
- // freeze authority matches owner
- Check::account(&mint2_key)
- .data_slice(50, owner_key.as_ref())
- .build(),
- ],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint2_account.data).unwrap();
- assert_eq!(mint.freeze_authority, COption::Some(owner_key));
- }
- #[test]
- fn test_initialize_mint_account() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(42, Account::get_packed_len(), &program_id);
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // account is not rent exempt
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- &[Check::err(TokenError::NotRentExempt.into())],
- )
- );
- account_account.lamports = account_minimum_balance();
- // mint is not valid (not initialized)
- assert_eq!(
- Err(TokenError::InvalidMint.into()),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- &[Check::err(TokenError::InvalidMint.into())],
- )
- );
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // mint not owned by program
- let not_program_id = Pubkey::new_unique();
- mint_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- mint_account.owner = program_id;
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create twice
- assert_eq!(
- Err(TokenError::AlreadyInUse.into()),
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar
- ],
- &[Check::err(TokenError::AlreadyInUse.into())],
- )
- );
- }
- #[test]
- fn test_transfer_dups() {
- let program_id = spl_token::id();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account2_info: AccountInfo = (&account2_key, false, &mut account2_account).into();
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_info: AccountInfo = (&account3_key, false, &mut account3_account).into();
- let account4_key = Pubkey::new_unique();
- let mut account4_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account4_info: AccountInfo = (&account4_key, true, &mut account4_account).into();
- let multisig_key = Pubkey::new_unique();
- let mut multisig_account = SolanaAccount::new(
- multisig_minimum_balance(),
- Multisig::get_packed_len(),
- &program_id,
- );
- let multisig_info: AccountInfo = (&multisig_key, true, &mut multisig_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction_dups(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- account2_info.clone(),
- mint_info.clone(),
- owner_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // source-owner transfer
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-owner TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-delegate transfer
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.amount = 1000;
- account.delegated_amount = 1000;
- account.delegate = COption::Some(account1_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-delegate TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // test destination-owner transfer
- do_process_instruction_dups(
- initialize_account(&program_id, &account3_key, &mint_key, &account2_key).unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account3_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account3_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- account1_info.is_signer = false;
- account2_info.is_signer = true;
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account3_key,
- &account2_key,
- &account2_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- account2_info.clone(),
- account2_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // destination-owner TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account3_key,
- &mint_key,
- &account2_key,
- &account2_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account2_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // test source-multisig signer
- do_process_instruction_dups(
- initialize_multisig(&program_id, &multisig_key, &[&account4_key], 1).unwrap(),
- vec![
- multisig_info.clone(),
- rent_info.clone(),
- account4_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction_dups(
- initialize_account(&program_id, &account4_key, &mint_key, &multisig_key).unwrap(),
- vec![
- account4_info.clone(),
- mint_info.clone(),
- multisig_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account4_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account4_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // source-multisig-signer transfer
- do_process_instruction_dups(
- transfer(
- &program_id,
- &account4_key,
- &account2_key,
- &multisig_key,
- &[&account4_key],
- 500,
- )
- .unwrap(),
- vec![
- account4_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account4_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-multisig-signer TransferChecked
- do_process_instruction_dups(
- transfer_checked(
- &program_id,
- &account4_key,
- &mint_key,
- &account2_key,
- &multisig_key,
- &[&account4_key],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account4_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account4_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_transfer() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let mismatch_key = Pubkey::new_unique();
- let mut mismatch_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create mismatch account
- do_process_instruction(
- initialize_account(&program_id, &mismatch_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut mismatch_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&mismatch_account.data).unwrap();
- account.mint = mint2_key;
- Account::pack(account, &mut mismatch_account.data).unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // missing signer
- let mut instruction = transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 1000,
- )
- .unwrap();
- instruction.accounts[2].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::err(ProgramError::MissingRequiredSignature)],
- )
- );
- // mismatch mint
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &mismatch_key,
- &owner_key,
- &[],
- 1000
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mismatch_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::MintMismatch.into())],
- )
- );
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner2_key,
- &[],
- 1000
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // account not owned by program
- let not_program_id = Pubkey::new_unique();
- account_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- transfer(&program_id, &account_key, &account2_key, &owner_key, &[], 0,).unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- account_account.owner = program_id;
- // account 2 not owned by program
- let not_program_id = Pubkey::new_unique();
- account2_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- transfer(&program_id, &account_key, &account2_key, &owner_key, &[], 0,).unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- account2_account.owner = program_id;
- // transfer
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 1000,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // insufficient funds
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(&program_id, &account_key, &account2_key, &owner_key, &[], 1).unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // transfer half back
- do_process_instruction(
- transfer(
- &program_id,
- &account2_key,
- &account_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut account_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- transfer_checked(
- &program_id,
- &account2_key,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 1,
- 10 // <-- incorrect decimals
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut account_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::MintDecimalsMismatch.into())],
- )
- );
- // incorrect mint
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- transfer_checked(
- &program_id,
- &account2_key,
- &account3_key, // <-- incorrect mint
- &account_key,
- &owner_key,
- &[],
- 1,
- 2
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut account3_account, // <-- incorrect mint
- &mut account_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::MintMismatch.into())],
- )
- );
- // transfer rest with explicit decimals
- do_process_instruction(
- transfer_checked(
- &program_id,
- &account2_key,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut account_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // insufficient funds
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(&program_id, &account2_key, &account_key, &owner_key, &[], 1).unwrap(),
- vec![
- &mut account2_account,
- &mut account_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // not a delegate of source account
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner2_key, // <-- incorrect owner or delegate
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // insufficient funds approved via delegate
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 101
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // transfer via delegate
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // insufficient funds approved via delegate
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 1
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // transfer rest
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 900,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // insufficient funds in source account via delegate
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &delegate_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut delegate_account,
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- }
- #[test]
- fn test_self_transfer() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- let account_info = (&account_key, false, &mut account_account).into_account_info();
- let account3_info = (&account3_key, false, &mut account3_account).into_account_info();
- let delegate_info = (&delegate_key, true, &mut delegate_account).into_account_info();
- let owner_info = (&owner_key, true, &mut owner_account).into_account_info();
- let owner2_info = (&owner2_key, true, &mut owner2_account).into_account_info();
- let mint_info = (&mint_key, false, &mut mint_account).into_account_info();
- // transfer
- let instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1000,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(account_info.key)
- .data_slice(64, &1000u64.to_le_bytes())
- .build()
- ],
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- // transfer checked
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(account_info.key)
- .data_slice(64, &1000u64.to_le_bytes())
- .build()
- ],
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- // missing signer
- let mut owner_no_sign_info = owner_info.clone();
- let mut instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- owner_no_sign_info.key,
- &[],
- 1000,
- )
- .unwrap();
- instruction.accounts[2].is_signer = false;
- owner_no_sign_info.is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- owner_no_sign_info.clone(),
- ],
- &[Check::err(ProgramError::MissingRequiredSignature)],
- )
- );
- // missing signer checked
- let mut instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- owner_no_sign_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- instruction.accounts[3].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_no_sign_info,
- ],
- &[Check::err(ProgramError::MissingRequiredSignature)],
- )
- );
- // missing owner
- let instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- owner2_info.key,
- &[],
- 1000,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- owner2_info.clone(),
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // missing owner checked
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- owner2_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner2_info.clone(),
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // insufficient funds
- let instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1001,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // insufficient funds checked
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1001,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // incorrect decimals
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1,
- 10, // <-- incorrect decimals
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[Check::err(TokenError::MintDecimalsMismatch.into())],
- )
- );
- // incorrect mint
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- account3_info.key, // <-- incorrect mint
- account_info.key,
- owner_info.key,
- &[],
- 1,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account3_info.clone(), // <-- incorrect mint
- account_info.clone(),
- owner_info.clone(),
- ],
- &[Check::err(TokenError::MintMismatch.into())],
- )
- );
- // approve delegate
- let instruction = approve(
- &program_id,
- account_info.key,
- delegate_info.key,
- owner_info.key,
- &[],
- 100,
- )
- .unwrap();
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- delegate_info.clone(),
- owner_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // delegate transfer
- let instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- delegate_info.key,
- &[],
- 100,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(account_info.key)
- .data_slice(64, &1000u64.to_le_bytes())
- .build(),
- Check::account(&account_key)
- .data_slice(121, &100u64.to_le_bytes())
- .build(),
- ],
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- assert_eq!(account.delegated_amount, 100);
- // delegate transfer checked
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- delegate_info.key,
- &[],
- 100,
- 2,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(account_info.key)
- .data_slice(64, &1000u64.to_le_bytes())
- .build(),
- Check::account(&account_key)
- .data_slice(121, &100u64.to_le_bytes())
- .build(),
- ],
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- assert_eq!(account.delegated_amount, 100);
- // delegate insufficient funds
- let instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- delegate_info.key,
- &[],
- 101,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // delegate insufficient funds checked
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- delegate_info.key,
- &[],
- 101,
- 2,
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- delegate_info.clone(),
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // owner transfer with delegate assigned
- let instruction = transfer(
- &program_id,
- account_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1000,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(account_info.key)
- .data_slice(64, &1000u64.to_le_bytes())
- .build(),
- ],
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- // owner transfer with delegate assigned checked
- let instruction = transfer_checked(
- &program_id,
- account_info.key,
- mint_info.key,
- account_info.key,
- owner_info.key,
- &[],
- 1000,
- 2,
- )
- .unwrap();
- assert_eq!(
- Ok(()),
- do_process_instruction_dups(
- instruction,
- vec![
- account_info.clone(),
- mint_info.clone(),
- account_info.clone(),
- owner_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(account_info.key)
- .data_slice(64, &1000u64.to_le_bytes())
- .build(),
- ],
- )
- );
- // no balance change...
- let account = Account::unpack_unchecked(&account_info.try_borrow_data().unwrap()).unwrap();
- assert_eq!(account.amount, 1000);
- }
- #[test]
- fn test_mintable_token_with_zero_supply() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint-able token with zero supply
- let decimals = 2;
- let expected_mint = Mint {
- mint_authority: COption::Some(owner_key),
- supply: 0,
- decimals,
- is_initialized: true,
- freeze_authority: COption::None,
- };
- let mut mint_data = [0u8; Mint::LEN];
- expected_mint.pack_into_slice(&mut mint_data);
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, decimals).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[
- Check::success(),
- Check::account(&mint_key).data(&mint_data).build(),
- ],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint, expected_mint);
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(64, &42u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let _ = Mint::unpack(&mint_account.data).unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // mint to 2, with incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- mint_to_checked(
- &program_id,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 42,
- decimals + 1
- )
- .unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[
- Check::err(TokenError::MintDecimalsMismatch.into()),
- Check::account(&account_key)
- .data_slice(64, &42u64.to_le_bytes())
- .build(),
- ],
- )
- );
- let _ = Mint::unpack(&mint_account.data).unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // mint to 2
- do_process_instruction(
- mint_to_checked(
- &program_id,
- &mint_key,
- &account_key,
- &owner_key,
- &[],
- 42,
- decimals,
- )
- .unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(64, &84u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let _ = Mint::unpack(&mint_account.data).unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 84);
- }
- #[test]
- fn test_approve_dups() {
- let program_id = spl_token::id();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_info: AccountInfo = (&account2_key, false, &mut account2_account).into();
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_info: AccountInfo = (&account3_key, true, &mut account3_account).into();
- let multisig_key = Pubkey::new_unique();
- let mut multisig_account = SolanaAccount::new(
- multisig_minimum_balance(),
- Multisig::get_packed_len(),
- &program_id,
- );
- let multisig_info: AccountInfo = (&multisig_key, true, &mut multisig_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction_dups(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- account2_info.clone(),
- mint_info.clone(),
- owner_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // source-owner approve
- do_process_instruction_dups(
- approve(
- &program_id,
- &account1_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-owner approve_checked
- do_process_instruction_dups(
- approve_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account2_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-owner revoke
- do_process_instruction_dups(
- revoke(&program_id, &account1_key, &account1_key, &[]).unwrap(),
- vec![account1_info.clone(), account1_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // test source-multisig signer
- do_process_instruction_dups(
- initialize_multisig(&program_id, &multisig_key, &[&account3_key], 1).unwrap(),
- vec![
- multisig_info.clone(),
- rent_info.clone(),
- account3_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction_dups(
- initialize_account(&program_id, &account3_key, &mint_key, &multisig_key).unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- multisig_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account3_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account3_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // source-multisig-signer approve
- do_process_instruction_dups(
- approve(
- &program_id,
- &account3_key,
- &account2_key,
- &multisig_key,
- &[&account3_key],
- 500,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account3_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-multisig-signer approve_checked
- do_process_instruction_dups(
- approve_checked(
- &program_id,
- &account3_key,
- &mint_key,
- &account2_key,
- &multisig_key,
- &[&account3_key],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account3_info.clone(),
- mint_info.clone(),
- account2_info.clone(),
- multisig_info.clone(),
- account3_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-owner multisig-signer
- do_process_instruction_dups(
- revoke(&program_id, &account3_key, &multisig_key, &[&account3_key]).unwrap(),
- vec![
- account3_info.clone(),
- multisig_info.clone(),
- account3_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_approve() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // missing signer
- let mut instruction = approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap();
- instruction.accounts[2].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::err(ProgramError::MissingRequiredSignature)],
- )
- );
- // no owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner2_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner2_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // approve delegate 2, with incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- approve_checked(
- &program_id,
- &account_key,
- &mint_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- 0 // <-- incorrect decimals
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::MintDecimalsMismatch.into())],
- )
- );
- // approve delegate 2, with incorrect mint
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- approve_checked(
- &program_id,
- &account_key,
- &account2_key, // <-- bad mint
- &delegate_key,
- &owner_key,
- &[],
- 100,
- 0
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account, // <-- bad mint
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::MintMismatch.into())],
- )
- );
- // approve delegate 2
- do_process_instruction(
- approve_checked(
- &program_id,
- &account_key,
- &mint_key,
- &delegate_key,
- &owner_key,
- &[],
- 100,
- 2,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // revoke delegate
- do_process_instruction(
- revoke(&program_id, &account_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_set_authority_dups() {
- let program_id = spl_token::id();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &mint_key, Some(&mint_key), 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // set mint_authority when currently self
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner_key),
- AuthorityType::MintTokens,
- &mint_key,
- &[],
- )
- .unwrap(),
- vec![mint_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // set freeze_authority when currently self
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner_key),
- AuthorityType::FreezeAccount,
- &mint_key,
- &[],
- )
- .unwrap(),
- vec![mint_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // set account owner when currently self
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &account1_key,
- Some(&owner_key),
- AuthorityType::AccountOwner,
- &account1_key,
- &[],
- )
- .unwrap(),
- vec![account1_info.clone(), account1_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // set close_authority when currently self
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.close_authority = COption::Some(account1_key);
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- set_authority(
- &program_id,
- &account1_key,
- Some(&owner_key),
- AuthorityType::CloseAccount,
- &account1_key,
- &[],
- )
- .unwrap(),
- vec![account1_info.clone(), account1_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_set_authority() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let owner3_key = Pubkey::new_unique();
- let mut owner3_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create mint with owner and freeze_authority
- do_process_instruction(
- initialize_mint(&program_id, &mint2_key, &owner_key, Some(&owner_key), 2).unwrap(),
- vec![&mut mint2_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // invalid account
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[Check::err(ProgramError::UninitializedAccount)],
- )
- );
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint2_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint2_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner_key),
- AuthorityType::AccountOwner,
- &owner2_key,
- &[]
- )
- .unwrap(),
- vec![&mut account_account, &mut owner2_account],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // owner did not sign
- let mut instruction = set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap();
- instruction.accounts[1].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![&mut account_account, &mut owner_account,],
- &[Check::err(ProgramError::MissingRequiredSignature)]
- ),
- );
- // wrong authority type
- assert_eq!(
- Err(TokenError::AuthorityTypeNotSupported.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[Check::err(TokenError::AuthorityTypeNotSupported.into())],
- )
- );
- // account owner may not be set to None
- assert_eq!(
- Err(TokenError::InvalidInstruction.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- None,
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[Check::err(TokenError::InvalidInstruction.into())],
- )
- );
- // set delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &owner2_key,
- &owner_key,
- &[],
- u64::MAX,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut owner2_account,
- &mut owner_account,
- ],
- &[
- Check::success(),
- // delegate set
- Check::account(&account_key)
- .data_slice(72, &[1, 0, 0, 0])
- .build(),
- // delegate
- Check::account(&account_key)
- .data_slice(76, owner2_key.as_ref())
- .build(),
- // delegated amount
- Check::account(&account_key)
- .data_slice(121, &u64::MAX.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.delegate, COption::Some(owner2_key));
- assert_eq!(account.delegated_amount, u64::MAX);
- // set owner
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner3_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[
- Check::success(),
- // delegate not set
- Check::account(&account_key)
- .data_slice(72, &[0, 0, 0, 0])
- .build(),
- // delegated amount
- Check::account(&account_key)
- .data_slice(121, &0u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- // check delegate cleared
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.delegate, COption::None);
- assert_eq!(account.delegated_amount, 0);
- // set owner without existing delegate
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner3_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner3_account],
- &[Check::success()],
- )
- .unwrap();
- // set close_authority
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::CloseAccount,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner2_account],
- &[Check::success()],
- )
- .unwrap();
- // close_authority may be set to None
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- None,
- AuthorityType::CloseAccount,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner2_account],
- &[Check::success()],
- )
- .unwrap();
- // wrong owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner3_key),
- AuthorityType::MintTokens,
- &owner2_key,
- &[]
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner2_account],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // owner did not sign
- let mut instruction = set_authority(
- &program_id,
- &mint_key,
- Some(&owner2_key),
- AuthorityType::MintTokens,
- &owner_key,
- &[],
- )
- .unwrap();
- instruction.accounts[1].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![&mut mint_account, &mut owner_account],
- &[Check::err(ProgramError::MissingRequiredSignature)]
- ),
- );
- // cannot freeze
- assert_eq!(
- Err(TokenError::MintCannotFreeze.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- &[Check::err(TokenError::MintCannotFreeze.into())],
- )
- );
- // set owner
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner2_key),
- AuthorityType::MintTokens,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // set owner to None
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- None,
- AuthorityType::MintTokens,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner2_account],
- &[Check::success()],
- )
- .unwrap();
- // test unsetting mint_authority is one-way operation
- assert_eq!(
- Err(TokenError::FixedSupply.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- Some(&owner2_key),
- AuthorityType::MintTokens,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- &[Check::err(TokenError::FixedSupply.into())],
- )
- );
- // set freeze_authority
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint2_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // test unsetting freeze_authority is one-way operation
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- None,
- AuthorityType::FreezeAccount,
- &owner2_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint2_account, &mut owner2_account],
- &[Check::success()],
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::MintCannotFreeze.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &mint2_key,
- Some(&owner2_key),
- AuthorityType::FreezeAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint2_account, &mut owner2_account],
- &[Check::err(TokenError::MintCannotFreeze.into())],
- )
- );
- }
- #[test]
- fn test_mint_to_dups() {
- let program_id = spl_token::id();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &mint_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &owner_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- owner_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint_to when mint_authority is self
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &mint_key, &[], 42).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // mint_to_checked when mint_authority is self
- do_process_instruction_dups(
- mint_to_checked(&program_id, &mint_key, &account1_key, &mint_key, &[], 42, 2).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // mint_to when mint_authority is account owner
- let mut mint = Mint::unpack_unchecked(&mint_info.data.borrow()).unwrap();
- mint.mint_authority = COption::Some(account1_key);
- Mint::pack(mint, &mut mint_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- mint_to(
- &program_id,
- &mint_key,
- &account1_key,
- &account1_key,
- &[],
- 42,
- )
- .unwrap(),
- vec![
- mint_info.clone(),
- account1_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint_to_checked when mint_authority is account owner
- do_process_instruction_dups(
- mint_to(
- &program_id,
- &mint_key,
- &account1_key,
- &account1_key,
- &[],
- 42,
- )
- .unwrap(),
- vec![
- mint_info.clone(),
- account1_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_mint_to() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mismatch_key = Pubkey::new_unique();
- let mut mismatch_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let uninitialized_key = Pubkey::new_unique();
- let mut uninitialized_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create mismatch account
- do_process_instruction(
- initialize_account(&program_id, &mismatch_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut mismatch_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&mismatch_account.data).unwrap();
- account.mint = mint2_key;
- Account::pack(account, &mut mismatch_account.data).unwrap();
- // mint to
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&mint_key)
- .data_slice(36, &42u64.to_le_bytes())
- .build(),
- Check::account(&account_key)
- .data_slice(64, &42u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 42);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // mint to another account to test supply accumulation
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account2_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account2_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&mint_key)
- .data_slice(36, &84u64.to_le_bytes())
- .build(),
- Check::account(&account2_key)
- .data_slice(64, &42u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 84);
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // missing signer
- let mut instruction =
- mint_to(&program_id, &mint_key, &account2_key, &owner_key, &[], 42).unwrap();
- instruction.accounts[2].is_signer = false;
- assert_eq!(
- Err(ProgramError::MissingRequiredSignature),
- do_process_instruction(
- instruction,
- vec![&mut mint_account, &mut account2_account, &mut owner_account],
- &[Check::err(ProgramError::MissingRequiredSignature)],
- )
- );
- // mismatch account
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &mismatch_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut mismatch_account, &mut owner_account],
- &[Check::err(TokenError::MintMismatch.into())],
- )
- );
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account2_key, &owner2_key, &[], 42).unwrap(),
- vec![
- &mut mint_account,
- &mut account2_account,
- &mut owner2_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // mint not owned by program
- let not_program_id = Pubkey::new_unique();
- mint_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 0).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- mint_account.owner = program_id;
- // account not owned by program
- let not_program_id = Pubkey::new_unique();
- account_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 0).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- account_account.owner = program_id;
- // uninitialized destination account
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &uninitialized_key,
- &owner_key,
- &[],
- 42
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut uninitialized_account,
- &mut owner_account,
- ],
- &[Check::err(ProgramError::UninitializedAccount)],
- )
- );
- // unset mint_authority and test minting fails
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- None,
- AuthorityType::MintTokens,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut mint_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::FixedSupply.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account2_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mint_account, &mut account2_account, &mut owner_account],
- &[Check::err(TokenError::FixedSupply.into())],
- )
- );
- }
- #[test]
- fn test_burn_dups() {
- let program_id = spl_token::id();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner_info: AccountInfo = (&owner_key, true, &mut owner_account).into();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // source-owner burn
- do_process_instruction_dups(
- burn(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-owner burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint-owner burn
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.owner = mint_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- burn(&program_id, &account1_key, &mint_key, &mint_key, &[], 500).unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // mint-owner burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &mint_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // source-delegate burn
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.delegated_amount = 1000;
- account.delegate = COption::Some(account1_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- burn(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-delegate burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &account1_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint-delegate burn
- do_process_instruction_dups(
- mint_to(&program_id, &mint_key, &account1_key, &owner_key, &[], 1000).unwrap(),
- vec![mint_info.clone(), account1_info.clone(), owner_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.delegated_amount = 1000;
- account.delegate = COption::Some(mint_key);
- account.owner = owner_key;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- burn(&program_id, &account1_key, &mint_key, &mint_key, &[], 500).unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // mint-delegate burn_checked
- do_process_instruction_dups(
- burn_checked(
- &program_id,
- &account1_key,
- &mint_key,
- &mint_key,
- &[],
- 500,
- 2,
- )
- .unwrap(),
- vec![account1_info.clone(), mint_info.clone(), mint_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_burn() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- let mismatch_key = Pubkey::new_unique();
- let mut mismatch_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint2_key = Pubkey::new_unique();
- let mut rent_sysvar = rent_sysvar();
- // create new mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create mismatch account
- do_process_instruction(
- initialize_account(&program_id, &mismatch_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut mismatch_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // mint to mismatch account and change mint key
- do_process_instruction(
- mint_to(&program_id, &mint_key, &mismatch_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut mismatch_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&mismatch_account.data).unwrap();
- account.mint = mint2_key;
- Account::pack(account, &mut mismatch_account.data).unwrap();
- // missing signer
- let mut instruction =
- burn(&program_id, &account_key, &mint_key, &delegate_key, &[], 42).unwrap();
- instruction.accounts[1].is_signer = false;
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- instruction,
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // missing owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner2_key, &[], 42).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // account not owned by program
- let not_program_id = Pubkey::new_unique();
- account_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 0).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- account_account.owner = program_id;
- // mint not owned by program
- let not_program_id = Pubkey::new_unique();
- mint_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 0).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(ProgramError::IncorrectProgramId)],
- )
- );
- mint_account.owner = program_id;
- // mint mismatch
- assert_eq!(
- Err(TokenError::MintMismatch.into()),
- do_process_instruction(
- burn(&program_id, &mismatch_key, &mint_key, &owner_key, &[], 42).unwrap(),
- vec![&mut mismatch_account, &mut mint_account, &mut owner_account],
- &[Check::err(TokenError::MintMismatch.into())],
- )
- );
- // burn
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 21).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // burn_checked, with incorrect decimals
- assert_eq!(
- Err(TokenError::MintDecimalsMismatch.into()),
- do_process_instruction(
- burn_checked(&program_id, &account_key, &mint_key, &owner_key, &[], 21, 3).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(TokenError::MintDecimalsMismatch.into())],
- )
- );
- // burn_checked
- do_process_instruction(
- burn_checked(&program_id, &account_key, &mint_key, &owner_key, &[], 21, 2).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&mint_key)
- .data_slice(36, &(2000u64 - 42).to_le_bytes())
- .build(),
- Check::account(&account_key)
- .data_slice(64, &(1000u64 - 42).to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 2000 - 42);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 1000 - 42);
- // insufficient funds
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &owner_key,
- &[],
- 100_000_000
- )
- .unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // approve delegate
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 84,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // not a delegate of source account
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &owner2_key, // <-- incorrect owner or delegate
- &[],
- 1,
- )
- .unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // insufficient funds approved via delegate
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &delegate_key, &[], 85).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // burn via delegate
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &delegate_key, &[], 84).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account,
- ],
- &[
- Check::success(),
- Check::account(&mint_key)
- .data_slice(36, &(2000u64 - 42 - 84).to_le_bytes())
- .build(),
- Check::account(&account_key)
- .data_slice(64, &(1000u64 - 42 - 84).to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- // match
- let mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- assert_eq!(mint.supply, 2000 - 42 - 84);
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 1000 - 42 - 84);
- // insufficient funds approved via delegate
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &delegate_key, &[], 1).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut delegate_account
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- }
- #[test]
- fn test_burn_and_close_system_and_incinerator_tokens() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let incinerator_account_key = Pubkey::new_unique();
- let mut incinerator_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let system_account_key = Pubkey::new_unique();
- let mut system_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let recipient_key = Pubkey::new_unique();
- let mut recipient_account = SolanaAccount::default();
- let mut mock_incinerator_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- // create new mint
- do_process_instruction(
- initialize_mint2(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account3(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![&mut account_account, &mut mint_account],
- &[Check::success()],
- )
- .unwrap();
- // create incinerator- and system-owned accounts
- do_process_instruction(
- initialize_account3(
- &program_id,
- &incinerator_account_key,
- &mint_key,
- &solana_sdk_ids::incinerator::id(),
- )
- .unwrap(),
- vec![&mut incinerator_account, &mut mint_account],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- initialize_account3(
- &program_id,
- &system_account_key,
- &mint_key,
- &solana_sdk_ids::system_program::id(),
- )
- .unwrap(),
- vec![&mut system_account, &mut mint_account],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // transfer half to incinerator, half to system program
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &incinerator_account_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut incinerator_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &system_account_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut system_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // close with balance fails
- assert_eq!(
- Err(TokenError::NonNativeHasBalance.into()),
- do_process_instruction(
- close_account(
- &program_id,
- &incinerator_account_key,
- &solana_sdk_ids::incinerator::id(),
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![
- &mut incinerator_account,
- &mut mock_incinerator_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::NonNativeHasBalance.into())],
- )
- );
- assert_eq!(
- Err(TokenError::NonNativeHasBalance.into()),
- do_process_instruction(
- close_account(
- &program_id,
- &system_account_key,
- &solana_sdk_ids::incinerator::id(),
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![
- &mut system_account,
- &mut mock_incinerator_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::NonNativeHasBalance.into())],
- )
- );
- // anyone can burn
- do_process_instruction(
- burn(
- &program_id,
- &incinerator_account_key,
- &mint_key,
- &recipient_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut incinerator_account,
- &mut mint_account,
- &mut recipient_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- burn(
- &program_id,
- &system_account_key,
- &mint_key,
- &recipient_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut system_account,
- &mut mint_account,
- &mut recipient_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- // closing fails if destination is not the incinerator
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- do_process_instruction(
- close_account(
- &program_id,
- &incinerator_account_key,
- &recipient_key,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![
- &mut incinerator_account,
- &mut recipient_account,
- &mut owner_account,
- ],
- &[Check::err(ProgramError::InvalidAccountData)],
- )
- );
- assert_eq!(
- Err(ProgramError::InvalidAccountData),
- do_process_instruction(
- close_account(
- &program_id,
- &system_account_key,
- &recipient_key,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![
- &mut system_account,
- &mut recipient_account,
- &mut owner_account,
- ],
- &[Check::err(ProgramError::InvalidAccountData)],
- )
- );
- // closing succeeds with incinerator recipient
- do_process_instruction(
- close_account(
- &program_id,
- &incinerator_account_key,
- &solana_sdk_ids::incinerator::id(),
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![
- &mut incinerator_account,
- &mut mock_incinerator_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- close_account(
- &program_id,
- &system_account_key,
- &solana_sdk_ids::incinerator::id(),
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![
- &mut system_account,
- &mut mock_incinerator_account,
- &mut owner_account,
- ],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_multisig() {
- let program_id = spl_token::id();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let account_key = Pubkey::new_unique();
- let mut account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let multisig_key = Pubkey::new_unique();
- let mut multisig_account = SolanaAccount::new(42, Multisig::get_packed_len(), &program_id);
- let multisig_delegate_key = Pubkey::new_unique();
- let mut multisig_delegate_account = SolanaAccount::new(
- multisig_minimum_balance(),
- Multisig::get_packed_len(),
- &program_id,
- );
- let signer_keys = vec![Pubkey::new_unique(); MAX_SIGNERS];
- let signer_key_refs: Vec<&Pubkey> = signer_keys.iter().collect();
- let mut signer_accounts = vec![SolanaAccount::new(0, 0, &program_id); MAX_SIGNERS];
- let mut rent_sysvar = rent_sysvar();
- // multisig is not rent exempt
- let account_info_iter = &mut signer_accounts.iter_mut();
- assert_eq!(
- Err(TokenError::NotRentExempt.into()),
- do_process_instruction(
- initialize_multisig(&program_id, &multisig_key, &[&signer_keys[0]], 1).unwrap(),
- vec![
- &mut multisig_account,
- &mut rent_sysvar,
- account_info_iter.next().unwrap(),
- ],
- &[Check::err(TokenError::NotRentExempt.into())],
- )
- );
- multisig_account.lamports = multisig_minimum_balance();
- let mut multisig_account2 = multisig_account.clone();
- // single signer
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- initialize_multisig(&program_id, &multisig_key, &[&signer_keys[0]], 1).unwrap(),
- vec![
- &mut multisig_account,
- &mut rent_sysvar,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // single signer using `initialize_multisig2`
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- initialize_multisig2(&program_id, &multisig_key, &[&signer_keys[0]], 1).unwrap(),
- vec![&mut multisig_account2, account_info_iter.next().unwrap()],
- &[Check::success()],
- )
- .unwrap();
- // multiple signer
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- initialize_multisig(
- &program_id,
- &multisig_delegate_key,
- &signer_key_refs,
- MAX_SIGNERS as u8,
- )
- .unwrap(),
- vec![
- &mut multisig_delegate_account,
- &mut rent_sysvar,
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // create new mint with multisig owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &multisig_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account with multisig owner
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &multisig_key).unwrap(),
- vec![
- &mut account,
- &mut mint_account,
- &mut multisig_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account with multisig owner
- do_process_instruction(
- initialize_account(
- &program_id,
- &account2_key,
- &mint_key,
- &multisig_delegate_key,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut multisig_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &multisig_key,
- &[&signer_keys[0]],
- 1000,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // approve
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &multisig_delegate_key,
- &multisig_key,
- &[&signer_keys[0]],
- 100,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut multisig_delegate_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // transfer
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &multisig_key,
- &[&signer_keys[0]],
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut account2_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // transfer via delegate
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &multisig_delegate_key,
- &signer_key_refs,
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut account2_account,
- &mut multisig_delegate_account,
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account2_key,
- &multisig_key,
- &[&signer_keys[0]],
- 42,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account2_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // burn
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &multisig_key,
- &[&signer_keys[0]],
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut mint_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // burn via delegate
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &mint_key,
- &multisig_delegate_key,
- &signer_key_refs,
- 42,
- )
- .unwrap(),
- vec![
- &mut account,
- &mut mint_account,
- &mut multisig_delegate_account,
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // freeze account
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mint2_key = Pubkey::new_unique();
- let mut mint2_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- do_process_instruction(
- initialize_mint(
- &program_id,
- &mint2_key,
- &multisig_key,
- Some(&multisig_key),
- 2,
- )
- .unwrap(),
- vec![&mut mint2_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- initialize_account(&program_id, &account3_key, &mint2_key, &owner_key).unwrap(),
- vec![
- &mut account3_account,
- &mut mint2_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- mint_to(
- &program_id,
- &mint2_key,
- &account3_key,
- &multisig_key,
- &[&signer_keys[0]],
- 1000,
- )
- .unwrap(),
- vec![
- &mut mint2_account,
- &mut account3_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- freeze_account(
- &program_id,
- &account3_key,
- &mint2_key,
- &multisig_key,
- &[&signer_keys[0]],
- )
- .unwrap(),
- vec![
- &mut account3_account,
- &mut mint2_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // do SetAuthority on mint
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- set_authority(
- &program_id,
- &mint_key,
- Some(&owner_key),
- AuthorityType::MintTokens,
- &multisig_key,
- &[&signer_keys[0]],
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // do SetAuthority on account
- let account_info_iter = &mut signer_accounts.iter_mut();
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner_key),
- AuthorityType::AccountOwner,
- &multisig_key,
- &[&signer_keys[0]],
- )
- .unwrap(),
- vec![
- &mut account,
- &mut multisig_account,
- account_info_iter.next().unwrap(),
- ],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_owner_close_account_dups() {
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- let to_close_key = Pubkey::new_unique();
- let mut to_close_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let to_close_account_info: AccountInfo = (&to_close_key, true, &mut to_close_account).into();
- let destination_account_key = Pubkey::new_unique();
- let mut destination_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let destination_account_info: AccountInfo =
- (&destination_account_key, true, &mut destination_account).into();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &to_close_key, &mint_key, &to_close_key).unwrap(),
- vec![
- to_close_account_info.clone(),
- mint_info.clone(),
- to_close_account_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // source-owner close
- do_process_instruction_dups(
- close_account(
- &program_id,
- &to_close_key,
- &destination_account_key,
- &to_close_key,
- &[],
- )
- .unwrap(),
- vec![
- to_close_account_info.clone(),
- destination_account_info.clone(),
- to_close_account_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(&to_close_key).data(&[]).build(),
- ],
- )
- .unwrap();
- assert_eq!(*to_close_account_info.data.borrow(), &[0u8; Account::LEN]);
- }
- #[test]
- fn test_close_authority_close_account_dups() {
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, false, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- let to_close_key = Pubkey::new_unique();
- let mut to_close_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let to_close_account_info: AccountInfo = (&to_close_key, true, &mut to_close_account).into();
- let destination_account_key = Pubkey::new_unique();
- let mut destination_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let destination_account_info: AccountInfo =
- (&destination_account_key, true, &mut destination_account).into();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &to_close_key, &mint_key, &to_close_key).unwrap(),
- vec![
- to_close_account_info.clone(),
- mint_info.clone(),
- to_close_account_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- let mut account = Account::unpack_unchecked(&to_close_account_info.data.borrow()).unwrap();
- account.close_authority = COption::Some(to_close_key);
- account.owner = owner_key;
- Account::pack(account, &mut to_close_account_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- close_account(
- &program_id,
- &to_close_key,
- &destination_account_key,
- &to_close_key,
- &[],
- )
- .unwrap(),
- vec![
- to_close_account_info.clone(),
- destination_account_info.clone(),
- to_close_account_info.clone(),
- ],
- &[
- Check::success(),
- Check::account(&to_close_key).data(&[]).build(),
- ],
- )
- .unwrap();
- assert_eq!(*to_close_account_info.data.borrow(), &[0u8; Account::LEN]);
- }
- #[test]
- fn test_close_account() {
- let program_id = spl_token::id();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance() + 42,
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mut rent_sysvar = rent_sysvar();
- // uninitialized
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- &[Check::err(ProgramError::UninitializedAccount)],
- )
- );
- // initialize and mint to non-native account
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 42).unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(64, &42u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, 42);
- // initialize native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &account2_key,
- &spl_token::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[
- Check::success(),
- Check::account(&account2_key)
- .data_slice(109, &[1, 0, 0, 0])
- .build(),
- Check::account(&account2_key)
- .data_slice(64, &42u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 42);
- // close non-native account with balance
- assert_eq!(
- Err(TokenError::NonNativeHasBalance.into()),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- &[
- Check::err(TokenError::NonNativeHasBalance.into()),
- Check::account(&account_key)
- .lamports(account_minimum_balance())
- .build()
- ],
- )
- );
- assert_eq!(account_account.lamports, account_minimum_balance());
- // empty account
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 42).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // wrong owner
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // close account
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- &[
- Check::success(),
- Check::account(&account_key).data(&[]).build(),
- Check::account(&account_key).lamports(0).build(),
- Check::account(&account3_key)
- .lamports(2 * account_minimum_balance())
- .build(),
- ],
- )
- .unwrap();
- assert!(account_account.data.is_empty());
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account3_account.lamports, 2 * account_minimum_balance());
- // fund and initialize new non-native account to test close authority
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- account_account.lamports = 2;
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::CloseAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // account owner cannot authorize close if close_authority is set
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // close non-native account with close_authority
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- &[
- Check::success(),
- Check::account(&account_key).data(&[]).build(),
- Check::account(&account_key).lamports(0).build(),
- Check::account(&account3_key)
- .lamports(2 * account_minimum_balance() + 2)
- .build(),
- ],
- )
- .unwrap();
- assert!(account_account.data.is_empty());
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account3_account.lamports, 2 * account_minimum_balance() + 2);
- // close native account
- do_process_instruction(
- close_account(&program_id, &account2_key, &account3_key, &owner_key, &[]).unwrap(),
- vec![
- &mut account2_account,
- &mut account3_account,
- &mut owner_account,
- ],
- &[
- Check::success(),
- Check::account(&account2_key).data(&[]).build(),
- Check::account(&account3_key)
- .lamports(3 * account_minimum_balance() + 2 + 42)
- .build(),
- ],
- )
- .unwrap();
- assert!(account2_account.data.is_empty());
- assert_eq!(
- account3_account.lamports,
- 3 * account_minimum_balance() + 2 + 42
- );
- }
- #[test]
- fn test_native_token() {
- let program_id = spl_token::id();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance() + 40,
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account3_key = Pubkey::new_unique();
- let mut account3_account = SolanaAccount::new(account_minimum_balance(), 0, &program_id);
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let owner3_key = Pubkey::new_unique();
- let mut rent_sysvar = rent_sysvar();
- // initialize native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &account_key,
- &spl_token::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(109, &[1, 0, 0, 0])
- .build(),
- Check::account(&account_key)
- .data_slice(64, &40u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 40);
- // initialize native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &account2_key,
- &spl_token::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[
- Check::success(),
- Check::account(&account2_key)
- .data_slice(109, &[1, 0, 0, 0])
- .build(),
- Check::account(&account2_key)
- .data_slice(64, &0u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 0);
- // mint_to unsupported
- assert_eq!(
- Err(TokenError::NativeNotSupported.into()),
- do_process_instruction(
- mint_to(
- &program_id,
- &spl_token::native_mint::id(),
- &account_key,
- &owner_key,
- &[],
- 42
- )
- .unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::err(TokenError::NativeNotSupported.into())],
- )
- );
- // burn unsupported
- let bogus_mint_key = Pubkey::new_unique();
- let mut bogus_mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- do_process_instruction(
- initialize_mint(&program_id, &bogus_mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut bogus_mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- assert_eq!(
- Err(TokenError::NativeNotSupported.into()),
- do_process_instruction(
- burn(
- &program_id,
- &account_key,
- &bogus_mint_key,
- &owner_key,
- &[],
- 42
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut bogus_mint_account,
- &mut owner_account
- ],
- &[Check::err(TokenError::NativeNotSupported.into())],
- )
- );
- // ensure can't transfer below rent-exempt reserve
- assert_eq!(
- Err(TokenError::InsufficientFunds.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 50,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::InsufficientFunds.into())],
- )
- );
- // transfer between native accounts
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 40,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[
- Check::success(),
- Check::account(&account_key)
- .lamports(account_minimum_balance())
- .build(),
- Check::account(&account_key)
- .data_slice(109, &[1, 0, 0, 0])
- .build(),
- Check::account(&account_key)
- .data_slice(64, &0u64.to_le_bytes())
- .build(),
- Check::account(&account2_key)
- .lamports(account_minimum_balance() + 40)
- .build(),
- Check::account(&account_key)
- .data_slice(109, &[1, 0, 0, 0])
- .build(),
- Check::account(&account2_key)
- .data_slice(64, &40u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- assert_eq!(account_account.lamports, account_minimum_balance());
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 0);
- assert_eq!(account2_account.lamports, account_minimum_balance() + 40);
- let account = Account::unpack_unchecked(&account2_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, 40);
- // set close authority
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner3_key),
- AuthorityType::CloseAccount,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(129, &[1, 0, 0, 0])
- .build(),
- Check::account(&account_key)
- .data_slice(133, owner3_key.as_ref())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.close_authority, COption::Some(owner3_key));
- // set new account owner
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&owner2_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[],
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(129, &[0, 0, 0, 0])
- .build(),
- ],
- )
- .unwrap();
- // close authority cleared
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.close_authority, COption::None);
- // close native account
- do_process_instruction(
- close_account(&program_id, &account_key, &account3_key, &owner2_key, &[]).unwrap(),
- vec![
- &mut account_account,
- &mut account3_account,
- &mut owner2_account,
- ],
- &[
- Check::success(),
- Check::account(&account_key).lamports(0).build(),
- Check::account(&account3_key)
- .lamports(2 * account_minimum_balance())
- .build(),
- Check::account(&account_key).data(&[]).build(),
- ],
- )
- .unwrap();
- assert_eq!(account_account.lamports, 0);
- assert_eq!(account3_account.lamports, 2 * account_minimum_balance());
- assert!(account_account.data.is_empty());
- }
- #[test]
- fn test_overflow() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_owner_key = Pubkey::new_unique();
- let mut mint_owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &mint_owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create an account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner2_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner2_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint the max to an account
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &mint_owner_key,
- &[],
- u64::MAX,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut mint_owner_account,
- ],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(64, &u64::MAX.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX);
- // attempt to mint one more to account
- assert_eq!(
- Err(TokenError::Overflow.into()),
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &mint_owner_key,
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut mint_owner_account,
- ],
- &[
- Check::err(TokenError::Overflow.into()),
- Check::account(&account_key)
- .data_slice(64, &u64::MAX.to_le_bytes())
- .build(),
- ],
- )
- );
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX);
- // attempt to mint one more to the other account
- assert_eq!(
- Err(TokenError::Overflow.into()),
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account2_key,
- &mint_owner_key,
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account2_account,
- &mut mint_owner_account,
- ],
- &[Check::err(TokenError::Overflow.into())],
- )
- );
- // burn some of the supply
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 100).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(64, &(u64::MAX - 100).to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX - 100);
- do_process_instruction(
- mint_to(
- &program_id,
- &mint_key,
- &account_key,
- &mint_owner_key,
- &[],
- 100,
- )
- .unwrap(),
- vec![
- &mut mint_account,
- &mut account_account,
- &mut mint_owner_account,
- ],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(64, &u64::MAX.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.amount, u64::MAX);
- // manipulate account balance to attempt overflow transfer
- let mut account = Account::unpack_unchecked(&account2_account.data).unwrap();
- account.amount = 1;
- Account::pack(account, &mut account2_account.data).unwrap();
- assert_eq!(
- Err(TokenError::Overflow.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account2_key,
- &account_key,
- &owner2_key,
- &[],
- 1,
- )
- .unwrap(),
- vec![
- &mut account2_account,
- &mut account_account,
- &mut owner2_account,
- ],
- &[Check::err(TokenError::Overflow.into())],
- )
- );
- }
- #[test]
- fn test_frozen() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account2_key = Pubkey::new_unique();
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint and fund first account
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // create another account
- do_process_instruction(
- initialize_account(&program_id, &account2_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account2_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // fund first account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // no transfer if either account is frozen
- let mut account = Account::unpack_unchecked(&account2_account.data).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account2_account.data).unwrap();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- let mut account = Account::unpack_unchecked(&account_account.data).unwrap();
- account.state = AccountState::Initialized;
- Account::pack(account, &mut account_account.data).unwrap();
- let mut account = Account::unpack_unchecked(&account2_account.data).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account2_account.data).unwrap();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- transfer(
- &program_id,
- &account_key,
- &account2_key,
- &owner_key,
- &[],
- 500,
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut account2_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- // no approve if account is frozen
- let mut account = Account::unpack_unchecked(&account_account.data).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account_account.data).unwrap();
- let delegate_key = Pubkey::new_unique();
- let mut delegate_account = SolanaAccount::default();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- approve(
- &program_id,
- &account_key,
- &delegate_key,
- &owner_key,
- &[],
- 100
- )
- .unwrap(),
- vec![
- &mut account_account,
- &mut delegate_account,
- &mut owner_account,
- ],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- // no revoke if account is frozen
- let mut account = Account::unpack_unchecked(&account_account.data).unwrap();
- account.delegate = COption::Some(delegate_key);
- account.delegated_amount = 100;
- Account::pack(account, &mut account_account.data).unwrap();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- revoke(&program_id, &account_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut owner_account],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- // no set authority if account is frozen
- let new_owner_key = Pubkey::new_unique();
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- set_authority(
- &program_id,
- &account_key,
- Some(&new_owner_key),
- AuthorityType::AccountOwner,
- &owner_key,
- &[]
- )
- .unwrap(),
- vec![&mut account_account, &mut owner_account,],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- // no mint_to if destination account is frozen
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 100).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account,],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- // no burn if account is frozen
- assert_eq!(
- Err(TokenError::AccountFrozen.into()),
- do_process_instruction(
- burn(&program_id, &account_key, &mint_key, &owner_key, &[], 100).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(TokenError::AccountFrozen.into())],
- )
- );
- }
- #[test]
- fn test_freeze_thaw_dups() {
- let program_id = spl_token::id();
- let account1_key = Pubkey::new_unique();
- let mut account1_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account1_info: AccountInfo = (&account1_key, true, &mut account1_account).into();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_info: AccountInfo = (&mint_key, true, &mut mint_account).into();
- let rent_key = rent::id();
- let mut rent_sysvar = rent_sysvar();
- let rent_info: AccountInfo = (&rent_key, false, &mut rent_sysvar).into();
- // create mint
- do_process_instruction_dups(
- initialize_mint(&program_id, &mint_key, &owner_key, Some(&account1_key), 2).unwrap(),
- vec![mint_info.clone(), rent_info.clone()],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction_dups(
- initialize_account(&program_id, &account1_key, &mint_key, &account1_key).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- rent_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // freeze where mint freeze_authority is account
- do_process_instruction_dups(
- freeze_account(&program_id, &account1_key, &mint_key, &account1_key, &[]).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- // thaw where mint freeze_authority is account
- let mut account = Account::unpack_unchecked(&account1_info.data.borrow()).unwrap();
- account.state = AccountState::Frozen;
- Account::pack(account, &mut account1_info.data.borrow_mut()).unwrap();
- do_process_instruction_dups(
- thaw_account(&program_id, &account1_key, &mint_key, &account1_key, &[]).unwrap(),
- vec![
- account1_info.clone(),
- mint_info.clone(),
- account1_info.clone(),
- ],
- &[Check::success()],
- )
- .unwrap();
- }
- #[test]
- fn test_freeze_account() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let account_owner_key = Pubkey::new_unique();
- let mut account_owner_account = SolanaAccount::default();
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let owner2_key = Pubkey::new_unique();
- let mut owner2_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create new mint with owner different from account owner
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &account_owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut account_owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // mint to account
- do_process_instruction(
- mint_to(&program_id, &mint_key, &account_key, &owner_key, &[], 1000).unwrap(),
- vec![&mut mint_account, &mut account_account, &mut owner_account],
- &[Check::success()],
- )
- .unwrap();
- // mint cannot freeze
- assert_eq!(
- Err(TokenError::MintCannotFreeze.into()),
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(TokenError::MintCannotFreeze.into())],
- )
- );
- // missing freeze_authority
- let mut mint = Mint::unpack_unchecked(&mint_account.data).unwrap();
- mint.freeze_authority = COption::Some(owner_key);
- Mint::pack(mint, &mut mint_account.data).unwrap();
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner2_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // check explicit thaw
- assert_eq!(
- Err(TokenError::InvalidState.into()),
- do_process_instruction(
- thaw_account(&program_id, &account_key, &mint_key, &owner2_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- &[Check::err(TokenError::InvalidState.into())],
- )
- );
- // freeze
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(108, &[AccountState::Frozen as u8])
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.state, AccountState::Frozen);
- // check explicit freeze
- assert_eq!(
- Err(TokenError::InvalidState.into()),
- do_process_instruction(
- freeze_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[Check::err(TokenError::InvalidState.into())],
- )
- );
- // check thaw authority
- assert_eq!(
- Err(TokenError::OwnerMismatch.into()),
- do_process_instruction(
- thaw_account(&program_id, &account_key, &mint_key, &owner2_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner2_account],
- &[Check::err(TokenError::OwnerMismatch.into())],
- )
- );
- // thaw
- do_process_instruction(
- thaw_account(&program_id, &account_key, &mint_key, &owner_key, &[]).unwrap(),
- vec![&mut account_account, &mut mint_account, &mut owner_account],
- &[
- Check::success(),
- Check::account(&account_key)
- .data_slice(108, &[AccountState::Initialized as u8])
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&account_account.data).unwrap();
- assert_eq!(account.state, AccountState::Initialized);
- }
- #[test]
- fn test_initialize_account2_and_3() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account2_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let mut account3_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- initialize_account2(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![&mut account2_account, &mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- assert_eq!(account_account, account2_account);
- do_process_instruction(
- initialize_account3(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![&mut account3_account, &mut mint_account],
- &[Check::success()],
- )
- .unwrap();
- assert_eq!(account_account, account3_account);
- }
- #[test]
- fn test_sync_native() {
- let program_id = spl_token::id();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let native_account_key = Pubkey::new_unique();
- let lamports = 40;
- let mut native_account = SolanaAccount::new(
- account_minimum_balance() + lamports,
- Account::get_packed_len(),
- &program_id,
- );
- let non_native_account_key = Pubkey::new_unique();
- let mut non_native_account = SolanaAccount::new(
- account_minimum_balance() + 50,
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mut rent_sysvar = rent_sysvar();
- // initialize non-native mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // initialize non-native account
- do_process_instruction(
- initialize_account(&program_id, &non_native_account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut non_native_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[
- Check::success(),
- Check::account(&non_native_account_key)
- .data_slice(109, &[0, 0, 0, 0])
- .build(),
- Check::account(&non_native_account_key)
- .data_slice(64, &0u64.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&non_native_account.data).unwrap();
- assert!(!account.is_native());
- assert_eq!(account.amount, 0);
- // fail sync non-native
- assert_eq!(
- Err(TokenError::NonNativeNotSupported.into()),
- do_process_instruction(
- sync_native(&program_id, &non_native_account_key,).unwrap(),
- vec![&mut non_native_account],
- &[Check::err(TokenError::NonNativeNotSupported.into())],
- )
- );
- // fail sync uninitialized
- assert_eq!(
- Err(ProgramError::UninitializedAccount),
- do_process_instruction(
- sync_native(&program_id, &native_account_key,).unwrap(),
- vec![&mut native_account],
- &[Check::err(ProgramError::UninitializedAccount)],
- )
- );
- // wrap native account
- do_process_instruction(
- initialize_account(
- &program_id,
- &native_account_key,
- &spl_token::native_mint::id(),
- &owner_key,
- )
- .unwrap(),
- vec![
- &mut native_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // fail sync, not owned by program
- let not_program_id = Pubkey::new_unique();
- native_account.owner = not_program_id;
- assert_eq!(
- Err(ProgramError::IncorrectProgramId),
- do_process_instruction(
- sync_native(&program_id, &native_account_key,).unwrap(),
- vec![&mut native_account],
- &[
- Check::err(ProgramError::IncorrectProgramId),
- Check::account(&native_account_key)
- .data_slice(109, &[1, 0, 0, 0])
- .build(),
- Check::account(&native_account_key)
- .data_slice(64, &lamports.to_le_bytes())
- .build()
- ],
- )
- );
- native_account.owner = program_id;
- let account = Account::unpack_unchecked(&native_account.data).unwrap();
- assert!(account.is_native());
- assert_eq!(account.amount, lamports);
- // sync, no change
- do_process_instruction(
- sync_native(&program_id, &native_account_key).unwrap(),
- vec![&mut native_account],
- &[
- Check::success(),
- Check::account(&native_account_key)
- .data_slice(64, &lamports.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&native_account.data).unwrap();
- assert_eq!(account.amount, lamports);
- // transfer sol
- let new_lamports = lamports + 50;
- native_account.lamports = account_minimum_balance() + new_lamports;
- // success sync
- do_process_instruction(
- sync_native(&program_id, &native_account_key).unwrap(),
- vec![&mut native_account],
- &[
- Check::success(),
- Check::account(&native_account_key)
- .data_slice(64, &new_lamports.to_le_bytes())
- .build(),
- ],
- )
- .unwrap();
- let account = Account::unpack_unchecked(&native_account.data).unwrap();
- assert_eq!(account.amount, new_lamports);
- // reduce sol
- native_account.lamports -= 1;
- // fail sync
- assert_eq!(
- Err(TokenError::InvalidState.into()),
- do_process_instruction(
- sync_native(&program_id, &native_account_key,).unwrap(),
- vec![&mut native_account],
- &[Check::err(TokenError::InvalidState.into())],
- )
- );
- }
- #[test]
- #[serial]
- fn test_get_account_data_size() {
- // see integration tests for return-data validity
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mut rent_sysvar = rent_sysvar();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mint_key = Pubkey::new_unique();
- // fail if an invalid mint is passed in
- assert_eq!(
- Err(TokenError::InvalidMint.into()),
- do_process_instruction(
- get_account_data_size(&program_id, &mint_key).unwrap(),
- vec![&mut mint_account],
- &[Check::err(TokenError::InvalidMint.into())],
- )
- );
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- get_account_data_size(&program_id, &mint_key).unwrap(),
- vec![&mut mint_account],
- &[
- Check::success(),
- Check::return_data(&Account::LEN.to_le_bytes()),
- ],
- )
- .unwrap();
- }
- #[test]
- fn test_initialize_immutable_owner() {
- let program_id = spl_token::id();
- let account_key = Pubkey::new_unique();
- let mut account_account = SolanaAccount::new(
- account_minimum_balance(),
- Account::get_packed_len(),
- &program_id,
- );
- let owner_key = Pubkey::new_unique();
- let mut owner_account = SolanaAccount::default();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- // success initialize immutable
- do_process_instruction(
- initialize_immutable_owner(&program_id, &account_key).unwrap(),
- vec![&mut account_account],
- &[Check::success()],
- )
- .unwrap();
- // create account
- do_process_instruction(
- initialize_account(&program_id, &account_key, &mint_key, &owner_key).unwrap(),
- vec![
- &mut account_account,
- &mut mint_account,
- &mut owner_account,
- &mut rent_sysvar,
- ],
- &[Check::success()],
- )
- .unwrap();
- // fail post-init
- assert_eq!(
- Err(TokenError::AlreadyInUse.into()),
- do_process_instruction(
- initialize_immutable_owner(&program_id, &account_key).unwrap(),
- vec![&mut account_account],
- &[Check::err(TokenError::AlreadyInUse.into())],
- )
- );
- }
- #[test]
- #[serial]
- fn test_amount_to_ui_amount() {
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // fail if an invalid mint is passed in
- assert_eq!(
- Err(TokenError::InvalidMint.into()),
- do_process_instruction(
- amount_to_ui_amount(&program_id, &mint_key, 110).unwrap(),
- vec![&mut mint_account],
- &[Check::err(TokenError::InvalidMint.into())],
- )
- );
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- amount_to_ui_amount(&program_id, &mint_key, 23).unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data("0.23".as_bytes())],
- )
- .unwrap();
- do_process_instruction(
- amount_to_ui_amount(&program_id, &mint_key, 110).unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data("1.1".as_bytes())],
- )
- .unwrap();
- do_process_instruction(
- amount_to_ui_amount(&program_id, &mint_key, 4200).unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data("42".as_bytes())],
- )
- .unwrap();
- do_process_instruction(
- amount_to_ui_amount(&program_id, &mint_key, 0).unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data("0".as_bytes())],
- )
- .unwrap();
- }
- #[test]
- #[serial]
- fn test_ui_amount_to_amount() {
- let program_id = spl_token::id();
- let owner_key = Pubkey::new_unique();
- let mint_key = Pubkey::new_unique();
- let mut mint_account =
- SolanaAccount::new(mint_minimum_balance(), Mint::get_packed_len(), &program_id);
- let mut rent_sysvar = rent_sysvar();
- // fail if an invalid mint is passed in
- assert_eq!(
- Err(TokenError::InvalidMint.into()),
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "1.1").unwrap(),
- vec![&mut mint_account],
- &[Check::err(TokenError::InvalidMint.into())],
- )
- );
- // create mint
- do_process_instruction(
- initialize_mint(&program_id, &mint_key, &owner_key, None, 2).unwrap(),
- vec![&mut mint_account, &mut rent_sysvar],
- &[Check::success()],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "0.23").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&23u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "0.20").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&20u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "0.2000").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&20u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, ".20").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&20u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "1.1").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&110u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "1.10").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&110u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "42").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&4200u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "42.").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&4200u64.to_le_bytes())],
- )
- .unwrap();
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "0").unwrap(),
- vec![&mut mint_account],
- &[Check::success(), Check::return_data(&0u64.to_le_bytes())],
- )
- .unwrap();
- // fail if invalid ui_amount passed in
- assert_eq!(
- Err(ProgramError::InvalidArgument),
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "").unwrap(),
- vec![&mut mint_account],
- &[Check::err(ProgramError::InvalidArgument)],
- )
- );
- assert_eq!(
- Err(ProgramError::InvalidArgument),
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, ".").unwrap(),
- vec![&mut mint_account],
- &[Check::err(ProgramError::InvalidArgument)],
- )
- );
- assert_eq!(
- Err(ProgramError::InvalidArgument),
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "0.111").unwrap(),
- vec![&mut mint_account],
- &[Check::err(ProgramError::InvalidArgument)],
- )
- );
- assert_eq!(
- Err(ProgramError::InvalidArgument),
- do_process_instruction(
- ui_amount_to_amount(&program_id, &mint_key, "0.t").unwrap(),
- vec![&mut mint_account],
- &[Check::err(ProgramError::InvalidArgument)],
- )
- );
- }
|