FIRUserTests.m 202 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321
  1. /*
  2. * Copyright 2017 Google
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #import <Foundation/Foundation.h>
  17. #import <OCMock/OCMock.h>
  18. #import <XCTest/XCTest.h>
  19. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRAuthTokenResult.h"
  20. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRCustomTokenProviderDelegate.h"
  21. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIREmailAuthProvider.h"
  22. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRFacebookAuthProvider.h"
  23. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRGoogleAuthProvider.h"
  24. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRUserInfo.h"
  25. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRUserMetadata.h"
  26. #import "FirebaseAuth/Sources/Auth/FIRAuthGlobalWorkQueue.h"
  27. #import "FirebaseAuth/Sources/Auth/FIRAuthOperationType.h"
  28. #import "FirebaseAuth/Sources/Auth/FIRAuth_Internal.h"
  29. #import "FirebaseAuth/Sources/Backend/FIRAuthBackend.h"
  30. #import "FirebaseAuth/Sources/Backend/RPC/FIREmailLinkSignInResponse.h"
  31. #import "FirebaseAuth/Sources/Backend/RPC/FIRGetAccountInfoRequest.h"
  32. #import "FirebaseAuth/Sources/Backend/RPC/FIRGetAccountInfoResponse.h"
  33. #import "FirebaseAuth/Sources/Backend/RPC/FIRSecureTokenRequest.h"
  34. #import "FirebaseAuth/Sources/Backend/RPC/FIRSecureTokenResponse.h"
  35. #import "FirebaseAuth/Sources/Backend/RPC/FIRSetAccountInfoRequest.h"
  36. #import "FirebaseAuth/Sources/Backend/RPC/FIRSetAccountInfoResponse.h"
  37. #import "FirebaseAuth/Sources/Backend/RPC/FIRSignUpNewUserResponse.h"
  38. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyAssertionRequest.h"
  39. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyAssertionResponse.h"
  40. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyCustomTokenRequest.h"
  41. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyCustomTokenResponse.h"
  42. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyPasswordRequest.h"
  43. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyPasswordResponse.h"
  44. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyPhoneNumberRequest.h"
  45. #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyPhoneNumberResponse.h"
  46. #import "FirebaseAuth/Sources/SystemService/FIRSecureTokenService.h"
  47. #import "FirebaseAuth/Sources/User/FIRAdditionalUserInfo_Internal.h"
  48. #import "FirebaseAuth/Sources/User/FIRUser_Internal.h"
  49. #import "FirebaseAuth/Sources/Utilities/FIRAuthErrorUtils.h"
  50. #import "FirebaseAuth/Tests/Unit/FIRApp+FIRAuthUnitTests.h"
  51. #import "FirebaseAuth/Tests/Unit/OCMStubRecorder+FIRAuthUnitTests.h"
  52. #if TARGET_OS_IOS
  53. #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRPhoneAuthProvider.h"
  54. #import "FirebaseAuth/Sources/AuthProvider/Phone/FIRPhoneAuthCredential_Internal.h"
  55. #endif
  56. NS_ASSUME_NONNULL_BEGIN
  57. /** @var kAPIKey
  58. @brief The fake API key.
  59. */
  60. static NSString *const kAPIKey = @"FAKE_API_KEY";
  61. /** @var kAccessToken
  62. @brief A fake access token. The user ID is "testuser".
  63. */
  64. static NSString *const kAccessToken =
  65. @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZDBhODBiYzh"
  66. "jM2U5NDM0OTc0ZDFmMWRiNTEifQ."
  67. "eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vZmItc2EtdXBncm"
  68. "FkZWQiLCJhdWQiOiJ0ZXN0X2F1ZCIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZCI"
  69. "s"
  70. "InN1YiI6InRlc3Rfc3ViIiwiaWF0IjoxNTIyMzYwNTU3LCJleHAiOjE1MjIzNjQxNTcsImVtYWlsIjoiYXVuaXRlc3R1c"
  71. "2"
  72. "VyQGdtYWlsLmNvbSIsImVtYWlsX3ZlcmlmaWVkIjpmYWxzZSwiZmlyZWJhc2UiOnsiaWRlbnRpdGllcyI6eyJlbWFpbCI"
  73. "6"
  74. "WyJhdW5pdGVzdHVzZXJAZ21haWwuY29tIl19LCJzaWduX2luX3Byb3ZpZGVyIjoicGFzc3dvcmQifX0=."
  75. "WFQqSrpVnxx7m"
  76. "UrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_"
  77. "BKkcbZuq8O7iDVpOvqKC"
  78. "3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-"
  79. "XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjN"
  80. "hHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-"
  81. "dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbW"
  82. "lddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  83. /** @var kAccessToken2
  84. @brief A fake access token. The user ID is "testuser2".
  85. */
  86. static NSString *const kAccessToken2 =
  87. @"eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9."
  88. @"eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vZmItc2EtdXBncmFkZWQiLCJhdWQiOiJ0ZXN0X2F1Z"
  89. @"CIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3R1c2VyMiIsInN1YiI6InRlc3Rfc3ViIiwiaWF0Ij"
  90. @"oxNTIyMzYwNTU3LCJleHAiOjE1MjIzNjQxNTcsImVtYWlsIjoiYXVuaXRlc3R1c2VyQGdtYWlsLmNvbSIsImVtYWlsX3Z"
  91. @"lcmlmaWVkIjpmYWxzZSwiZmlyZWJhc2UiOnsiaWRlbnRpdGllcyI6eyJlbWFpbCI6WyJhdW5pdGVzdHVzZXJAZ21haWwu"
  92. @"Y29tIl19LCJzaWduX2luX3Byb3ZpZGVyIjoicGFzc3dvcmQifX0.pLFQHyjOyN_"
  93. @"ibaiPJssFIt1jq5kisazDOA2wRiPtUzZ707FZMl-7F1aFsMnYD5Ha2tNFrCOUnlLlOZRYkyxkIVDTyZfSpCHhn_"
  94. @"Axs1XZwandin--gIFsxAf5Ygsf7Qkp0BioVnEd4Bcwc9CrZSRdPLuGoIPl6p6b9PM-"
  95. @"4HLSj8DVuw5A0iVkPG4DHn43TUFt8gzha4A2H91UE8oMn1lWUV2RZDnIoBewKs_"
  96. @"QSY0wnh6cWNbWQjHaS7t59wQQvH1YC2Scovsz2N4lkMvrRnzgnx3MfCw0BuK27MJ8iUeF0gIH2aSCYeurJuo5uvSuagDJ"
  97. @"6OaUQJ6xZK_BmEzn-33H9g";
  98. /** @var kAccessTokenWithBase64URLCharacters
  99. @brief The fake access where the AUD value is "??????????>>>>>>>>>>" and the email value is
  100. ">>>>>>>>????????@gmail.com".
  101. */
  102. static NSString *const kAccessTokenWithBase64URLCharacter =
  103. @"ey?hbGciOimnuzI1NiIsImtpZCI6ImY1YjE4M"
  104. "jc2YTQ4NjYxZDBhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ."
  105. "eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2ds"
  106. "ZS5jb20vZmItc2EtdXBncmFkZWQiLCJhdWQiOiI_Pz8_Pz8_Pz8_Pj4-Pj4-Pj4-"
  107. "PiIsImF1dGhfdGltZSI6MTUyMjM2MD"
  108. "U0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZCIsInN1YiI6InRlc3Rfc3ViIiwiaWF0IjoxNTIyMzYwNTU3LCJleHAiOjE"
  109. "1"
  110. "MjIzNjQxNTcsImVtYWlsIjoiPj4-Pj4-Pj4_Pz8_Pz8_"
  111. "P0BnbWFpbC5jb20iLCJlbWFpbF92ZXJpZmllZCI6ZmFsc2UsIm"
  112. "ZpcmViYXNlIjp7ImlkZW50aXRpZXMiOnsiZW1haWwiOlsiYXVuaXRlc3R1c2VyQGdtYWlsLmNvbSJdfSwic2lnbl9pbl9"
  113. "w"
  114. "cm92aWRlciI6IlBhc3N3b3JkIn19.WFQqSrpVnxx7mUrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-"
  115. "aTYMvND3-"
  116. "4aS3qRnN2kvk9KJAaF3eI_BKkcbZuq8O7iDVpOvqKC3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-"
  117. "XtBt"
  118. "1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjNhHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-"
  119. "dDV7"
  120. "jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbWlddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  121. /** @var kbase64URLEncodedEmail
  122. @brief The fake email address with a value containing non-valid base64 encoded characters.
  123. @remarks This is used to ensure that the token parser is able to handle base64 URL encoded
  124. strings. Note that consecutive question marks in a string without being escaped is not
  125. legal C99.
  126. */
  127. static NSString *const kbase64URLEncodedEmail = @">>>>>>>>\?\?\?\?\?\?\?\?@gmail.com";
  128. /** @var kbase64URLEncodedAUD
  129. @brief The fake AUD with a value containing non-valid base64 encoded characters.
  130. @remarks This is used to ensure that the token parser is able to handle base64 URL encoded
  131. strings. Note that consecutive question marks in a string without being escaped is not
  132. legal C99.
  133. */
  134. static NSString *const kbase64URLEncodedAUD = @"\?\?\?\?\?\?\?\?\?\?>>>>>>>>>>";
  135. /** @var kAccessTokenLength415
  136. @brief The fake access token with 415 characters in the claims potion of the token.
  137. */
  138. static NSString *const kAccessTokenLength415 =
  139. @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZD"
  140. "BhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ."
  141. "eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vdGVzd"
  142. "CIsImF1ZCI6InRlc3RfYXVkIiwiYXV0aF90aW1lIjoxNTIyMzYwNTQ5LCJ1c2VyX2lkIjoidGVzdF91c2VyX2lkIiwic3"
  143. "V"
  144. "iIjoidGVzdF9zdWIiLCJpYXQiOjE1MjIzNjA1NTcsImV4cCI6MTUyMjM2NDE1NywiZW1haWwiOiJhdW5pdGVzdHVzZXJA"
  145. "Z"
  146. "21haWwuY29tIiwiZW1haWxfdmVyaWZpZWQiOmZhbHNlLCJmaXJlYmFzZSI6eyJpZGVudGl0aWVzIjp7ImVtYWlsIjpbIm"
  147. "F"
  148. "1bml0ZXN0dXNlckBnbWFpbC5jb20iXX0sInNpZ25faW5fcHJvdmlkZXIiOiJwYXNzd29yZCJ9fQ=.WFQqSrpVnxx7m"
  149. "UrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_"
  150. "BKkcbZuq8O7iDVpOvqKC"
  151. "3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-"
  152. "XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjN"
  153. "hHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-"
  154. "dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbW"
  155. "lddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  156. /** @var kAccessTokenLength416
  157. @brief The fake access token with 416 characters in the claims potion of the token.
  158. */
  159. static NSString *const kAccessTokenLength416 =
  160. @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZD"
  161. "BhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ."
  162. "eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vdGVzd"
  163. "DIiLCJhdWQiOiJ0ZXN0X2F1ZCIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZCIsIn"
  164. "N"
  165. "1YiI6InRlc3Rfc3ViIiwiaWF0IjoxNTIyMzYwNTU3LCJleHAiOjE1MjIzNjQxNTcsImVtYWlsIjoiYXVuaXRlc3R1c2Vy"
  166. "Q"
  167. "GdtYWlsLmNvbSIsImVtYWlsX3ZlcmlmaWVkIjpmYWxzZSwiZmlyZWJhc2UiOnsiaWRlbnRpdGllcyI6eyJlbWFpbCI6Wy"
  168. "J"
  169. "hdW5pdGVzdHVzZXJAZ21haWwuY29tIl19LCJzaWduX2luX3Byb3ZpZGVyIjoicGFzc3dvcmQifX0=.WFQqSrpVnxx7m"
  170. "UrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_"
  171. "BKkcbZuq8O7iDVpOvqKC"
  172. "3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-"
  173. "XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjN"
  174. "hHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-"
  175. "dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbW"
  176. "lddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  177. /** @var kAccessTokenLength4523
  178. @brief The fake access token with 523 characters in the claims potion of the token.
  179. */
  180. static NSString *const kAccessTokenLength523 =
  181. @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZD"
  182. "BhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ."
  183. "eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vdGVzd"
  184. "DQiLCJhdWQiOiJ0ZXN0X2F1ZCIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZF81ND"
  185. "M"
  186. "yIiwic3ViIjoidGVzdF9zdWIiLCJpYXQiOjE1MjIzNjA1NTcsImV4cCI6MTUyMjM2NDE1OSwiZW1haWwiOiJhdW5pdGVz"
  187. "d"
  188. "HVzZXI0QGdtYWlsLmNvbSIsImVtYWlsX3ZlcmlmaWVkIjp0cnVlLCJmaXJlYmFzZSI6eyJpZGVudGl0aWVzIjp7ImVtYW"
  189. "l"
  190. "sIjpbImF1bml0ZXN0dXNlckBnbWFpbC5jb20iXX0sInNpZ25faW5fcHJvdmlkZXIiOiJwYXNzd29yZCJ9fQ=."
  191. "WFQqSrpVn"
  192. "xx7mUrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_"
  193. "BKkcbZuq8O7iDVpO"
  194. "vqKC3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-"
  195. "XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH"
  196. "7jjNhHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-"
  197. "dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqV"
  198. "kPbWlddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  199. /** @var kNewAccessToken
  200. @brief A new value for the fake access token.
  201. */
  202. static NSString *const kNewAccessToken = @"NEW_ACCESS_TOKEN";
  203. /** @var kAccessTokenValidInterval
  204. @brief The time to live for the fake access token.
  205. */
  206. static const NSTimeInterval kAccessTokenTimeToLive = 60 * 60;
  207. /** @var kRefreshToken
  208. @brief The fake refresh token.
  209. */
  210. static NSString *const kRefreshToken = @"REFRESH_TOKEN";
  211. /** @var kLocalID
  212. @brief The fake local user ID.
  213. */
  214. static NSString *const kLocalID = @"LOCAL_ID";
  215. /** @var kAnotherLocalID
  216. @brief The fake local ID of another user.
  217. */
  218. static NSString *const kAnotherLocalID = @"ANOTHER_LOCAL_ID";
  219. /** @var kGoogleIDToken
  220. @brief The fake ID token from Google Sign-In.
  221. */
  222. static NSString *const kGoogleIDToken = @"GOOGLE_ID_TOKEN";
  223. /** @var kFacebookIDToken
  224. @brief The fake ID token from Facebook Sign-In. Facebook provider ID token is always nil.
  225. */
  226. static NSString *const kFacebookIDToken = nil;
  227. /** @var kGoogleAccessToken
  228. @brief The fake access token from Google Sign-In.
  229. */
  230. static NSString *const kGoogleAccessToken = @"GOOGLE_ACCESS_TOKEN";
  231. /** @var kFacebookAccessToken
  232. @brief The fake access token from Facebook Sign-In.
  233. */
  234. static NSString *const kFacebookAccessToken = @"FACEBOOK_ACCESS_TOKEN";
  235. /** @var kEmail
  236. @brief The fake user email.
  237. */
  238. static NSString *const kEmail = @"user@company.com";
  239. /** @var kPhoneNumber
  240. @brief The fake user phone number.
  241. */
  242. static NSString *const kPhoneNumber = @"12345658";
  243. /** @var kTemporaryProof
  244. @brief The fake temporary proof.
  245. */
  246. static NSString *const kTemporaryProof = @"12345658";
  247. /** @var kNewEmail
  248. @brief A new value for the fake user email.
  249. */
  250. static NSString *const kNewEmail = @"newuser@company.com";
  251. /** @var kUserName
  252. @brief The fake user name.
  253. */
  254. static NSString *const kUserName = @"User Doe";
  255. /** @var kNewDisplayName
  256. @brief A new value for the fake user display name.
  257. */
  258. static NSString *const kNewDisplayName = @"New User Doe";
  259. /** @var kPhotoURL
  260. @brief The fake user profile image URL string.
  261. */
  262. static NSString *const kPhotoURL = @"https://host.domain/image";
  263. /** @var kNewPhotoURL
  264. @brief A new value for the fake user profile image URL string..
  265. */
  266. static NSString *const kNewPhotoURL = @"https://host.domain/new/image";
  267. /** @var kFakePassword
  268. @brief The fake user password.
  269. */
  270. static NSString *const kFakePassword = @"123456";
  271. /** @var kNewPassword
  272. @brief The fake new user password.
  273. */
  274. static NSString *const kNewPassword = @"1234567";
  275. /** @var kPasswordHash
  276. @brief The fake user password hash.
  277. */
  278. static NSString *const kPasswordHash = @"UkVEQUNURUQ=";
  279. /** @var kGoogleUD
  280. @brief The fake user ID under Google Sign-In.
  281. */
  282. static NSString *const kGoogleID = @"GOOGLE_ID";
  283. /** @var kGoogleEmail
  284. @brief The fake user email under Google Sign-In.
  285. */
  286. static NSString *const kGoogleEmail = @"user@gmail.com";
  287. /** @var kGoogleDisplayName
  288. @brief The fake user display name under Google Sign-In.
  289. */
  290. static NSString *const kGoogleDisplayName = @"Google Doe";
  291. /** @var kEmailDisplayName
  292. @brief The fake user display name for email password user.
  293. */
  294. static NSString *const kEmailDisplayName = @"Email Doe";
  295. /** @var kFacebookDisplayName
  296. @brief The fake user display name under Facebook Sign-In.
  297. */
  298. static NSString *const kFacebookDisplayName = @"Facebook Doe";
  299. /** @var kGooglePhotoURL
  300. @brief The fake user profile image URL string under Google Sign-In.
  301. */
  302. static NSString *const kGooglePhotoURL = @"https://googleusercontents.com/user/profile";
  303. /** @var kFacebookID
  304. @brief The fake user ID under Facebook Login.
  305. */
  306. static NSString *const kFacebookID = @"FACEBOOK_ID";
  307. /** @var kFacebookEmail
  308. @brief The fake user email under Facebook Login.
  309. */
  310. static NSString *const kFacebookEmail = @"user@facebook.com";
  311. /** @var kVerificationCode
  312. @brief Fake verification code used for testing.
  313. */
  314. static NSString *const kVerificationCode = @"12345678";
  315. /** @var kVerificationID
  316. @brief Fake verification ID for testing.
  317. */
  318. static NSString *const kVerificationID = @"55432";
  319. /** @var kUserArchiverKey
  320. @brief The key used to archive and unarchive the user object for test NSSecureCoding.
  321. */
  322. static NSString *const kUserArchiverKey = @"userArchiverKey";
  323. /** @var kCustomToken
  324. @brief The fake custom token to sign in.
  325. */
  326. static NSString *const kCustomToken = @"CUSTOM_TOKEN";
  327. /** @var kCreationDateInSeconds
  328. @brief The fake creation date.
  329. */
  330. static NSTimeInterval const kCreationDateTimeIntervalInSeconds = 1505858500;
  331. /** @var kLastSignInDateTimeIntervalInSeconds
  332. @brief The fake last sign in date date.
  333. */
  334. static NSTimeInterval const kLastSignInDateTimeIntervalInSeconds = 1505858583;
  335. /** @var kExpectationTimeout
  336. @brief The maximum time waiting for expectations to fulfill.
  337. */
  338. static const NSTimeInterval kExpectationTimeout = 2;
  339. /** @extention FIRSecureTokenService
  340. @brief Extends the FIRSecureTokenService class to expose one private method for testing only.
  341. */
  342. @interface FIRSecureTokenService ()
  343. /** @fn hasValidAccessToken
  344. @brief private method exposed so it can be mocked to prevent the fake expiration date from
  345. affecting the the unit tests.
  346. */
  347. - (BOOL)hasValidAccessToken;
  348. @end
  349. /** @class FIRUserTests
  350. @brief Tests for @c FIRUser .
  351. */
  352. @interface FIRUserTests : XCTestCase
  353. @end
  354. @implementation FIRUserTests {
  355. /** @var _mockBackend
  356. @brief The mock @c FIRAuthBackendImplementation .
  357. */
  358. id _mockBackend;
  359. }
  360. /** @fn googleProfile
  361. @brief The fake user profile under additional user data in @c FIRVerifyAssertionResponse.
  362. */
  363. + (NSDictionary *)googleProfile {
  364. static NSDictionary *kGoogleProfile = nil;
  365. static dispatch_once_t onceToken;
  366. dispatch_once(&onceToken, ^{
  367. kGoogleProfile = @{@"email" : kGoogleEmail, @"given_name" : @"User", @"family_name" : @"Doe"};
  368. });
  369. return kGoogleProfile;
  370. }
  371. - (void)setUp {
  372. [super setUp];
  373. _mockBackend = OCMProtocolMock(@protocol(FIRAuthBackendImplementation));
  374. [FIRAuthBackend setBackendImplementation:_mockBackend];
  375. [FIRApp resetAppForAuthUnitTests];
  376. }
  377. - (void)tearDown {
  378. [FIRAuthBackend setDefaultBackendImplementationWithRPCIssuer:nil];
  379. [super tearDown];
  380. }
  381. #pragma mark - Tests
  382. /** @fn testUserPropertiesAndNSSecureCoding
  383. @brief Tests properties of the @c FIRUser instance before and after being
  384. serialized/deserialized.
  385. */
  386. - (void)testUserPropertiesAndNSSecureCoding {
  387. // Mock auth provider user info for email/password for GetAccountInfo.
  388. id mockPasswordUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  389. OCMStub([mockPasswordUserInfo providerID]).andReturn(FIREmailAuthProviderID);
  390. OCMStub([mockPasswordUserInfo federatedID]).andReturn(kEmail);
  391. OCMStub([mockPasswordUserInfo email]).andReturn(kEmail);
  392. // Mock auth provider user info from Google for GetAccountInfo.
  393. id mockGoogleUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  394. OCMStub([mockGoogleUserInfo providerID]).andReturn(FIRGoogleAuthProviderID);
  395. OCMStub([mockGoogleUserInfo displayName]).andReturn(kGoogleDisplayName);
  396. OCMStub([mockGoogleUserInfo photoURL]).andReturn([NSURL URLWithString:kGooglePhotoURL]);
  397. OCMStub([mockGoogleUserInfo federatedID]).andReturn(kGoogleID);
  398. OCMStub([mockGoogleUserInfo email]).andReturn(kGoogleEmail);
  399. // Mock auth provider user info from Facebook for GetAccountInfo.
  400. id mockFacebookUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  401. OCMStub([mockFacebookUserInfo providerID]).andReturn(FIRFacebookAuthProviderID);
  402. OCMStub([mockFacebookUserInfo federatedID]).andReturn(kFacebookID);
  403. OCMStub([mockFacebookUserInfo email]).andReturn(kFacebookEmail);
  404. #if TARGET_OS_IOS
  405. // Mock auth provider user info from Phone auth provider for GetAccountInfo.
  406. id mockPhoneUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  407. OCMStub([mockPhoneUserInfo providerID]).andReturn(FIRPhoneAuthProviderID);
  408. OCMStub([mockPhoneUserInfo phoneNumber]).andReturn(kPhoneNumber);
  409. #endif
  410. // Mock the root user info object for GetAccountInfo.
  411. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  412. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  413. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  414. OCMStub([mockGetAccountInfoResponseUser emailVerified]).andReturn(YES);
  415. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  416. OCMStub([mockGetAccountInfoResponseUser photoURL]).andReturn([NSURL URLWithString:kPhotoURL]);
  417. OCMStub([mockGetAccountInfoResponseUser creationDate])
  418. .andReturn([NSDate dateWithTimeIntervalSince1970:kCreationDateTimeIntervalInSeconds]);
  419. OCMStub([mockGetAccountInfoResponseUser lastLoginDate])
  420. .andReturn([NSDate dateWithTimeIntervalSince1970:kLastSignInDateTimeIntervalInSeconds]);
  421. NSArray *providerUserInfos = @[
  422. #if TARGET_OS_IOS
  423. mockPhoneUserInfo,
  424. #endif
  425. mockPasswordUserInfo, mockGoogleUserInfo, mockFacebookUserInfo
  426. ];
  427. OCMStub([mockGetAccountInfoResponseUser providerUserInfo]).andReturn(providerUserInfos);
  428. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  429. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  430. [self
  431. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  432. completion:^(FIRUser *user) {
  433. // Verify FIRUserInfo properties on FIRUser itself.
  434. XCTAssertEqualObjects(user.providerID, @"Firebase");
  435. XCTAssertEqualObjects(user.uid, kLocalID);
  436. XCTAssertEqualObjects(user.displayName,
  437. kGoogleDisplayName);
  438. XCTAssertEqualObjects(user.photoURL,
  439. [NSURL URLWithString:kPhotoURL]);
  440. XCTAssertEqualObjects(user.email, kEmail);
  441. XCTAssertEqualObjects(
  442. user.metadata.creationDate,
  443. [NSDate dateWithTimeIntervalSince1970:
  444. kCreationDateTimeIntervalInSeconds]);
  445. XCTAssertEqualObjects(
  446. user.metadata.lastSignInDate,
  447. [NSDate dateWithTimeIntervalSince1970:
  448. kLastSignInDateTimeIntervalInSeconds]);
  449. // Verify FIRUser properties besides providerData
  450. // contents.
  451. XCTAssertFalse(user.anonymous);
  452. XCTAssertTrue(user.emailVerified);
  453. XCTAssertEqualObjects(user.refreshToken,
  454. kRefreshToken);
  455. XCTAssertEqual(user.providerData.count,
  456. providerUserInfos.count);
  457. NSDictionary<NSString *, id<FIRUserInfo>>
  458. *providerMap = [self
  459. dictionaryWithUserInfoArray:user.providerData];
  460. // Verify FIRUserInfo properties from email/password.
  461. id<FIRUserInfo> passwordUserInfo =
  462. providerMap[FIREmailAuthProviderID];
  463. XCTAssertEqualObjects(passwordUserInfo.uid, kEmail);
  464. XCTAssertNil(passwordUserInfo.displayName);
  465. XCTAssertNil(passwordUserInfo.photoURL);
  466. XCTAssertEqualObjects(passwordUserInfo.email, kEmail);
  467. // Verify FIRUserInfo properties from the Google auth
  468. // provider.
  469. id<FIRUserInfo> googleUserInfo =
  470. providerMap[FIRGoogleAuthProviderID];
  471. XCTAssertEqualObjects(googleUserInfo.uid, kGoogleID);
  472. XCTAssertEqualObjects(googleUserInfo.displayName,
  473. kGoogleDisplayName);
  474. XCTAssertEqualObjects(
  475. googleUserInfo.photoURL,
  476. [NSURL URLWithString:kGooglePhotoURL]);
  477. XCTAssertEqualObjects(googleUserInfo.email,
  478. kGoogleEmail);
  479. // Verify FIRUserInfo properties from the Facebook auth
  480. // provider.
  481. id<FIRUserInfo> facebookUserInfo =
  482. providerMap[FIRFacebookAuthProviderID];
  483. XCTAssertEqualObjects(facebookUserInfo.uid,
  484. kFacebookID);
  485. XCTAssertNil(facebookUserInfo.displayName);
  486. XCTAssertNil(facebookUserInfo.photoURL);
  487. XCTAssertEqualObjects(facebookUserInfo.email,
  488. kFacebookEmail);
  489. #if TARGET_OS_IOS
  490. // Verify FIRUserInfo properties from the phone auth
  491. // provider.
  492. id<FIRUserInfo> phoneUserInfo =
  493. providerMap[FIRPhoneAuthProviderID];
  494. XCTAssertEqualObjects(phoneUserInfo.phoneNumber,
  495. kPhoneNumber);
  496. #endif
  497. // Test NSSecureCoding
  498. XCTAssertTrue([FIRUser supportsSecureCoding]);
  499. NSMutableData *data = [NSMutableData data];
  500. NSKeyedArchiver *archiver = [[NSKeyedArchiver alloc]
  501. initForWritingWithMutableData:data];
  502. [archiver encodeObject:user forKey:kUserArchiverKey];
  503. [archiver finishEncoding];
  504. NSKeyedUnarchiver *unarchiver =
  505. [[NSKeyedUnarchiver alloc]
  506. initForReadingWithData:data];
  507. FIRUser *unarchivedUser =
  508. [unarchiver decodeObjectForKey:kUserArchiverKey];
  509. // Verify NSSecureCoding for FIRUser
  510. XCTAssertEqualObjects(unarchivedUser.providerID,
  511. user.providerID);
  512. XCTAssertEqualObjects(unarchivedUser.uid, user.uid);
  513. XCTAssertEqualObjects(unarchivedUser.email,
  514. user.email);
  515. XCTAssertEqualObjects(unarchivedUser.photoURL,
  516. user.photoURL);
  517. XCTAssertEqualObjects(unarchivedUser.displayName,
  518. user.displayName);
  519. // Verify NSSecureCoding properties besides
  520. // providerData contents.
  521. XCTAssertEqual(unarchivedUser.anonymous,
  522. user.anonymous);
  523. XCTAssertEqual(unarchivedUser.emailVerified,
  524. user.emailVerified);
  525. XCTAssertEqualObjects(unarchivedUser.refreshToken,
  526. user.refreshToken);
  527. XCTAssertEqualObjects(
  528. unarchivedUser.metadata.creationDate,
  529. user.metadata.creationDate);
  530. XCTAssertEqualObjects(
  531. unarchivedUser.metadata.lastSignInDate,
  532. user.metadata.lastSignInDate);
  533. XCTAssertEqual(unarchivedUser.providerData.count,
  534. user.providerData.count);
  535. NSDictionary<NSString *, id<FIRUserInfo>>
  536. *unarchivedProviderMap = [self
  537. dictionaryWithUserInfoArray:unarchivedUser
  538. .providerData];
  539. // Verify NSSecureCoding properties from
  540. // email/password.
  541. id<FIRUserInfo> unarchivedPasswordUserInfo =
  542. unarchivedProviderMap[FIREmailAuthProviderID];
  543. XCTAssertEqualObjects(unarchivedPasswordUserInfo.uid,
  544. passwordUserInfo.uid);
  545. XCTAssertEqualObjects(
  546. unarchivedPasswordUserInfo.displayName,
  547. passwordUserInfo.displayName);
  548. XCTAssertEqualObjects(
  549. unarchivedPasswordUserInfo.photoURL,
  550. passwordUserInfo.photoURL);
  551. XCTAssertEqualObjects(unarchivedPasswordUserInfo.email,
  552. passwordUserInfo.email);
  553. // Verify NSSecureCoding properties from the Google
  554. // auth provider.
  555. id<FIRUserInfo> unarchivedGoogleUserInfo =
  556. unarchivedProviderMap[FIRGoogleAuthProviderID];
  557. XCTAssertEqualObjects(unarchivedGoogleUserInfo.uid,
  558. googleUserInfo.uid);
  559. XCTAssertEqualObjects(
  560. unarchivedGoogleUserInfo.displayName,
  561. googleUserInfo.displayName);
  562. XCTAssertEqualObjects(
  563. unarchivedGoogleUserInfo.photoURL,
  564. googleUserInfo.photoURL);
  565. XCTAssertEqualObjects(unarchivedGoogleUserInfo.email,
  566. googleUserInfo.email);
  567. // Verify NSSecureCoding properties from the Facebook
  568. // auth provider.
  569. id<FIRUserInfo> unarchivedFacebookUserInfo =
  570. unarchivedProviderMap[FIRFacebookAuthProviderID];
  571. XCTAssertEqualObjects(unarchivedFacebookUserInfo.uid,
  572. facebookUserInfo.uid);
  573. XCTAssertEqualObjects(
  574. unarchivedFacebookUserInfo.displayName,
  575. facebookUserInfo.displayName);
  576. XCTAssertEqualObjects(
  577. unarchivedFacebookUserInfo.photoURL,
  578. facebookUserInfo.photoURL);
  579. XCTAssertEqualObjects(unarchivedFacebookUserInfo.email,
  580. facebookUserInfo.email);
  581. #if TARGET_OS_IOS
  582. // Verify FIRUserInfo properties from the phone auth
  583. // provider.
  584. id<FIRUserInfo> unarchivedPhoneUserInfo =
  585. unarchivedProviderMap[FIRPhoneAuthProviderID];
  586. XCTAssertEqualObjects(
  587. unarchivedPhoneUserInfo.phoneNumber,
  588. phoneUserInfo.phoneNumber);
  589. #endif
  590. [expectation fulfill];
  591. }];
  592. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  593. OCMVerifyAll(_mockBackend);
  594. }
  595. /** @fn testUpdateEmailSuccess
  596. @brief Tests the flow of a successful @c updateEmail:completion: call.
  597. */
  598. - (void)testUpdateEmailSuccess {
  599. id (^mockUserInfoWithDisplayName)(NSString *) = ^(NSString *displayName) {
  600. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  601. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  602. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  603. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(displayName);
  604. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  605. return mockGetAccountInfoResponseUser;
  606. };
  607. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  608. id userInfoResponse = mockUserInfoWithDisplayName(kGoogleDisplayName);
  609. [self
  610. signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  611. completion:^(FIRUser *user) {
  612. // Pretend that the display name on the server has been
  613. // changed since last request.
  614. [self
  615. expectGetAccountInfoWithMockUserInfoResponse:
  616. mockUserInfoWithDisplayName(kNewDisplayName)];
  617. OCMExpect([self->_mockBackend
  618. setAccountInfo:[OCMArg any]
  619. callback:[OCMArg any]])
  620. .andCallBlock2(^(
  621. FIRSetAccountInfoRequest *_Nullable request,
  622. FIRSetAccountInfoResponseCallback callback) {
  623. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  624. XCTAssertEqualObjects(request.accessToken,
  625. kAccessToken);
  626. XCTAssertEqualObjects(request.email, kNewEmail);
  627. XCTAssertNil(request.localID);
  628. XCTAssertNil(request.displayName);
  629. XCTAssertNil(request.photoURL);
  630. XCTAssertNil(request.password);
  631. XCTAssertNil(request.providers);
  632. XCTAssertNil(request.deleteAttributes);
  633. XCTAssertNil(request.deleteProviders);
  634. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  635. id mockSetAccountInfoResponse = OCMClassMock(
  636. [FIRSetAccountInfoResponse class]);
  637. OCMStub([mockSetAccountInfoResponse email])
  638. .andReturn(kNewEmail);
  639. OCMStub(
  640. [mockSetAccountInfoResponse displayName])
  641. .andReturn(kNewDisplayName);
  642. callback(mockSetAccountInfoResponse, nil);
  643. });
  644. });
  645. [user updateEmail:kNewEmail
  646. completion:^(NSError *_Nullable error) {
  647. XCTAssertNil(error);
  648. XCTAssertEqualObjects(user.email, kNewEmail);
  649. XCTAssertEqualObjects(user.displayName,
  650. kNewDisplayName);
  651. [expectation fulfill];
  652. }];
  653. }];
  654. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  655. OCMVerifyAll(_mockBackend);
  656. }
  657. /** @fn testUpdateEmailWithAuthLinkAccountSuccess
  658. @brief Tests a successful @c updateEmail:completion: call updates provider info.
  659. */
  660. - (void)testUpdateEmailWithAuthLinkAccountSuccess {
  661. id (^mockUserInfoWithDisplayName)(NSString *) = ^(NSString *displayName) {
  662. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  663. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  664. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  665. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(displayName);
  666. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  667. return mockGetAccountInfoResponseUser;
  668. };
  669. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  670. id userInfoResponse = mockUserInfoWithDisplayName(kGoogleDisplayName);
  671. [self
  672. signInWithEmailLinkWithMockUserInfoResponse:userInfoResponse
  673. completion:^(FIRUser *user) {
  674. // Pretend that the display name on the server has been
  675. // changed since last request.
  676. [self expectGetAccountInfoWithMockUserInfoResponse:
  677. mockUserInfoWithDisplayName(kNewDisplayName)];
  678. OCMExpect([self->_mockBackend setAccountInfo:[OCMArg any]
  679. callback:[OCMArg any]])
  680. .andCallBlock2(^(
  681. FIRSetAccountInfoRequest *_Nullable request,
  682. FIRSetAccountInfoResponseCallback callback) {
  683. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  684. XCTAssertEqualObjects(request.accessToken,
  685. kAccessToken);
  686. XCTAssertEqualObjects(request.email, kNewEmail);
  687. XCTAssertNil(request.localID);
  688. XCTAssertNil(request.displayName);
  689. XCTAssertNil(request.photoURL);
  690. XCTAssertNil(request.password);
  691. XCTAssertNil(request.providers);
  692. XCTAssertNil(request.deleteAttributes);
  693. XCTAssertNil(request.deleteProviders);
  694. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  695. id mockSetAccountInfoResponse = OCMClassMock(
  696. [FIRSetAccountInfoResponse class]);
  697. OCMStub([mockSetAccountInfoResponse email])
  698. .andReturn(kNewEmail);
  699. OCMStub([mockSetAccountInfoResponse displayName])
  700. .andReturn(kNewDisplayName);
  701. callback(mockSetAccountInfoResponse, nil);
  702. });
  703. });
  704. [user updateEmail:kNewEmail
  705. completion:^(NSError *_Nullable error) {
  706. XCTAssertNil(error);
  707. XCTAssertEqualObjects(user.email, kNewEmail);
  708. XCTAssertEqualObjects(user.displayName,
  709. kNewDisplayName);
  710. XCTAssertFalse(user.isAnonymous);
  711. [expectation fulfill];
  712. }];
  713. }];
  714. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  715. OCMVerifyAll(_mockBackend);
  716. }
  717. /** @fn testUpdateEmailFailure
  718. @brief Tests the flow of a failed @c updateEmail:completion: call.
  719. */
  720. - (void)testUpdateEmailFailure {
  721. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  722. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  723. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  724. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  725. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  726. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  727. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  728. completion:^(FIRUser *user) {
  729. [self expectGetAccountInfoWithMockUserInfoResponse:
  730. mockGetAccountInfoResponseUser];
  731. OCMExpect([self->_mockBackend
  732. setAccountInfo:[OCMArg any]
  733. callback:[OCMArg any]])
  734. .andDispatchError2([FIRAuthErrorUtils
  735. invalidEmailErrorWithMessage:nil]);
  736. [user
  737. updateEmail:kNewEmail
  738. completion:^(NSError *_Nullable error) {
  739. XCTAssertTrue([NSThread isMainThread]);
  740. XCTAssertEqual(error.code,
  741. FIRAuthErrorCodeInvalidEmail);
  742. // Email should not have changed on the client
  743. // side.
  744. XCTAssertEqualObjects(user.email, kEmail);
  745. // User is still signed in.
  746. XCTAssertEqual([FIRAuth auth].currentUser,
  747. user);
  748. [expectation fulfill];
  749. }];
  750. }];
  751. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  752. OCMVerifyAll(_mockBackend);
  753. }
  754. /** @fn testUpdateEmailAutoSignOut
  755. @brief Tests the flow of a failed @c updateEmail:completion: call that automatically signs out.
  756. */
  757. - (void)testUpdateEmailAutoSignOut {
  758. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  759. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  760. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  761. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  762. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  763. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  764. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  765. completion:^(FIRUser *user) {
  766. [self expectGetAccountInfoWithMockUserInfoResponse:
  767. mockGetAccountInfoResponseUser];
  768. OCMExpect([self->_mockBackend
  769. setAccountInfo:[OCMArg any]
  770. callback:[OCMArg any]])
  771. .andDispatchError2([FIRAuthErrorUtils
  772. invalidUserTokenErrorWithMessage:nil]);
  773. [user updateEmail:kNewEmail
  774. completion:^(NSError *_Nullable error) {
  775. XCTAssertTrue([NSThread isMainThread]);
  776. XCTAssertEqual(
  777. error.code,
  778. FIRAuthErrorCodeInvalidUserToken);
  779. // Email should not have changed on the
  780. // client side.
  781. XCTAssertEqualObjects(user.email, kEmail);
  782. // User is no longer signed in.
  783. XCTAssertNil([FIRAuth auth].currentUser);
  784. [expectation fulfill];
  785. }];
  786. }];
  787. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  788. OCMVerifyAll(_mockBackend);
  789. }
  790. #if TARGET_OS_IOS
  791. /** @fn testUpdatePhoneSuccess
  792. @brief Tests the flow of a successful @c updatePhoneNumberCredential:completion: call.
  793. */
  794. - (void)testUpdatePhoneSuccess {
  795. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  796. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  797. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  798. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  799. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  800. if (phoneNumber.length) {
  801. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  802. }
  803. return mockGetAccountInfoResponseUser;
  804. };
  805. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  806. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  807. [self
  808. signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  809. completion:^(FIRUser *user) {
  810. [self
  811. expectVerifyPhoneNumberRequestWithPhoneNumber:
  812. kPhoneNumber
  813. isLinkOperation:NO
  814. error:nil];
  815. id userInfoResponseUpdate =
  816. mockUserInfoWithPhoneNumber(kPhoneNumber);
  817. [self expectGetAccountInfoWithMockUserInfoResponse:
  818. userInfoResponseUpdate];
  819. FIRPhoneAuthCredential *credential =
  820. [[FIRPhoneAuthProvider provider]
  821. credentialWithVerificationID:kVerificationID
  822. verificationCode:
  823. kVerificationCode];
  824. [user updatePhoneNumberCredential:credential
  825. completion:^(
  826. NSError *_Nullable error) {
  827. XCTAssertTrue(
  828. [NSThread isMainThread]);
  829. XCTAssertNil(error);
  830. XCTAssertEqualObjects(
  831. [FIRAuth auth]
  832. .currentUser
  833. .phoneNumber,
  834. kPhoneNumber);
  835. [expectation fulfill];
  836. }];
  837. }];
  838. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  839. OCMVerifyAll(_mockBackend);
  840. }
  841. /** @fn testUpdatePhoneNumberFailure
  842. @brief Tests the flow of a failed @c updatePhoneNumberCredential:completion: call.
  843. */
  844. - (void)testUpdatePhoneNumberFailure {
  845. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  846. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  847. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  848. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  849. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  850. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  851. [self
  852. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  853. completion:^(FIRUser *user) {
  854. OCMExpect([self->_mockBackend
  855. verifyPhoneNumber:[OCMArg any]
  856. callback:[OCMArg any]])
  857. .andDispatchError2([FIRAuthErrorUtils
  858. invalidPhoneNumberErrorWithMessage:nil]);
  859. FIRPhoneAuthCredential *credential =
  860. [[FIRPhoneAuthProvider provider]
  861. credentialWithVerificationID:kVerificationID
  862. verificationCode:
  863. kVerificationCode];
  864. [user
  865. updatePhoneNumberCredential:credential
  866. completion:^(
  867. NSError *_Nullable error) {
  868. XCTAssertTrue(
  869. [NSThread isMainThread]);
  870. XCTAssertEqual(
  871. error.code,
  872. FIRAuthErrorCodeInvalidPhoneNumber);
  873. XCTAssertEqual(
  874. [FIRAuth auth].currentUser,
  875. user);
  876. [expectation fulfill];
  877. }];
  878. }];
  879. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  880. OCMVerifyAll(_mockBackend);
  881. }
  882. /** @fn testUpdatePhoneNumberFailureAutoSignOut
  883. @brief Tests the flow of a failed @c updatePhoneNumberCredential:completion: call that
  884. automatically signs out.
  885. */
  886. - (void)testUpdatePhoneNumberFailureAutoSignOut {
  887. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  888. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  889. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  890. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  891. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  892. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  893. [self
  894. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  895. completion:^(FIRUser *user) {
  896. OCMExpect([self->_mockBackend
  897. verifyPhoneNumber:[OCMArg any]
  898. callback:[OCMArg any]])
  899. .andDispatchError2([FIRAuthErrorUtils
  900. userTokenExpiredErrorWithMessage:nil]);
  901. FIRPhoneAuthCredential *credential =
  902. [[FIRPhoneAuthProvider provider]
  903. credentialWithVerificationID:kVerificationID
  904. verificationCode:
  905. kVerificationCode];
  906. [user
  907. updatePhoneNumberCredential:credential
  908. completion:^(
  909. NSError *_Nullable error) {
  910. XCTAssertTrue(
  911. [NSThread isMainThread]);
  912. XCTAssertEqual(
  913. error.code,
  914. FIRAuthErrorCodeUserTokenExpired);
  915. XCTAssertNil(
  916. [FIRAuth auth].currentUser);
  917. [expectation fulfill];
  918. }];
  919. }];
  920. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  921. OCMVerifyAll(_mockBackend);
  922. }
  923. #endif
  924. /** @fn testUpdatePasswordSuccess
  925. @brief Tests the flow of a successful @c updatePassword:completion: call.
  926. */
  927. - (void)testUpdatePasswordSuccess {
  928. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  929. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  930. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  931. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  932. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  933. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  934. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  935. completion:^(FIRUser *user) {
  936. [self expectGetAccountInfoWithMockUserInfoResponse:
  937. mockGetAccountInfoResponseUser];
  938. OCMExpect([self->_mockBackend
  939. setAccountInfo:[OCMArg any]
  940. callback:[OCMArg any]])
  941. .andCallBlock2(^(
  942. FIRSetAccountInfoRequest *_Nullable request,
  943. FIRSetAccountInfoResponseCallback callback) {
  944. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  945. XCTAssertEqualObjects(request.accessToken,
  946. kAccessToken);
  947. XCTAssertEqualObjects(request.password,
  948. kNewPassword);
  949. XCTAssertNil(request.localID);
  950. XCTAssertNil(request.displayName);
  951. XCTAssertNil(request.photoURL);
  952. XCTAssertNil(request.email);
  953. XCTAssertNil(request.providers);
  954. XCTAssertNil(request.deleteAttributes);
  955. XCTAssertNil(request.deleteProviders);
  956. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  957. id mockSetAccountInfoResponse = OCMClassMock(
  958. [FIRSetAccountInfoResponse class]);
  959. OCMStub(
  960. [mockSetAccountInfoResponse displayName])
  961. .andReturn(kNewDisplayName);
  962. callback(mockSetAccountInfoResponse, nil);
  963. });
  964. });
  965. [user updatePassword:kNewPassword
  966. completion:^(NSError *_Nullable error) {
  967. XCTAssertTrue([NSThread isMainThread]);
  968. XCTAssertNil(error);
  969. XCTAssertFalse(user.isAnonymous);
  970. [expectation fulfill];
  971. }];
  972. }];
  973. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  974. OCMVerifyAll(_mockBackend);
  975. }
  976. /** @fn testUpdatePasswordFailure
  977. @brief Tests the flow of a failed @c updatePassword:completion: call.
  978. */
  979. - (void)testUpdatePasswordFailure {
  980. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  981. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  982. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  983. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  984. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  985. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  986. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  987. completion:^(FIRUser *user) {
  988. [self expectGetAccountInfoWithMockUserInfoResponse:
  989. mockGetAccountInfoResponseUser];
  990. OCMExpect([self->_mockBackend
  991. setAccountInfo:[OCMArg any]
  992. callback:[OCMArg any]])
  993. .andDispatchError2([FIRAuthErrorUtils
  994. requiresRecentLoginErrorWithMessage:nil]);
  995. [user updatePassword:kNewPassword
  996. completion:^(NSError *_Nullable error) {
  997. XCTAssertTrue([NSThread isMainThread]);
  998. XCTAssertEqual(
  999. error.code,
  1000. FIRAuthErrorCodeRequiresRecentLogin);
  1001. XCTAssertEqual(
  1002. [FIRAuth auth].currentUser, user);
  1003. [expectation fulfill];
  1004. }];
  1005. }];
  1006. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1007. OCMVerifyAll(_mockBackend);
  1008. }
  1009. /** @fn testUpdateEmptyPasswordFailure
  1010. @brief Tests the flow of a failed @c updatePassword:completion: call due to an empty password.
  1011. */
  1012. - (void)testUpdateEmptyPasswordFailure {
  1013. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1014. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1015. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1016. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1017. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1018. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1019. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1020. completion:^(FIRUser *user) {
  1021. [user updatePassword:@""
  1022. completion:^(NSError *_Nullable error) {
  1023. XCTAssertTrue([NSThread isMainThread]);
  1024. XCTAssertEqual(
  1025. error.code,
  1026. FIRAuthErrorCodeWeakPassword);
  1027. [expectation fulfill];
  1028. }];
  1029. }];
  1030. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1031. }
  1032. /** @fn testUpdatePasswordFailureAutoSignOut
  1033. @brief Tests the flow of a failed @c updatePassword:completion: call that automatically signs
  1034. out.
  1035. */
  1036. - (void)testUpdatePasswordFailureAutoSignOut {
  1037. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1038. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1039. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1040. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1041. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1042. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1043. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1044. completion:^(FIRUser *user) {
  1045. [self expectGetAccountInfoWithMockUserInfoResponse:
  1046. mockGetAccountInfoResponseUser];
  1047. OCMExpect([self->_mockBackend
  1048. setAccountInfo:[OCMArg any]
  1049. callback:[OCMArg any]])
  1050. .andDispatchError2([FIRAuthErrorUtils
  1051. userDisabledErrorWithMessage:nil]);
  1052. [user updatePassword:kNewPassword
  1053. completion:^(NSError *_Nullable error) {
  1054. XCTAssertTrue([NSThread isMainThread]);
  1055. XCTAssertEqual(
  1056. error.code,
  1057. FIRAuthErrorCodeUserDisabled);
  1058. XCTAssertNil([FIRAuth auth].currentUser);
  1059. [expectation fulfill];
  1060. }];
  1061. }];
  1062. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1063. OCMVerifyAll(_mockBackend);
  1064. }
  1065. /** @fn testChangeProfileSuccess
  1066. @brief Tests a successful user profile change flow.
  1067. */
  1068. - (void)testChangeProfileSuccess {
  1069. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1070. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1071. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1072. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1073. OCMStub([mockGetAccountInfoResponseUser photoURL]).andReturn(kPhotoURL);
  1074. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1075. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1076. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1077. completion:^(FIRUser *user) {
  1078. [self expectGetAccountInfoWithMockUserInfoResponse:
  1079. mockGetAccountInfoResponseUser];
  1080. OCMExpect([self->_mockBackend
  1081. setAccountInfo:[OCMArg any]
  1082. callback:[OCMArg any]])
  1083. .andCallBlock2(^(
  1084. FIRSetAccountInfoRequest *_Nullable request,
  1085. FIRSetAccountInfoResponseCallback callback) {
  1086. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1087. XCTAssertEqualObjects(request.accessToken,
  1088. kAccessToken);
  1089. XCTAssertEqualObjects(request.displayName,
  1090. kNewDisplayName);
  1091. XCTAssertEqualObjects(
  1092. request.photoURL,
  1093. [NSURL URLWithString:kNewPhotoURL]);
  1094. XCTAssertNil(request.localID);
  1095. XCTAssertNil(request.email);
  1096. XCTAssertNil(request.password);
  1097. XCTAssertNil(request.providers);
  1098. XCTAssertNil(request.deleteAttributes);
  1099. XCTAssertNil(request.deleteProviders);
  1100. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1101. id mockSetAccountInfoResponse = OCMClassMock(
  1102. [FIRSetAccountInfoResponse class]);
  1103. OCMStub(
  1104. [mockSetAccountInfoResponse displayName])
  1105. .andReturn(kNewDisplayName);
  1106. callback(mockSetAccountInfoResponse, nil);
  1107. });
  1108. });
  1109. FIRUserProfileChangeRequest *profileChange =
  1110. [user profileChangeRequest];
  1111. profileChange.photoURL =
  1112. [NSURL URLWithString:kNewPhotoURL];
  1113. profileChange.displayName = kNewDisplayName;
  1114. [profileChange commitChangesWithCompletion:^(
  1115. NSError *_Nullable error) {
  1116. XCTAssertTrue([NSThread isMainThread]);
  1117. XCTAssertNil(error);
  1118. XCTAssertEqualObjects(user.displayName,
  1119. kNewDisplayName);
  1120. XCTAssertEqualObjects(
  1121. user.photoURL,
  1122. [NSURL URLWithString:kNewPhotoURL]);
  1123. [expectation fulfill];
  1124. }];
  1125. }];
  1126. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1127. OCMVerifyAll(_mockBackend);
  1128. }
  1129. /** @fn testChangeProfileFailure
  1130. @brief Tests a failed user profile change flow.
  1131. */
  1132. - (void)testChangeProfileFailure {
  1133. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1134. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1135. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1136. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1137. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1138. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1139. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1140. completion:^(FIRUser *user) {
  1141. [self expectGetAccountInfoWithMockUserInfoResponse:
  1142. mockGetAccountInfoResponseUser];
  1143. OCMExpect([self->_mockBackend
  1144. setAccountInfo:[OCMArg any]
  1145. callback:[OCMArg any]])
  1146. .andDispatchError2([FIRAuthErrorUtils
  1147. tooManyRequestsErrorWithMessage:nil]);
  1148. FIRUserProfileChangeRequest *profileChange =
  1149. [user profileChangeRequest];
  1150. profileChange.displayName = kNewDisplayName;
  1151. [profileChange commitChangesWithCompletion:^(
  1152. NSError *_Nullable error) {
  1153. XCTAssertTrue([NSThread isMainThread]);
  1154. XCTAssertEqual(error.code,
  1155. FIRAuthErrorCodeTooManyRequests);
  1156. XCTAssertEqualObjects(user.displayName,
  1157. kGoogleDisplayName);
  1158. XCTAssertEqual([FIRAuth auth].currentUser, user);
  1159. [expectation fulfill];
  1160. }];
  1161. }];
  1162. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1163. OCMVerifyAll(_mockBackend);
  1164. }
  1165. /** @fn testChangeProfileFailureAutoSignOut
  1166. @brief Tests a failed user profile change flow that automatically signs out.
  1167. */
  1168. - (void)testChangeProfileFailureAutoSignOut {
  1169. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1170. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1171. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1172. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1173. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1174. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1175. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1176. completion:^(FIRUser *user) {
  1177. [self expectGetAccountInfoWithMockUserInfoResponse:
  1178. mockGetAccountInfoResponseUser];
  1179. OCMExpect([self->_mockBackend
  1180. setAccountInfo:[OCMArg any]
  1181. callback:[OCMArg any]])
  1182. .andDispatchError2([FIRAuthErrorUtils
  1183. userNotFoundErrorWithMessage:nil]);
  1184. FIRUserProfileChangeRequest *profileChange =
  1185. [user profileChangeRequest];
  1186. profileChange.displayName = kNewDisplayName;
  1187. [profileChange commitChangesWithCompletion:^(
  1188. NSError *_Nullable error) {
  1189. XCTAssertTrue([NSThread isMainThread]);
  1190. XCTAssertEqual(error.code,
  1191. FIRAuthErrorCodeUserNotFound);
  1192. XCTAssertEqualObjects(user.displayName,
  1193. kGoogleDisplayName);
  1194. XCTAssertNil([FIRAuth auth].currentUser);
  1195. [expectation fulfill];
  1196. }];
  1197. }];
  1198. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1199. OCMVerifyAll(_mockBackend);
  1200. }
  1201. /** @fn testReloadSuccess
  1202. @brief Tests the flow of a successful @c reloadWithCompletion: call.
  1203. */
  1204. - (void)testReloadSuccess {
  1205. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1206. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1207. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1208. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1209. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1210. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1211. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1212. completion:^(FIRUser *user) {
  1213. id mockGetAccountInfoResponseUserNew = OCMClassMock(
  1214. [FIRGetAccountInfoResponseUser class]);
  1215. OCMStub([mockGetAccountInfoResponseUserNew localID])
  1216. .andReturn(kLocalID);
  1217. OCMStub([mockGetAccountInfoResponseUserNew email])
  1218. .andReturn(kNewEmail);
  1219. OCMStub(
  1220. [mockGetAccountInfoResponseUserNew displayName])
  1221. .andReturn(kNewDisplayName);
  1222. OCMStub(
  1223. [mockGetAccountInfoResponseUserNew passwordHash])
  1224. .andReturn(kPasswordHash);
  1225. [self expectGetAccountInfoWithMockUserInfoResponse:
  1226. mockGetAccountInfoResponseUserNew];
  1227. [user reloadWithCompletion:^(
  1228. NSError *_Nullable error) {
  1229. XCTAssertTrue([NSThread isMainThread]);
  1230. XCTAssertNil(error);
  1231. XCTAssertEqualObjects(user.email, kNewEmail);
  1232. XCTAssertEqualObjects(user.displayName,
  1233. kNewDisplayName);
  1234. [expectation fulfill];
  1235. }];
  1236. }];
  1237. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1238. OCMVerifyAll(_mockBackend);
  1239. }
  1240. /** @fn testGetIDTokenResultSuccess
  1241. @brief Tests the flow of a successful @c getIDTokenResultWithCompletion: call.
  1242. */
  1243. - (void)testGetIDTokenResultSuccess {
  1244. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1245. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1246. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1247. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1248. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1249. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1250. id mockSecureTokenService = OCMClassMock([FIRSecureTokenService class]);
  1251. OCMStub([mockSecureTokenService hasValidAccessToken]).andReturn(YES);
  1252. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1253. completion:^(FIRUser *user) {
  1254. [user getIDTokenResultWithCompletion:^(
  1255. FIRAuthTokenResult *_Nullable tokenResult,
  1256. NSError *_Nullable error) {
  1257. XCTAssertTrue([NSThread isMainThread]);
  1258. XCTAssertNil(error);
  1259. XCTAssertEqualObjects(tokenResult.token,
  1260. kAccessToken);
  1261. XCTAssertTrue(tokenResult.issuedAtDate &&
  1262. [tokenResult.issuedAtDate
  1263. isKindOfClass:[NSDate class]]);
  1264. XCTAssertTrue(tokenResult.authDate &&
  1265. [tokenResult.authDate
  1266. isKindOfClass:[NSDate class]]);
  1267. XCTAssertTrue(tokenResult.expirationDate &&
  1268. [tokenResult.expirationDate
  1269. isKindOfClass:[NSDate class]]);
  1270. XCTAssertEqualObjects(tokenResult.signInProvider,
  1271. FIREmailAuthProviderID);
  1272. XCTAssertTrue(
  1273. tokenResult.claims &&
  1274. [tokenResult.claims
  1275. isKindOfClass:[NSDictionary class]]);
  1276. [expectation fulfill];
  1277. }];
  1278. }];
  1279. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1280. OCMVerifyAll(_mockBackend);
  1281. }
  1282. /** @fn testGetIDTokenResultForcingRefreshNoCustomTokenDelegateNoRefreshTokenFailure
  1283. @brief Tests the flow of a failed @c getIDTokenResultForcingRefresh:completion: call when there
  1284. is no refresh token nor custom token provider.
  1285. */
  1286. - (void)testGetIDTokenResultForcingRefreshNoCustomTokenDelegateNoRefreshTokenFailure {
  1287. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1288. [self signInWithCustomTokenNoRefreshTokenWithCompletion:^(FIRUser *user) {
  1289. [user getIDTokenResultForcingRefresh:YES
  1290. completion:^(FIRAuthTokenResult *_Nullable tokenResult,
  1291. NSError *_Nullable error) {
  1292. XCTAssertTrue([NSThread isMainThread]);
  1293. XCTAssertNil(tokenResult);
  1294. XCTAssertEqual(error.code, FIRAuthErrorCodeTokenRefreshUnavailable);
  1295. XCTAssertEqualObjects(error.userInfo[NSLocalizedDescriptionKey],
  1296. @"No refresh token is available.");
  1297. [expectation fulfill];
  1298. }];
  1299. }];
  1300. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1301. OCMVerifyAll(_mockBackend);
  1302. }
  1303. /** @fn testGetIDTokenResultForcingRefreshCustomTokenDelegateNoRefreshTokenSuccess
  1304. @brief Tests the flow of a successful @c getIDTokenResultForcingRefresh:completion: call when no
  1305. refresh token is available, but a custom token provider is set.
  1306. */
  1307. - (void)testGetIDTokenResultForcingRefreshCustomTokenDelegateNoRefreshTokenSuccess {
  1308. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1309. [FIRAuth auth].customTokenProviderDelegate = (id<FIRCustomTokenProviderDelegate>)self;
  1310. [self signInWithCustomTokenNoRefreshTokenWithCompletion:^(FIRUser *user) {
  1311. OCMExpect([self->_mockBackend verifyCustomToken:[OCMArg any] callback:[OCMArg any]])
  1312. .andCallBlock2(^(FIRVerifyCustomTokenRequest *_Nullable request,
  1313. FIRVerifyCustomTokenResponseCallback callback) {
  1314. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1315. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1316. id mockVerifyCustomTokenResponse = OCMClassMock([FIRVerifyCustomTokenResponse class]);
  1317. OCMStub([mockVerifyCustomTokenResponse IDToken]).andReturn(kAccessToken);
  1318. callback(mockVerifyCustomTokenResponse, nil);
  1319. });
  1320. });
  1321. [user getIDTokenResultForcingRefresh:YES
  1322. completion:^(FIRAuthTokenResult *_Nullable tokenResult,
  1323. NSError *_Nullable error) {
  1324. XCTAssertTrue([NSThread isMainThread]);
  1325. XCTAssertNil(error);
  1326. XCTAssertEqualObjects(tokenResult.token, kAccessToken);
  1327. XCTAssertTrue(
  1328. tokenResult.issuedAtDate &&
  1329. [tokenResult.issuedAtDate isKindOfClass:[NSDate class]]);
  1330. XCTAssertTrue(tokenResult.authDate &&
  1331. [tokenResult.authDate isKindOfClass:[NSDate class]]);
  1332. XCTAssertTrue(
  1333. tokenResult.expirationDate &&
  1334. [tokenResult.expirationDate isKindOfClass:[NSDate class]]);
  1335. XCTAssertTrue(
  1336. tokenResult.claims &&
  1337. [tokenResult.claims isKindOfClass:[NSDictionary class]]);
  1338. [expectation fulfill];
  1339. }];
  1340. }];
  1341. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1342. OCMVerifyAll(_mockBackend);
  1343. }
  1344. /** @fn testGetIDTokenResultForcingRefreshCustomTokenDelegateUserIDMismatch
  1345. @brief Tests the flow of a failed @c getIDTokenResultForcingRefresh:completion: call with a
  1346. custom token provider, when the user ID does not match the existing signed-in user.
  1347. */
  1348. - (void)testGetIDTokenResultForcingRefreshCustomTokenDelegateUserIDMismatch {
  1349. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1350. [FIRAuth auth].customTokenProviderDelegate = (id<FIRCustomTokenProviderDelegate>)self;
  1351. [self signInWithCustomTokenNoRefreshTokenWithCompletion:^(FIRUser *user) {
  1352. OCMExpect([self->_mockBackend verifyCustomToken:[OCMArg any] callback:[OCMArg any]])
  1353. .andCallBlock2(^(FIRVerifyCustomTokenRequest *_Nullable request,
  1354. FIRVerifyCustomTokenResponseCallback callback) {
  1355. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1356. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1357. id mockVerifyCustomTokenResponse = OCMClassMock([FIRVerifyCustomTokenResponse class]);
  1358. OCMStub([mockVerifyCustomTokenResponse IDToken]).andReturn(kAccessToken2);
  1359. callback(mockVerifyCustomTokenResponse, nil);
  1360. });
  1361. });
  1362. [user getIDTokenResultForcingRefresh:YES
  1363. completion:^(FIRAuthTokenResult *_Nullable tokenResult,
  1364. NSError *_Nullable error) {
  1365. XCTAssertTrue([NSThread isMainThread]);
  1366. XCTAssertNil(tokenResult);
  1367. XCTAssertEqual(error.code, FIRAuthErrorCodeUserMismatch);
  1368. [expectation fulfill];
  1369. }];
  1370. }];
  1371. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1372. OCMVerifyAll(_mockBackend);
  1373. }
  1374. /** @fn testGetIDTokenResultForcingRefreshFailure
  1375. @brief Tests the flow successful @c getIDTokenResultForcingRefresh:completion: calls.
  1376. */
  1377. - (void)testGetIDTokenResultForcingRefreshSuccess {
  1378. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessToken];
  1379. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenLength415];
  1380. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenLength416];
  1381. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenLength523];
  1382. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenWithBase64URLCharacter];
  1383. }
  1384. /** @fn testGetIDTokenResultSuccessWithBase64EncodedURL
  1385. @brief Tests the flow of a successful @c getIDTokenResultWithCompletion: call using a base64 url
  1386. encoded string.
  1387. */
  1388. - (void)testGetIDTokenResultSuccessWithBase64EncodedURL {
  1389. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1390. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1391. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1392. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1393. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1394. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1395. [self
  1396. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1397. completion:^(FIRUser *user) {
  1398. OCMExpect([self->_mockBackend
  1399. secureToken:[OCMArg any]
  1400. callback:[OCMArg any]])
  1401. .andCallBlock2(^(
  1402. FIRSecureTokenRequest *_Nullable request,
  1403. FIRSecureTokenResponseCallback callback) {
  1404. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1405. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1406. id mockSecureTokenResponse = OCMClassMock(
  1407. [FIRSecureTokenResponse class]);
  1408. OCMStub([mockSecureTokenResponse accessToken])
  1409. .andReturn(
  1410. kAccessTokenWithBase64URLCharacter);
  1411. callback(mockSecureTokenResponse, nil);
  1412. });
  1413. });
  1414. [user
  1415. getIDTokenResultForcingRefresh:YES
  1416. completion:^(
  1417. FIRAuthTokenResult
  1418. *_Nullable tokenResult,
  1419. NSError *_Nullable error) {
  1420. XCTAssertTrue(
  1421. [NSThread isMainThread]);
  1422. XCTAssertNil(error);
  1423. XCTAssertEqualObjects(
  1424. tokenResult.token,
  1425. kAccessTokenWithBase64URLCharacter);
  1426. XCTAssertTrue(
  1427. tokenResult
  1428. .issuedAtDate &&
  1429. [tokenResult.issuedAtDate
  1430. isKindOfClass:
  1431. [NSDate class]]);
  1432. XCTAssertTrue(
  1433. tokenResult.authDate &&
  1434. [tokenResult.authDate
  1435. isKindOfClass:
  1436. [NSDate class]]);
  1437. XCTAssertTrue(
  1438. tokenResult
  1439. .expirationDate &&
  1440. [tokenResult
  1441. .expirationDate
  1442. isKindOfClass:
  1443. [NSDate class]]);
  1444. XCTAssertTrue(
  1445. tokenResult.claims &&
  1446. [tokenResult.claims
  1447. isKindOfClass:
  1448. [NSDictionary
  1449. class]]);
  1450. NSDictionary *claims =
  1451. tokenResult.claims;
  1452. XCTAssertEqualObjects(
  1453. claims[@"email"],
  1454. kbase64URLEncodedEmail);
  1455. XCTAssertEqualObjects(
  1456. claims[@"aud"],
  1457. kbase64URLEncodedAUD);
  1458. [expectation fulfill];
  1459. }];
  1460. }];
  1461. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1462. OCMVerifyAll(_mockBackend);
  1463. }
  1464. /** @fn testGetIDTokenResultForcingRefreshFailure
  1465. @brief Tests the flow of a failed @c getIDTokenResultForcingRefresh:completion: call.
  1466. */
  1467. - (void)testGetIDTokenResultForcingRefreshFailure {
  1468. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1469. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1470. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1471. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1472. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1473. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1474. [self
  1475. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1476. completion:^(FIRUser *user) {
  1477. OCMExpect([self->_mockBackend
  1478. secureToken:[OCMArg any]
  1479. callback:[OCMArg any]])
  1480. .andCallBlock2(^(
  1481. FIRSecureTokenRequest *_Nullable request,
  1482. FIRSecureTokenResponseCallback callback) {
  1483. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1484. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1485. NSError *underlying =
  1486. [NSError errorWithDomain:@"Test Error"
  1487. code:1
  1488. userInfo:nil];
  1489. callback(nil,
  1490. [FIRAuthErrorUtils
  1491. networkErrorWithUnderlyingError:
  1492. underlying]);
  1493. });
  1494. });
  1495. [user
  1496. getIDTokenResultForcingRefresh:YES
  1497. completion:^(
  1498. FIRAuthTokenResult
  1499. *_Nullable tokenResult,
  1500. NSError *_Nullable error) {
  1501. XCTAssertTrue(
  1502. [NSThread isMainThread]);
  1503. XCTAssertNil(tokenResult);
  1504. XCTAssertEqual(
  1505. error.code,
  1506. FIRAuthErrorCodeNetworkError);
  1507. [expectation fulfill];
  1508. }];
  1509. }];
  1510. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1511. OCMVerifyAll(_mockBackend);
  1512. }
  1513. /** @fn testReloadFailure
  1514. @brief Tests the flow of a failed @c reloadWithCompletion: call.
  1515. */
  1516. - (void)testReloadFailure {
  1517. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1518. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1519. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1520. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1521. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1522. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1523. completion:^(FIRUser *user) {
  1524. OCMExpect([self->_mockBackend
  1525. getAccountInfo:[OCMArg any]
  1526. callback:[OCMArg any]])
  1527. .andDispatchError2([FIRAuthErrorUtils
  1528. quotaExceededErrorWithMessage:nil]);
  1529. [user reloadWithCompletion:^(
  1530. NSError *_Nullable error) {
  1531. XCTAssertTrue([NSThread isMainThread]);
  1532. XCTAssertEqual(error.code,
  1533. FIRAuthErrorCodeQuotaExceeded);
  1534. XCTAssertEqual([FIRAuth auth].currentUser, user);
  1535. [expectation fulfill];
  1536. }];
  1537. }];
  1538. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1539. OCMVerifyAll(_mockBackend);
  1540. }
  1541. /** @fn testReloadFailureAutoSignOut
  1542. @brief Tests the flow of a failed @c reloadWithCompletion: call that automtatically signs out.
  1543. */
  1544. - (void)testReloadFailureAutoSignOut {
  1545. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1546. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1547. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1548. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1549. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1550. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1551. completion:^(FIRUser *user) {
  1552. OCMExpect([self->_mockBackend
  1553. getAccountInfo:[OCMArg any]
  1554. callback:[OCMArg any]])
  1555. .andDispatchError2([FIRAuthErrorUtils
  1556. userTokenExpiredErrorWithMessage:nil]);
  1557. [user reloadWithCompletion:^(
  1558. NSError *_Nullable error) {
  1559. XCTAssertTrue([NSThread isMainThread]);
  1560. XCTAssertEqual(error.code,
  1561. FIRAuthErrorCodeUserTokenExpired);
  1562. XCTAssertNil([FIRAuth auth].currentUser);
  1563. [expectation fulfill];
  1564. }];
  1565. }];
  1566. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1567. OCMVerifyAll(_mockBackend);
  1568. }
  1569. /** @fn testReauthenticateSuccess
  1570. @brief Tests the flow of a successful @c reauthenticateWithCredential:completion: call.
  1571. */
  1572. - (void)testReauthenticateSuccess {
  1573. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1574. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1575. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1576. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1577. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1578. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1579. [self
  1580. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1581. completion:^(FIRUser *user) {
  1582. OCMExpect([self->_mockBackend
  1583. verifyPassword:[OCMArg any]
  1584. callback:[OCMArg any]])
  1585. .andCallBlock2(^(
  1586. FIRVerifyPasswordRequest *_Nullable request,
  1587. FIRVerifyPasswordResponseCallback callback) {
  1588. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1589. id mockVerifyPasswordResponse = OCMClassMock(
  1590. [FIRVerifyPasswordResponse class]);
  1591. // New authentication comes back with new
  1592. // access token.
  1593. OCMStub([mockVerifyPasswordResponse IDToken])
  1594. .andReturn(kNewAccessToken);
  1595. OCMStub([mockVerifyPasswordResponse
  1596. approximateExpirationDate])
  1597. .andReturn(
  1598. [NSDate dateWithTimeIntervalSinceNow:
  1599. kAccessTokenTimeToLive]);
  1600. OCMStub(
  1601. [mockVerifyPasswordResponse refreshToken])
  1602. .andReturn(kRefreshToken);
  1603. callback(mockVerifyPasswordResponse, nil);
  1604. });
  1605. });
  1606. OCMExpect([self->_mockBackend
  1607. getAccountInfo:[OCMArg any]
  1608. callback:[OCMArg any]])
  1609. .andCallBlock2(^(
  1610. FIRGetAccountInfoRequest *_Nullable request,
  1611. FIRGetAccountInfoResponseCallback callback) {
  1612. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1613. // Verify that the new access token is being used
  1614. // for subsequent requests.
  1615. XCTAssertEqualObjects(request.accessToken,
  1616. kNewAccessToken);
  1617. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1618. id mockGetAccountInfoResponse = OCMClassMock(
  1619. [FIRGetAccountInfoResponse class]);
  1620. OCMStub([mockGetAccountInfoResponse users])
  1621. .andReturn(
  1622. @[ mockGetAccountInfoResponseUser ]);
  1623. callback(mockGetAccountInfoResponse, nil);
  1624. });
  1625. });
  1626. FIRAuthCredential *emailCredential =
  1627. [FIREmailAuthProvider
  1628. credentialWithEmail:kEmail
  1629. password:kFakePassword];
  1630. [user
  1631. reauthenticateWithCredential:emailCredential
  1632. completion:^(
  1633. FIRAuthDataResult
  1634. *_Nullable result,
  1635. NSError *_Nullable error) {
  1636. XCTAssertTrue(
  1637. [NSThread isMainThread]);
  1638. XCTAssertNil(error);
  1639. // Verify that the current user
  1640. // is unchanged.
  1641. XCTAssertEqual(
  1642. [FIRAuth auth].currentUser,
  1643. user);
  1644. [expectation fulfill];
  1645. }];
  1646. }];
  1647. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1648. OCMVerifyAll(_mockBackend);
  1649. }
  1650. /** @fn testReauthenticateWithCredentialSuccess
  1651. @brief Tests the flow of a successful @c reauthenticateWithCredential:completion:
  1652. call.
  1653. */
  1654. - (void)testReauthenticateWithCredentialSuccess {
  1655. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1656. federatedID:kGoogleID
  1657. displayName:kGoogleDisplayName
  1658. profile:[[self class] googleProfile]
  1659. providerIDToken:kGoogleIDToken
  1660. providerAccessToken:kGoogleAccessToken];
  1661. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1662. [[FIRAuth auth] signOut:NULL];
  1663. FIRAuthCredential *googleCredential =
  1664. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1665. [[FIRAuth auth]
  1666. signInWithCredential:googleCredential
  1667. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  1668. XCTAssertTrue([NSThread isMainThread]);
  1669. [self assertUserGoogle:authResult.user];
  1670. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  1671. [[self class] googleProfile]);
  1672. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1673. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  1674. FIRGoogleAuthProviderID);
  1675. XCTAssertNil(error);
  1676. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1677. federatedID:kGoogleID
  1678. displayName:kGoogleDisplayName
  1679. profile:[[self class] googleProfile]
  1680. providerIDToken:kGoogleIDToken
  1681. providerAccessToken:kGoogleAccessToken];
  1682. FIRAuthCredential *reauthenticateGoogleCredential =
  1683. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken
  1684. accessToken:kGoogleAccessToken];
  1685. [authResult.user
  1686. reauthenticateWithCredential:reauthenticateGoogleCredential
  1687. completion:^(
  1688. FIRAuthDataResult *_Nullable reauthenticateAuthResult,
  1689. NSError *_Nullable error) {
  1690. XCTAssertTrue([NSThread isMainThread]);
  1691. XCTAssertNil(error);
  1692. // Verify that the current user is unchanged.
  1693. XCTAssertEqual([FIRAuth auth].currentUser,
  1694. authResult.user);
  1695. // Verify that the current user and reauthenticated user
  1696. // are not same pointers.
  1697. XCTAssertNotEqualObjects(authResult.user,
  1698. reauthenticateAuthResult.user);
  1699. // Verify that anyway the current user and reauthenticated
  1700. // user have same IDs.
  1701. XCTAssertEqualObjects(authResult.user.uid,
  1702. reauthenticateAuthResult.user.uid);
  1703. XCTAssertEqualObjects(
  1704. authResult.user.displayName,
  1705. reauthenticateAuthResult.user.displayName);
  1706. XCTAssertEqualObjects(
  1707. reauthenticateAuthResult.additionalUserInfo.profile,
  1708. [[self class] googleProfile]);
  1709. XCTAssertEqualObjects(
  1710. reauthenticateAuthResult.additionalUserInfo.username,
  1711. kUserName);
  1712. XCTAssertEqualObjects(reauthenticateAuthResult
  1713. .additionalUserInfo.providerID,
  1714. FIRGoogleAuthProviderID);
  1715. [expectation fulfill];
  1716. }];
  1717. }];
  1718. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1719. [self assertUserGoogle:[FIRAuth auth].currentUser];
  1720. OCMVerifyAll(_mockBackend);
  1721. }
  1722. /** @fn testReauthenticateFailure
  1723. @brief Tests the flow of a failed @c reauthenticateWithCredential:completion: call.
  1724. */
  1725. - (void)testReauthenticateFailure {
  1726. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1727. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1728. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1729. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1730. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1731. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1732. [self
  1733. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1734. completion:^(FIRUser *user) {
  1735. OCMExpect([self->_mockBackend
  1736. verifyPassword:[OCMArg any]
  1737. callback:[OCMArg any]])
  1738. .andCallBlock2(^(
  1739. FIRVerifyPasswordRequest *_Nullable request,
  1740. FIRVerifyPasswordResponseCallback callback) {
  1741. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1742. id mockVerifyPasswordResponse = OCMClassMock(
  1743. [FIRVerifyPasswordResponse class]);
  1744. OCMStub([mockVerifyPasswordResponse IDToken])
  1745. .andReturn(kNewAccessToken);
  1746. OCMStub([mockVerifyPasswordResponse
  1747. approximateExpirationDate])
  1748. .andReturn(
  1749. [NSDate dateWithTimeIntervalSinceNow:
  1750. kAccessTokenTimeToLive]);
  1751. OCMStub(
  1752. [mockVerifyPasswordResponse refreshToken])
  1753. .andReturn(kRefreshToken);
  1754. callback(mockVerifyPasswordResponse, nil);
  1755. });
  1756. });
  1757. OCMExpect([self->_mockBackend
  1758. getAccountInfo:[OCMArg any]
  1759. callback:[OCMArg any]])
  1760. .andCallBlock2(^(
  1761. FIRGetAccountInfoRequest *_Nullable request,
  1762. FIRGetAccountInfoResponseCallback callback) {
  1763. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1764. id mockGetAccountInfoResponseUserNew =
  1765. OCMClassMock(
  1766. [FIRGetAccountInfoResponseUser class]);
  1767. // The newly-signed-in user has a different ID.
  1768. OCMStub([mockGetAccountInfoResponseUserNew
  1769. localID])
  1770. .andReturn(kAnotherLocalID);
  1771. OCMStub(
  1772. [mockGetAccountInfoResponseUserNew email])
  1773. .andReturn(kNewEmail);
  1774. OCMStub([mockGetAccountInfoResponseUserNew
  1775. displayName])
  1776. .andReturn(kNewDisplayName);
  1777. OCMStub([mockGetAccountInfoResponseUserNew
  1778. passwordHash])
  1779. .andReturn(kPasswordHash);
  1780. id mockGetAccountInfoResponse = OCMClassMock(
  1781. [FIRGetAccountInfoResponse class]);
  1782. OCMStub([mockGetAccountInfoResponse users])
  1783. .andReturn(@[
  1784. mockGetAccountInfoResponseUserNew
  1785. ]);
  1786. callback(mockGetAccountInfoResponse, nil);
  1787. });
  1788. });
  1789. FIRAuthCredential *emailCredential =
  1790. [FIREmailAuthProvider
  1791. credentialWithEmail:kEmail
  1792. password:kFakePassword];
  1793. [user
  1794. reauthenticateWithCredential:emailCredential
  1795. completion:^(
  1796. FIRAuthDataResult
  1797. *_Nullable result,
  1798. NSError *_Nullable error) {
  1799. XCTAssertTrue(
  1800. [NSThread isMainThread]);
  1801. // Verify user mismatch error.
  1802. XCTAssertEqual(
  1803. error.code,
  1804. FIRAuthErrorCodeUserMismatch);
  1805. // Verify that the current user
  1806. // is unchanged.
  1807. XCTAssertEqual(
  1808. [FIRAuth auth].currentUser,
  1809. user);
  1810. [expectation fulfill];
  1811. }];
  1812. }];
  1813. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1814. OCMVerifyAll(_mockBackend);
  1815. }
  1816. /** @fn testReauthenticateUserMismatchFailure
  1817. @brief Tests the flow of a failed @c reauthenticateWithCredential:completion: call due to trying
  1818. to reauthenticate a user that does not exist.
  1819. */
  1820. - (void)testReauthenticateUserMismatchFailure {
  1821. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1822. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1823. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1824. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1825. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1826. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1827. [self
  1828. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1829. completion:^(FIRUser *user) {
  1830. OCMExpect([self->_mockBackend
  1831. verifyAssertion:[OCMArg any]
  1832. callback:[OCMArg any]])
  1833. .andCallBlock2(^(
  1834. FIRVerifyAssertionRequest *_Nullable request,
  1835. FIRVerifyAssertionResponseCallback callback) {
  1836. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1837. callback(
  1838. nil,
  1839. [FIRAuthErrorUtils
  1840. userNotFoundErrorWithMessage:nil]);
  1841. });
  1842. });
  1843. FIRAuthCredential *googleCredential =
  1844. [FIRGoogleAuthProvider
  1845. credentialWithIDToken:kGoogleIDToken
  1846. accessToken:kGoogleAccessToken];
  1847. [user
  1848. reauthenticateWithCredential:googleCredential
  1849. completion:^(
  1850. FIRAuthDataResult
  1851. *_Nullable result,
  1852. NSError *_Nullable error) {
  1853. XCTAssertTrue(
  1854. [NSThread isMainThread]);
  1855. // Verify user mismatch error.
  1856. XCTAssertEqual(
  1857. error.code,
  1858. FIRAuthErrorCodeUserMismatch);
  1859. // Verify that the current user
  1860. // is unchanged.
  1861. XCTAssertEqual(
  1862. [FIRAuth auth].currentUser,
  1863. user);
  1864. [expectation fulfill];
  1865. }];
  1866. }];
  1867. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1868. OCMVerifyAll(_mockBackend);
  1869. }
  1870. /** @fn testlinkAndRetrieveDataSuccess
  1871. @brief Tests the flow of a successful @c linkWithCredential:completion:
  1872. call.
  1873. */
  1874. - (void)testlinkAndRetrieveDataSuccess {
  1875. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1876. federatedID:kFacebookID
  1877. displayName:kFacebookDisplayName
  1878. profile:[[self class] googleProfile]
  1879. providerIDToken:kFacebookIDToken
  1880. providerAccessToken:kFacebookAccessToken];
  1881. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1882. [[FIRAuth auth] signOut:NULL];
  1883. FIRAuthCredential *facebookCredential =
  1884. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1885. [[FIRAuth auth]
  1886. signInWithCredential:facebookCredential
  1887. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  1888. XCTAssertTrue([NSThread isMainThread]);
  1889. [self assertUserFacebook:authResult.user];
  1890. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  1891. [[self class] googleProfile]);
  1892. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1893. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  1894. FIRFacebookAuthProviderID);
  1895. XCTAssertNil(error);
  1896. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1897. federatedID:kGoogleID
  1898. displayName:kGoogleDisplayName
  1899. profile:[[self class] googleProfile]
  1900. providerIDToken:kGoogleIDToken
  1901. providerAccessToken:kGoogleAccessToken];
  1902. FIRAuthCredential *linkGoogleCredential =
  1903. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken
  1904. accessToken:kGoogleAccessToken];
  1905. [authResult.user
  1906. linkWithCredential:linkGoogleCredential
  1907. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  1908. NSError *_Nullable error) {
  1909. XCTAssertTrue([NSThread isMainThread]);
  1910. XCTAssertNil(error);
  1911. // Verify that the current user is unchanged.
  1912. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1913. // Verify that the current user and reauthenticated user are same
  1914. // pointers.
  1915. XCTAssertEqualObjects(authResult.user, linkAuthResult.user);
  1916. // Verify that anyway the current user and linked user have same
  1917. // IDs.
  1918. XCTAssertEqualObjects(authResult.user.uid, linkAuthResult.user.uid);
  1919. XCTAssertEqualObjects(authResult.user.displayName,
  1920. linkAuthResult.user.displayName);
  1921. XCTAssertEqualObjects(linkAuthResult.additionalUserInfo.profile,
  1922. [[self class] googleProfile]);
  1923. XCTAssertEqualObjects(linkAuthResult.additionalUserInfo.username,
  1924. kUserName);
  1925. XCTAssertEqualObjects(linkAuthResult.additionalUserInfo.providerID,
  1926. FIRGoogleAuthProviderID);
  1927. [expectation fulfill];
  1928. }];
  1929. }];
  1930. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1931. [self assertUserGoogle:[FIRAuth auth].currentUser];
  1932. OCMVerifyAll(_mockBackend);
  1933. }
  1934. /** @fn testlinkAndRetrieveDataError
  1935. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion:
  1936. call with an error from the backend.
  1937. */
  1938. - (void)testlinkAndRetrieveDataError {
  1939. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1940. federatedID:kFacebookID
  1941. displayName:kFacebookDisplayName
  1942. profile:[[self class] googleProfile]
  1943. providerIDToken:kFacebookIDToken
  1944. providerAccessToken:kFacebookAccessToken];
  1945. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1946. [[FIRAuth auth] signOut:NULL];
  1947. FIRAuthCredential *facebookCredential =
  1948. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1949. [[FIRAuth auth]
  1950. signInWithCredential:facebookCredential
  1951. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  1952. XCTAssertTrue([NSThread isMainThread]);
  1953. [self assertUserFacebook:authResult.user];
  1954. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  1955. [[self class] googleProfile]);
  1956. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1957. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  1958. FIRFacebookAuthProviderID);
  1959. XCTAssertNil(error);
  1960. OCMExpect([self->_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  1961. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  1962. FIRVerifyAssertionResponseCallback callback) {
  1963. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1964. callback(nil,
  1965. [FIRAuthErrorUtils
  1966. accountExistsWithDifferentCredentialErrorWithEmail:kEmail
  1967. updatedCredential:nil]);
  1968. });
  1969. });
  1970. FIRAuthCredential *linkGoogleCredential =
  1971. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken
  1972. accessToken:kGoogleAccessToken];
  1973. [authResult.user
  1974. linkWithCredential:linkGoogleCredential
  1975. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  1976. NSError *_Nullable error) {
  1977. XCTAssertTrue([NSThread isMainThread]);
  1978. XCTAssertNil(linkAuthResult);
  1979. XCTAssertEqual(
  1980. error.code,
  1981. FIRAuthErrorCodeAccountExistsWithDifferentCredential);
  1982. XCTAssertEqual(error.userInfo[FIRAuthErrorUserInfoEmailKey],
  1983. kEmail);
  1984. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1985. [expectation fulfill];
  1986. }];
  1987. }];
  1988. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1989. OCMVerifyAll(_mockBackend);
  1990. }
  1991. /** @fn testlinkAndRetrieveDataProviderAlreadyLinked
  1992. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion:
  1993. call with FIRAuthErrorCodeProviderAlreadyLinked, which is a client side error.
  1994. */
  1995. - (void)testlinkAndRetrieveDataProviderAlreadyLinked {
  1996. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1997. federatedID:kFacebookID
  1998. displayName:kFacebookDisplayName
  1999. profile:[[self class] googleProfile]
  2000. providerIDToken:kFacebookIDToken
  2001. providerAccessToken:kFacebookAccessToken];
  2002. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2003. [[FIRAuth auth] signOut:NULL];
  2004. FIRAuthCredential *facebookCredential =
  2005. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2006. [[FIRAuth auth]
  2007. signInWithCredential:facebookCredential
  2008. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2009. XCTAssertTrue([NSThread isMainThread]);
  2010. [self assertUserFacebook:authResult.user];
  2011. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2012. [[self class] googleProfile]);
  2013. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2014. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2015. FIRFacebookAuthProviderID);
  2016. XCTAssertNil(error);
  2017. FIRAuthCredential *linkFacebookCredential =
  2018. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2019. [authResult.user
  2020. linkWithCredential:linkFacebookCredential
  2021. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  2022. NSError *_Nullable error) {
  2023. XCTAssertTrue([NSThread isMainThread]);
  2024. XCTAssertNil(linkAuthResult);
  2025. XCTAssertEqual(error.code, FIRAuthErrorCodeProviderAlreadyLinked);
  2026. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  2027. [expectation fulfill];
  2028. }];
  2029. }];
  2030. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2031. OCMVerifyAll(_mockBackend);
  2032. }
  2033. /** @fn testlinkAndRetrieveDataErrorAutoSignOut
  2034. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion:
  2035. call that automatically signs out.
  2036. */
  2037. - (void)testlinkAndRetrieveDataErrorAutoSignOut {
  2038. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2039. federatedID:kFacebookID
  2040. displayName:kFacebookDisplayName
  2041. profile:[[self class] googleProfile]
  2042. providerIDToken:kFacebookIDToken
  2043. providerAccessToken:kFacebookAccessToken];
  2044. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2045. [[FIRAuth auth] signOut:NULL];
  2046. FIRAuthCredential *facebookCredential =
  2047. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2048. [[FIRAuth auth]
  2049. signInWithCredential:facebookCredential
  2050. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2051. XCTAssertTrue([NSThread isMainThread]);
  2052. [self assertUserFacebook:authResult.user];
  2053. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2054. [[self class] googleProfile]);
  2055. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2056. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2057. FIRFacebookAuthProviderID);
  2058. XCTAssertNil(error);
  2059. OCMExpect([self->_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  2060. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  2061. FIRVerifyAssertionResponseCallback callback) {
  2062. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2063. callback(nil, [FIRAuthErrorUtils userDisabledErrorWithMessage:nil]);
  2064. });
  2065. });
  2066. FIRAuthCredential *linkGoogleCredential =
  2067. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken
  2068. accessToken:kGoogleAccessToken];
  2069. [authResult.user
  2070. linkWithCredential:linkGoogleCredential
  2071. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  2072. NSError *_Nullable error) {
  2073. XCTAssertTrue([NSThread isMainThread]);
  2074. XCTAssertNil(linkAuthResult);
  2075. XCTAssertEqual(error.code, FIRAuthErrorCodeUserDisabled);
  2076. XCTAssertNil([FIRAuth auth].currentUser);
  2077. [expectation fulfill];
  2078. }];
  2079. }];
  2080. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2081. OCMVerifyAll(_mockBackend);
  2082. }
  2083. /** @fn testlinkEmailAndRetrieveDataSuccess
  2084. @brief Tests the flow of a successful @c linkWithCredential:completion:
  2085. invocation for email credential.
  2086. */
  2087. - (void)testlinkEmailAndRetrieveDataSuccess {
  2088. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2089. federatedID:kFacebookID
  2090. displayName:kFacebookDisplayName
  2091. profile:[[self class] googleProfile]
  2092. providerIDToken:kFacebookIDToken
  2093. providerAccessToken:kFacebookAccessToken];
  2094. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2095. [[FIRAuth auth] signOut:NULL];
  2096. FIRAuthCredential *facebookCredential =
  2097. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2098. [[FIRAuth auth]
  2099. signInWithCredential:facebookCredential
  2100. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2101. XCTAssertTrue([NSThread isMainThread]);
  2102. [self assertUserFacebook:authResult.user];
  2103. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2104. [[self class] googleProfile]);
  2105. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2106. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2107. FIRFacebookAuthProviderID);
  2108. XCTAssertNil(error);
  2109. id mockGetAccountInfoResponseUser =
  2110. OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2111. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2112. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2113. OCMStub([mockGetAccountInfoResponseUser displayName])
  2114. .andReturn(kEmailDisplayName);
  2115. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2116. // Get account info is expected to be invoked twice.
  2117. [self
  2118. expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  2119. [self
  2120. expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  2121. OCMExpect([self->_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2122. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  2123. FIRSetAccountInfoResponseCallback callback) {
  2124. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2125. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2126. XCTAssertEqualObjects(request.password, kFakePassword);
  2127. XCTAssertNil(request.localID);
  2128. XCTAssertNil(request.displayName);
  2129. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2130. id mockSetAccountInfoResponse =
  2131. OCMClassMock([FIRSetAccountInfoResponse class]);
  2132. callback(mockSetAccountInfoResponse, nil);
  2133. });
  2134. });
  2135. FIRAuthCredential *linkEmailCredential =
  2136. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  2137. [authResult.user
  2138. linkWithCredential:linkEmailCredential
  2139. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  2140. NSError *_Nullable error) {
  2141. XCTAssertTrue([NSThread isMainThread]);
  2142. XCTAssertNil(error);
  2143. XCTAssertEqualObjects(linkAuthResult.user.email, kEmail);
  2144. XCTAssertEqualObjects(linkAuthResult.user.displayName,
  2145. kEmailDisplayName);
  2146. [expectation fulfill];
  2147. }];
  2148. }];
  2149. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2150. OCMVerifyAll(_mockBackend);
  2151. }
  2152. /** @fn testlinkEmailProviderAlreadyLinkedError
  2153. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion:
  2154. invocation for email credential and FIRAuthErrorCodeProviderAlreadyLinked which is a client
  2155. side error.
  2156. */
  2157. - (void)testlinkEmailProviderAlreadyLinkedError {
  2158. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2159. federatedID:kFacebookID
  2160. displayName:kFacebookDisplayName
  2161. profile:[[self class] googleProfile]
  2162. providerIDToken:kFacebookIDToken
  2163. providerAccessToken:kFacebookAccessToken];
  2164. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2165. [[FIRAuth auth] signOut:NULL];
  2166. FIRAuthCredential *facebookCredential =
  2167. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2168. [[FIRAuth auth]
  2169. signInWithCredential:facebookCredential
  2170. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2171. XCTAssertTrue([NSThread isMainThread]);
  2172. [self assertUserFacebook:authResult.user];
  2173. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2174. [[self class] googleProfile]);
  2175. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2176. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2177. FIRFacebookAuthProviderID);
  2178. XCTAssertNil(error);
  2179. id mockGetAccountInfoResponseUser =
  2180. OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2181. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2182. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2183. OCMStub([mockGetAccountInfoResponseUser displayName])
  2184. .andReturn(kEmailDisplayName);
  2185. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2186. // Get account info is expected to be invoked twice.
  2187. [self
  2188. expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  2189. [self
  2190. expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  2191. OCMExpect([self->_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2192. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  2193. FIRSetAccountInfoResponseCallback callback) {
  2194. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2195. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2196. XCTAssertEqualObjects(request.password, kFakePassword);
  2197. XCTAssertNil(request.localID);
  2198. XCTAssertNil(request.displayName);
  2199. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2200. id mockSetAccountInfoResponse =
  2201. OCMClassMock([FIRSetAccountInfoResponse class]);
  2202. callback(mockSetAccountInfoResponse, nil);
  2203. });
  2204. });
  2205. FIRAuthCredential *linkEmailCredential =
  2206. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  2207. [authResult.user
  2208. linkWithCredential:linkEmailCredential
  2209. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  2210. NSError *_Nullable error) {
  2211. XCTAssertNil(error);
  2212. XCTAssertEqualObjects(linkAuthResult.user.email, kEmail);
  2213. XCTAssertEqualObjects(linkAuthResult.user.displayName,
  2214. kEmailDisplayName);
  2215. // Try linking same credential a second time to trigger client side
  2216. // error.
  2217. [authResult.user
  2218. linkWithCredential:linkEmailCredential
  2219. completion:^(
  2220. FIRAuthDataResult *_Nullable linkAuthResult,
  2221. NSError *_Nullable error) {
  2222. XCTAssertTrue([NSThread isMainThread]);
  2223. XCTAssertNil(linkAuthResult);
  2224. XCTAssertEqual(error.code,
  2225. FIRAuthErrorCodeProviderAlreadyLinked);
  2226. [expectation fulfill];
  2227. }];
  2228. }];
  2229. }];
  2230. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2231. OCMVerifyAll(_mockBackend);
  2232. }
  2233. /** @fn testlinkEmailAndRetrieveDataError
  2234. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion:
  2235. invocation for email credential and an error from the backend.
  2236. */
  2237. - (void)testlinkEmailAndRetrieveDataError {
  2238. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2239. federatedID:kFacebookID
  2240. displayName:kFacebookDisplayName
  2241. profile:[[self class] googleProfile]
  2242. providerIDToken:kFacebookIDToken
  2243. providerAccessToken:kFacebookAccessToken];
  2244. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2245. [[FIRAuth auth] signOut:NULL];
  2246. FIRAuthCredential *facebookCredential =
  2247. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2248. [[FIRAuth auth]
  2249. signInWithCredential:facebookCredential
  2250. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2251. XCTAssertTrue([NSThread isMainThread]);
  2252. [self assertUserFacebook:authResult.user];
  2253. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2254. [[self class] googleProfile]);
  2255. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2256. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2257. FIRFacebookAuthProviderID);
  2258. XCTAssertNil(error);
  2259. OCMExpect([self->_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2260. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  2261. FIRGetAccountInfoResponseCallback callback) {
  2262. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2263. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2264. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2265. callback(nil, [FIRAuthErrorUtils tooManyRequestsErrorWithMessage:nil]);
  2266. });
  2267. });
  2268. FIRAuthCredential *linkEmailCredential =
  2269. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  2270. [authResult.user
  2271. linkWithCredential:linkEmailCredential
  2272. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  2273. NSError *_Nullable error) {
  2274. XCTAssertTrue([NSThread isMainThread]);
  2275. XCTAssertNil(linkAuthResult);
  2276. XCTAssertEqual(error.code, FIRAuthErrorCodeTooManyRequests);
  2277. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  2278. [expectation fulfill];
  2279. }];
  2280. }];
  2281. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2282. OCMVerifyAll(_mockBackend);
  2283. }
  2284. /** @fn testlinkEmailAndRetrieveDataError
  2285. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion:
  2286. invocation that automatically signs out.
  2287. */
  2288. - (void)testlinkEmailAndRetrieveDataErrorAutoSignOut {
  2289. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2290. federatedID:kFacebookID
  2291. displayName:kFacebookDisplayName
  2292. profile:[[self class] googleProfile]
  2293. providerIDToken:kFacebookIDToken
  2294. providerAccessToken:kFacebookAccessToken];
  2295. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2296. [[FIRAuth auth] signOut:NULL];
  2297. FIRAuthCredential *facebookCredential =
  2298. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2299. [[FIRAuth auth]
  2300. signInWithCredential:facebookCredential
  2301. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2302. XCTAssertTrue([NSThread isMainThread]);
  2303. [self assertUserFacebook:authResult.user];
  2304. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2305. [[self class] googleProfile]);
  2306. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2307. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2308. FIRFacebookAuthProviderID);
  2309. XCTAssertNil(error);
  2310. OCMExpect([self->_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2311. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  2312. FIRGetAccountInfoResponseCallback callback) {
  2313. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2314. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2315. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2316. callback(nil, [FIRAuthErrorUtils userTokenExpiredErrorWithMessage:nil]);
  2317. });
  2318. });
  2319. FIRAuthCredential *linkEmailCredential =
  2320. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  2321. [authResult.user
  2322. linkWithCredential:linkEmailCredential
  2323. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  2324. NSError *_Nullable error) {
  2325. XCTAssertTrue([NSThread isMainThread]);
  2326. XCTAssertNil(linkAuthResult);
  2327. XCTAssertEqual(error.code, FIRAuthErrorCodeUserTokenExpired);
  2328. XCTAssertNil([FIRAuth auth].currentUser);
  2329. [expectation fulfill];
  2330. }];
  2331. }];
  2332. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2333. OCMVerifyAll(_mockBackend);
  2334. }
  2335. /** @fn testlinkCredentialSuccess
  2336. @brief Tests the flow of a successful @c linkWithCredential:completion: call, without additional
  2337. IDP data.
  2338. */
  2339. - (void)testlinkCredentialSuccess {
  2340. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2341. federatedID:kFacebookID
  2342. displayName:kFacebookDisplayName
  2343. profile:[[self class] googleProfile]
  2344. providerIDToken:kFacebookIDToken
  2345. providerAccessToken:kFacebookAccessToken];
  2346. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2347. [[FIRAuth auth] signOut:NULL];
  2348. FIRAuthCredential *facebookCredential =
  2349. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2350. [[FIRAuth auth]
  2351. signInWithCredential:facebookCredential
  2352. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2353. XCTAssertTrue([NSThread isMainThread]);
  2354. [self assertUserFacebook:authResult.user];
  2355. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2356. [[self class] googleProfile]);
  2357. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2358. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2359. FIRFacebookAuthProviderID);
  2360. XCTAssertNil(error);
  2361. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  2362. federatedID:kGoogleID
  2363. displayName:kGoogleDisplayName
  2364. profile:[[self class] googleProfile]
  2365. providerIDToken:kGoogleIDToken
  2366. providerAccessToken:kGoogleAccessToken];
  2367. FIRAuthCredential *linkGoogleCredential =
  2368. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken
  2369. accessToken:kGoogleAccessToken];
  2370. [authResult.user
  2371. linkWithCredential:linkGoogleCredential
  2372. completion:^(FIRAuthDataResult *_Nullable result,
  2373. NSError *_Nullable error) {
  2374. XCTAssertNil(error);
  2375. id<FIRUserInfo> userInfo = result.user.providerData.firstObject;
  2376. XCTAssertEqual(userInfo.providerID, FIRGoogleAuthProviderID);
  2377. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  2378. [expectation fulfill];
  2379. }];
  2380. }];
  2381. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2382. [self assertUserGoogle:[FIRAuth auth].currentUser];
  2383. OCMVerifyAll(_mockBackend);
  2384. }
  2385. /** @fn testlinkCredentialError
  2386. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion: call, with an error
  2387. from the backend.
  2388. */
  2389. - (void)testlinkCredentialError {
  2390. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2391. federatedID:kFacebookID
  2392. displayName:kFacebookDisplayName
  2393. profile:[[self class] googleProfile]
  2394. providerIDToken:kFacebookIDToken
  2395. providerAccessToken:kFacebookAccessToken];
  2396. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2397. [[FIRAuth auth] signOut:NULL];
  2398. FIRAuthCredential *facebookCredential =
  2399. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2400. [[FIRAuth auth]
  2401. signInWithCredential:facebookCredential
  2402. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2403. XCTAssertTrue([NSThread isMainThread]);
  2404. [self assertUserFacebook:authResult.user];
  2405. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2406. [[self class] googleProfile]);
  2407. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2408. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2409. FIRFacebookAuthProviderID);
  2410. XCTAssertNil(error);
  2411. OCMExpect([self->_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  2412. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  2413. FIRVerifyAssertionResponseCallback callback) {
  2414. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2415. callback(nil, [FIRAuthErrorUtils userDisabledErrorWithMessage:nil]);
  2416. });
  2417. });
  2418. FIRAuthCredential *linkGoogleCredential =
  2419. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken
  2420. accessToken:kGoogleAccessToken];
  2421. [authResult.user
  2422. linkWithCredential:linkGoogleCredential
  2423. completion:^(FIRAuthDataResult *_Nullable result,
  2424. NSError *_Nullable error) {
  2425. XCTAssertNil(result.user);
  2426. XCTAssertEqual(error.code, FIRAuthErrorCodeUserDisabled);
  2427. [expectation fulfill];
  2428. }];
  2429. }];
  2430. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2431. OCMVerifyAll(_mockBackend);
  2432. }
  2433. /** @fn testlinkCredentialProviderAlreadyLinkedError
  2434. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion: call, with a client
  2435. side error.
  2436. */
  2437. - (void)testlinkCredentialProviderAlreadyLinkedError {
  2438. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  2439. federatedID:kFacebookID
  2440. displayName:kFacebookDisplayName
  2441. profile:[[self class] googleProfile]
  2442. providerIDToken:kFacebookIDToken
  2443. providerAccessToken:kFacebookAccessToken];
  2444. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2445. [[FIRAuth auth] signOut:NULL];
  2446. FIRAuthCredential *facebookCredential =
  2447. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  2448. [[FIRAuth auth]
  2449. signInWithCredential:facebookCredential
  2450. completion:^(FIRAuthDataResult *_Nullable authResult, NSError *_Nullable error) {
  2451. XCTAssertTrue([NSThread isMainThread]);
  2452. [self assertUserFacebook:authResult.user];
  2453. XCTAssertEqualObjects(authResult.additionalUserInfo.profile,
  2454. [[self class] googleProfile]);
  2455. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  2456. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID,
  2457. FIRFacebookAuthProviderID);
  2458. XCTAssertNil(error);
  2459. FIRAuthCredential *linkFacebookCredential =
  2460. [FIRFacebookAuthProvider credentialWithAccessToken:kGoogleAccessToken];
  2461. [authResult.user
  2462. linkWithCredential:linkFacebookCredential
  2463. completion:^(FIRAuthDataResult *_Nullable result,
  2464. NSError *_Nullable error) {
  2465. XCTAssertNil(result.user);
  2466. XCTAssertEqual(error.code, FIRAuthErrorCodeProviderAlreadyLinked);
  2467. [expectation fulfill];
  2468. }];
  2469. }];
  2470. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2471. OCMVerifyAll(_mockBackend);
  2472. }
  2473. #if TARGET_OS_IOS
  2474. /** @fn testlinkPhoneAuthCredentialSuccess
  2475. @brief Tests the flow of a successful @c linkWithCredential:completion:
  2476. call using a phoneAuthCredential.
  2477. */
  2478. - (void)testlinkPhoneAuthCredentialSuccess {
  2479. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  2480. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2481. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2482. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2483. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2484. if (phoneNumber.length) {
  2485. NSDictionary *userInfoDictionary = @{@"providerId" : FIRPhoneAuthProviderID};
  2486. FIRGetAccountInfoResponseProviderUserInfo *userInfo =
  2487. [[FIRGetAccountInfoResponseProviderUserInfo alloc] initWithDictionary:userInfoDictionary];
  2488. OCMStub([mockGetAccountInfoResponseUser providerUserInfo]).andReturn(@[ userInfo ]);
  2489. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  2490. }
  2491. return mockGetAccountInfoResponseUser;
  2492. };
  2493. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2494. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  2495. [self
  2496. signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  2497. completion:^(FIRUser *user) {
  2498. [self
  2499. expectVerifyPhoneNumberRequestWithPhoneNumber:
  2500. kPhoneNumber
  2501. isLinkOperation:YES
  2502. error:nil];
  2503. id userInfoResponseUpdate =
  2504. mockUserInfoWithPhoneNumber(kPhoneNumber);
  2505. [self expectGetAccountInfoWithMockUserInfoResponse:
  2506. userInfoResponseUpdate];
  2507. FIRPhoneAuthCredential *credential =
  2508. [[FIRPhoneAuthProvider provider]
  2509. credentialWithVerificationID:kVerificationID
  2510. verificationCode:
  2511. kVerificationCode];
  2512. [user
  2513. linkWithCredential:credential
  2514. completion:^(FIRAuthDataResult
  2515. *_Nullable linkAuthResult,
  2516. NSError *_Nullable error) {
  2517. XCTAssertNil(error);
  2518. XCTAssertEqualObjects(
  2519. [FIRAuth auth]
  2520. .currentUser.providerData
  2521. .firstObject.providerID,
  2522. FIRPhoneAuthProviderID);
  2523. XCTAssertEqualObjects(
  2524. [FIRAuth auth]
  2525. .currentUser.phoneNumber,
  2526. kPhoneNumber);
  2527. [expectation fulfill];
  2528. }];
  2529. }];
  2530. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2531. OCMVerifyAll(_mockBackend);
  2532. }
  2533. /** @fn testUnlinkPhoneAuthCredentialSuccess
  2534. @brief Tests the flow of a successful @c unlinkFromProvider:completion: call using a
  2535. @c FIRPhoneAuthProvider.
  2536. */
  2537. - (void)testUnlinkPhoneAuthCredentialSuccess {
  2538. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  2539. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2540. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2541. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2542. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2543. if (phoneNumber.length) {
  2544. NSDictionary *userInfoDictionary = @{@"providerId" : FIRPhoneAuthProviderID};
  2545. FIRGetAccountInfoResponseProviderUserInfo *userInfo =
  2546. [[FIRGetAccountInfoResponseProviderUserInfo alloc] initWithDictionary:userInfoDictionary];
  2547. OCMStub([mockGetAccountInfoResponseUser providerUserInfo]).andReturn(@[ userInfo ]);
  2548. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  2549. }
  2550. return mockGetAccountInfoResponseUser;
  2551. };
  2552. OCMExpect([self->_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2553. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  2554. FIRSetAccountInfoResponseCallback callback) {
  2555. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2556. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2557. XCTAssertNotNil(request.deleteProviders);
  2558. XCTAssertNil(request.email);
  2559. XCTAssertNil(request.localID);
  2560. XCTAssertNil(request.displayName);
  2561. XCTAssertNil(request.photoURL);
  2562. XCTAssertNil(request.password);
  2563. XCTAssertNil(request.providers);
  2564. XCTAssertNil(request.deleteAttributes);
  2565. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2566. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  2567. callback(mockSetAccountInfoResponse, nil);
  2568. });
  2569. });
  2570. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2571. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  2572. [self
  2573. signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  2574. completion:^(FIRUser *user) {
  2575. [self
  2576. expectVerifyPhoneNumberRequestWithPhoneNumber:
  2577. kPhoneNumber
  2578. isLinkOperation:YES
  2579. error:nil];
  2580. id userInfoResponseUpdate =
  2581. mockUserInfoWithPhoneNumber(kPhoneNumber);
  2582. [self expectGetAccountInfoWithMockUserInfoResponse:
  2583. userInfoResponseUpdate];
  2584. FIRPhoneAuthCredential *credential =
  2585. [[FIRPhoneAuthProvider provider]
  2586. credentialWithVerificationID:kVerificationID
  2587. verificationCode:
  2588. kVerificationCode];
  2589. // Link phone credential.
  2590. [user
  2591. linkWithCredential:credential
  2592. completion:^(FIRAuthDataResult
  2593. *_Nullable linkAuthResult,
  2594. NSError *_Nullable error) {
  2595. XCTAssertNil(error);
  2596. XCTAssertEqualObjects(
  2597. [FIRAuth auth]
  2598. .currentUser.providerData
  2599. .firstObject.providerID,
  2600. FIRPhoneAuthProviderID);
  2601. XCTAssertEqualObjects(
  2602. [FIRAuth auth]
  2603. .currentUser.phoneNumber,
  2604. kPhoneNumber);
  2605. // Immediately unlink the phone auth
  2606. // provider.
  2607. [user
  2608. unlinkFromProvider:
  2609. FIRPhoneAuthProviderID
  2610. completion:^(
  2611. FIRUser *_Nullable user,
  2612. NSError
  2613. *_Nullable error) {
  2614. XCTAssertNil(error);
  2615. XCTAssertNil(
  2616. [FIRAuth auth]
  2617. .currentUser
  2618. .phoneNumber);
  2619. [expectation fulfill];
  2620. }];
  2621. }];
  2622. }];
  2623. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2624. OCMVerifyAll(_mockBackend);
  2625. }
  2626. /** @fn testlinkPhoneAuthCredentialFailure
  2627. @brief Tests the flow of a failed call to @c linkWithCredential:completion: due
  2628. to a phone provider already being linked.
  2629. */
  2630. - (void)testlinkPhoneAuthCredentialFailure {
  2631. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  2632. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2633. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2634. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2635. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2636. if (phoneNumber.length) {
  2637. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  2638. }
  2639. return mockGetAccountInfoResponseUser;
  2640. };
  2641. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2642. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  2643. [self
  2644. signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  2645. completion:^(FIRUser *user) {
  2646. NSError *error =
  2647. [FIRAuthErrorUtils providerAlreadyLinkedError];
  2648. [self
  2649. expectVerifyPhoneNumberRequestWithPhoneNumber:nil
  2650. isLinkOperation:YES
  2651. error:
  2652. error];
  2653. FIRPhoneAuthCredential *credential =
  2654. [[FIRPhoneAuthProvider provider]
  2655. credentialWithVerificationID:kVerificationID
  2656. verificationCode:
  2657. kVerificationCode];
  2658. [user
  2659. linkWithCredential:credential
  2660. completion:^(FIRAuthDataResult
  2661. *_Nullable linkAuthResult,
  2662. NSError *_Nullable error) {
  2663. XCTAssertEqual(
  2664. error.code,
  2665. FIRAuthErrorCodeProviderAlreadyLinked);
  2666. [expectation fulfill];
  2667. }];
  2668. }];
  2669. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2670. OCMVerifyAll(_mockBackend);
  2671. }
  2672. /** @fn testlinkPhoneCredentialAlreadyExistsError
  2673. @brief Tests the flow of @c linkWithCredential:completion:
  2674. call using a phoneAuthCredential and a credential already exisits error. In this case we
  2675. should get a FIRAuthCredential in the error object.
  2676. */
  2677. - (void)testlinkPhoneCredentialAlreadyExistsError {
  2678. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  2679. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2680. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2681. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2682. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2683. if (phoneNumber.length) {
  2684. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  2685. }
  2686. return mockGetAccountInfoResponseUser;
  2687. };
  2688. void (^expectVerifyPhoneNumberRequest)(NSString *) = ^(NSString *phoneNumber) {
  2689. OCMExpect([self->_mockBackend verifyPhoneNumber:[OCMArg any] callback:[OCMArg any]])
  2690. .andCallBlock2(^(FIRVerifyPhoneNumberRequest *_Nullable request,
  2691. FIRVerifyPhoneNumberResponseCallback callback) {
  2692. XCTAssertEqualObjects(request.verificationID, kVerificationID);
  2693. XCTAssertEqualObjects(request.verificationCode, kVerificationCode);
  2694. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2695. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2696. FIRPhoneAuthCredential *credential =
  2697. [[FIRPhoneAuthCredential alloc] initWithTemporaryProof:kTemporaryProof
  2698. phoneNumber:kPhoneNumber
  2699. providerID:FIRPhoneAuthProviderID];
  2700. callback(nil, [FIRAuthErrorUtils credentialAlreadyInUseErrorWithMessage:nil
  2701. credential:credential
  2702. email:nil]);
  2703. });
  2704. });
  2705. };
  2706. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2707. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  2708. [self
  2709. signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  2710. completion:^(FIRUser *user) {
  2711. expectVerifyPhoneNumberRequest(kPhoneNumber);
  2712. FIRPhoneAuthCredential *credential =
  2713. [[FIRPhoneAuthProvider provider]
  2714. credentialWithVerificationID:kVerificationID
  2715. verificationCode:
  2716. kVerificationCode];
  2717. [user
  2718. linkWithCredential:credential
  2719. completion:^(FIRAuthDataResult
  2720. *_Nullable linkAuthResult,
  2721. NSError *_Nullable error) {
  2722. XCTAssertNil(linkAuthResult);
  2723. XCTAssertEqual(
  2724. error.code,
  2725. FIRAuthErrorCodeCredentialAlreadyInUse);
  2726. FIRPhoneAuthCredential *credential =
  2727. error.userInfo
  2728. [FIRAuthErrorUserInfoUpdatedCredentialKey];
  2729. XCTAssertEqual(credential.temporaryProof,
  2730. kTemporaryProof);
  2731. XCTAssertEqual(credential.phoneNumber,
  2732. kPhoneNumber);
  2733. [expectation fulfill];
  2734. }];
  2735. }];
  2736. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2737. OCMVerifyAll(_mockBackend);
  2738. }
  2739. #endif
  2740. #pragma mark - FIRCustomTokenProviderDelegate
  2741. - (void)getCustomTokenWithCompletion:(void (^)(NSString *_Nullable customToken,
  2742. NSError *_Nullable error))completion {
  2743. completion(kCustomToken, nil);
  2744. }
  2745. #pragma mark - Helpers
  2746. /** @fn getIDTokenResultForcingRefreshSuccess
  2747. @brief Helper for testing the flow of a successful @c
  2748. getIDTokenResultForcingRefresh:completion: call.
  2749. */
  2750. - (void)getIDTokenResultForcingRefreshSuccessWithIDToken:(NSString *)IDToken {
  2751. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2752. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2753. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2754. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  2755. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2756. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2757. [self
  2758. signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  2759. completion:^(FIRUser *user) {
  2760. OCMExpect([self->_mockBackend
  2761. secureToken:[OCMArg any]
  2762. callback:[OCMArg any]])
  2763. .andCallBlock2(^(
  2764. FIRSecureTokenRequest *_Nullable request,
  2765. FIRSecureTokenResponseCallback callback) {
  2766. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2767. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2768. id mockSecureTokenResponse = OCMClassMock(
  2769. [FIRSecureTokenResponse class]);
  2770. OCMStub([mockSecureTokenResponse accessToken])
  2771. .andReturn(IDToken);
  2772. callback(mockSecureTokenResponse, nil);
  2773. });
  2774. });
  2775. [user
  2776. getIDTokenResultForcingRefresh:YES
  2777. completion:^(
  2778. FIRAuthTokenResult
  2779. *_Nullable tokenResult,
  2780. NSError *_Nullable error) {
  2781. XCTAssertTrue(
  2782. [NSThread isMainThread]);
  2783. XCTAssertNil(error);
  2784. XCTAssertEqualObjects(
  2785. tokenResult.token,
  2786. IDToken);
  2787. XCTAssertTrue(
  2788. tokenResult
  2789. .issuedAtDate &&
  2790. [tokenResult.issuedAtDate
  2791. isKindOfClass:
  2792. [NSDate class]]);
  2793. XCTAssertTrue(
  2794. tokenResult.authDate &&
  2795. [tokenResult.authDate
  2796. isKindOfClass:
  2797. [NSDate class]]);
  2798. XCTAssertTrue(
  2799. tokenResult
  2800. .expirationDate &&
  2801. [tokenResult
  2802. .expirationDate
  2803. isKindOfClass:
  2804. [NSDate class]]);
  2805. XCTAssertTrue(
  2806. tokenResult.claims &&
  2807. [tokenResult.claims
  2808. isKindOfClass:
  2809. [NSDictionary
  2810. class]]);
  2811. [expectation fulfill];
  2812. }];
  2813. }];
  2814. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2815. OCMVerifyAll(_mockBackend);
  2816. }
  2817. /** @fn signInAnonymouslyWithMockGetAccountInfoResponse:completion:
  2818. @brief Signs in with an anonymous account with mocked backend end calls.
  2819. @param mockUserInfoResponse A mocked FIRGetAccountInfoResponseUser object.
  2820. @param completion The completion block that takes the newly signed-in user as the only
  2821. parameter.
  2822. */
  2823. - (void)signInAnonymouslyWithMockGetAccountInfoResponse:(id)mockUserInfoResponse
  2824. completion:(void (^)(FIRUser *user))completion {
  2825. OCMExpect([_mockBackend signUpNewUser:[OCMArg any] callback:[OCMArg any]])
  2826. .andCallBlock2(
  2827. ^(FIRSignUpNewUserRequest *_Nullable request, FIRSignupNewUserCallback callback) {
  2828. id mockSignUpResponse = OCMClassMock([FIRSignUpNewUserResponse class]);
  2829. OCMStub([mockSignUpResponse IDToken]).andReturn(kAccessToken);
  2830. OCMStub([mockSignUpResponse approximateExpirationDate])
  2831. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2832. OCMStub([mockSignUpResponse refreshToken]).andReturn(kRefreshToken);
  2833. callback(mockSignUpResponse, nil);
  2834. });
  2835. [self expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoResponse];
  2836. [[FIRAuth auth] signOut:NULL];
  2837. [[FIRAuth auth] signInAnonymouslyWithCompletion:^(FIRAuthDataResult *_Nullable result,
  2838. NSError *_Nullable error) {
  2839. XCTAssertNotNil(result.user);
  2840. XCTAssertNil(error);
  2841. completion(result.user);
  2842. }];
  2843. }
  2844. /** @fn signInWithEmailPasswordWithMockGetAccountInfoResponse:completion:
  2845. @brief Signs in with an email and password account with mocked backend end calls.
  2846. @param mockUserInfoResponse A mocked FIRGetAccountInfoResponseUser object.
  2847. @param completion The completion block that takes the newly signed-in user as the only
  2848. parameter.
  2849. */
  2850. - (void)signInWithEmailPasswordWithMockUserInfoResponse:(id)mockUserInfoResponse
  2851. completion:(void (^)(FIRUser *user))completion {
  2852. OCMExpect([_mockBackend verifyPassword:[OCMArg any] callback:[OCMArg any]])
  2853. .andCallBlock2(^(FIRVerifyPasswordRequest *_Nullable request,
  2854. FIRVerifyPasswordResponseCallback callback) {
  2855. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2856. id mockVerifyPasswordResponse = OCMClassMock([FIRVerifyPasswordResponse class]);
  2857. OCMStub([mockVerifyPasswordResponse IDToken]).andReturn(kAccessToken);
  2858. OCMStub([mockVerifyPasswordResponse approximateExpirationDate])
  2859. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2860. OCMStub([mockVerifyPasswordResponse refreshToken]).andReturn(kRefreshToken);
  2861. callback(mockVerifyPasswordResponse, nil);
  2862. });
  2863. });
  2864. [self expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoResponse];
  2865. [[FIRAuth auth] signOut:NULL];
  2866. [[FIRAuth auth] signInWithEmail:kEmail
  2867. password:kFakePassword
  2868. completion:^(FIRAuthDataResult *_Nullable result, NSError *_Nullable error) {
  2869. XCTAssertNotNil(result.user);
  2870. XCTAssertNil(error);
  2871. completion(result.user);
  2872. }];
  2873. }
  2874. /** @fn signInWithEmailLinkWithMockGetAccountInfoResponse:completion:
  2875. @brief Signs in with an email link auth account with mocked backend end calls.
  2876. @param mockUserInfoResponse A mocked FIRGetAccountInfoResponseUser object.
  2877. @param completion The completion block that takes the newly signed-in user as the only
  2878. parameter.
  2879. */
  2880. - (void)signInWithEmailLinkWithMockUserInfoResponse:(id)mockUserInfoResponse
  2881. completion:(void (^)(FIRUser *user))completion {
  2882. OCMExpect([_mockBackend emailLinkSignin:[OCMArg any] callback:[OCMArg any]])
  2883. .andCallBlock2(^(FIREmailLinkSignInRequest *_Nullable request,
  2884. FIREmailLinkSigninResponseCallback callback) {
  2885. id mockVerifyLinkResponse = OCMClassMock([FIREmailLinkSignInResponse class]);
  2886. OCMStub([mockVerifyLinkResponse IDToken]).andReturn(kAccessToken);
  2887. OCMStub([mockVerifyLinkResponse approximateExpirationDate])
  2888. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2889. OCMStub([mockVerifyLinkResponse refreshToken]).andReturn(kRefreshToken);
  2890. callback(mockVerifyLinkResponse, nil);
  2891. });
  2892. [self expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoResponse];
  2893. [[FIRAuth auth] signOut:NULL];
  2894. [[FIRAuth auth] signInWithEmail:kEmail
  2895. link:@"https://www.google.com?oobCode=aCode&mode=signIn"
  2896. completion:^(FIRAuthDataResult *_Nullable result, NSError *_Nullable error) {
  2897. XCTAssertNotNil(result.user);
  2898. XCTAssertNil(error);
  2899. completion(result.user);
  2900. }];
  2901. }
  2902. /** @fn expectGetAccountInfoWithMockUserInfoResponse:
  2903. @brief Expects a GetAccountInfo request on the mock backend and calls back with provided
  2904. fake account data.
  2905. @param mockUserInfoResponse A mock @c FIRGetAccountInfoResponseUser object containing user info.
  2906. */
  2907. - (void)expectGetAccountInfoWithMockUserInfoResponse:(id)mockUserInfoResponse {
  2908. OCMExpect([self->_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2909. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  2910. FIRGetAccountInfoResponseCallback callback) {
  2911. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2912. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2913. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2914. id mockGetAccountInfoResponse = OCMClassMock([FIRGetAccountInfoResponse class]);
  2915. OCMStub([mockGetAccountInfoResponse users]).andReturn(@[ mockUserInfoResponse ]);
  2916. callback(mockGetAccountInfoResponse, nil);
  2917. });
  2918. });
  2919. }
  2920. /** @fn signInWithCustomTokenNoRefreshTokenWithCompletion:
  2921. @brief Signs in with a custom token with mocked backend calls, without returning a refresh
  2922. token in the response.
  2923. @param completion The completion block that takes the newly signed-in user as the only
  2924. parameter.
  2925. */
  2926. - (void)signInWithCustomTokenNoRefreshTokenWithCompletion:(void (^)(FIRUser *user))completion {
  2927. OCMExpect([_mockBackend verifyCustomToken:[OCMArg any] callback:[OCMArg any]])
  2928. .andCallBlock2(^(FIRVerifyCustomTokenRequest *_Nullable request,
  2929. FIRVerifyCustomTokenResponseCallback callback) {
  2930. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2931. XCTAssertEqualObjects(request.token, kCustomToken);
  2932. XCTAssertTrue(request.returnSecureToken);
  2933. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2934. id mockVerifyCustomTokenResponse = OCMClassMock([FIRVerifyCustomTokenResponse class]);
  2935. [self stubTokensWithMockResponseNoRefreshToken:mockVerifyCustomTokenResponse];
  2936. callback(mockVerifyCustomTokenResponse, nil);
  2937. });
  2938. });
  2939. [[FIRAuth auth] signOut:NULL];
  2940. [[FIRAuth auth]
  2941. signInWithCustomToken:kCustomToken
  2942. completion:^(FIRAuthDataResult *_Nullable result, NSError *_Nullable error) {
  2943. XCTAssertTrue([NSThread isMainThread]);
  2944. [self assertUserFromIDToken:result.user];
  2945. XCTAssertNil(error);
  2946. completion(result.user);
  2947. }];
  2948. }
  2949. /** @fn dictionaryWithUserInfoArray:
  2950. @brief Converts an array of @c FIRUserInfo into a dictionary that indexed by provider IDs.
  2951. @param userInfoArray An array of @c FIRUserInfo objects.
  2952. @return A dictionary contains same values as @c userInfoArray does but keyed by their
  2953. @c providerID .
  2954. */
  2955. - (NSDictionary<NSString *, id<FIRUserInfo>> *)dictionaryWithUserInfoArray:
  2956. (NSArray<id<FIRUserInfo>> *)userInfoArray {
  2957. NSMutableDictionary<NSString *, id<FIRUserInfo>> *map =
  2958. [NSMutableDictionary dictionaryWithCapacity:userInfoArray.count];
  2959. for (id<FIRUserInfo> userInfo in userInfoArray) {
  2960. XCTAssertNil(map[userInfo.providerID]);
  2961. map[userInfo.providerID] = userInfo;
  2962. }
  2963. return map;
  2964. }
  2965. /** @fn stubSecureTokensWithMockResponse
  2966. @brief Creates stubs on the mock response object with access and refresh tokens
  2967. @param mockResponse The mock response object.
  2968. */
  2969. - (void)stubTokensWithMockResponse:(id)mockResponse {
  2970. OCMStub([mockResponse IDToken]).andReturn(kAccessToken);
  2971. OCMStub([mockResponse approximateExpirationDate])
  2972. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2973. OCMStub([mockResponse refreshToken]).andReturn(kRefreshToken);
  2974. }
  2975. /** @fn stubTokensWithMockResponseNoRefreshToken
  2976. @brief Creates stubs on the mock response object with only access token, no refresh token.
  2977. @param mockResponse The mock response object.
  2978. */
  2979. - (void)stubTokensWithMockResponseNoRefreshToken:(id)mockResponse {
  2980. OCMStub([mockResponse IDToken]).andReturn(kAccessToken);
  2981. OCMStub([mockResponse approximateExpirationDate])
  2982. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2983. OCMStub([mockResponse refreshToken]).andReturn(nil);
  2984. }
  2985. /** @fn assertUserGoogle
  2986. @brief Asserts the given FIRUser matching the fake data returned by
  2987. @c expectGetAccountInfo:federatedID:displayName: .
  2988. @param user The user object to be verified.
  2989. */
  2990. - (void)assertUserGoogle:(FIRUser *)user {
  2991. XCTAssertNotNil(user);
  2992. XCTAssertEqualObjects(user.uid, kLocalID);
  2993. XCTAssertEqualObjects(user.displayName, kGoogleDisplayName);
  2994. XCTAssertEqual(user.providerData.count, 1u);
  2995. id<FIRUserInfo> googleUserInfo = user.providerData[0];
  2996. XCTAssertEqualObjects(googleUserInfo.providerID, FIRGoogleAuthProviderID);
  2997. XCTAssertEqualObjects(googleUserInfo.uid, kGoogleID);
  2998. XCTAssertEqualObjects(googleUserInfo.displayName, kGoogleDisplayName);
  2999. XCTAssertEqualObjects(googleUserInfo.email, kGoogleEmail);
  3000. }
  3001. /** @fn assertUserFacebook
  3002. @brief Asserts the given FIRUser matching the fake data returned by
  3003. @c expectGetAccountInfo:federatedID:displayName: .
  3004. @param user The user object to be verified.
  3005. */
  3006. - (void)assertUserFacebook:(FIRUser *)user {
  3007. XCTAssertNotNil(user);
  3008. XCTAssertEqualObjects(user.uid, kLocalID);
  3009. XCTAssertEqualObjects(user.displayName, kFacebookDisplayName);
  3010. XCTAssertEqual(user.providerData.count, 1u);
  3011. id<FIRUserInfo> googleUserInfo = user.providerData[0];
  3012. XCTAssertEqualObjects(googleUserInfo.providerID, FIRFacebookAuthProviderID);
  3013. XCTAssertEqualObjects(googleUserInfo.uid, kFacebookID);
  3014. XCTAssertEqualObjects(googleUserInfo.displayName, kFacebookDisplayName);
  3015. XCTAssertEqualObjects(googleUserInfo.email, kGoogleEmail);
  3016. }
  3017. /** @fn assertUserFromIDToken
  3018. @brief Asserts the given FIRUser matching the fake data returned by a IDToken.
  3019. @param user The user object to be verified.
  3020. */
  3021. - (void)assertUserFromIDToken:(FIRUser *)user {
  3022. XCTAssertNotNil(user);
  3023. XCTAssertEqualObjects(user.uid, @"test_user_id");
  3024. XCTAssertFalse(user.anonymous);
  3025. XCTAssertEqual(user.providerData.count, 0u);
  3026. }
  3027. /** @fn expectGetAccountInfo:federatedID:displayName:
  3028. @brief Expects a GetAccountInfo request on the mock backend and calls back with fake account
  3029. data for a Google Sign-In user.
  3030. */
  3031. - (void)expectGetAccountInfo:(NSString *)providerId
  3032. federatedID:(NSString *)federatedID
  3033. displayName:(NSString *)displayName {
  3034. OCMExpect([self->_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  3035. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  3036. FIRGetAccountInfoResponseCallback callback) {
  3037. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  3038. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  3039. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  3040. id mockGoogleUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  3041. OCMStub([mockGoogleUserInfo providerID]).andReturn(providerId);
  3042. OCMStub([mockGoogleUserInfo displayName]).andReturn(displayName);
  3043. OCMStub([mockGoogleUserInfo federatedID]).andReturn(federatedID);
  3044. OCMStub([mockGoogleUserInfo email]).andReturn(kGoogleEmail);
  3045. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  3046. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  3047. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(displayName);
  3048. OCMStub([mockGetAccountInfoResponseUser providerUserInfo])
  3049. .andReturn((@[ mockGoogleUserInfo ]));
  3050. id mockGetAccountInfoResponse = OCMClassMock([FIRGetAccountInfoResponse class]);
  3051. OCMStub([mockGetAccountInfoResponse users]).andReturn(@[
  3052. mockGetAccountInfoResponseUser
  3053. ]);
  3054. callback(mockGetAccountInfoResponse, nil);
  3055. });
  3056. });
  3057. }
  3058. /** @fn expectVerifyAssertionRequest:federatedID:displayName:profile:providerAccessToken:
  3059. @brief Expects a Verify Assertion request on the mock backend and calls back with fake account
  3060. data.
  3061. */
  3062. - (void)expectVerifyAssertionRequest:(NSString *)providerId
  3063. federatedID:(NSString *)federatedID
  3064. displayName:(NSString *)displayName
  3065. profile:(NSDictionary *)profile
  3066. providerIDToken:(nullable NSString *)providerIDToken
  3067. providerAccessToken:(NSString *)providerAccessToken {
  3068. OCMExpect([_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  3069. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  3070. FIRVerifyAssertionResponseCallback callback) {
  3071. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  3072. XCTAssertEqualObjects(request.providerID, providerId);
  3073. XCTAssertEqualObjects(request.providerIDToken, providerIDToken);
  3074. XCTAssertEqualObjects(request.providerAccessToken, providerAccessToken);
  3075. XCTAssertTrue(request.returnSecureToken);
  3076. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  3077. id mockVerifyAssertionResponse = OCMClassMock([FIRVerifyAssertionResponse class]);
  3078. OCMStub([mockVerifyAssertionResponse federatedID]).andReturn(federatedID);
  3079. OCMStub([mockVerifyAssertionResponse providerID]).andReturn(providerId);
  3080. OCMStub([mockVerifyAssertionResponse localID]).andReturn(kLocalID);
  3081. OCMStub([mockVerifyAssertionResponse displayName]).andReturn(displayName);
  3082. OCMStub([mockVerifyAssertionResponse profile]).andReturn(profile);
  3083. OCMStub([mockVerifyAssertionResponse username]).andReturn(kUserName);
  3084. [self stubTokensWithMockResponse:mockVerifyAssertionResponse];
  3085. callback(mockVerifyAssertionResponse, nil);
  3086. });
  3087. });
  3088. [self expectGetAccountInfo:providerId federatedID:federatedID displayName:displayName];
  3089. }
  3090. #if TARGET_OS_IOS
  3091. /** @fn expectVerifyPhoneNumberRequestWithPhoneNumber:error:
  3092. @brief Expects a verify phone numner request on the mock backend and calls back with fake
  3093. account data or an error.
  3094. @param phoneNumber Optionally; The phone number to use in the mocked response.
  3095. @param isLinkOperation Boolean value that indicates whether or not this method is triggered by
  3096. a link operation.
  3097. @param error Optionally; The error to return in the mocked response.
  3098. */
  3099. - (void)expectVerifyPhoneNumberRequestWithPhoneNumber:(nullable NSString *)phoneNumber
  3100. isLinkOperation:(BOOL)isLinkOperation
  3101. error:(nullable NSError *)error {
  3102. OCMExpect([self->_mockBackend verifyPhoneNumber:[OCMArg any] callback:[OCMArg any]])
  3103. .andCallBlock2(^(FIRVerifyPhoneNumberRequest *_Nullable request,
  3104. FIRVerifyPhoneNumberResponseCallback callback) {
  3105. XCTAssertEqualObjects(request.verificationID, kVerificationID);
  3106. XCTAssertEqualObjects(request.verificationCode, kVerificationCode);
  3107. if (isLinkOperation) {
  3108. XCTAssertEqual(request.operation, FIRAuthOperationTypeLink);
  3109. } else {
  3110. XCTAssertEqual(request.operation, FIRAuthOperationTypeUpdate);
  3111. }
  3112. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  3113. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  3114. if (error) {
  3115. callback(nil, error);
  3116. return;
  3117. }
  3118. id mockVerifyPhoneNumberResponse = OCMClassMock([FIRVerifyPhoneNumberResponse class]);
  3119. OCMStub([mockVerifyPhoneNumberResponse phoneNumber]).andReturn(phoneNumber);
  3120. OCMStub([mockVerifyPhoneNumberResponse IDToken]).andReturn(kAccessToken);
  3121. OCMStub([mockVerifyPhoneNumberResponse approximateExpirationDate])
  3122. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  3123. OCMStub([mockVerifyPhoneNumberResponse refreshToken]).andReturn(kRefreshToken);
  3124. callback(mockVerifyPhoneNumberResponse, nil);
  3125. });
  3126. });
  3127. }
  3128. #endif
  3129. @end
  3130. NS_ASSUME_NONNULL_END