FIRUserTests.m 127 KB

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