FIRUserTests.m 203 KB

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