FIRUserTests.m 229 KB

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