FIRUserTests.m 239 KB

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