FIRUserTests.m 123 KB

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