FIRUserTests.m 211 KB

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