FIRUserTests.m 193 KB

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