FIRUserTests.m 212 KB

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