FIRUserTests.m 224 KB

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