FIRUserTests.m 122 KB

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