FIRUserTests.m 131 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606
  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 <FirebaseAuth/FirebaseAuth.h>
  19. #import "FIRAuth_Internal.h"
  20. #import "FIRAuthErrorUtils.h"
  21. #import "FIRAuthBackend.h"
  22. #import "FIRAuthGlobalWorkQueue.h"
  23. #import "FIRAuthOperationType.h"
  24. #import "FIRAuthTokenResult.h"
  25. #import "FIREmailLinkSignInResponse.m"
  26. #import "FIRSecureTokenService.h"
  27. #import "FIRSecureTokenRequest.h"
  28. #import "FIRSecureTokenResponse.h"
  29. #import "FIRSignUpNewUserResponse.h"
  30. #import "FIRGetAccountInfoRequest.h"
  31. #import "FIRGetAccountInfoResponse.h"
  32. #import "FIRSetAccountInfoRequest.h"
  33. #import "FIRSetAccountInfoResponse.h"
  34. #import "FIRUser_Internal.h"
  35. #import "FIRUserInfo.h"
  36. #import "FIRUserMetadata.h"
  37. #import "FIRVerifyAssertionResponse.h"
  38. #import "FIRVerifyAssertionRequest.h"
  39. #import "FIRVerifyPasswordRequest.h"
  40. #import "FIRVerifyPasswordResponse.h"
  41. #import "FIRVerifyPhoneNumberRequest.h"
  42. #import "FIRVerifyPhoneNumberResponse.h"
  43. #import "FIRApp+FIRAuthUnitTests.h"
  44. #import "OCMStubRecorder+FIRAuthUnitTests.h"
  45. #import <OCMock/OCMock.h>
  46. #if TARGET_OS_IOS
  47. #import "Phone/FIRPhoneAuthCredential_Internal.h"
  48. #import "FIRPhoneAuthProvider.h"
  49. #endif
  50. NS_ASSUME_NONNULL_BEGIN
  51. /** @var kAPIKey
  52. @brief The fake API key.
  53. */
  54. static NSString *const kAPIKey = @"FAKE_API_KEY";
  55. /** @var kAccessToken
  56. @brief The fake access token.
  57. */
  58. static NSString *const kAccessToken = @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZDBhODBiYzh"
  59. "jM2U5NDM0OTc0ZDFmMWRiNTEifQ.eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vZmItc2EtdXBncm"
  60. "FkZWQiLCJhdWQiOiJ0ZXN0X2F1ZCIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZCIs"
  61. "InN1YiI6InRlc3Rfc3ViIiwiaWF0IjoxNTIyMzYwNTU3LCJleHAiOjE1MjIzNjQxNTcsImVtYWlsIjoiYXVuaXRlc3R1c2"
  62. "VyQGdtYWlsLmNvbSIsImVtYWlsX3ZlcmlmaWVkIjpmYWxzZSwiZmlyZWJhc2UiOnsiaWRlbnRpdGllcyI6eyJlbWFpbCI6"
  63. "WyJhdW5pdGVzdHVzZXJAZ21haWwuY29tIl19LCJzaWduX2luX3Byb3ZpZGVyIjoicGFzc3dvcmQifX0=.WFQqSrpVnxx7m"
  64. "UrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_BKkcbZuq8O7iDVpOvqKC"
  65. "3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjN"
  66. "hHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbW"
  67. "lddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  68. /** @var kAccessTokenWithBase64URLCharacters
  69. @brief The fake access where the AUD value is "??????????>>>>>>>>>>" and the email value is
  70. ">>>>>>>>????????@gmail.com".
  71. */
  72. static NSString *const kAccessTokenWithBase64URLCharacter = @"ey?hbGciOimnuzI1NiIsImtpZCI6ImY1YjE4M"
  73. "jc2YTQ4NjYxZDBhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ.eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2ds"
  74. "ZS5jb20vZmItc2EtdXBncmFkZWQiLCJhdWQiOiI_Pz8_Pz8_Pz8_Pj4-Pj4-Pj4-PiIsImF1dGhfdGltZSI6MTUyMjM2MD"
  75. "U0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZCIsInN1YiI6InRlc3Rfc3ViIiwiaWF0IjoxNTIyMzYwNTU3LCJleHAiOjE1"
  76. "MjIzNjQxNTcsImVtYWlsIjoiPj4-Pj4-Pj4_Pz8_Pz8_P0BnbWFpbC5jb20iLCJlbWFpbF92ZXJpZmllZCI6ZmFsc2UsIm"
  77. "ZpcmViYXNlIjp7ImlkZW50aXRpZXMiOnsiZW1haWwiOlsiYXVuaXRlc3R1c2VyQGdtYWlsLmNvbSJdfSwic2lnbl9pbl9w"
  78. "cm92aWRlciI6IlBhc3N3b3JkIn19.WFQqSrpVnxx7mUrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-"
  79. "4aS3qRnN2kvk9KJAaF3eI_BKkcbZuq8O7iDVpOvqKC3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-XtBt"
  80. "1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjNhHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-dDV7"
  81. "jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbWlddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  82. /** @var kbase64URLEncodedEmail
  83. @brief The fake email address with a value containing non-valid base64 encoded characters.
  84. @remarks This is used to ensure that the token parser is able to handle base64 URL enconded
  85. strings.
  86. */
  87. static NSString *const kbase64URLEncodedEmail = @">>>>>>>>????????@gmail.com";
  88. /** @var kbase64URLEncodedAUD
  89. @brief The fake AUD with a value containing non-valid base64 encoded characters.
  90. @remarks This is used to ensure that the token parser is able to handle base64 URL enconded
  91. strings.
  92. */
  93. static NSString *const kbase64URLEncodedAUD = @"??????????>>>>>>>>>>";
  94. /** @var kAccessTokenLength415
  95. @brief The fake access token with 415 characters in the claims potion of the token.
  96. */
  97. static NSString *const kAccessTokenLength415 = @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZD"
  98. "BhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ.eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vdGVzd"
  99. "CIsImF1ZCI6InRlc3RfYXVkIiwiYXV0aF90aW1lIjoxNTIyMzYwNTQ5LCJ1c2VyX2lkIjoidGVzdF91c2VyX2lkIiwic3V"
  100. "iIjoidGVzdF9zdWIiLCJpYXQiOjE1MjIzNjA1NTcsImV4cCI6MTUyMjM2NDE1NywiZW1haWwiOiJhdW5pdGVzdHVzZXJAZ"
  101. "21haWwuY29tIiwiZW1haWxfdmVyaWZpZWQiOmZhbHNlLCJmaXJlYmFzZSI6eyJpZGVudGl0aWVzIjp7ImVtYWlsIjpbImF"
  102. "1bml0ZXN0dXNlckBnbWFpbC5jb20iXX0sInNpZ25faW5fcHJvdmlkZXIiOiJwYXNzd29yZCJ9fQ=.WFQqSrpVnxx7m"
  103. "UrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_BKkcbZuq8O7iDVpOvqKC"
  104. "3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjN"
  105. "hHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbW"
  106. "lddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  107. /** @var kAccessTokenLength416
  108. @brief The fake access token with 416 characters in the claims potion of the token.
  109. */
  110. static NSString *const kAccessTokenLength416 = @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZD"
  111. "BhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ.eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vdGVzd"
  112. "DIiLCJhdWQiOiJ0ZXN0X2F1ZCIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZCIsInN"
  113. "1YiI6InRlc3Rfc3ViIiwiaWF0IjoxNTIyMzYwNTU3LCJleHAiOjE1MjIzNjQxNTcsImVtYWlsIjoiYXVuaXRlc3R1c2VyQ"
  114. "GdtYWlsLmNvbSIsImVtYWlsX3ZlcmlmaWVkIjpmYWxzZSwiZmlyZWJhc2UiOnsiaWRlbnRpdGllcyI6eyJlbWFpbCI6WyJ"
  115. "hdW5pdGVzdHVzZXJAZ21haWwuY29tIl19LCJzaWduX2luX3Byb3ZpZGVyIjoicGFzc3dvcmQifX0=.WFQqSrpVnxx7m"
  116. "UrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_BKkcbZuq8O7iDVpOvqKC"
  117. "3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH7jjN"
  118. "hHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqVkPbW"
  119. "lddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  120. /** @var kAccessTokenLength4523
  121. @brief The fake access token with 523 characters in the claims potion of the token.
  122. */
  123. static NSString *const kAccessTokenLength523 = @"eyJhbGciOimnuzI1NiIsImtpZCI6ImY1YjE4Mjc2YTQ4NjYxZD"
  124. "BhODBiYzhjM2U5NDM0OTc0ZDFmMWRiNTEifQ.eyJpc3MiOiJodHRwczovL3NlY3VyZXRva2VuLmdvb2dsZS5jb20vdGVzd"
  125. "DQiLCJhdWQiOiJ0ZXN0X2F1ZCIsImF1dGhfdGltZSI6MTUyMjM2MDU0OSwidXNlcl9pZCI6InRlc3RfdXNlcl9pZF81NDM"
  126. "yIiwic3ViIjoidGVzdF9zdWIiLCJpYXQiOjE1MjIzNjA1NTcsImV4cCI6MTUyMjM2NDE1OSwiZW1haWwiOiJhdW5pdGVzd"
  127. "HVzZXI0QGdtYWlsLmNvbSIsImVtYWlsX3ZlcmlmaWVkIjp0cnVlLCJmaXJlYmFzZSI6eyJpZGVudGl0aWVzIjp7ImVtYWl"
  128. "sIjpbImF1bml0ZXN0dXNlckBnbWFpbC5jb20iXX0sInNpZ25faW5fcHJvdmlkZXIiOiJwYXNzd29yZCJ9fQ=.WFQqSrpVn"
  129. "xx7mUrdKZA517Sp4ZBt-l2xQzGKNMVE90JB3vuNa-NyWZC-aTYMvND3-4aS3qRnN2kvk9KJAaF3eI_BKkcbZuq8O7iDVpO"
  130. "vqKC3QcW0PnwqSPChL3XqoDF322FcBEgemwwgaEVZMuo7GhJvHw-XtBt1KRXOoGHcr3P6RsvoulUouKQmqt6TP27eZtrgH"
  131. "7jjNhHm7gjX_WaRmgTOvYsuDbBBGdE15yIVZ3acI4cFUgwMRhaW-dDV7jTOqZGYJlTsI5oRMehphoVnYnEedJga28r4mqV"
  132. "kPbWlddL4dVVm85FYmQcRc0b2CLMnSevBDlwu754ZUZmRgnuvDA";
  133. /** @var kNewAccessToken
  134. @brief A new value for the fake access token.
  135. */
  136. static NSString *const kNewAccessToken = @"NEW_ACCESS_TOKEN";
  137. /** @var kAccessTokenValidInterval
  138. @brief The time to live for the fake access token.
  139. */
  140. static const NSTimeInterval kAccessTokenTimeToLive = 60 * 60;
  141. /** @var kRefreshToken
  142. @brief The fake refresh token.
  143. */
  144. static NSString *const kRefreshToken = @"REFRESH_TOKEN";
  145. /** @var kLocalID
  146. @brief The fake local user ID.
  147. */
  148. static NSString *const kLocalID = @"LOCAL_ID";
  149. /** @var kAnotherLocalID
  150. @brief The fake local ID of another user.
  151. */
  152. static NSString *const kAnotherLocalID = @"ANOTHER_LOCAL_ID";
  153. /** @var kGoogleIDToken
  154. @brief The fake ID token from Google Sign-In.
  155. */
  156. static NSString *const kGoogleIDToken = @"GOOGLE_ID_TOKEN";
  157. /** @var kFacebookIDToken
  158. @brief The fake ID token from Facebook Sign-In. Facebook provider ID token is always nil.
  159. */
  160. static NSString *const kFacebookIDToken = nil;
  161. /** @var kGoogleAccessToken
  162. @brief The fake access token from Google Sign-In.
  163. */
  164. static NSString *const kGoogleAccessToken = @"GOOGLE_ACCESS_TOKEN";
  165. /** @var kFacebookAccessToken
  166. @brief The fake access token from Facebook Sign-In.
  167. */
  168. static NSString *const kFacebookAccessToken = @"FACEBOOK_ACCESS_TOKEN";
  169. /** @var kEmail
  170. @brief The fake user email.
  171. */
  172. static NSString *const kEmail = @"user@company.com";
  173. /** @var kPhoneNumber
  174. @brief The fake user phone number.
  175. */
  176. static NSString *const kPhoneNumber = @"12345658";
  177. /** @var kTemporaryProof
  178. @brief The fake temporary proof.
  179. */
  180. static NSString *const kTemporaryProof = @"12345658";
  181. /** @var kNewEmail
  182. @brief A new value for the fake user email.
  183. */
  184. static NSString *const kNewEmail = @"newuser@company.com";
  185. /** @var kUserName
  186. @brief The fake user name.
  187. */
  188. static NSString *const kUserName = @"User Doe";
  189. /** @var kNewDisplayName
  190. @brief A new value for the fake user display name.
  191. */
  192. static NSString *const kNewDisplayName = @"New User Doe";
  193. /** @var kPhotoURL
  194. @brief The fake user profile image URL string.
  195. */
  196. static NSString *const kPhotoURL = @"https://host.domain/image";
  197. /** @var kNewPhotoURL
  198. @brief A new value for the fake user profile image URL string..
  199. */
  200. static NSString *const kNewPhotoURL = @"https://host.domain/new/image";
  201. /** @var kFakePassword
  202. @brief The fake user password.
  203. */
  204. static NSString *const kFakePassword = @"123456";
  205. /** @var kNewPassword
  206. @brief The fake new user password.
  207. */
  208. static NSString *const kNewPassword = @"1234567";
  209. /** @var kPasswordHash
  210. @brief The fake user password hash.
  211. */
  212. static NSString *const kPasswordHash = @"UkVEQUNURUQ=";
  213. /** @var kGoogleUD
  214. @brief The fake user ID under Google Sign-In.
  215. */
  216. static NSString *const kGoogleID = @"GOOGLE_ID";
  217. /** @var kGoogleEmail
  218. @brief The fake user email under Google Sign-In.
  219. */
  220. static NSString *const kGoogleEmail = @"user@gmail.com";
  221. /** @var kGoogleDisplayName
  222. @brief The fake user display name under Google Sign-In.
  223. */
  224. static NSString *const kGoogleDisplayName = @"Google Doe";
  225. /** @var kEmailDisplayName
  226. @brief The fake user display name for email password user.
  227. */
  228. static NSString *const kEmailDisplayName = @"Email Doe";
  229. /** @var kFacebookDisplayName
  230. @brief The fake user display name under Facebook Sign-In.
  231. */
  232. static NSString *const kFacebookDisplayName = @"Facebook Doe";
  233. /** @var kGooglePhotoURL
  234. @brief The fake user profile image URL string under Google Sign-In.
  235. */
  236. static NSString *const kGooglePhotoURL = @"https://googleusercontents.com/user/profile";
  237. /** @var kFacebookID
  238. @brief The fake user ID under Facebook Login.
  239. */
  240. static NSString *const kFacebookID = @"FACEBOOK_ID";
  241. /** @var kFacebookEmail
  242. @brief The fake user email under Facebook Login.
  243. */
  244. static NSString *const kFacebookEmail = @"user@facebook.com";
  245. /** @var kVerificationCode
  246. @brief Fake verification code used for testing.
  247. */
  248. static NSString *const kVerificationCode = @"12345678";
  249. /** @var kVerificationID
  250. @brief Fake verification ID for testing.
  251. */
  252. static NSString *const kVerificationID = @"55432";
  253. /** @var kUserArchiverKey
  254. @brief The key used to archive and unarchive the user object for test NSSecureCoding.
  255. */
  256. static NSString *const kUserArchiverKey = @"userArchiverKey";
  257. /** @var kCreationDateInSeconds
  258. @brief The fake creation date.
  259. */
  260. static NSTimeInterval const kCreationDateTimeIntervalInSeconds = 1505858500;
  261. /** @var kLastSignInDateTimeIntervalInSeconds
  262. @brief The fake last sign in date date.
  263. */
  264. static NSTimeInterval const kLastSignInDateTimeIntervalInSeconds = 1505858583;
  265. /** @var kExpectationTimeout
  266. @brief The maximum time waiting for expectations to fulfill.
  267. */
  268. static const NSTimeInterval kExpectationTimeout = 2;
  269. /** @extention FIRSecureTokenService
  270. @brief Extends the FIRSecureTokenService class to expose one private method for testing only.
  271. */
  272. @interface FIRSecureTokenService ()
  273. /** @fn hasValidAccessToken
  274. @brief private method exposed so it can be mocked to prevent the fake expiration date from
  275. affecting the the unit tests.
  276. */
  277. - (BOOL)hasValidAccessToken;
  278. @end
  279. /** @class FIRUserTests
  280. @brief Tests for @c FIRUser .
  281. */
  282. @interface FIRUserTests : XCTestCase
  283. @end
  284. @implementation FIRUserTests {
  285. /** @var _mockBackend
  286. @brief The mock @c FIRAuthBackendImplementation .
  287. */
  288. id _mockBackend;
  289. }
  290. /** @fn googleProfile
  291. @brief The fake user profile under additional user data in @c FIRVerifyAssertionResponse.
  292. */
  293. + (NSDictionary *)googleProfile {
  294. static NSDictionary *kGoogleProfile = nil;
  295. static dispatch_once_t onceToken;
  296. dispatch_once(&onceToken, ^{
  297. kGoogleProfile = @{
  298. @"email": kGoogleEmail,
  299. @"given_name": @"User",
  300. @"family_name": @"Doe"
  301. };
  302. });
  303. return kGoogleProfile;
  304. }
  305. - (void)setUp {
  306. [super setUp];
  307. _mockBackend = OCMProtocolMock(@protocol(FIRAuthBackendImplementation));
  308. [FIRAuthBackend setBackendImplementation:_mockBackend];
  309. [FIRApp resetAppForAuthUnitTests];
  310. }
  311. - (void)tearDown {
  312. [FIRAuthBackend setDefaultBackendImplementationWithRPCIssuer:nil];
  313. [super tearDown];
  314. }
  315. #pragma mark - Tests
  316. /** @fn testUserPropertiesAndNSSecureCoding
  317. @brief Tests properties of the @c FIRUser instance before and after being
  318. serialized/deserialized.
  319. */
  320. - (void)testUserPropertiesAndNSSecureCoding {
  321. // Mock auth provider user info for email/password for GetAccountInfo.
  322. id mockPasswordUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  323. OCMStub([mockPasswordUserInfo providerID]).andReturn(FIREmailAuthProviderID);
  324. OCMStub([mockPasswordUserInfo federatedID]).andReturn(kEmail);
  325. OCMStub([mockPasswordUserInfo email]).andReturn(kEmail);
  326. // Mock auth provider user info from Google for GetAccountInfo.
  327. id mockGoogleUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  328. OCMStub([mockGoogleUserInfo providerID]).andReturn(FIRGoogleAuthProviderID);
  329. OCMStub([mockGoogleUserInfo displayName]).andReturn(kGoogleDisplayName);
  330. OCMStub([mockGoogleUserInfo photoURL]).andReturn([NSURL URLWithString:kGooglePhotoURL]);
  331. OCMStub([mockGoogleUserInfo federatedID]).andReturn(kGoogleID);
  332. OCMStub([mockGoogleUserInfo email]).andReturn(kGoogleEmail);
  333. // Mock auth provider user info from Facebook for GetAccountInfo.
  334. id mockFacebookUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  335. OCMStub([mockFacebookUserInfo providerID]).andReturn(FIRFacebookAuthProviderID);
  336. OCMStub([mockFacebookUserInfo federatedID]).andReturn(kFacebookID);
  337. OCMStub([mockFacebookUserInfo email]).andReturn(kFacebookEmail);
  338. #if TARGET_OS_IOS
  339. // Mock auth provider user info from Phone auth provider for GetAccountInfo.
  340. id mockPhoneUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  341. OCMStub([mockPhoneUserInfo providerID]).andReturn(FIRPhoneAuthProviderID);
  342. OCMStub([mockPhoneUserInfo phoneNumber]).andReturn(kPhoneNumber);
  343. #endif
  344. // Mock the root user info object for GetAccountInfo.
  345. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  346. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  347. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  348. OCMStub([mockGetAccountInfoResponseUser emailVerified]).andReturn(YES);
  349. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  350. OCMStub([mockGetAccountInfoResponseUser photoURL]).andReturn([NSURL URLWithString:kPhotoURL]);
  351. OCMStub([mockGetAccountInfoResponseUser creationDate]).
  352. andReturn([NSDate dateWithTimeIntervalSince1970:kCreationDateTimeIntervalInSeconds]);
  353. OCMStub([mockGetAccountInfoResponseUser lastLoginDate]).
  354. andReturn([NSDate dateWithTimeIntervalSince1970:kLastSignInDateTimeIntervalInSeconds]);
  355. NSArray *providerUserInfos = @[
  356. #if TARGET_OS_IOS
  357. mockPhoneUserInfo,
  358. #endif
  359. mockPasswordUserInfo,
  360. mockGoogleUserInfo,
  361. mockFacebookUserInfo
  362. ];
  363. OCMStub([mockGetAccountInfoResponseUser providerUserInfo]).andReturn(providerUserInfos);
  364. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  365. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  366. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  367. completion:^(FIRUser *user) {
  368. // Verify FIRUserInfo properties on FIRUser itself.
  369. XCTAssertEqualObjects(user.providerID, @"Firebase");
  370. XCTAssertEqualObjects(user.uid, kLocalID);
  371. XCTAssertEqualObjects(user.displayName, kGoogleDisplayName);
  372. XCTAssertEqualObjects(user.photoURL, [NSURL URLWithString:kPhotoURL]);
  373. XCTAssertEqualObjects(user.email, kEmail);
  374. XCTAssertEqualObjects(user.metadata.creationDate,
  375. [NSDate dateWithTimeIntervalSince1970:kCreationDateTimeIntervalInSeconds]);
  376. XCTAssertEqualObjects(user.metadata.lastSignInDate,
  377. [NSDate dateWithTimeIntervalSince1970:kLastSignInDateTimeIntervalInSeconds]);
  378. // Verify FIRUser properties besides providerData contents.
  379. XCTAssertFalse(user.anonymous);
  380. XCTAssertTrue(user.emailVerified);
  381. XCTAssertEqualObjects(user.refreshToken, kRefreshToken);
  382. XCTAssertEqual(user.providerData.count, providerUserInfos.count);
  383. NSDictionary<NSString *, id<FIRUserInfo>> *providerMap =
  384. [self dictionaryWithUserInfoArray:user.providerData];
  385. // Verify FIRUserInfo properties from email/password.
  386. id<FIRUserInfo> passwordUserInfo = providerMap[FIREmailAuthProviderID];
  387. XCTAssertEqualObjects(passwordUserInfo.uid, kEmail);
  388. XCTAssertNil(passwordUserInfo.displayName);
  389. XCTAssertNil(passwordUserInfo.photoURL);
  390. XCTAssertEqualObjects(passwordUserInfo.email, kEmail);
  391. // Verify FIRUserInfo properties from the Google auth provider.
  392. id<FIRUserInfo> googleUserInfo = providerMap[FIRGoogleAuthProviderID];
  393. XCTAssertEqualObjects(googleUserInfo.uid, kGoogleID);
  394. XCTAssertEqualObjects(googleUserInfo.displayName, kGoogleDisplayName);
  395. XCTAssertEqualObjects(googleUserInfo.photoURL, [NSURL URLWithString:kGooglePhotoURL]);
  396. XCTAssertEqualObjects(googleUserInfo.email, kGoogleEmail);
  397. // Verify FIRUserInfo properties from the Facebook auth provider.
  398. id<FIRUserInfo> facebookUserInfo = providerMap[FIRFacebookAuthProviderID];
  399. XCTAssertEqualObjects(facebookUserInfo.uid, kFacebookID);
  400. XCTAssertNil(facebookUserInfo.displayName);
  401. XCTAssertNil(facebookUserInfo.photoURL);
  402. XCTAssertEqualObjects(facebookUserInfo.email, kFacebookEmail);
  403. #if TARGET_OS_IOS
  404. // Verify FIRUserInfo properties from the phone auth provider.
  405. id<FIRUserInfo> phoneUserInfo = providerMap[FIRPhoneAuthProviderID];
  406. XCTAssertEqualObjects(phoneUserInfo.phoneNumber, kPhoneNumber);
  407. #endif
  408. // Test NSSecureCoding
  409. XCTAssertTrue([FIRUser supportsSecureCoding]);
  410. NSMutableData *data = [NSMutableData data];
  411. NSKeyedArchiver *archiver = [[NSKeyedArchiver alloc] initForWritingWithMutableData:data];
  412. [archiver encodeObject:user forKey:kUserArchiverKey];
  413. [archiver finishEncoding];
  414. NSKeyedUnarchiver *unarchiver = [[NSKeyedUnarchiver alloc] initForReadingWithData:data];
  415. FIRUser *unarchivedUser = [unarchiver decodeObjectForKey:kUserArchiverKey];
  416. // Verify NSSecureCoding for FIRUser
  417. XCTAssertEqualObjects(unarchivedUser.providerID, user.providerID);
  418. XCTAssertEqualObjects(unarchivedUser.uid, user.uid);
  419. XCTAssertEqualObjects(unarchivedUser.email, user.email);
  420. XCTAssertEqualObjects(unarchivedUser.photoURL, user.photoURL);
  421. XCTAssertEqualObjects(unarchivedUser.displayName, user.displayName);
  422. // Verify NSSecureCoding properties besides providerData contents.
  423. XCTAssertEqual(unarchivedUser.anonymous, user.anonymous);
  424. XCTAssertEqual(unarchivedUser.emailVerified, user.emailVerified);
  425. XCTAssertEqualObjects(unarchivedUser.refreshToken, user.refreshToken);
  426. XCTAssertEqualObjects(unarchivedUser.metadata.creationDate, user.metadata.creationDate);
  427. XCTAssertEqualObjects(unarchivedUser.metadata.lastSignInDate, user.metadata.lastSignInDate);
  428. XCTAssertEqual(unarchivedUser.providerData.count, user.providerData.count);
  429. NSDictionary<NSString *, id<FIRUserInfo>> *unarchivedProviderMap =
  430. [self dictionaryWithUserInfoArray:unarchivedUser.providerData];
  431. // Verify NSSecureCoding properties from email/password.
  432. id<FIRUserInfo> unarchivedPasswordUserInfo = unarchivedProviderMap[FIREmailAuthProviderID];
  433. XCTAssertEqualObjects(unarchivedPasswordUserInfo.uid, passwordUserInfo.uid);
  434. XCTAssertEqualObjects(unarchivedPasswordUserInfo.displayName, passwordUserInfo.displayName);
  435. XCTAssertEqualObjects(unarchivedPasswordUserInfo.photoURL, passwordUserInfo.photoURL);
  436. XCTAssertEqualObjects(unarchivedPasswordUserInfo.email, passwordUserInfo.email);
  437. // Verify NSSecureCoding properties from the Google auth provider.
  438. id<FIRUserInfo> unarchivedGoogleUserInfo = unarchivedProviderMap[FIRGoogleAuthProviderID];
  439. XCTAssertEqualObjects(unarchivedGoogleUserInfo.uid, googleUserInfo.uid);
  440. XCTAssertEqualObjects(unarchivedGoogleUserInfo.displayName, googleUserInfo.displayName);
  441. XCTAssertEqualObjects(unarchivedGoogleUserInfo.photoURL, googleUserInfo.photoURL);
  442. XCTAssertEqualObjects(unarchivedGoogleUserInfo.email, googleUserInfo.email);
  443. // Verify NSSecureCoding properties from the Facebook auth provider.
  444. id<FIRUserInfo> unarchivedFacebookUserInfo = unarchivedProviderMap[FIRFacebookAuthProviderID];
  445. XCTAssertEqualObjects(unarchivedFacebookUserInfo.uid, facebookUserInfo.uid);
  446. XCTAssertEqualObjects(unarchivedFacebookUserInfo.displayName, facebookUserInfo.displayName);
  447. XCTAssertEqualObjects(unarchivedFacebookUserInfo.photoURL, facebookUserInfo.photoURL);
  448. XCTAssertEqualObjects(unarchivedFacebookUserInfo.email, facebookUserInfo.email);
  449. #if TARGET_OS_IOS
  450. // Verify FIRUserInfo properties from the phone auth provider.
  451. id<FIRUserInfo> unarchivedPhoneUserInfo = unarchivedProviderMap[FIRPhoneAuthProviderID];
  452. XCTAssertEqualObjects(unarchivedPhoneUserInfo.phoneNumber, phoneUserInfo.phoneNumber);
  453. #endif
  454. [expectation fulfill];
  455. }];
  456. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  457. OCMVerifyAll(_mockBackend);
  458. }
  459. /** @fn testUpdateEmailSuccess
  460. @brief Tests the flow of a successful @c updateEmail:completion: call.
  461. */
  462. - (void)testUpdateEmailSuccess {
  463. id (^mockUserInfoWithDisplayName)(NSString *) = ^(NSString *displayName) {
  464. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  465. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  466. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  467. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(displayName);
  468. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  469. return mockGetAccountInfoResponseUser;
  470. };
  471. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  472. id userInfoResponse = mockUserInfoWithDisplayName(kGoogleDisplayName);
  473. [self signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  474. completion:^(FIRUser *user) {
  475. // Pretend that the display name on the server has been changed since last request.
  476. [self
  477. expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoWithDisplayName(kNewDisplayName)];
  478. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  479. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  480. FIRSetAccountInfoResponseCallback callback) {
  481. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  482. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  483. XCTAssertEqualObjects(request.email, kNewEmail);
  484. XCTAssertNil(request.localID);
  485. XCTAssertNil(request.displayName);
  486. XCTAssertNil(request.photoURL);
  487. XCTAssertNil(request.password);
  488. XCTAssertNil(request.providers);
  489. XCTAssertNil(request.deleteAttributes);
  490. XCTAssertNil(request.deleteProviders);
  491. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  492. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  493. OCMStub([mockSetAccountInfoResponse email]).andReturn(kNewEmail);
  494. OCMStub([mockSetAccountInfoResponse displayName]).andReturn(kNewDisplayName);
  495. callback(mockSetAccountInfoResponse, nil);
  496. });
  497. });
  498. [user updateEmail:kNewEmail completion:^(NSError *_Nullable error) {
  499. XCTAssertNil(error);
  500. XCTAssertEqualObjects(user.email, kNewEmail);
  501. XCTAssertEqualObjects(user.displayName, kNewDisplayName);
  502. [expectation fulfill];
  503. }];
  504. }];
  505. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  506. OCMVerifyAll(_mockBackend);
  507. }
  508. /** @fn testUpdateEmailWithAuthLinkAccountSuccess
  509. @brief Tests a successful @c updateEmail:completion: call updates provider info.
  510. */
  511. - (void)testUpdateEmailWithAuthLinkAccountSuccess {
  512. id (^mockUserInfoWithDisplayName)(NSString *) = ^(NSString *displayName) {
  513. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  514. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  515. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  516. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(displayName);
  517. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  518. return mockGetAccountInfoResponseUser;
  519. };
  520. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  521. id userInfoResponse = mockUserInfoWithDisplayName(kGoogleDisplayName);
  522. [self signInWithEmailLinkWithMockUserInfoResponse:userInfoResponse
  523. completion:^(FIRUser *user) {
  524. // Pretend that the display name on the server has been changed since last request.
  525. [self
  526. expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoWithDisplayName(kNewDisplayName)];
  527. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  528. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  529. FIRSetAccountInfoResponseCallback callback) {
  530. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  531. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  532. XCTAssertEqualObjects(request.email, kNewEmail);
  533. XCTAssertNil(request.localID);
  534. XCTAssertNil(request.displayName);
  535. XCTAssertNil(request.photoURL);
  536. XCTAssertNil(request.password);
  537. XCTAssertNil(request.providers);
  538. XCTAssertNil(request.deleteAttributes);
  539. XCTAssertNil(request.deleteProviders);
  540. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  541. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  542. OCMStub([mockSetAccountInfoResponse email]).andReturn(kNewEmail);
  543. OCMStub([mockSetAccountInfoResponse displayName]).andReturn(kNewDisplayName);
  544. callback(mockSetAccountInfoResponse, nil);
  545. });
  546. });
  547. [user updateEmail:kNewEmail completion:^(NSError *_Nullable error) {
  548. XCTAssertNil(error);
  549. XCTAssertEqualObjects(user.email, kNewEmail);
  550. XCTAssertEqualObjects(user.displayName, kNewDisplayName);
  551. XCTAssertFalse(user.isAnonymous);
  552. [expectation fulfill];
  553. }];
  554. }];
  555. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  556. OCMVerifyAll(_mockBackend);
  557. }
  558. /** @fn testUpdateEmailFailure
  559. @brief Tests the flow of a failed @c updateEmail:completion: call.
  560. */
  561. - (void)testUpdateEmailFailure {
  562. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  563. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  564. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  565. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  566. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  567. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  568. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  569. completion:^(FIRUser *user) {
  570. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  571. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  572. .andDispatchError2([FIRAuthErrorUtils invalidEmailErrorWithMessage:nil]);
  573. [user updateEmail:kNewEmail completion:^(NSError *_Nullable error) {
  574. XCTAssertTrue([NSThread isMainThread]);
  575. XCTAssertEqual(error.code, FIRAuthErrorCodeInvalidEmail);
  576. // Email should not have changed on the client side.
  577. XCTAssertEqualObjects(user.email, kEmail);
  578. // User is still signed in.
  579. XCTAssertEqual([FIRAuth auth].currentUser, user);
  580. [expectation fulfill];
  581. }];
  582. }];
  583. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  584. OCMVerifyAll(_mockBackend);
  585. }
  586. /** @fn testUpdateEmailAutoSignOut
  587. @brief Tests the flow of a failed @c updateEmail:completion: call that automatically signs out.
  588. */
  589. - (void)testUpdateEmailAutoSignOut {
  590. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  591. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  592. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  593. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  594. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  595. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  596. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  597. completion:^(FIRUser *user) {
  598. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  599. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  600. .andDispatchError2([FIRAuthErrorUtils invalidUserTokenErrorWithMessage:nil]);
  601. [user updateEmail:kNewEmail completion:^(NSError *_Nullable error) {
  602. XCTAssertTrue([NSThread isMainThread]);
  603. XCTAssertEqual(error.code, FIRAuthErrorCodeInvalidUserToken);
  604. // Email should not have changed on the client side.
  605. XCTAssertEqualObjects(user.email, kEmail);
  606. // User is no longer signed in.
  607. XCTAssertNil([FIRAuth auth].currentUser);
  608. [expectation fulfill];
  609. }];
  610. }];
  611. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  612. OCMVerifyAll(_mockBackend);
  613. }
  614. #if TARGET_OS_IOS
  615. /** @fn testUpdatePhoneSuccess
  616. @brief Tests the flow of a successful @c updatePhoneNumberCredential:completion: call.
  617. */
  618. - (void)testUpdatePhoneSuccess {
  619. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  620. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  621. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  622. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  623. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  624. if (phoneNumber.length) {
  625. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  626. }
  627. return mockGetAccountInfoResponseUser;
  628. };
  629. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  630. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  631. [self signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  632. completion:^(FIRUser *user) {
  633. [self expectVerifyPhoneNumberRequestWithPhoneNumber:kPhoneNumber isLinkOperation:NO error:nil];
  634. id userInfoResponseUpdate = mockUserInfoWithPhoneNumber(kPhoneNumber);
  635. [self expectGetAccountInfoWithMockUserInfoResponse:userInfoResponseUpdate];
  636. FIRPhoneAuthCredential *credential =
  637. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  638. verificationCode:kVerificationCode];
  639. [user updatePhoneNumberCredential:credential
  640. completion:^(NSError * _Nullable error) {
  641. XCTAssertTrue([NSThread isMainThread]);
  642. XCTAssertNil(error);
  643. XCTAssertEqualObjects([FIRAuth auth].currentUser.phoneNumber, kPhoneNumber);
  644. [expectation fulfill];
  645. }];
  646. }];
  647. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  648. OCMVerifyAll(_mockBackend);
  649. }
  650. /** @fn testUpdatePhoneNumberFailure
  651. @brief Tests the flow of a failed @c updatePhoneNumberCredential:completion: call.
  652. */
  653. - (void)testUpdatePhoneNumberFailure {
  654. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  655. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  656. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  657. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  658. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  659. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  660. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  661. completion:^(FIRUser *user) {
  662. OCMExpect([_mockBackend verifyPhoneNumber:[OCMArg any] callback:[OCMArg any]])
  663. .andDispatchError2([FIRAuthErrorUtils invalidPhoneNumberErrorWithMessage:nil]);
  664. FIRPhoneAuthCredential *credential =
  665. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  666. verificationCode:kVerificationCode];
  667. [user updatePhoneNumberCredential:credential completion:^(NSError *_Nullable error) {
  668. XCTAssertTrue([NSThread isMainThread]);
  669. XCTAssertEqual(error.code, FIRAuthErrorCodeInvalidPhoneNumber);
  670. XCTAssertEqual([FIRAuth auth].currentUser, user);
  671. [expectation fulfill];
  672. }];
  673. }];
  674. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  675. OCMVerifyAll(_mockBackend);
  676. }
  677. /** @fn testUpdatePhoneNumberFailureAutoSignOut
  678. @brief Tests the flow of a failed @c updatePhoneNumberCredential:completion: call that
  679. automatically signs out.
  680. */
  681. - (void)testUpdatePhoneNumberFailureAutoSignOut {
  682. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  683. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  684. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  685. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  686. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  687. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  688. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  689. completion:^(FIRUser *user) {
  690. OCMExpect([_mockBackend verifyPhoneNumber:[OCMArg any] callback:[OCMArg any]])
  691. .andDispatchError2([FIRAuthErrorUtils userTokenExpiredErrorWithMessage:nil]);
  692. FIRPhoneAuthCredential *credential =
  693. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  694. verificationCode:kVerificationCode];
  695. [user updatePhoneNumberCredential:credential completion:^(NSError *_Nullable error) {
  696. XCTAssertTrue([NSThread isMainThread]);
  697. XCTAssertEqual(error.code, FIRAuthErrorCodeUserTokenExpired);
  698. XCTAssertNil([FIRAuth auth].currentUser);
  699. [expectation fulfill];
  700. }];
  701. }];
  702. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  703. OCMVerifyAll(_mockBackend);
  704. }
  705. #endif
  706. /** @fn testUpdatePasswordSuccess
  707. @brief Tests the flow of a successful @c updatePassword:completion: call.
  708. */
  709. - (void)testUpdatePasswordSuccess {
  710. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  711. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  712. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  713. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  714. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  715. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  716. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  717. completion:^(FIRUser *user) {
  718. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  719. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  720. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  721. FIRSetAccountInfoResponseCallback callback) {
  722. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  723. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  724. XCTAssertEqualObjects(request.password, kNewPassword);
  725. XCTAssertNil(request.localID);
  726. XCTAssertNil(request.displayName);
  727. XCTAssertNil(request.photoURL);
  728. XCTAssertNil(request.email);
  729. XCTAssertNil(request.providers);
  730. XCTAssertNil(request.deleteAttributes);
  731. XCTAssertNil(request.deleteProviders);
  732. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  733. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  734. OCMStub([mockSetAccountInfoResponse displayName]).andReturn(kNewDisplayName);
  735. callback(mockSetAccountInfoResponse, nil);
  736. });
  737. });
  738. [user updatePassword:kNewPassword completion:^(NSError *_Nullable error) {
  739. XCTAssertTrue([NSThread isMainThread]);
  740. XCTAssertNil(error);
  741. XCTAssertFalse(user.isAnonymous);
  742. [expectation fulfill];
  743. }];
  744. }];
  745. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  746. OCMVerifyAll(_mockBackend);
  747. }
  748. /** @fn testUpdatePasswordFailure
  749. @brief Tests the flow of a failed @c updatePassword:completion: call.
  750. */
  751. - (void)testUpdatePasswordFailure {
  752. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  753. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  754. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  755. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  756. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  757. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  758. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  759. completion:^(FIRUser *user) {
  760. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  761. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  762. .andDispatchError2([FIRAuthErrorUtils requiresRecentLoginErrorWithMessage:nil]);
  763. [user updatePassword:kNewPassword completion:^(NSError *_Nullable error) {
  764. XCTAssertTrue([NSThread isMainThread]);
  765. XCTAssertEqual(error.code, FIRAuthErrorCodeRequiresRecentLogin);
  766. XCTAssertEqual([FIRAuth auth].currentUser, user);
  767. [expectation fulfill];
  768. }];
  769. }];
  770. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  771. OCMVerifyAll(_mockBackend);
  772. }
  773. /** @fn testUpdateEmptyPasswordFailure
  774. @brief Tests the flow of a failed @c updatePassword:completion: call due to an empty password.
  775. */
  776. - (void)testUpdateEmptyPasswordFailure {
  777. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  778. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  779. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  780. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  781. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  782. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  783. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  784. completion:^(FIRUser *user) {
  785. [user updatePassword:@"" completion:^(NSError *_Nullable error) {
  786. XCTAssertTrue([NSThread isMainThread]);
  787. XCTAssertEqual(error.code, FIRAuthErrorCodeWeakPassword);
  788. [expectation fulfill];
  789. }];
  790. }];
  791. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  792. }
  793. /** @fn testUpdatePasswordFailureAutoSignOut
  794. @brief Tests the flow of a failed @c updatePassword:completion: call that automatically signs
  795. out.
  796. */
  797. - (void)testUpdatePasswordFailureAutoSignOut {
  798. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  799. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  800. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  801. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  802. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  803. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  804. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  805. completion:^(FIRUser *user) {
  806. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  807. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  808. .andDispatchError2([FIRAuthErrorUtils userDisabledErrorWithMessage:nil]);
  809. [user updatePassword:kNewPassword completion:^(NSError *_Nullable error) {
  810. XCTAssertTrue([NSThread isMainThread]);
  811. XCTAssertEqual(error.code, FIRAuthErrorCodeUserDisabled);
  812. XCTAssertNil([FIRAuth auth].currentUser);
  813. [expectation fulfill];
  814. }];
  815. }];
  816. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  817. OCMVerifyAll(_mockBackend);
  818. }
  819. /** @fn testChangeProfileSuccess
  820. @brief Tests a successful user profile change flow.
  821. */
  822. - (void)testChangeProfileSuccess {
  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 photoURL]).andReturn(kPhotoURL);
  828. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  829. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  830. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  831. completion:^(FIRUser *user) {
  832. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  833. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  834. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  835. FIRSetAccountInfoResponseCallback callback) {
  836. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  837. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  838. XCTAssertEqualObjects(request.displayName, kNewDisplayName);
  839. XCTAssertEqualObjects(request.photoURL, [NSURL URLWithString:kNewPhotoURL]);
  840. XCTAssertNil(request.localID);
  841. XCTAssertNil(request.email);
  842. XCTAssertNil(request.password);
  843. XCTAssertNil(request.providers);
  844. XCTAssertNil(request.deleteAttributes);
  845. XCTAssertNil(request.deleteProviders);
  846. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  847. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  848. OCMStub([mockSetAccountInfoResponse displayName]).andReturn(kNewDisplayName);
  849. callback(mockSetAccountInfoResponse, nil);
  850. });
  851. });
  852. FIRUserProfileChangeRequest *profileChange = [user profileChangeRequest];
  853. profileChange.photoURL = [NSURL URLWithString:kNewPhotoURL];
  854. profileChange.displayName = kNewDisplayName;
  855. [profileChange commitChangesWithCompletion:^(NSError *_Nullable error) {
  856. XCTAssertTrue([NSThread isMainThread]);
  857. XCTAssertNil(error);
  858. XCTAssertEqualObjects(user.displayName, kNewDisplayName);
  859. XCTAssertEqualObjects(user.photoURL, [NSURL URLWithString:kNewPhotoURL]);
  860. [expectation fulfill];
  861. }];
  862. }];
  863. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  864. OCMVerifyAll(_mockBackend);
  865. }
  866. /** @fn testChangeProfileFailure
  867. @brief Tests a failed user profile change flow.
  868. */
  869. - (void)testChangeProfileFailure {
  870. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  871. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  872. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  873. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  874. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  875. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  876. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  877. completion:^(FIRUser *user) {
  878. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  879. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  880. .andDispatchError2([FIRAuthErrorUtils tooManyRequestsErrorWithMessage:nil]);
  881. FIRUserProfileChangeRequest *profileChange = [user profileChangeRequest];
  882. profileChange.displayName = kNewDisplayName;
  883. [profileChange commitChangesWithCompletion:^(NSError *_Nullable error) {
  884. XCTAssertTrue([NSThread isMainThread]);
  885. XCTAssertEqual(error.code, FIRAuthErrorCodeTooManyRequests);
  886. XCTAssertEqualObjects(user.displayName, kGoogleDisplayName);
  887. XCTAssertEqual([FIRAuth auth].currentUser, user);
  888. [expectation fulfill];
  889. }];
  890. }];
  891. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  892. OCMVerifyAll(_mockBackend);
  893. }
  894. /** @fn testChangeProfileFailureAutoSignOut
  895. @brief Tests a failed user profile change flow that automatically signs out.
  896. */
  897. - (void)testChangeProfileFailureAutoSignOut {
  898. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  899. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  900. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  901. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  902. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  903. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  904. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  905. completion:^(FIRUser *user) {
  906. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  907. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  908. .andDispatchError2([FIRAuthErrorUtils userNotFoundErrorWithMessage:nil]);
  909. FIRUserProfileChangeRequest *profileChange = [user profileChangeRequest];
  910. profileChange.displayName = kNewDisplayName;
  911. [profileChange commitChangesWithCompletion:^(NSError *_Nullable error) {
  912. XCTAssertTrue([NSThread isMainThread]);
  913. XCTAssertEqual(error.code, FIRAuthErrorCodeUserNotFound);
  914. XCTAssertEqualObjects(user.displayName, kGoogleDisplayName);
  915. XCTAssertNil([FIRAuth auth].currentUser);
  916. [expectation fulfill];
  917. }];
  918. }];
  919. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  920. OCMVerifyAll(_mockBackend);
  921. }
  922. /** @fn testReloadSuccess
  923. @brief Tests the flow of a successful @c reloadWithCompletion: call.
  924. */
  925. - (void)testReloadSuccess {
  926. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  927. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  928. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  929. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  930. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  931. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  932. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  933. completion:^(FIRUser *user) {
  934. id mockGetAccountInfoResponseUserNew = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  935. OCMStub([mockGetAccountInfoResponseUserNew localID]).andReturn(kLocalID);
  936. OCMStub([mockGetAccountInfoResponseUserNew email]).andReturn(kNewEmail);
  937. OCMStub([mockGetAccountInfoResponseUserNew displayName]).andReturn(kNewDisplayName);
  938. OCMStub([mockGetAccountInfoResponseUserNew passwordHash]).andReturn(kPasswordHash);
  939. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUserNew];
  940. [user reloadWithCompletion:^(NSError *_Nullable error) {
  941. XCTAssertTrue([NSThread isMainThread]);
  942. XCTAssertNil(error);
  943. XCTAssertEqualObjects(user.email, kNewEmail);
  944. XCTAssertEqualObjects(user.displayName, kNewDisplayName);
  945. [expectation fulfill];
  946. }];
  947. }];
  948. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  949. OCMVerifyAll(_mockBackend);
  950. }
  951. /** @fn testGetIDTokenResultSuccess
  952. @brief Tests the flow of a successful @c getIDTokenResultWithCompletion: call.
  953. */
  954. - (void)testGetIDTokenResultSuccess {
  955. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  956. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  957. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  958. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  959. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  960. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  961. id mockSecureTokenService = OCMClassMock([FIRSecureTokenService class]);
  962. OCMStub([mockSecureTokenService hasValidAccessToken]).andReturn(YES);
  963. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  964. completion:^(FIRUser *user) {
  965. [user getIDTokenResultWithCompletion:^(FIRAuthTokenResult *_Nullable tokenResult,
  966. NSError *_Nullable error) {
  967. XCTAssertTrue([NSThread isMainThread]);
  968. XCTAssertNil(error);
  969. XCTAssertEqualObjects(tokenResult.token, kAccessToken);
  970. XCTAssertTrue(tokenResult.issuedAtDate &&
  971. [tokenResult.issuedAtDate isKindOfClass:[NSDate class]]);
  972. XCTAssertTrue(tokenResult.authDate && [tokenResult.authDate isKindOfClass:[NSDate class]]);
  973. XCTAssertTrue(tokenResult.expirationDate &&
  974. [tokenResult.expirationDate isKindOfClass:[NSDate class]]);
  975. XCTAssertTrue(tokenResult.claims && [tokenResult.claims isKindOfClass:[NSDictionary class]]);
  976. [expectation fulfill];
  977. }];
  978. }];
  979. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  980. OCMVerifyAll(_mockBackend);
  981. }
  982. /** @fn testGetIDTokenResultForcingRefreshFailure
  983. @brief Tests the flow successful @c getIDTokenResultForcingRefresh:completion: calls.
  984. */
  985. - (void)testGetIDTokenResultForcingRefreshSuccess {
  986. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessToken];
  987. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenLength415];
  988. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenLength416];
  989. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenLength523];
  990. [self getIDTokenResultForcingRefreshSuccessWithIDToken:kAccessTokenWithBase64URLCharacter];
  991. }
  992. /** @fn testGetIDTokenResultSuccessWithBase64EncodedURL
  993. @brief Tests the flow of a successful @c getIDTokenResultWithCompletion: call using a base64 url
  994. encoded string.
  995. */
  996. - (void)testGetIDTokenResultSuccessWithBase64EncodedURL {
  997. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  998. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  999. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1000. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1001. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1002. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1003. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1004. completion:^(FIRUser *user) {
  1005. OCMExpect([_mockBackend secureToken:[OCMArg any] callback:[OCMArg any]])
  1006. .andCallBlock2(^(FIRSecureTokenRequest *_Nullable request,
  1007. FIRSecureTokenResponseCallback callback) {
  1008. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1009. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1010. id mockSecureTokenResponse = OCMClassMock([FIRSecureTokenResponse class]);
  1011. OCMStub([mockSecureTokenResponse accessToken]).andReturn(kAccessTokenWithBase64URLCharacter);
  1012. callback(mockSecureTokenResponse, nil);
  1013. });
  1014. });
  1015. [user getIDTokenResultForcingRefresh:YES
  1016. completion:^(FIRAuthTokenResult *_Nullable tokenResult,
  1017. NSError *_Nullable error) {
  1018. XCTAssertTrue([NSThread isMainThread]);
  1019. XCTAssertNil(error);
  1020. XCTAssertEqualObjects(tokenResult.token, kAccessTokenWithBase64URLCharacter);
  1021. XCTAssertTrue(tokenResult.issuedAtDate &&
  1022. [tokenResult.issuedAtDate isKindOfClass:[NSDate class]]);
  1023. XCTAssertTrue(tokenResult.authDate && [tokenResult.authDate isKindOfClass:[NSDate class]]);
  1024. XCTAssertTrue(tokenResult.expirationDate &&
  1025. [tokenResult.expirationDate isKindOfClass:[NSDate class]]);
  1026. XCTAssertTrue(tokenResult.claims && [tokenResult.claims isKindOfClass:[NSDictionary class]]);
  1027. NSDictionary *claims = tokenResult.claims;
  1028. XCTAssertEqualObjects(claims[@"email"], kbase64URLEncodedEmail);
  1029. XCTAssertEqualObjects(claims[@"aud"], kbase64URLEncodedAUD);
  1030. [expectation fulfill];
  1031. }];
  1032. }];
  1033. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1034. OCMVerifyAll(_mockBackend);
  1035. }
  1036. /** @fn testGetIDTokenResultForcingRefreshFailure
  1037. @brief Tests the flow of a failed @c getIDTokenResultForcingRefresh:completion: call.
  1038. */
  1039. - (void)testGetIDTokenResultForcingRefreshFailure {
  1040. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1041. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1042. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1043. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1044. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1045. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1046. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1047. completion:^(FIRUser *user) {
  1048. OCMExpect([_mockBackend secureToken:[OCMArg any] callback:[OCMArg any]])
  1049. .andCallBlock2(^(FIRSecureTokenRequest *_Nullable request,
  1050. FIRSecureTokenResponseCallback callback) {
  1051. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1052. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1053. callback(nil, [FIRAuthErrorUtils networkErrorWithUnderlyingError:[NSError new]]);
  1054. });
  1055. });
  1056. [user getIDTokenResultForcingRefresh:YES
  1057. completion:^(FIRAuthTokenResult *_Nullable tokenResult,
  1058. NSError *_Nullable error) {
  1059. XCTAssertTrue([NSThread isMainThread]);
  1060. XCTAssertNil(tokenResult);
  1061. XCTAssertEqual(error.code, FIRAuthErrorCodeNetworkError);
  1062. [expectation fulfill];
  1063. }];
  1064. }];
  1065. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1066. OCMVerifyAll(_mockBackend);
  1067. }
  1068. /** @fn testReloadFailure
  1069. @brief Tests the flow of a failed @c reloadWithCompletion: call.
  1070. */
  1071. - (void)testReloadFailure {
  1072. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1073. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1074. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1075. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1076. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1077. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1078. completion:^(FIRUser *user) {
  1079. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1080. .andDispatchError2([FIRAuthErrorUtils quotaExceededErrorWithMessage:nil]);
  1081. [user reloadWithCompletion:^(NSError *_Nullable error) {
  1082. XCTAssertTrue([NSThread isMainThread]);
  1083. XCTAssertEqual(error.code, FIRAuthErrorCodeQuotaExceeded);
  1084. XCTAssertEqual([FIRAuth auth].currentUser, user);
  1085. [expectation fulfill];
  1086. }];
  1087. }];
  1088. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1089. OCMVerifyAll(_mockBackend);
  1090. }
  1091. /** @fn testReloadFailureAutoSignOut
  1092. @brief Tests the flow of a failed @c reloadWithCompletion: call that automtatically signs out.
  1093. */
  1094. - (void)testReloadFailureAutoSignOut {
  1095. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1096. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1097. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1098. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1099. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1100. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1101. completion:^(FIRUser *user) {
  1102. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1103. .andDispatchError2([FIRAuthErrorUtils userTokenExpiredErrorWithMessage:nil]);
  1104. [user reloadWithCompletion:^(NSError *_Nullable error) {
  1105. XCTAssertTrue([NSThread isMainThread]);
  1106. XCTAssertEqual(error.code, FIRAuthErrorCodeUserTokenExpired);
  1107. XCTAssertNil([FIRAuth auth].currentUser);
  1108. [expectation fulfill];
  1109. }];
  1110. }];
  1111. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1112. OCMVerifyAll(_mockBackend);
  1113. }
  1114. /** @fn testReauthenticateSuccess
  1115. @brief Tests the flow of a successful @c reauthenticateWithCredential:completion: call.
  1116. */
  1117. - (void)testReauthenticateSuccess {
  1118. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1119. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1120. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1121. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1122. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1123. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1124. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1125. completion:^(FIRUser *user) {
  1126. OCMExpect([_mockBackend verifyPassword:[OCMArg any] callback:[OCMArg any]])
  1127. .andCallBlock2(^(FIRVerifyPasswordRequest *_Nullable request,
  1128. FIRVerifyPasswordResponseCallback callback) {
  1129. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1130. id mockVeriyPasswordResponse = OCMClassMock([FIRVerifyPasswordResponse class]);
  1131. // New authentication comes back with new access token.
  1132. OCMStub([mockVeriyPasswordResponse IDToken]).andReturn(kNewAccessToken);
  1133. OCMStub([mockVeriyPasswordResponse approximateExpirationDate])
  1134. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  1135. OCMStub([mockVeriyPasswordResponse refreshToken]).andReturn(kRefreshToken);
  1136. callback(mockVeriyPasswordResponse, nil);
  1137. });
  1138. });
  1139. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1140. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  1141. FIRGetAccountInfoResponseCallback callback) {
  1142. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1143. // Verify that the new access token is being used for subsequent requests.
  1144. XCTAssertEqualObjects(request.accessToken, kNewAccessToken);
  1145. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1146. id mockGetAccountInfoResponse = OCMClassMock([FIRGetAccountInfoResponse class]);
  1147. OCMStub([mockGetAccountInfoResponse users]).andReturn(@[ mockGetAccountInfoResponseUser ]);
  1148. callback(mockGetAccountInfoResponse, nil);
  1149. });
  1150. });
  1151. FIRAuthCredential *emailCredential =
  1152. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  1153. [user reauthenticateWithCredential:emailCredential completion:^(NSError *_Nullable error) {
  1154. XCTAssertTrue([NSThread isMainThread]);
  1155. XCTAssertNil(error);
  1156. // Verify that the current user is unchanged.
  1157. XCTAssertEqual([FIRAuth auth].currentUser, user);
  1158. [expectation fulfill];
  1159. }];
  1160. }];
  1161. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1162. OCMVerifyAll(_mockBackend);
  1163. }
  1164. /** @fn testReauthenticateAndRetrieveDataSuccess
  1165. @brief Tests the flow of a successful @c reauthenticateAndRetrieveDataWithCredential:completion:
  1166. call.
  1167. */
  1168. - (void)testReauthenticateAndRetrieveDataSuccess {
  1169. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1170. federatedID:kGoogleID
  1171. displayName:kGoogleDisplayName
  1172. profile:[[self class] googleProfile]
  1173. providerIDToken:kGoogleIDToken
  1174. providerAccessToken:kGoogleAccessToken];
  1175. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1176. [[FIRAuth auth] signOut:NULL];
  1177. FIRAuthCredential *googleCredential =
  1178. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1179. [[FIRAuth auth] signInAndRetrieveDataWithCredential:googleCredential
  1180. completion:^(FIRAuthDataResult *_Nullable authResult,
  1181. NSError *_Nullable error) {
  1182. XCTAssertTrue([NSThread isMainThread]);
  1183. [self assertUserGoogle:authResult.user];
  1184. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1185. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1186. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRGoogleAuthProviderID);
  1187. XCTAssertNil(error);
  1188. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1189. federatedID:kGoogleID
  1190. displayName:kGoogleDisplayName
  1191. profile:[[self class] googleProfile]
  1192. providerIDToken:kGoogleIDToken
  1193. providerAccessToken:kGoogleAccessToken];
  1194. FIRAuthCredential *reauthenticateGoogleCredential =
  1195. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1196. [authResult.user
  1197. reauthenticateAndRetrieveDataWithCredential:reauthenticateGoogleCredential
  1198. completion:^(FIRAuthDataResult *_Nullable
  1199. reauthenticateAuthResult,
  1200. NSError *_Nullable error) {
  1201. XCTAssertTrue([NSThread isMainThread]);
  1202. XCTAssertNil(error);
  1203. // Verify that the current user is unchanged.
  1204. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1205. // Verify that the current user and reauthenticated user are not same pointers.
  1206. XCTAssertNotEqualObjects(authResult.user, reauthenticateAuthResult.user);
  1207. // Verify that anyway the current user and reauthenticated user have same IDs.
  1208. XCTAssertEqualObjects(authResult.user.uid, reauthenticateAuthResult.user.uid);
  1209. XCTAssertEqualObjects(authResult.user.displayName, reauthenticateAuthResult.user.displayName);
  1210. XCTAssertEqualObjects(reauthenticateAuthResult.additionalUserInfo.profile,
  1211. [[self class] googleProfile]);
  1212. XCTAssertEqualObjects(reauthenticateAuthResult.additionalUserInfo.username, kUserName);
  1213. XCTAssertEqualObjects(reauthenticateAuthResult.additionalUserInfo.providerID,
  1214. FIRGoogleAuthProviderID);
  1215. [expectation fulfill];
  1216. }];
  1217. }];
  1218. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1219. [self assertUserGoogle:[FIRAuth auth].currentUser];
  1220. OCMVerifyAll(_mockBackend);
  1221. }
  1222. /** @fn testReauthenticateFailure
  1223. @brief Tests the flow of a failed @c reauthenticateWithCredential:completion: call.
  1224. */
  1225. - (void)testReauthenticateFailure {
  1226. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1227. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1228. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1229. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1230. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1231. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1232. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1233. completion:^(FIRUser *user) {
  1234. OCMExpect([_mockBackend verifyPassword:[OCMArg any] callback:[OCMArg any]])
  1235. .andCallBlock2(^(FIRVerifyPasswordRequest *_Nullable request,
  1236. FIRVerifyPasswordResponseCallback callback) {
  1237. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1238. id mockVeriyPasswordResponse = OCMClassMock([FIRVerifyPasswordResponse class]);
  1239. OCMStub([mockVeriyPasswordResponse IDToken]).andReturn(kNewAccessToken);
  1240. OCMStub([mockVeriyPasswordResponse approximateExpirationDate])
  1241. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  1242. OCMStub([mockVeriyPasswordResponse refreshToken]).andReturn(kRefreshToken);
  1243. callback(mockVeriyPasswordResponse, nil);
  1244. });
  1245. });
  1246. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1247. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  1248. FIRGetAccountInfoResponseCallback callback) {
  1249. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1250. id mockGetAccountInfoResponseUserNew = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1251. // The newly-signed-in user has a different ID.
  1252. OCMStub([mockGetAccountInfoResponseUserNew localID]).andReturn(kAnotherLocalID);
  1253. OCMStub([mockGetAccountInfoResponseUserNew email]).andReturn(kNewEmail);
  1254. OCMStub([mockGetAccountInfoResponseUserNew displayName]).andReturn(kNewDisplayName);
  1255. OCMStub([mockGetAccountInfoResponseUserNew passwordHash]).andReturn(kPasswordHash);
  1256. id mockGetAccountInfoResponse = OCMClassMock([FIRGetAccountInfoResponse class]);
  1257. OCMStub([mockGetAccountInfoResponse users])
  1258. .andReturn(@[ mockGetAccountInfoResponseUserNew ]);
  1259. callback(mockGetAccountInfoResponse, nil);
  1260. });
  1261. });
  1262. FIRAuthCredential *emailCredential =
  1263. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  1264. [user reauthenticateWithCredential:emailCredential completion:^(NSError *_Nullable error) {
  1265. XCTAssertTrue([NSThread isMainThread]);
  1266. // Verify user mismatch error.
  1267. XCTAssertEqual(error.code, FIRAuthErrorCodeUserMismatch);
  1268. // Verify that the current user is unchanged.
  1269. XCTAssertEqual([FIRAuth auth].currentUser, user);
  1270. [expectation fulfill];
  1271. }];
  1272. }];
  1273. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1274. OCMVerifyAll(_mockBackend);
  1275. }
  1276. /** @fn testReauthenticateUserMismatchFailure
  1277. @brief Tests the flow of a failed @c reauthenticateWithCredential:completion: call due to trying
  1278. to reauthenticate a user that does not exist.
  1279. */
  1280. - (void)testReauthenticateUserMismatchFailure {
  1281. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1282. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1283. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1284. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  1285. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1286. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1287. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  1288. completion:^(FIRUser *user) {
  1289. OCMExpect([_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  1290. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  1291. FIRVerifyAssertionResponseCallback callback) {
  1292. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1293. callback(nil, [FIRAuthErrorUtils userNotFoundErrorWithMessage:nil]);
  1294. });
  1295. });
  1296. FIRAuthCredential *googleCredential =
  1297. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1298. [user reauthenticateWithCredential:googleCredential completion:^(NSError *_Nullable error) {
  1299. XCTAssertTrue([NSThread isMainThread]);
  1300. // Verify user mismatch error.
  1301. XCTAssertEqual(error.code, FIRAuthErrorCodeUserMismatch);
  1302. // Verify that the current user is unchanged.
  1303. XCTAssertEqual([FIRAuth auth].currentUser, user);
  1304. [expectation fulfill];
  1305. }];
  1306. }];
  1307. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1308. OCMVerifyAll(_mockBackend);
  1309. }
  1310. /** @fn testlinkAndRetrieveDataSuccess
  1311. @brief Tests the flow of a successful @c linkAndRetrieveDataWithCredential:completion:
  1312. call.
  1313. */
  1314. - (void)testlinkAndRetrieveDataSuccess {
  1315. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1316. federatedID:kFacebookID
  1317. displayName:kFacebookDisplayName
  1318. profile:[[self class] googleProfile]
  1319. providerIDToken:kFacebookIDToken
  1320. providerAccessToken:kFacebookAccessToken];
  1321. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1322. [[FIRAuth auth] signOut:NULL];
  1323. FIRAuthCredential *facebookCredential =
  1324. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1325. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1326. completion:^(FIRAuthDataResult *_Nullable authResult,
  1327. NSError *_Nullable error) {
  1328. XCTAssertTrue([NSThread isMainThread]);
  1329. [self assertUserFacebook:authResult.user];
  1330. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1331. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1332. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1333. XCTAssertNil(error);
  1334. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1335. federatedID:kGoogleID
  1336. displayName:kGoogleDisplayName
  1337. profile:[[self class] googleProfile]
  1338. providerIDToken:kGoogleIDToken
  1339. providerAccessToken:kGoogleAccessToken];
  1340. FIRAuthCredential *linkGoogleCredential =
  1341. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1342. [authResult.user linkAndRetrieveDataWithCredential:linkGoogleCredential
  1343. completion:^(FIRAuthDataResult *_Nullable
  1344. linkAuthResult,
  1345. NSError *_Nullable error) {
  1346. XCTAssertTrue([NSThread isMainThread]);
  1347. XCTAssertNil(error);
  1348. // Verify that the current user is unchanged.
  1349. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1350. // Verify that the current user and reauthenticated user are same pointers.
  1351. XCTAssertEqualObjects(authResult.user, linkAuthResult.user);
  1352. // Verify that anyway the current user and linked user have same IDs.
  1353. XCTAssertEqualObjects(authResult.user.uid, linkAuthResult.user.uid);
  1354. XCTAssertEqualObjects(authResult.user.displayName, linkAuthResult.user.displayName);
  1355. XCTAssertEqualObjects(linkAuthResult.additionalUserInfo.profile,
  1356. [[self class] googleProfile]);
  1357. XCTAssertEqualObjects(linkAuthResult.additionalUserInfo.username, kUserName);
  1358. XCTAssertEqualObjects(linkAuthResult.additionalUserInfo.providerID,
  1359. FIRGoogleAuthProviderID);
  1360. [expectation fulfill];
  1361. }];
  1362. }];
  1363. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1364. [self assertUserGoogle:[FIRAuth auth].currentUser];
  1365. OCMVerifyAll(_mockBackend);
  1366. }
  1367. /** @fn testlinkAndRetrieveDataError
  1368. @brief Tests the flow of an unsuccessful @c linkAndRetrieveDataWithCredential:completion:
  1369. call with an error from the backend.
  1370. */
  1371. - (void)testlinkAndRetrieveDataError {
  1372. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1373. federatedID:kFacebookID
  1374. displayName:kFacebookDisplayName
  1375. profile:[[self class] googleProfile]
  1376. providerIDToken:kFacebookIDToken
  1377. providerAccessToken:kFacebookAccessToken];
  1378. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1379. [[FIRAuth auth] signOut:NULL];
  1380. FIRAuthCredential *facebookCredential =
  1381. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1382. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1383. completion:^(FIRAuthDataResult *_Nullable authResult,
  1384. NSError *_Nullable error) {
  1385. XCTAssertTrue([NSThread isMainThread]);
  1386. [self assertUserFacebook:authResult.user];
  1387. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1388. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1389. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1390. XCTAssertNil(error);
  1391. OCMExpect([_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  1392. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  1393. FIRVerifyAssertionResponseCallback callback) {
  1394. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1395. callback(nil,
  1396. [FIRAuthErrorUtils accountExistsWithDifferentCredentialErrorWithEmail:kEmail]);
  1397. });
  1398. });
  1399. FIRAuthCredential *linkGoogleCredential =
  1400. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1401. [authResult.user linkAndRetrieveDataWithCredential:linkGoogleCredential
  1402. completion:^(FIRAuthDataResult *_Nullable
  1403. linkAuthResult,
  1404. NSError *_Nullable error) {
  1405. XCTAssertTrue([NSThread isMainThread]);
  1406. XCTAssertNil(linkAuthResult);
  1407. XCTAssertEqual(error.code, FIRAuthErrorCodeAccountExistsWithDifferentCredential);
  1408. XCTAssertEqual(error.userInfo[FIRAuthErrorUserInfoEmailKey], kEmail);
  1409. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1410. [expectation fulfill];
  1411. }];
  1412. }];
  1413. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1414. OCMVerifyAll(_mockBackend);
  1415. }
  1416. /** @fn testlinkAndRetrieveDataProviderAlreadyLinked
  1417. @brief Tests the flow of an unsuccessful @c linkAndRetrieveDataWithCredential:completion:
  1418. call with FIRAuthErrorCodeProviderAlreadyLinked, which is a client side error.
  1419. */
  1420. - (void)testlinkAndRetrieveDataProviderAlreadyLinked {
  1421. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1422. federatedID:kFacebookID
  1423. displayName:kFacebookDisplayName
  1424. profile:[[self class] googleProfile]
  1425. providerIDToken:kFacebookIDToken
  1426. providerAccessToken:kFacebookAccessToken];
  1427. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1428. [[FIRAuth auth] signOut:NULL];
  1429. FIRAuthCredential *facebookCredential =
  1430. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1431. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1432. completion:^(FIRAuthDataResult *_Nullable authResult,
  1433. NSError *_Nullable error) {
  1434. XCTAssertTrue([NSThread isMainThread]);
  1435. [self assertUserFacebook:authResult.user];
  1436. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1437. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1438. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1439. XCTAssertNil(error);
  1440. FIRAuthCredential *linkFacebookCredential =
  1441. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1442. [authResult.user linkAndRetrieveDataWithCredential:linkFacebookCredential
  1443. completion:^(FIRAuthDataResult *_Nullable
  1444. linkAuthResult,
  1445. NSError *_Nullable error) {
  1446. XCTAssertTrue([NSThread isMainThread]);
  1447. XCTAssertNil(linkAuthResult);
  1448. XCTAssertEqual(error.code, FIRAuthErrorCodeProviderAlreadyLinked);
  1449. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1450. [expectation fulfill];
  1451. }];
  1452. }];
  1453. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1454. OCMVerifyAll(_mockBackend);
  1455. }
  1456. /** @fn testlinkAndRetrieveDataErrorAutoSignOut
  1457. @brief Tests the flow of an unsuccessful @c linkAndRetrieveDataWithCredential:completion:
  1458. call that automatically signs out.
  1459. */
  1460. - (void)testlinkAndRetrieveDataErrorAutoSignOut {
  1461. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1462. federatedID:kFacebookID
  1463. displayName:kFacebookDisplayName
  1464. profile:[[self class] googleProfile]
  1465. providerIDToken:kFacebookIDToken
  1466. providerAccessToken:kFacebookAccessToken];
  1467. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1468. [[FIRAuth auth] signOut:NULL];
  1469. FIRAuthCredential *facebookCredential =
  1470. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1471. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1472. completion:^(FIRAuthDataResult *_Nullable authResult,
  1473. NSError *_Nullable error) {
  1474. XCTAssertTrue([NSThread isMainThread]);
  1475. [self assertUserFacebook:authResult.user];
  1476. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1477. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1478. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1479. XCTAssertNil(error);
  1480. OCMExpect([_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  1481. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  1482. FIRVerifyAssertionResponseCallback callback) {
  1483. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1484. callback(nil, [FIRAuthErrorUtils userDisabledErrorWithMessage:nil]);
  1485. });
  1486. });
  1487. FIRAuthCredential *linkGoogleCredential =
  1488. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1489. [authResult.user linkAndRetrieveDataWithCredential:linkGoogleCredential
  1490. completion:^(FIRAuthDataResult *_Nullable
  1491. linkAuthResult,
  1492. NSError *_Nullable error) {
  1493. XCTAssertTrue([NSThread isMainThread]);
  1494. XCTAssertNil(linkAuthResult);
  1495. XCTAssertEqual(error.code, FIRAuthErrorCodeUserDisabled);
  1496. XCTAssertNil([FIRAuth auth].currentUser);
  1497. [expectation fulfill];
  1498. }];
  1499. }];
  1500. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1501. OCMVerifyAll(_mockBackend);
  1502. }
  1503. /** @fn testLinkingAnonymousAccountsUpdatesIsAnonymous
  1504. @brief Tests the flow of a successful @c linkAndRetrieveDataWithCredential:completion:
  1505. invocation for email credential.
  1506. */
  1507. - (void)testLinkingAnonymousAccountsUpdatesIsAnonymous {
  1508. FIRAuthCredential *linkEmailCredential =
  1509. [FIREmailAuthProvider credentialWithEmail:kEmail
  1510. link:@"https://google.com?oobCode=aCode&mode=signIn"];
  1511. id (^mockUserInfoWithDisplayName)(NSString *, BOOL) = ^(NSString *displayName,
  1512. BOOL hasProviders) {
  1513. NSArray *providers = hasProviders ? @[ @{
  1514. @"providerId": FIREmailAuthProviderID,
  1515. @"email": kEmail
  1516. } ] : @[];
  1517. FIRGetAccountInfoResponseUser *responseUser =
  1518. [[FIRGetAccountInfoResponseUser alloc] initWithDictionary:@{
  1519. @"providerUserInfo": providers,
  1520. @"localId": kLocalID,
  1521. @"displayName": displayName,
  1522. @"email": kEmail
  1523. }];
  1524. return responseUser;
  1525. };
  1526. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1527. id userInfoResponse = mockUserInfoWithDisplayName(kGoogleDisplayName, NO);
  1528. [self signInAnonymouslyWithMockGetAccountInfoResponse:userInfoResponse
  1529. completion:^(FIRUser *user) {
  1530. // Pretend that the display name and providers on the server have been updated.
  1531. // Get account info is expected to be invoked twice.
  1532. id updatedMockUser = mockUserInfoWithDisplayName(kNewDisplayName, YES);
  1533. [self expectGetAccountInfoWithMockUserInfoResponse:updatedMockUser];
  1534. [self expectGetAccountInfoWithMockUserInfoResponse:updatedMockUser];
  1535. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1536. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  1537. FIRSetAccountInfoResponseCallback callback) {
  1538. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  1539. OCMStub([mockSetAccountInfoResponse email]).andReturn(kNewEmail);
  1540. OCMStub([mockSetAccountInfoResponse displayName]).andReturn(kNewDisplayName);
  1541. callback(mockSetAccountInfoResponse, nil);
  1542. });
  1543. XCTAssertTrue(user.isAnonymous);
  1544. [user linkAndRetrieveDataWithCredential:linkEmailCredential
  1545. completion:^(FIRAuthDataResult *_Nullable linkAuthResult,
  1546. NSError *_Nullable error) {
  1547. XCTAssertTrue([NSThread isMainThread]);
  1548. XCTAssertNil(error);
  1549. XCTAssertEqualObjects(user.email, kEmail);
  1550. XCTAssertFalse(user.isAnonymous);
  1551. [expectation fulfill];
  1552. }];
  1553. }];
  1554. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1555. OCMVerifyAll(_mockBackend);
  1556. }
  1557. /** @fn testlinkEmailAndRetrieveDataSuccess
  1558. @brief Tests the flow of a successful @c linkAndRetrieveDataWithCredential:completion:
  1559. invocation for email credential.
  1560. */
  1561. - (void)testlinkEmailAndRetrieveDataSuccess {
  1562. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1563. federatedID:kFacebookID
  1564. displayName:kFacebookDisplayName
  1565. profile:[[self class] googleProfile]
  1566. providerIDToken:kFacebookIDToken
  1567. providerAccessToken:kFacebookAccessToken];
  1568. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1569. [[FIRAuth auth] signOut:NULL];
  1570. FIRAuthCredential *facebookCredential =
  1571. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1572. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1573. completion:^(FIRAuthDataResult *_Nullable authResult,
  1574. NSError *_Nullable error) {
  1575. XCTAssertTrue([NSThread isMainThread]);
  1576. [self assertUserFacebook:authResult.user];
  1577. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1578. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1579. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1580. XCTAssertNil(error);
  1581. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1582. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1583. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1584. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kEmailDisplayName);
  1585. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1586. // Get account info is expected to be invoked twice.
  1587. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  1588. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  1589. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1590. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  1591. FIRSetAccountInfoResponseCallback callback) {
  1592. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1593. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  1594. XCTAssertEqualObjects(request.password, kFakePassword);
  1595. XCTAssertNil(request.localID);
  1596. XCTAssertNil(request.displayName);
  1597. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1598. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  1599. callback(mockSetAccountInfoResponse, nil);
  1600. });
  1601. });
  1602. FIRAuthCredential *linkEmailCredential =
  1603. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  1604. [authResult.user linkAndRetrieveDataWithCredential:linkEmailCredential
  1605. completion:^(FIRAuthDataResult *_Nullable
  1606. linkAuthResult,
  1607. NSError *_Nullable error) {
  1608. XCTAssertTrue([NSThread isMainThread]);
  1609. XCTAssertNil(error);
  1610. XCTAssertEqualObjects(linkAuthResult.user.email, kEmail);
  1611. XCTAssertEqualObjects(linkAuthResult.user.displayName, kEmailDisplayName);
  1612. [expectation fulfill];
  1613. }];
  1614. }];
  1615. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1616. OCMVerifyAll(_mockBackend);
  1617. }
  1618. /** @fn testlinkEmailProviderAlreadyLinkedError
  1619. @brief Tests the flow of an unsuccessful @c linkAndRetrieveDataWithCredential:completion:
  1620. invocation for email credential and FIRAuthErrorCodeProviderAlreadyLinked which is a client
  1621. side error.
  1622. */
  1623. - (void)testlinkEmailProviderAlreadyLinkedError {
  1624. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1625. federatedID:kFacebookID
  1626. displayName:kFacebookDisplayName
  1627. profile:[[self class] googleProfile]
  1628. providerIDToken:kFacebookIDToken
  1629. providerAccessToken:kFacebookAccessToken];
  1630. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1631. [[FIRAuth auth] signOut:NULL];
  1632. FIRAuthCredential *facebookCredential =
  1633. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1634. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1635. completion:^(FIRAuthDataResult *_Nullable authResult,
  1636. NSError *_Nullable error) {
  1637. XCTAssertTrue([NSThread isMainThread]);
  1638. [self assertUserFacebook:authResult.user];
  1639. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1640. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1641. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1642. XCTAssertNil(error);
  1643. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1644. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1645. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1646. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kEmailDisplayName);
  1647. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1648. // Get account info is expected to be invoked twice.
  1649. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  1650. [self expectGetAccountInfoWithMockUserInfoResponse:mockGetAccountInfoResponseUser];
  1651. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1652. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  1653. FIRSetAccountInfoResponseCallback callback) {
  1654. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1655. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  1656. XCTAssertEqualObjects(request.password, kFakePassword);
  1657. XCTAssertNil(request.localID);
  1658. XCTAssertNil(request.displayName);
  1659. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1660. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  1661. callback(mockSetAccountInfoResponse, nil);
  1662. });
  1663. });
  1664. FIRAuthCredential *linkEmailCredential =
  1665. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  1666. [authResult.user linkAndRetrieveDataWithCredential:linkEmailCredential
  1667. completion:^(FIRAuthDataResult *_Nullable
  1668. linkAuthResult,
  1669. NSError *_Nullable error) {
  1670. XCTAssertNil(error);
  1671. XCTAssertEqualObjects(linkAuthResult.user.email, kEmail);
  1672. XCTAssertEqualObjects(linkAuthResult.user.displayName, kEmailDisplayName);
  1673. // Try linking same credential a second time to trigger client side error.
  1674. [authResult.user linkAndRetrieveDataWithCredential:linkEmailCredential
  1675. completion:^(FIRAuthDataResult *_Nullable
  1676. linkAuthResult,
  1677. NSError *_Nullable error) {
  1678. XCTAssertTrue([NSThread isMainThread]);
  1679. XCTAssertNil(linkAuthResult);
  1680. XCTAssertEqual(error.code, FIRAuthErrorCodeProviderAlreadyLinked);
  1681. [expectation fulfill];
  1682. }];
  1683. }];
  1684. }];
  1685. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1686. OCMVerifyAll(_mockBackend);
  1687. }
  1688. /** @fn testlinkEmailAndRetrieveDataError
  1689. @brief Tests the flow of an unsuccessful @c linkAndRetrieveDataWithCredential:completion:
  1690. invocation for email credential and an error from the backend.
  1691. */
  1692. - (void)testlinkEmailAndRetrieveDataError {
  1693. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1694. federatedID:kFacebookID
  1695. displayName:kFacebookDisplayName
  1696. profile:[[self class] googleProfile]
  1697. providerIDToken:kFacebookIDToken
  1698. providerAccessToken:kFacebookAccessToken];
  1699. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1700. [[FIRAuth auth] signOut:NULL];
  1701. FIRAuthCredential *facebookCredential =
  1702. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1703. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1704. completion:^(FIRAuthDataResult *_Nullable authResult,
  1705. NSError *_Nullable error) {
  1706. XCTAssertTrue([NSThread isMainThread]);
  1707. [self assertUserFacebook:authResult.user];
  1708. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1709. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1710. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1711. XCTAssertNil(error);
  1712. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1713. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  1714. FIRGetAccountInfoResponseCallback callback) {
  1715. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1716. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  1717. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1718. callback(nil, [FIRAuthErrorUtils tooManyRequestsErrorWithMessage:nil]);
  1719. });
  1720. });
  1721. FIRAuthCredential *linkEmailCredential =
  1722. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  1723. [authResult.user linkAndRetrieveDataWithCredential:linkEmailCredential
  1724. completion:^(FIRAuthDataResult *_Nullable
  1725. linkAuthResult,
  1726. NSError *_Nullable error) {
  1727. XCTAssertTrue([NSThread isMainThread]);
  1728. XCTAssertNil(linkAuthResult);
  1729. XCTAssertEqual(error.code, FIRAuthErrorCodeTooManyRequests);
  1730. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1731. [expectation fulfill];
  1732. }];
  1733. }];
  1734. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1735. OCMVerifyAll(_mockBackend);
  1736. }
  1737. /** @fn testlinkEmailAndRetrieveDataError
  1738. @brief Tests the flow of an unsuccessful @c linkAndRetrieveDataWithCredential:completion:
  1739. invocation that automatically signs out.
  1740. */
  1741. - (void)testlinkEmailAndRetrieveDataErrorAutoSignOut {
  1742. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1743. federatedID:kFacebookID
  1744. displayName:kFacebookDisplayName
  1745. profile:[[self class] googleProfile]
  1746. providerIDToken:kFacebookIDToken
  1747. providerAccessToken:kFacebookAccessToken];
  1748. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1749. [[FIRAuth auth] signOut:NULL];
  1750. FIRAuthCredential *facebookCredential =
  1751. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1752. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1753. completion:^(FIRAuthDataResult *_Nullable authResult,
  1754. NSError *_Nullable error) {
  1755. XCTAssertTrue([NSThread isMainThread]);
  1756. [self assertUserFacebook:authResult.user];
  1757. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1758. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1759. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1760. XCTAssertNil(error);
  1761. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1762. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  1763. FIRGetAccountInfoResponseCallback callback) {
  1764. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1765. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  1766. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1767. callback(nil, [FIRAuthErrorUtils userTokenExpiredErrorWithMessage:nil]);
  1768. });
  1769. });
  1770. FIRAuthCredential *linkEmailCredential =
  1771. [FIREmailAuthProvider credentialWithEmail:kEmail password:kFakePassword];
  1772. [authResult.user linkAndRetrieveDataWithCredential:linkEmailCredential
  1773. completion:^(FIRAuthDataResult *_Nullable
  1774. linkAuthResult,
  1775. NSError *_Nullable error) {
  1776. XCTAssertTrue([NSThread isMainThread]);
  1777. XCTAssertNil(linkAuthResult);
  1778. XCTAssertEqual(error.code, FIRAuthErrorCodeUserTokenExpired);
  1779. XCTAssertNil([FIRAuth auth].currentUser);
  1780. [expectation fulfill];
  1781. }];
  1782. }];
  1783. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1784. OCMVerifyAll(_mockBackend);
  1785. }
  1786. /** @fn testlinkCredentialSuccess
  1787. @brief Tests the flow of a successful @c linkWithCredential:completion: call, without additional
  1788. IDP data.
  1789. */
  1790. - (void)testlinkCredentialSuccess {
  1791. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1792. federatedID:kFacebookID
  1793. displayName:kFacebookDisplayName
  1794. profile:[[self class] googleProfile]
  1795. providerIDToken:kFacebookIDToken
  1796. providerAccessToken:kFacebookAccessToken];
  1797. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1798. [[FIRAuth auth] signOut:NULL];
  1799. FIRAuthCredential *facebookCredential =
  1800. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1801. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1802. completion:^(FIRAuthDataResult *_Nullable authResult,
  1803. NSError *_Nullable error) {
  1804. XCTAssertTrue([NSThread isMainThread]);
  1805. [self assertUserFacebook:authResult.user];
  1806. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1807. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1808. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1809. XCTAssertNil(error);
  1810. [self expectVerifyAssertionRequest:FIRGoogleAuthProviderID
  1811. federatedID:kGoogleID
  1812. displayName:kGoogleDisplayName
  1813. profile:[[self class] googleProfile]
  1814. providerIDToken:kGoogleIDToken
  1815. providerAccessToken:kGoogleAccessToken];
  1816. FIRAuthCredential *linkGoogleCredential =
  1817. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1818. [authResult.user linkWithCredential:linkGoogleCredential
  1819. completion:^(FIRUser *_Nullable user,
  1820. NSError *_Nullable error) {
  1821. XCTAssertNil(error);
  1822. id<FIRUserInfo> userInfo = user.providerData.firstObject;
  1823. XCTAssertEqual(userInfo.providerID, FIRGoogleAuthProviderID);
  1824. XCTAssertEqual([FIRAuth auth].currentUser, authResult.user);
  1825. [expectation fulfill];
  1826. }];
  1827. }];
  1828. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1829. [self assertUserGoogle:[FIRAuth auth].currentUser];
  1830. OCMVerifyAll(_mockBackend);
  1831. }
  1832. /** @fn testlinkCredentialError
  1833. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion: call, with an error
  1834. from the backend.
  1835. */
  1836. - (void)testlinkCredentialError {
  1837. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1838. federatedID:kFacebookID
  1839. displayName:kFacebookDisplayName
  1840. profile:[[self class] googleProfile]
  1841. providerIDToken:kFacebookIDToken
  1842. providerAccessToken:kFacebookAccessToken];
  1843. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1844. [[FIRAuth auth] signOut:NULL];
  1845. FIRAuthCredential *facebookCredential =
  1846. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1847. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1848. completion:^(FIRAuthDataResult *_Nullable authResult,
  1849. NSError *_Nullable error) {
  1850. XCTAssertTrue([NSThread isMainThread]);
  1851. [self assertUserFacebook:authResult.user];
  1852. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1853. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1854. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1855. XCTAssertNil(error);
  1856. OCMExpect([_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  1857. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  1858. FIRVerifyAssertionResponseCallback callback) {
  1859. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1860. callback(nil, [FIRAuthErrorUtils userDisabledErrorWithMessage:nil]);
  1861. });
  1862. });
  1863. FIRAuthCredential *linkGoogleCredential =
  1864. [FIRGoogleAuthProvider credentialWithIDToken:kGoogleIDToken accessToken:kGoogleAccessToken];
  1865. [authResult.user linkWithCredential:linkGoogleCredential
  1866. completion:^(FIRUser *_Nullable user,
  1867. NSError *_Nullable error) {
  1868. XCTAssertNil(user);
  1869. XCTAssertEqual(error.code, FIRAuthErrorCodeUserDisabled);
  1870. [expectation fulfill];
  1871. }];
  1872. }];
  1873. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1874. OCMVerifyAll(_mockBackend);
  1875. }
  1876. /** @fn testlinkCredentialProviderAlreadyLinkedError
  1877. @brief Tests the flow of an unsuccessful @c linkWithCredential:completion: call, with a client
  1878. side error.
  1879. */
  1880. - (void)testlinkCredentialProviderAlreadyLinkedError {
  1881. [self expectVerifyAssertionRequest:FIRFacebookAuthProviderID
  1882. federatedID:kFacebookID
  1883. displayName:kFacebookDisplayName
  1884. profile:[[self class] googleProfile]
  1885. providerIDToken:kFacebookIDToken
  1886. providerAccessToken:kFacebookAccessToken];
  1887. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1888. [[FIRAuth auth] signOut:NULL];
  1889. FIRAuthCredential *facebookCredential =
  1890. [FIRFacebookAuthProvider credentialWithAccessToken:kFacebookAccessToken];
  1891. [[FIRAuth auth] signInAndRetrieveDataWithCredential:facebookCredential
  1892. completion:^(FIRAuthDataResult *_Nullable authResult,
  1893. NSError *_Nullable error) {
  1894. XCTAssertTrue([NSThread isMainThread]);
  1895. [self assertUserFacebook:authResult.user];
  1896. XCTAssertEqualObjects(authResult.additionalUserInfo.profile, [[self class] googleProfile]);
  1897. XCTAssertEqualObjects(authResult.additionalUserInfo.username, kUserName);
  1898. XCTAssertEqualObjects(authResult.additionalUserInfo.providerID, FIRFacebookAuthProviderID);
  1899. XCTAssertNil(error);
  1900. FIRAuthCredential *linkFacebookCredential =
  1901. [FIRFacebookAuthProvider credentialWithAccessToken:kGoogleAccessToken];
  1902. [authResult.user linkWithCredential:linkFacebookCredential
  1903. completion:^(FIRUser *_Nullable user,
  1904. NSError *_Nullable error) {
  1905. XCTAssertNil(user);
  1906. XCTAssertEqual(error.code, FIRAuthErrorCodeProviderAlreadyLinked);
  1907. [expectation fulfill];
  1908. }];
  1909. }];
  1910. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1911. OCMVerifyAll(_mockBackend);
  1912. }
  1913. #if TARGET_OS_IOS
  1914. /** @fn testlinkPhoneAuthCredentialSuccess
  1915. @brief Tests the flow of a successful @c linkAndRetrieveDataWithCredential:completion:
  1916. call using a phoneAuthCredential.
  1917. */
  1918. - (void)testlinkPhoneAuthCredentialSuccess {
  1919. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  1920. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1921. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1922. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1923. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1924. if (phoneNumber.length) {
  1925. NSDictionary *userInfoDictionary = @{ @"providerId" : FIRPhoneAuthProviderID };
  1926. FIRGetAccountInfoResponseProviderUserInfo *userInfo =
  1927. [[FIRGetAccountInfoResponseProviderUserInfo alloc] initWithDictionary:userInfoDictionary];
  1928. OCMStub([mockGetAccountInfoResponseUser providerUserInfo]).andReturn(@[ userInfo ]);
  1929. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  1930. }
  1931. return mockGetAccountInfoResponseUser;
  1932. };
  1933. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1934. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  1935. [self signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  1936. completion:^(FIRUser *user) {
  1937. [self expectVerifyPhoneNumberRequestWithPhoneNumber:kPhoneNumber isLinkOperation:YES error:nil];
  1938. id userInfoResponseUpdate = mockUserInfoWithPhoneNumber(kPhoneNumber);
  1939. [self expectGetAccountInfoWithMockUserInfoResponse:userInfoResponseUpdate];
  1940. FIRPhoneAuthCredential *credential =
  1941. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  1942. verificationCode:kVerificationCode];
  1943. [user linkAndRetrieveDataWithCredential:credential
  1944. completion:^(FIRAuthDataResult *_Nullable
  1945. linkAuthResult,
  1946. NSError *_Nullable error) {
  1947. XCTAssertNil(error);
  1948. XCTAssertEqualObjects([FIRAuth auth].currentUser.providerData.firstObject.providerID,
  1949. FIRPhoneAuthProviderID);
  1950. XCTAssertEqualObjects([FIRAuth auth].currentUser.phoneNumber, kPhoneNumber);
  1951. [expectation fulfill];
  1952. }];
  1953. }];
  1954. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  1955. OCMVerifyAll(_mockBackend);
  1956. }
  1957. /** @fn testUnlinkPhoneAuthCredentialSuccess
  1958. @brief Tests the flow of a successful @c unlinkFromProvider:completion: call using a
  1959. @c FIRPhoneAuthProvider.
  1960. */
  1961. - (void)testUnlinkPhoneAuthCredentialSuccess {
  1962. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  1963. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  1964. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  1965. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  1966. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  1967. if (phoneNumber.length) {
  1968. NSDictionary *userInfoDictionary = @{ @"providerId" : FIRPhoneAuthProviderID };
  1969. FIRGetAccountInfoResponseProviderUserInfo *userInfo =
  1970. [[FIRGetAccountInfoResponseProviderUserInfo alloc] initWithDictionary:userInfoDictionary];
  1971. OCMStub([mockGetAccountInfoResponseUser providerUserInfo]).andReturn(@[ userInfo ]);
  1972. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  1973. }
  1974. return mockGetAccountInfoResponseUser;
  1975. };
  1976. OCMExpect([_mockBackend setAccountInfo:[OCMArg any] callback:[OCMArg any]])
  1977. .andCallBlock2(^(FIRSetAccountInfoRequest *_Nullable request,
  1978. FIRSetAccountInfoResponseCallback callback) {
  1979. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  1980. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  1981. XCTAssertNotNil(request.deleteProviders);
  1982. XCTAssertNil(request.email);
  1983. XCTAssertNil(request.localID);
  1984. XCTAssertNil(request.displayName);
  1985. XCTAssertNil(request.photoURL);
  1986. XCTAssertNil(request.password);
  1987. XCTAssertNil(request.providers);
  1988. XCTAssertNil(request.deleteAttributes);
  1989. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  1990. id mockSetAccountInfoResponse = OCMClassMock([FIRSetAccountInfoResponse class]);
  1991. callback(mockSetAccountInfoResponse, nil);
  1992. });
  1993. });
  1994. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  1995. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  1996. [self signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  1997. completion:^(FIRUser *user) {
  1998. [self expectVerifyPhoneNumberRequestWithPhoneNumber:kPhoneNumber isLinkOperation:YES error:nil];
  1999. id userInfoResponseUpdate = mockUserInfoWithPhoneNumber(kPhoneNumber);
  2000. [self expectGetAccountInfoWithMockUserInfoResponse:userInfoResponseUpdate];
  2001. FIRPhoneAuthCredential *credential =
  2002. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  2003. verificationCode:kVerificationCode];
  2004. // Link phone credential.
  2005. [user linkAndRetrieveDataWithCredential:credential
  2006. completion:^(FIRAuthDataResult *_Nullable
  2007. linkAuthResult,
  2008. NSError *_Nullable error) {
  2009. XCTAssertNil(error);
  2010. XCTAssertEqualObjects([FIRAuth auth].currentUser.providerData.firstObject.providerID,
  2011. FIRPhoneAuthProviderID);
  2012. XCTAssertEqualObjects([FIRAuth auth].currentUser.phoneNumber, kPhoneNumber);
  2013. // Immediately unlink the phone auth provider.
  2014. [user unlinkFromProvider:FIRPhoneAuthProviderID
  2015. completion:^(FIRUser *_Nullable user, NSError *_Nullable error) {
  2016. XCTAssertNil(error);
  2017. XCTAssertNil([FIRAuth auth].currentUser.phoneNumber);
  2018. [expectation fulfill];
  2019. }];
  2020. }];
  2021. }];
  2022. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2023. OCMVerifyAll(_mockBackend);
  2024. }
  2025. /** @fn testlinkPhoneAuthCredentialFailure
  2026. @brief Tests the flow of a failed call to @c linkAndRetrieveDataWithCredential:completion: due
  2027. to a phone provider already being linked.
  2028. */
  2029. - (void)testlinkPhoneAuthCredentialFailure {
  2030. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  2031. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2032. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2033. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2034. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2035. if (phoneNumber.length) {
  2036. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  2037. }
  2038. return mockGetAccountInfoResponseUser;
  2039. };
  2040. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2041. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  2042. [self signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  2043. completion:^(FIRUser *user) {
  2044. NSError *error = [FIRAuthErrorUtils providerAlreadyLinkedError];
  2045. [self expectVerifyPhoneNumberRequestWithPhoneNumber:nil isLinkOperation:YES error:error];
  2046. FIRPhoneAuthCredential *credential =
  2047. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  2048. verificationCode:kVerificationCode];
  2049. [user linkAndRetrieveDataWithCredential:credential
  2050. completion:^(FIRAuthDataResult *_Nullable
  2051. linkAuthResult,
  2052. NSError *_Nullable error) {
  2053. XCTAssertEqual(error.code, FIRAuthErrorCodeProviderAlreadyLinked);
  2054. [expectation fulfill];
  2055. }];
  2056. }];
  2057. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2058. OCMVerifyAll(_mockBackend);
  2059. }
  2060. /** @fn testlinkPhoneCredentialAlreadyExistsError
  2061. @brief Tests the flow of @c linkAndRetrieveDataWithCredential:completion:
  2062. call using a phoneAuthCredential and a credential already exisits error. In this case we
  2063. should get a FIRAuthCredential in the error object.
  2064. */
  2065. - (void)testlinkPhoneCredentialAlreadyExistsError {
  2066. id (^mockUserInfoWithPhoneNumber)(NSString *) = ^(NSString *phoneNumber) {
  2067. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2068. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2069. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2070. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2071. if (phoneNumber.length) {
  2072. OCMStub([mockGetAccountInfoResponseUser phoneNumber]).andReturn(phoneNumber);
  2073. }
  2074. return mockGetAccountInfoResponseUser;
  2075. };
  2076. void (^expectVerifyPhoneNumberRequest)(NSString *) = ^(NSString *phoneNumber) {
  2077. OCMExpect([_mockBackend verifyPhoneNumber:[OCMArg any] callback:[OCMArg any]])
  2078. .andCallBlock2(^(FIRVerifyPhoneNumberRequest *_Nullable request,
  2079. FIRVerifyPhoneNumberResponseCallback callback) {
  2080. XCTAssertEqualObjects(request.verificationID, kVerificationID);
  2081. XCTAssertEqualObjects(request.verificationCode, kVerificationCode);
  2082. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2083. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2084. FIRPhoneAuthCredential *credential =
  2085. [[FIRPhoneAuthCredential alloc] initWithTemporaryProof:kTemporaryProof
  2086. phoneNumber:kPhoneNumber
  2087. providerID:FIRPhoneAuthProviderID];
  2088. callback(nil,
  2089. [FIRAuthErrorUtils credentialAlreadyInUseErrorWithMessage:nil
  2090. credential:credential]);
  2091. });
  2092. });
  2093. };
  2094. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2095. id userInfoResponse = mockUserInfoWithPhoneNumber(nil);
  2096. [self signInWithEmailPasswordWithMockUserInfoResponse:userInfoResponse
  2097. completion:^(FIRUser *user) {
  2098. expectVerifyPhoneNumberRequest(kPhoneNumber);
  2099. FIRPhoneAuthCredential *credential =
  2100. [[FIRPhoneAuthProvider provider] credentialWithVerificationID:kVerificationID
  2101. verificationCode:kVerificationCode];
  2102. [user linkAndRetrieveDataWithCredential:credential
  2103. completion:^(FIRAuthDataResult *_Nullable
  2104. linkAuthResult,
  2105. NSError *_Nullable error) {
  2106. XCTAssertNil(linkAuthResult);
  2107. XCTAssertEqual(error.code, FIRAuthErrorCodeCredentialAlreadyInUse);
  2108. FIRPhoneAuthCredential *credential = error.userInfo[FIRAuthUpdatedCredentialKey];
  2109. XCTAssertEqual(credential.temporaryProof, kTemporaryProof);
  2110. XCTAssertEqual(credential.phoneNumber, kPhoneNumber);
  2111. [expectation fulfill];
  2112. }];
  2113. }];
  2114. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2115. OCMVerifyAll(_mockBackend);
  2116. }
  2117. #endif
  2118. #pragma mark - Helpers
  2119. /** @fn getIDTokenResultForcingRefreshSuccess
  2120. @brief Helper for testing the flow of a successful @c
  2121. getIDTokenResultForcingRefresh:completion: call.
  2122. */
  2123. - (void)getIDTokenResultForcingRefreshSuccessWithIDToken:(NSString *)idToken {
  2124. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2125. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2126. OCMStub([mockGetAccountInfoResponseUser email]).andReturn(kEmail);
  2127. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(kGoogleDisplayName);
  2128. OCMStub([mockGetAccountInfoResponseUser passwordHash]).andReturn(kPasswordHash);
  2129. XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
  2130. [self signInWithEmailPasswordWithMockUserInfoResponse:mockGetAccountInfoResponseUser
  2131. completion:^(FIRUser *user) {
  2132. OCMExpect([_mockBackend secureToken:[OCMArg any] callback:[OCMArg any]])
  2133. .andCallBlock2(^(FIRSecureTokenRequest *_Nullable request,
  2134. FIRSecureTokenResponseCallback callback) {
  2135. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2136. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2137. id mockSecureTokenResponse = OCMClassMock([FIRSecureTokenResponse class]);
  2138. OCMStub([mockSecureTokenResponse accessToken]).andReturn(idToken);
  2139. callback(mockSecureTokenResponse, nil);
  2140. });
  2141. });
  2142. [user getIDTokenResultForcingRefresh:YES
  2143. completion:^(FIRAuthTokenResult *_Nullable tokenResult,
  2144. NSError *_Nullable error) {
  2145. XCTAssertTrue([NSThread isMainThread]);
  2146. XCTAssertNil(error);
  2147. XCTAssertEqualObjects(tokenResult.token, idToken);
  2148. XCTAssertTrue(tokenResult.issuedAtDate &&
  2149. [tokenResult.issuedAtDate isKindOfClass:[NSDate class]]);
  2150. XCTAssertTrue(tokenResult.authDate && [tokenResult.authDate isKindOfClass:[NSDate class]]);
  2151. XCTAssertTrue(tokenResult.expirationDate &&
  2152. [tokenResult.expirationDate isKindOfClass:[NSDate class]]);
  2153. XCTAssertTrue(tokenResult.claims && [tokenResult.claims isKindOfClass:[NSDictionary class]]);
  2154. [expectation fulfill];
  2155. }];
  2156. }];
  2157. [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
  2158. OCMVerifyAll(_mockBackend);
  2159. }
  2160. /** @fn signInAnonymouslyWithMockGetAccountInfoResponse:completion:
  2161. @brief Signs in with an anonymous account with mocked backend end calls.
  2162. @param mockUserInfoResponse A mocked FIRGetAccountInfoResponseUser object.
  2163. @param completion The completion block that takes the newly signed-in user as the only
  2164. parameter.
  2165. */
  2166. - (void)signInAnonymouslyWithMockGetAccountInfoResponse:(id)mockUserInfoResponse
  2167. completion:(void (^)(FIRUser *user))completion {
  2168. OCMExpect([_mockBackend signUpNewUser:[OCMArg any] callback:[OCMArg any]])
  2169. .andCallBlock2(^(FIRSignUpNewUserRequest *_Nullable request,
  2170. FIRSignupNewUserCallback callback) {
  2171. id mockSignUpResponse = OCMClassMock([FIRSignUpNewUserResponse class]);
  2172. OCMStub([mockSignUpResponse IDToken]).andReturn(kAccessToken);
  2173. OCMStub([mockSignUpResponse approximateExpirationDate])
  2174. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2175. OCMStub([mockSignUpResponse refreshToken]).andReturn(kRefreshToken);
  2176. callback(mockSignUpResponse, nil);
  2177. });
  2178. [self expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoResponse];
  2179. [[FIRAuth auth] signOut:NULL];
  2180. [[FIRAuth auth] signInAnonymouslyWithCompletion:^(FIRAuthDataResult *_Nullable result,
  2181. NSError *_Nullable error) {
  2182. XCTAssertNotNil(result.user);
  2183. XCTAssertNil(error);
  2184. completion(result.user);
  2185. }];
  2186. }
  2187. /** @fn signInWithEmailPasswordWithMockGetAccountInfoResponse:completion:
  2188. @brief Signs in with an email and password account with mocked backend end calls.
  2189. @param mockUserInfoResponse A mocked FIRGetAccountInfoResponseUser object.
  2190. @param completion The completion block that takes the newly signed-in user as the only
  2191. parameter.
  2192. */
  2193. - (void)signInWithEmailPasswordWithMockUserInfoResponse:(id)mockUserInfoResponse
  2194. completion:(void (^)(FIRUser *user))completion {
  2195. OCMExpect([_mockBackend verifyPassword:[OCMArg any] callback:[OCMArg any]])
  2196. .andCallBlock2(^(FIRVerifyPasswordRequest *_Nullable request,
  2197. FIRVerifyPasswordResponseCallback callback) {
  2198. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2199. id mockVeriyPasswordResponse = OCMClassMock([FIRVerifyPasswordResponse class]);
  2200. OCMStub([mockVeriyPasswordResponse IDToken]).andReturn(kAccessToken);
  2201. OCMStub([mockVeriyPasswordResponse approximateExpirationDate])
  2202. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2203. OCMStub([mockVeriyPasswordResponse refreshToken]).andReturn(kRefreshToken);
  2204. callback(mockVeriyPasswordResponse, nil);
  2205. });
  2206. });
  2207. [self expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoResponse];
  2208. [[FIRAuth auth] signOut:NULL];
  2209. [[FIRAuth auth] signInWithEmail:kEmail
  2210. password:kFakePassword
  2211. completion:^(FIRAuthDataResult *_Nullable result, NSError *_Nullable error) {
  2212. XCTAssertNotNil(result.user);
  2213. XCTAssertNil(error);
  2214. completion(result.user);
  2215. }];
  2216. }
  2217. /** @fn signInWithEmailLinkWithMockGetAccountInfoResponse:completion:
  2218. @brief Signs in with an email link auth account with mocked backend end calls.
  2219. @param mockUserInfoResponse A mocked FIRGetAccountInfoResponseUser object.
  2220. @param completion The completion block that takes the newly signed-in user as the only
  2221. parameter.
  2222. */
  2223. - (void)signInWithEmailLinkWithMockUserInfoResponse:(id)mockUserInfoResponse
  2224. completion:(void (^)(FIRUser *user))completion {
  2225. OCMExpect([_mockBackend emailLinkSignin:[OCMArg any] callback:[OCMArg any]])
  2226. .andCallBlock2(^(FIREmailLinkSignInRequest *_Nullable request,
  2227. FIREmailLinkSigninResponseCallback callback) {
  2228. id mockVerifyLinkResponse = OCMClassMock([FIREmailLinkSignInResponse class]);
  2229. OCMStub([mockVerifyLinkResponse IDToken]).andReturn(kAccessToken);
  2230. OCMStub([mockVerifyLinkResponse approximateExpirationDate])
  2231. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2232. OCMStub([mockVerifyLinkResponse refreshToken]).andReturn(kRefreshToken);
  2233. callback(mockVerifyLinkResponse, nil);
  2234. });
  2235. [self expectGetAccountInfoWithMockUserInfoResponse:mockUserInfoResponse];
  2236. [[FIRAuth auth] signOut:NULL];
  2237. [[FIRAuth auth] signInWithEmail:kEmail
  2238. link:@"https://www.google.com?oobCode=aCode&mode=signIn"
  2239. completion:^(FIRAuthDataResult *_Nullable result, NSError *_Nullable error) {
  2240. XCTAssertNotNil(result.user);
  2241. XCTAssertNil(error);
  2242. completion(result.user);
  2243. }];
  2244. }
  2245. /** @fn expectGetAccountInfoWithMockUserInfoResponse:
  2246. @brief Expects a GetAccountInfo request on the mock backend and calls back with provided
  2247. fake account data.
  2248. @param mockUserInfoResponse A mock @c FIRGetAccountInfoResponseUser object containing user info.
  2249. */
  2250. - (void)expectGetAccountInfoWithMockUserInfoResponse:(id)mockUserInfoResponse {
  2251. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2252. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  2253. FIRGetAccountInfoResponseCallback callback) {
  2254. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2255. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2256. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2257. id mockGetAccountInfoResponse = OCMClassMock([FIRGetAccountInfoResponse class]);
  2258. OCMStub([mockGetAccountInfoResponse users]).andReturn(@[ mockUserInfoResponse ]);
  2259. callback(mockGetAccountInfoResponse, nil);
  2260. });
  2261. });
  2262. }
  2263. /** @fn dictionaryWithUserInfoArray:
  2264. @brief Converts an array of @c FIRUserInfo into a dictionary that indexed by provider IDs.
  2265. @param userInfoArray An array of @c FIRUserInfo objects.
  2266. @return A dictionary contains same values as @c userInfoArray does but keyed by their
  2267. @c providerID .
  2268. */
  2269. - (NSDictionary<NSString *, id<FIRUserInfo>> *)
  2270. dictionaryWithUserInfoArray:(NSArray<id<FIRUserInfo>> *)userInfoArray {
  2271. NSMutableDictionary<NSString *, id<FIRUserInfo>> *map =
  2272. [NSMutableDictionary dictionaryWithCapacity:userInfoArray.count];
  2273. for (id<FIRUserInfo> userInfo in userInfoArray) {
  2274. XCTAssertNil(map[userInfo.providerID]);
  2275. map[userInfo.providerID] = userInfo;
  2276. }
  2277. return map;
  2278. }
  2279. /** @fn stubSecureTokensWithMockResponse
  2280. @brief Creates stubs on the mock response object with access and refresh tokens
  2281. @param mockResponse The mock response object.
  2282. */
  2283. - (void)stubTokensWithMockResponse:(id)mockResponse {
  2284. OCMStub([mockResponse IDToken]).andReturn(kAccessToken);
  2285. OCMStub([mockResponse approximateExpirationDate])
  2286. .andReturn([NSDate dateWithTimeIntervalSinceNow:kAccessTokenTimeToLive]);
  2287. OCMStub([mockResponse refreshToken]).andReturn(kRefreshToken);
  2288. }
  2289. /** @fn assertUserGoogle
  2290. @brief Asserts the given FIRUser matching the fake data returned by
  2291. @c expectGetAccountInfo:federatedID:displayName: .
  2292. @param user The user object to be verified.
  2293. */
  2294. - (void)assertUserGoogle:(FIRUser *)user {
  2295. XCTAssertNotNil(user);
  2296. XCTAssertEqualObjects(user.uid, kLocalID);
  2297. XCTAssertEqualObjects(user.displayName, kGoogleDisplayName);
  2298. XCTAssertEqual(user.providerData.count, 1u);
  2299. id<FIRUserInfo> googleUserInfo = user.providerData[0];
  2300. XCTAssertEqualObjects(googleUserInfo.providerID, FIRGoogleAuthProviderID);
  2301. XCTAssertEqualObjects(googleUserInfo.uid, kGoogleID);
  2302. XCTAssertEqualObjects(googleUserInfo.displayName, kGoogleDisplayName);
  2303. XCTAssertEqualObjects(googleUserInfo.email, kGoogleEmail);
  2304. }
  2305. /** @fn assertUserFacebook
  2306. @brief Asserts the given FIRUser matching the fake data returned by
  2307. @c expectGetAccountInfo:federatedID:displayName: .
  2308. @param user The user object to be verified.
  2309. */
  2310. - (void)assertUserFacebook:(FIRUser *)user {
  2311. XCTAssertNotNil(user);
  2312. XCTAssertEqualObjects(user.uid, kLocalID);
  2313. XCTAssertEqualObjects(user.displayName, kFacebookDisplayName);
  2314. XCTAssertEqual(user.providerData.count, 1u);
  2315. id<FIRUserInfo> googleUserInfo = user.providerData[0];
  2316. XCTAssertEqualObjects(googleUserInfo.providerID, FIRFacebookAuthProviderID);
  2317. XCTAssertEqualObjects(googleUserInfo.uid, kFacebookID);
  2318. XCTAssertEqualObjects(googleUserInfo.displayName, kFacebookDisplayName);
  2319. XCTAssertEqualObjects(googleUserInfo.email, kGoogleEmail);
  2320. }
  2321. /** @fn expectGetAccountInfo:federatedID:displayName:
  2322. @brief Expects a GetAccountInfo request on the mock backend and calls back with fake account
  2323. data for a Google Sign-In user.
  2324. */
  2325. - (void)expectGetAccountInfo:(NSString *)providerId
  2326. federatedID:(NSString *)federatedID
  2327. displayName:(NSString *)displayName {
  2328. OCMExpect([_mockBackend getAccountInfo:[OCMArg any] callback:[OCMArg any]])
  2329. .andCallBlock2(^(FIRGetAccountInfoRequest *_Nullable request,
  2330. FIRGetAccountInfoResponseCallback callback) {
  2331. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2332. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2333. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2334. id mockGoogleUserInfo = OCMClassMock([FIRGetAccountInfoResponseProviderUserInfo class]);
  2335. OCMStub([mockGoogleUserInfo providerID]).andReturn(providerId);
  2336. OCMStub([mockGoogleUserInfo displayName]).andReturn(displayName);
  2337. OCMStub([mockGoogleUserInfo federatedID]).andReturn(federatedID);
  2338. OCMStub([mockGoogleUserInfo email]).andReturn(kGoogleEmail);
  2339. id mockGetAccountInfoResponseUser = OCMClassMock([FIRGetAccountInfoResponseUser class]);
  2340. OCMStub([mockGetAccountInfoResponseUser localID]).andReturn(kLocalID);
  2341. OCMStub([mockGetAccountInfoResponseUser displayName]).andReturn(displayName);
  2342. OCMStub([mockGetAccountInfoResponseUser providerUserInfo])
  2343. .andReturn((@[ mockGoogleUserInfo ]));
  2344. id mockGetAccountInfoResponse = OCMClassMock([FIRGetAccountInfoResponse class]);
  2345. OCMStub([mockGetAccountInfoResponse users]).andReturn(@[ mockGetAccountInfoResponseUser ]);
  2346. callback(mockGetAccountInfoResponse, nil);
  2347. });
  2348. });
  2349. }
  2350. /** @fn expectVerifyAssertionRequest:federatedID:displayName:profile:providerAccessToken:
  2351. @brief Expects a Verify Assertion request on the mock backend and calls back with fake account
  2352. data.
  2353. */
  2354. - (void)expectVerifyAssertionRequest:(NSString *)providerId
  2355. federatedID:(NSString *)federatedID
  2356. displayName:(NSString *)displayName
  2357. profile:(NSDictionary *)profile
  2358. providerIDToken:(nullable NSString *)providerIDToken
  2359. providerAccessToken:(NSString *)providerAccessToken {
  2360. OCMExpect([_mockBackend verifyAssertion:[OCMArg any] callback:[OCMArg any]])
  2361. .andCallBlock2(^(FIRVerifyAssertionRequest *_Nullable request,
  2362. FIRVerifyAssertionResponseCallback callback) {
  2363. XCTAssertEqualObjects(request.APIKey, kAPIKey);
  2364. XCTAssertEqualObjects(request.providerID, providerId);
  2365. XCTAssertEqualObjects(request.providerIDToken, providerIDToken);
  2366. XCTAssertEqualObjects(request.providerAccessToken, providerAccessToken);
  2367. XCTAssertTrue(request.returnSecureToken);
  2368. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2369. id mockVeriyAssertionResponse = OCMClassMock([FIRVerifyAssertionResponse class]);
  2370. OCMStub([mockVeriyAssertionResponse federatedID]).andReturn(federatedID);
  2371. OCMStub([mockVeriyAssertionResponse providerID]).andReturn(providerId);
  2372. OCMStub([mockVeriyAssertionResponse localID]).andReturn(kLocalID);
  2373. OCMStub([mockVeriyAssertionResponse displayName]).andReturn(displayName);
  2374. OCMStub([mockVeriyAssertionResponse profile]).andReturn(profile);
  2375. OCMStub([mockVeriyAssertionResponse username]).andReturn(kUserName);
  2376. [self stubTokensWithMockResponse:mockVeriyAssertionResponse];
  2377. callback(mockVeriyAssertionResponse, nil);
  2378. });
  2379. });
  2380. [self expectGetAccountInfo:providerId federatedID:federatedID displayName:displayName];
  2381. }
  2382. #if TARGET_OS_IOS
  2383. /** @fn expectVerifyPhoneNumberRequestWithPhoneNumber:error:
  2384. @brief Expects a verify phone numner request on the mock backend and calls back with fake
  2385. account data or an error.
  2386. @param phoneNumber Optionally; The phone number to use in the mocked response.
  2387. @param isLinkOperation Boolean value that indicates whether or not this method is triggered by
  2388. a link operation.
  2389. @param error Optionally; The error to return in the mocked response.
  2390. */
  2391. - (void)expectVerifyPhoneNumberRequestWithPhoneNumber:(nullable NSString *)phoneNumber
  2392. isLinkOperation:(BOOL)isLinkOperation
  2393. error:(nullable NSError*)error {
  2394. OCMExpect([_mockBackend verifyPhoneNumber:[OCMArg any] callback:[OCMArg any]])
  2395. .andCallBlock2(^(FIRVerifyPhoneNumberRequest *_Nullable request,
  2396. FIRVerifyPhoneNumberResponseCallback callback) {
  2397. XCTAssertEqualObjects(request.verificationID, kVerificationID);
  2398. XCTAssertEqualObjects(request.verificationCode, kVerificationCode);
  2399. if (isLinkOperation) {
  2400. XCTAssertEqual(request.operation, FIRAuthOperationTypeLink);
  2401. } else {
  2402. XCTAssertEqual(request.operation, FIRAuthOperationTypeUpdate);
  2403. }
  2404. XCTAssertEqualObjects(request.accessToken, kAccessToken);
  2405. dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
  2406. if (error) {
  2407. callback(nil, error);
  2408. return;
  2409. }
  2410. id mockVerifyPhoneNumberResponse = OCMClassMock([FIRVerifyPhoneNumberResponse class]);
  2411. OCMStub([mockVerifyPhoneNumberResponse phoneNumber]).andReturn(phoneNumber);
  2412. callback(mockVerifyPhoneNumberResponse, nil);
  2413. });
  2414. });
  2415. }
  2416. #endif
  2417. @end
  2418. NS_ASSUME_NONNULL_END