FIRUserTests.m 200 KB

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