| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552 |
- /*
- * Copyright 2017 Google
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- #import <TargetConditionals.h>
- #if TARGET_OS_IOS
- #import <OCMock/OCMock.h>
- #import <SafariServices/SafariServices.h>
- #import <XCTest/XCTest.h>
- #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRAuth.h"
- #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRAuthSettings.h"
- #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRAuthUIDelegate.h"
- #import "FirebaseAuth/Sources/Public/FirebaseAuth/FIRPhoneAuthProvider.h"
- #import "FirebaseCore/Sources/Private/FirebaseCoreInternal.h"
- #import "FirebaseAuth/Sources/Auth/FIRAuthGlobalWorkQueue.h"
- #import "FirebaseAuth/Sources/Auth/FIRAuth_Internal.h"
- #import "FirebaseAuth/Sources/AuthProvider/FIRAuthCredential_Internal.h"
- #import "FirebaseAuth/Sources/AuthProvider/Phone/FIRPhoneAuthCredential_Internal.h"
- #import "FirebaseAuth/Sources/Backend/FIRAuthBackend.h"
- #import "FirebaseAuth/Sources/Backend/FIRAuthRequestConfiguration.h"
- #import "FirebaseAuth/Sources/Backend/RPC/FIRGetProjectConfigRequest.h"
- #import "FirebaseAuth/Sources/Backend/RPC/FIRGetProjectConfigResponse.h"
- #import "FirebaseAuth/Sources/Backend/RPC/FIRSendVerificationCodeRequest.h"
- #import "FirebaseAuth/Sources/Backend/RPC/FIRSendVerificationCodeResponse.h"
- #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyClientRequest.h"
- #import "FirebaseAuth/Sources/Backend/RPC/FIRVerifyClientResponse.h"
- #import "FirebaseAuth/Sources/SystemService/FIRAuthAPNSToken.h"
- #import "FirebaseAuth/Sources/SystemService/FIRAuthAPNSTokenManager.h"
- #import "FirebaseAuth/Sources/SystemService/FIRAuthAppCredential.h"
- #import "FirebaseAuth/Sources/SystemService/FIRAuthAppCredentialManager.h"
- #import "FirebaseAuth/Sources/SystemService/FIRAuthNotificationManager.h"
- #import "FirebaseAuth/Sources/Utilities/FIRAuthErrorUtils.h"
- #import "FirebaseAuth/Sources/Utilities/FIRAuthURLPresenter.h"
- #import "FirebaseAuth/Sources/Utilities/FIRAuthWebUtils.h"
- #import "FirebaseAuth/Tests/Unit/OCMStubRecorder+FIRAuthUnitTests.h"
- NS_ASSUME_NONNULL_BEGIN
- /** @var kTestPhoneNumber
- @brief A testing phone number.
- */
- static NSString *const kTestPhoneNumber = @"55555555";
- /** @var kTestInvalidPhoneNumber
- @brief An invalid testing phone number.
- */
- static NSString *const kTestInvalidPhoneNumber = @"555+!*55555";
- /** @var kTestVerificationID
- @brief A testing verfication ID.
- */
- static NSString *const kTestVerificationID = @"verificationID";
- /** @var kTestReceipt
- @brief A fake receipt for testing.
- */
- static NSString *const kTestReceipt = @"receipt";
- /** @var kTestSecret
- @brief A fake secret for testing.
- */
- static NSString *const kTestSecret = @"secret";
- /** @var kTestOldReceipt
- @brief A fake old receipt for testing.
- */
- static NSString *const kTestOldReceipt = @"old_receipt";
- /** @var kTestOldSecret
- @brief A fake old secret for testing.
- */
- static NSString *const kTestOldSecret = @"old_secret";
- /** @var kTestVerificationCode
- @brief A fake verfication code.
- */
- static NSString *const kTestVerificationCode = @"verificationCode";
- /** @var kFakeClientID
- @brief A fake client ID.
- */
- static NSString *const kFakeClientID = @"123456.apps.googleusercontent.com";
- /** @var kFakeReverseClientID
- @brief The dot-reversed version of the fake client ID.
- */
- static NSString *const kFakeReverseClientID = @"com.googleusercontent.apps.123456";
- /** @var kFakeFirebaseAppID
- @brief A fake Firebase app ID.
- */
- static NSString *const kFakeFirebaseAppID = @"1:123456789:ios:123abc456def";
- /** @var kFakeEncodedFirebaseAppID
- @brief A fake encoded Firebase app ID to be used as a custom URL scheme.
- */
- static NSString *const kFakeEncodedFirebaseAppID = @"app-1-123456789-ios-123abc456def";
- /** @var kFakeBundleID
- @brief A fake bundle ID.
- */
- static NSString *const kFakeBundleID = @"com.firebaseapp.example";
- /** @var kFakeAPIKey
- @brief A fake API key.
- */
- static NSString *const kFakeAPIKey = @"asdfghjkl";
- /** @var kFakeAuthorizedDomain
- @brief A fake authorized domain for the app.
- */
- static NSString *const kFakeAuthorizedDomain = @"test.firebaseapp.com";
- /** @var kFakeReCAPTCHAToken
- @brief A fake reCAPTCHA token.
- */
- static NSString *const kFakeReCAPTCHAToken = @"fakeReCAPTCHAToken";
- /** @var kFakeRedirectURLStringWithReCAPTCHAToken
- @brief The format for a fake redirect URL string (minus the scheme) that contains the fake
- reCAPTCHA token above.
- */
- static NSString *const kFakeRedirectURLStringWithReCAPTCHAToken =
- @"://firebaseauth/"
- @"link?deep_link_id=https%3A%2F%2Fexample.firebaseapp.com%2F__%2Fauth%2Fcallback%3FauthType%"
- @"3DverifyApp%26recaptchaToken%3DfakeReCAPTCHAToken";
- /** @var kFakeRedirectURLStringInvalidClientID
- @brief The format for a fake redirect URL string with an invalid client error.
- */
- static NSString *const kFakeRedirectURLStringInvalidClientID =
- @"com.googleusercontent.apps.1"
- "23456://firebaseauth/"
- "link?deep_link_id=https%3A%2F%2Fexample.firebaseapp.com%2F__%2Fauth%2Fcal"
- "lback%3FfirebaseError%3D%257B%2522code%2522%253A%2522auth%252Finvalid-oauth-client-id%2522%"
- "252"
- "C%2522message%2522%253A%2522The%2520OAuth%2520client%2520ID%2520provided%2520is%2520either%"
- "252"
- "0invalid%2520or%2520does%2520not%2520match%2520the%2520specified%2520API%2520key.%2522%257D%"
- "26"
- "authType%3DverifyApp";
- /** @var kFakeRedirectURLStringWebNetworkRequestFailed
- @brief The format for a fake redirect URL string with a web network request failed error.
- */
- static NSString *const kFakeRedirectURLStringWebNetworkRequestFailed =
- @"com.googleusercontent.apps"
- ".123456://firebaseauth/"
- "link?deep_link_id=https%3A%2F%2Fexample.firebaseapp.com%2F__%2Fauth%2Fc"
- "allback%3FfirebaseError%3D%257B%2522code%2522%253A%2522auth%252Fnetwork-request-failed%2522%"
- "25"
- "2C%2522message%2522%253A%2522The%2520network%2520request%2520failed%2520.%2522%257D%"
- "26authType"
- "%3DverifyApp";
- /** @var kFakeRedirectURLStringWebInternalError
- @brief The format for a fake redirect URL string with an internal web error.
- */
- static NSString *const kFakeRedirectURLStringWebInternalError =
- @"com.googleusercontent.apps.1"
- "23456://firebaseauth/"
- "link?deep_link_id=https%3A%2F%2Fexample.firebaseapp.com%2F__%2Fauth%2Fcal"
- "lback%3FfirebaseError%3D%257B%2522code%2522%253A%2522auth%252Finternal-error%2522%252C%"
- "2522mes"
- "sage%2522%253A%2522Internal%2520error%2520.%2522%257D%26authType%3DverifyApp";
- /** @var kFakeRedirectURLStringUnknownError
- @brief The format for a fake redirect URL string with unknown error response.
- */
- static NSString *const kFakeRedirectURLStringUnknownError =
- @"com.googleusercontent.apps.1"
- "23456://firebaseauth/"
- "link?deep_link_id=https%3A%2F%2Fexample.firebaseapp.com%2F__%2Fauth%2Fcal"
- "lback%3FfirebaseError%3D%257B%2522code%2522%253A%2522auth%252Funknown-error-id%2522%252"
- "C%2522message%2522%253A%2522The%2520OAuth%2520client%2520ID%2520provided%2520is%2520either%"
- "252"
- "0invalid%2520or%2520does%2520not%2520match%2520the%2520specified%2520API%2520key.%2522%257D%"
- "26"
- "authType%3DverifyApp";
- /** @var kFakeRedirectURLStringUnstructuredError
- @brief The format for a fake redirect URL string with unstructured error response.
- */
- static NSString *const kFakeRedirectURLStringUnstructuredError =
- @"com.googleusercontent.apps.1"
- "23456://firebaseauth/"
- "link?deep_link_id=https%3A%2F%2Fexample.firebaseapp.com%2F__%2Fauth%2Fcal"
- "lback%3FfirebaseError%3D%257B%2522unstructuredcode%2522%253A%2522auth%252Funknown-error-id%"
- "2522%252"
- "C%2522unstructuredmessage%2522%253A%2522The%2520OAuth%2520client%2520ID%2520provided%2520is%"
- "2520either%252"
- "0invalid%2520or%2520does%2520not%2520match%2520the%2520specified%2520API%2520key.%2522%257D%"
- "26"
- "authType%3DverifyApp";
- /** @var kTestTimeout
- @brief A fake timeout value for waiting for push notification.
- */
- static const NSTimeInterval kTestTimeout = 5;
- /** @var kExpectationTimeout
- @brief The maximum time waiting for expectations to fulfill.
- */
- static const NSTimeInterval kExpectationTimeout = 2;
- /** @class FIRPhoneAuthProviderTests
- @brief Tests for @c FIRPhoneAuthProvider
- */
- @interface FIRPhoneAuthProviderTests : XCTestCase
- @end
- @implementation FIRPhoneAuthProviderTests {
- /** @var _mockBackend
- @brief The mock @c FIRAuthBackendImplementation .
- */
- id _mockBackend;
- /** @var _provider
- @brief The @c FIRPhoneAuthProvider instance under test.
- */
- FIRPhoneAuthProvider *_provider;
- /** @var _mockAuth
- @brief The mock @c FIRAuth instance associated with @c _provider .
- */
- id _mockAuth;
- /** @var _mockApp
- @brief The mock @c FIRApp instance associated with @c _mockAuth .
- */
- id _mockApp;
- /** @var _mockOptions
- @brief The mock @c FIROptions instance associated with @c _mockApp.
- */
- id _mockOptions;
- /** @var _mockAPNSTokenManager
- @brief The mock @c FIRAuthAPNSTokenManager instance associated with @c _mockAuth .
- */
- id _mockAPNSTokenManager;
- /** @var _mockAppCredentialManager
- @brief The mock @c FIRAuthAppCredentialManager instance associated with @c _mockAuth .
- */
- id _mockAppCredentialManager;
- /** @var _mockNotificationManager
- @brief The mock @c FIRAuthNotificationManager instance associated with @c _mockAuth .
- */
- id _mockNotificationManager;
- /** @var _mockURLPresenter
- @brief The mock @c FIRAuthURLPresenter instance associated with @c _mockAuth .
- */
- id _mockURLPresenter;
- }
- - (void)setUp {
- [super setUp];
- _mockBackend = OCMProtocolMock(@protocol(FIRAuthBackendImplementation));
- [FIRAuthBackend setBackendImplementation:_mockBackend];
- _mockAuth = OCMClassMock([FIRAuth class]);
- _mockApp = OCMClassMock([FIRApp class]);
- OCMStub([_mockAuth app]).andReturn(_mockApp);
- _mockOptions = OCMClassMock([FIROptions class]);
- OCMStub([(FIRApp *)_mockApp options]).andReturn(_mockOptions);
- OCMStub([_mockOptions googleAppID]).andReturn(kFakeFirebaseAppID);
- _mockAPNSTokenManager = OCMClassMock([FIRAuthAPNSTokenManager class]);
- OCMStub([_mockAuth tokenManager]).andReturn(_mockAPNSTokenManager);
- _mockAppCredentialManager = OCMClassMock([FIRAuthAppCredentialManager class]);
- OCMStub([_mockAuth appCredentialManager]).andReturn(_mockAppCredentialManager);
- _mockNotificationManager = OCMClassMock([FIRAuthNotificationManager class]);
- OCMStub([_mockAuth notificationManager]).andReturn(_mockNotificationManager);
- _mockURLPresenter = OCMClassMock([FIRAuthURLPresenter class]);
- OCMStub([_mockAuth authURLPresenter]).andReturn(_mockURLPresenter);
- id mockRequestConfiguration = OCMClassMock([FIRAuthRequestConfiguration class]);
- OCMStub([_mockAuth requestConfiguration]).andReturn(mockRequestConfiguration);
- OCMStub([mockRequestConfiguration APIKey]).andReturn(kFakeAPIKey);
- }
- - (void)tearDown {
- [FIRAuthBackend setDefaultBackendImplementationWithRPCIssuer:nil];
- [super tearDown];
- }
- // We're still testing deprecated `verifyPhoneNumber:completion:` extensively.
- #pragma clang diagnostic push
- #pragma clang diagnostic ignored "-Wdeprecated-declarations"
- /** @fn testCredentialWithVerificationID
- @brief Tests the @c credentialWithToken method to make sure that it returns a valid
- FIRAuthCredential instance.
- */
- - (void)testCredentialWithVerificationID {
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- FIRPhoneAuthCredential *credential =
- [_provider credentialWithVerificationID:kTestVerificationID
- verificationCode:kTestVerificationCode];
- XCTAssertEqualObjects(credential.verificationID, kTestVerificationID);
- XCTAssertEqualObjects(credential.verificationCode, kTestVerificationCode);
- XCTAssertNil(credential.temporaryProof);
- XCTAssertNil(credential.phoneNumber);
- }
- /** @fn testVerifyEmptyPhoneNumber
- @brief Tests a failed invocation @c verifyPhoneNumber:completion: because an empty phone
- number was provided.
- */
- - (void)testVerifyEmptyPhoneNumber {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Empty phone number is checked on the client side so no backend RPC is mocked.
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:@""
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertNotNil(error);
- XCTAssertEqual(error.code, FIRAuthErrorCodeMissingPhoneNumber);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- }
- /** @fn testVerifyInvalidPhoneNumber
- @brief Tests a failed invocation @c verifyPhoneNumber:completion: because an invalid phone
- number was provided.
- */
- - (void)testVerifyInvalidPhoneNumber {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- OCMStub([_mockAppCredentialManager credential])
- .andReturn([[FIRAuthAppCredential alloc] initWithReceipt:kTestReceipt secret:kTestSecret]);
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback(nil, [FIRAuthErrorUtils invalidPhoneNumberErrorWithMessage:nil]);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(verificationID);
- XCTAssertEqual(error.code, FIRAuthErrorCodeInvalidPhoneNumber);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- }
- /** @fn testVerifyPhoneNumber
- @brief Tests a successful invocation of @c verifyPhoneNumber:completion:.
- */
- - (void)testVerifyPhoneNumber {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- OCMStub([_mockAppCredentialManager credential])
- .andReturn([[FIRAuthAppCredential alloc] initWithReceipt:kTestReceipt secret:kTestSecret]);
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- }
- /** @fn testVerifyPhoneNumberInTestMode
- @brief Tests a successful invocation of @c verifyPhoneNumber:completion: when app verification
- is disabled.
- */
- - (void)testVerifyPhoneNumberInTestMode {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Disable app verification.
- FIRAuthSettings *settings = [[FIRAuthSettings alloc] init];
- settings.appVerificationDisabledForTesting = YES;
- OCMStub([_mockAuth settings]).andReturn(settings);
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- // Assert that the app credential is nil when in test mode.
- XCTAssertNil(request.appCredential);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- }
- /** @fn testVerifyPhoneNumberInTestModeFailure
- @brief Tests a failed invocation of @c verifyPhoneNumber:completion: when app verification
- is disabled.
- */
- - (void)testVerifyPhoneNumberInTestModeFailure {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Disable app verification.
- FIRAuthSettings *settings = [[FIRAuthSettings alloc] init];
- settings.appVerificationDisabledForTesting = YES;
- OCMStub([_mockAuth settings]).andReturn(settings);
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- // Assert that the app credential is nil when in test mode.
- XCTAssertNil(request.appCredential);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- NSError *underlying = [NSError errorWithDomain:@"Test Error" code:1 userInfo:nil];
- callback(nil, [FIRAuthErrorUtils networkErrorWithUnderlyingError:underlying]);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(verificationID);
- XCTAssertEqual(error.code, FIRAuthErrorCodeNetworkError);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateFirebaseAppIdFlow
- @brief Tests a successful invocation of @c verifyPhoneNumber:UIDelegate:completion:.
- */
- - (void)testVerifyPhoneNumberUIDelegateFirebaseAppIdFlow {
- [self mockBundleWithURLScheme:kFakeEncodedFirebaseAppID];
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- // Fall back to the reCAPTCHA flow.
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- [self verifyReCAPTCHAVerificationFlowWithUIDelegate:mockUIDelegate
- clientID:nil
- firebaseAppID:kFakeFirebaseAppID];
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertNil(request.appCredential);
- XCTAssertEqualObjects(request.reCAPTCHAToken, kFakeReCAPTCHAToken);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateFirebaseAppIdWhileClientIdPresentFlow
- @brief Tests a successful invocation of @c verifyPhoneNumber:UIDelegate:completion: when the
- client ID is present in the plist file, but the encoded app ID is the registered custom URL
- scheme.
- */
- - (void)testVerifyPhoneNumberUIDelegateFirebaseAppIdWhileClientIdPresentFlow {
- [self mockBundleWithURLScheme:kFakeEncodedFirebaseAppID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- // Fall back to the reCAPTCHA flow.
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- [self verifyReCAPTCHAVerificationFlowWithUIDelegate:mockUIDelegate
- clientID:nil
- firebaseAppID:kFakeFirebaseAppID];
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertNil(request.appCredential);
- XCTAssertEqualObjects(request.reCAPTCHAToken, kFakeReCAPTCHAToken);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateClientIdFlow
- @brief Tests a successful invocation of @c verifyPhoneNumber:UIDelegate:completion:.
- */
- - (void)testVerifyPhoneNumberUIDelegateClientIdFlow {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- // Fall back to the reCAPTCHA flow.
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- [self verifyReCAPTCHAVerificationFlowWithUIDelegate:mockUIDelegate
- clientID:kFakeClientID
- firebaseAppID:nil];
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertNil(request.appCredential);
- XCTAssertEqualObjects(request.reCAPTCHAToken, kFakeReCAPTCHAToken);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateInvalidClientID
- @brief Tests a invocation of @c verifyPhoneNumber:UIDelegate:completion: which results in an
- invalid client ID error.
- */
- - (void)testVerifyPhoneNumberUIDelegateInvalidClientID {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:mockUIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion([NSURL URLWithString:kFakeRedirectURLStringInvalidClientID], nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertEqual(error.code, FIRAuthErrorCodeInvalidClientID);
- XCTAssertNil(verificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateWebNetworkRequestFailed
- @brief Tests a invocation of @c verifyPhoneNumber:UIDelegate:completion: which results in a web
- network request failed error.
- */
- - (void)testVerifyPhoneNumberUIDelegateNetworkRequestFailed {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:mockUIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion([NSURL URLWithString:kFakeRedirectURLStringWebNetworkRequestFailed], nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertEqual(error.code, FIRAuthErrorCodeWebNetworkRequestFailed);
- XCTAssertNil(verificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateWebInternalError
- @brief Tests a invocation of @c verifyPhoneNumber:UIDelegate:completion: which results in a web
- internal error.
- */
- - (void)testVerifyPhoneNumberUIDelegateWebInternalError {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:mockUIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion([NSURL URLWithString:kFakeRedirectURLStringWebInternalError], nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertEqual(error.code, FIRAuthErrorCodeWebInternalError);
- XCTAssertNil(verificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateUnexpectedError
- @brief Tests a invocation of @c verifyPhoneNumber:UIDelegate:completion: which results in an
- invalid client ID.
- */
- - (void)testVerifyPhoneNumberUIDelegateUnexpectedError {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:mockUIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion([NSURL URLWithString:kFakeRedirectURLStringUnknownError], nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider
- verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertEqual(error.code, FIRAuthErrorCodeAppVerificationUserInteractionFailure);
- XCTAssertNil(verificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateUnstructuredError
- @brief Tests a invocation of @c verifyPhoneNumber:UIDelegate:completion: which results in an
- error being surfaced with a default NSLocalizedFailureReasonErrorKey due to an unexpected
- structure of the error response.
- */
- - (void)testVerifyPhoneNumberUIDelegateUnstructuredError {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:mockUIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion([NSURL URLWithString:kFakeRedirectURLStringUnstructuredError], nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider
- verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertEqual(error.code, FIRAuthErrorCodeAppVerificationUserInteractionFailure);
- XCTAssertNil(verificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateRaiseException
- @brief Tests a invocation of @c verifyPhoneNumber:UIDelegate:completion: which results in an
- exception.
- */
- - (void)testVerifyPhoneNumberUIDelegateRaiseException {
- [self mockBundleWithURLScheme:@"badscheme"];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- XCTAssertThrows([_provider
- verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTFail(@"Shouldn't call completion here.");
- }]);
- }
- /** @fn testNotForwardingNotification
- @brief Tests returning an error for the app failing to forward notification.
- */
- - (void)testNotForwardingNotification {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(NO);
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(verificationID);
- XCTAssertEqual(error.code, FIRAuthErrorCodeNotificationNotForwarded);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testMissingAPNSToken
- @brief Tests returning an error for the app failing to provide an APNS device token.
- */
- - (void)testMissingAPNSToken {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- // Simulate missing app token error.
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- [self mockMissingAPNSToken];
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:mockUIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion(nil, [NSError errorWithDomain:FIRAuthErrorDomain
- code:FIRAuthErrorCodeMissingAppToken
- userInfo:nil]);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertEqual(error.code, FIRAuthErrorCodeMissingAppToken);
- XCTAssertNil(verificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyPhoneNumberUIDelegateiOSSecretMissingFlow
- @brief Tests a successful invocation of @c verifyPhoneNumber:UIDelegate:completion: that falls
- back to the reCAPTCHA flow when the push notification is not received before the timeout.
- */
- - (void)testVerifyPhoneNumberUIDelegateiOSSecretMissingFlow {
- [self mockBundleWithURLScheme:kFakeEncodedFirebaseAppID];
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- OCMExpect([_mockAppCredentialManager credential]).andReturn(nil);
- NSData *data = [@"!@#$%^" dataUsingEncoding:NSUTF8StringEncoding];
- FIRAuthAPNSToken *token = [[FIRAuthAPNSToken alloc] initWithData:data
- type:FIRAuthAPNSTokenTypeProd];
- OCMExpect([_mockAPNSTokenManager getTokenWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthAPNSTokenCallback callback) {
- callback(token, nil);
- });
- // Expect verify client request to the backend.
- OCMExpect([_mockBackend verifyClient:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRVerifyClientRequest *request, FIRVerifyClientResponseCallback callback) {
- XCTAssertEqualObjects(request.appToken, @"21402324255E");
- XCTAssertFalse(request.isSandbox);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockVerifyClientResponse = OCMClassMock([FIRVerifyClientResponse class]);
- OCMStub([mockVerifyClientResponse receipt]).andReturn(kTestReceipt);
- OCMStub([mockVerifyClientResponse suggestedTimeOutDate])
- .andReturn([NSDate dateWithTimeIntervalSinceNow:kTestTimeout]);
- callback(mockVerifyClientResponse, nil);
- });
- });
- // Mock failing to receive the push notification before the timeout.
- OCMExpect([_mockAppCredentialManager didStartVerificationWithReceipt:OCMOCK_ANY
- timeout:0
- callback:OCMOCK_ANY])
- .ignoringNonObjectArgs()
- .andCallIdDoubleIdBlock(
- ^(NSString *receipt, NSTimeInterval timeout, FIRAuthAppCredentialCallback callback) {
- XCTAssertEqualObjects(receipt, kTestReceipt);
- // Unfortunately 'ignoringNonObjectArgs' means the real value for 'timeout' doesn't get
- // passed into the block either, so we can't verify it here.
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback([[FIRAuthAppCredential alloc] initWithReceipt:kTestReceipt secret:nil]);
- });
- });
- // Fall back to the reCAPTCHA flow.
- id mockUIDelegate = OCMProtocolMock(@protocol(FIRAuthUIDelegate));
- [self verifyReCAPTCHAVerificationFlowWithUIDelegate:mockUIDelegate
- clientID:nil
- firebaseAppID:kFakeFirebaseAppID];
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertNil(request.appCredential);
- XCTAssertEqualObjects(request.reCAPTCHAToken, kFakeReCAPTCHAToken);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:mockUIDelegate
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- }
- /** @fn testVerifyClient
- @brief Tests verifying client before sending verification code.
- */
- - (void)testVerifyClient {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- OCMExpect([_mockAppCredentialManager credential]).andReturn(nil);
- NSData *data = [@"!@#$%^" dataUsingEncoding:NSUTF8StringEncoding];
- FIRAuthAPNSToken *token = [[FIRAuthAPNSToken alloc] initWithData:data
- type:FIRAuthAPNSTokenTypeProd];
- OCMExpect([_mockAPNSTokenManager getTokenWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthAPNSTokenCallback callback) {
- callback(token, nil);
- });
- // Expect verify client request to the backend.
- OCMExpect([_mockBackend verifyClient:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRVerifyClientRequest *request, FIRVerifyClientResponseCallback callback) {
- XCTAssertEqualObjects(request.appToken, @"21402324255E");
- XCTAssertFalse(request.isSandbox);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockVerifyClientResponse = OCMClassMock([FIRVerifyClientResponse class]);
- OCMStub([mockVerifyClientResponse receipt]).andReturn(kTestReceipt);
- OCMStub([mockVerifyClientResponse suggestedTimeOutDate])
- .andReturn([NSDate dateWithTimeIntervalSinceNow:kTestTimeout]);
- callback(mockVerifyClientResponse, nil);
- });
- });
- // Mock receiving of push notification.
- OCMExpect([_mockAppCredentialManager didStartVerificationWithReceipt:OCMOCK_ANY
- timeout:0
- callback:OCMOCK_ANY])
- .ignoringNonObjectArgs()
- .andCallIdDoubleIdBlock(^(NSString *receipt, NSTimeInterval timeout,
- FIRAuthAppCredentialCallback callback) {
- XCTAssertEqualObjects(receipt, kTestReceipt);
- // Unfortunately 'ignoringNonObjectArgs' means the real value for 'timeout' doesn't get
- // passed into the block either, so we can't verify it here.
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback([[FIRAuthAppCredential alloc] initWithReceipt:kTestReceipt secret:kTestSecret]);
- });
- });
- // Expect send verification code request to the backend.
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- OCMVerifyAll(_mockAPNSTokenManager);
- }
- /** @fn testSendVerificationCodeFailedRetry
- @brief Tests failed retry after failing to send verification code.
- */
- - (void)testSendVerificationCodeFailedRetry {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- // Expect twice due to null check consumes one expectation.
- OCMExpect([_mockAppCredentialManager credential])
- .andReturn([[FIRAuthAppCredential alloc] initWithReceipt:kTestOldReceipt
- secret:kTestOldSecret]);
- OCMExpect([_mockAppCredentialManager credential])
- .andReturn([[FIRAuthAppCredential alloc] initWithReceipt:kTestOldReceipt
- secret:kTestOldSecret]);
- NSData *data = [@"!@#$%^" dataUsingEncoding:NSUTF8StringEncoding];
- FIRAuthAPNSToken *token = [[FIRAuthAPNSToken alloc] initWithData:data
- type:FIRAuthAPNSTokenTypeProd];
- // Expect first sendVerificationCode request to the backend, with request containing old app
- // credential.
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestOldReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestOldSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback(nil, [FIRAuthErrorUtils invalidAppCredentialWithMessage:nil]);
- });
- });
- // Expect send verification code request to the backend, with request containing new app
- // credential data.
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback(nil, [FIRAuthErrorUtils invalidAppCredentialWithMessage:nil]);
- });
- });
- OCMExpect([_mockAPNSTokenManager getTokenWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthAPNSTokenCallback callback) {
- callback(token, nil);
- });
- // Expect verify client request to the backend.
- OCMExpect([_mockBackend verifyClient:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRVerifyClientRequest *request, FIRVerifyClientResponseCallback callback) {
- XCTAssertEqualObjects(request.appToken, @"21402324255E");
- XCTAssertFalse(request.isSandbox);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockVerifyClientResponse = OCMClassMock([FIRVerifyClientResponse class]);
- OCMStub([mockVerifyClientResponse receipt]).andReturn(kTestReceipt);
- OCMStub([mockVerifyClientResponse suggestedTimeOutDate])
- .andReturn([NSDate dateWithTimeIntervalSinceNow:kTestTimeout]);
- callback(mockVerifyClientResponse, nil);
- });
- });
- // Mock receiving of push notification.
- OCMStub([_mockAppCredentialManager didStartVerificationWithReceipt:OCMOCK_ANY
- timeout:0
- callback:OCMOCK_ANY])
- .ignoringNonObjectArgs()
- .andCallIdDoubleIdBlock(^(NSString *receipt, NSTimeInterval timeout,
- FIRAuthAppCredentialCallback callback) {
- XCTAssertEqualObjects(receipt, kTestReceipt);
- // Unfortunately 'ignoringNonObjectArgs' means the real value for 'timeout' doesn't get
- // passed into the block either, so we can't verify it here.
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback([[FIRAuthAppCredential alloc] initWithReceipt:kTestReceipt secret:kTestSecret]);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertTrue([NSThread isMainThread]);
- XCTAssertNil(verificationID);
- XCTAssertEqual(error.code, FIRAuthErrorCodeInternalError);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- OCMVerifyAll(_mockAPNSTokenManager);
- }
- /** @fn testSendVerificationCodeSuccessFulRetry
- @brief Tests successful retry after failing to send verification code.
- */
- - (void)testSendVerificationCodeSuccessfulRetry {
- [self mockBundleWithURLScheme:kFakeReverseClientID];
- OCMStub([_mockOptions clientID]).andReturn(kFakeClientID);
- _provider = [FIRPhoneAuthProvider providerWithAuth:_mockAuth];
- OCMExpect([_mockNotificationManager checkNotificationForwardingWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthNotificationForwardingCallback callback) {
- callback(YES);
- });
- // Expect twice due to null check consumes one expectation.
- OCMExpect([_mockAppCredentialManager credential])
- .andReturn([[FIRAuthAppCredential alloc] initWithReceipt:kTestOldReceipt
- secret:kTestOldSecret]);
- OCMExpect([_mockAppCredentialManager credential])
- .andReturn([[FIRAuthAppCredential alloc] initWithReceipt:kTestOldReceipt
- secret:kTestOldSecret]);
- NSData *data = [@"!@#$%^" dataUsingEncoding:NSUTF8StringEncoding];
- FIRAuthAPNSToken *token = [[FIRAuthAPNSToken alloc] initWithData:data
- type:FIRAuthAPNSTokenTypeProd];
- // Expect first sendVerificationCode request to the backend, with request containing old app
- // credential.
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestOldReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestOldSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback(nil, [FIRAuthErrorUtils invalidAppCredentialWithMessage:nil]);
- });
- });
- // Expect send verification code request to the backend, with request containing new app
- // credential data.
- OCMExpect([_mockBackend sendVerificationCode:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRSendVerificationCodeRequest *request,
- FIRSendVerificationCodeResponseCallback callback) {
- XCTAssertEqualObjects(request.phoneNumber, kTestPhoneNumber);
- XCTAssertEqualObjects(request.appCredential.receipt, kTestReceipt);
- XCTAssertEqualObjects(request.appCredential.secret, kTestSecret);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockSendVerificationCodeResponse =
- OCMClassMock([FIRSendVerificationCodeResponse class]);
- OCMStub([mockSendVerificationCodeResponse verificationID]).andReturn(kTestVerificationID);
- callback(mockSendVerificationCodeResponse, nil);
- });
- });
- OCMExpect([_mockAPNSTokenManager getTokenWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthAPNSTokenCallback callback) {
- callback(token, nil);
- });
- // Expect verify client request to the backend.
- OCMExpect([_mockBackend verifyClient:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(^(FIRVerifyClientRequest *request, FIRVerifyClientResponseCallback callback) {
- XCTAssertEqualObjects(request.appToken, @"21402324255E");
- XCTAssertFalse(request.isSandbox);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockVerifyClientResponse = OCMClassMock([FIRVerifyClientResponse class]);
- OCMStub([mockVerifyClientResponse receipt]).andReturn(kTestReceipt);
- OCMStub([mockVerifyClientResponse suggestedTimeOutDate])
- .andReturn([NSDate dateWithTimeIntervalSinceNow:kTestTimeout]);
- callback(mockVerifyClientResponse, nil);
- });
- });
- // Mock receiving of push notification.
- OCMStub([_mockAppCredentialManager didStartVerificationWithReceipt:OCMOCK_ANY
- timeout:0
- callback:OCMOCK_ANY])
- .ignoringNonObjectArgs()
- .andCallIdDoubleIdBlock(^(NSString *receipt, NSTimeInterval timeout,
- FIRAuthAppCredentialCallback callback) {
- XCTAssertEqualObjects(receipt, kTestReceipt);
- // Unfortunately 'ignoringNonObjectArgs' means the real value for 'timeout' doesn't get
- // passed into the block either, so we can't verify it here.
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- callback([[FIRAuthAppCredential alloc] initWithReceipt:kTestReceipt secret:kTestSecret]);
- });
- });
- XCTestExpectation *expectation = [self expectationWithDescription:@"callback"];
- [_provider verifyPhoneNumber:kTestPhoneNumber
- UIDelegate:nil
- completion:^(NSString *_Nullable verificationID, NSError *_Nullable error) {
- XCTAssertNil(error);
- XCTAssertEqualObjects(verificationID, kTestVerificationID);
- [expectation fulfill];
- }];
- [self waitForExpectationsWithTimeout:kExpectationTimeout handler:nil];
- OCMVerifyAll(_mockBackend);
- OCMVerifyAll(_mockNotificationManager);
- OCMVerifyAll(_mockAppCredentialManager);
- OCMVerifyAll(_mockAPNSTokenManager);
- }
- - (void)mockBundleWithURLScheme:(NSString *)URLScheme {
- id mockBundle = OCMClassMock([NSBundle class]);
- OCMStub(ClassMethod([mockBundle mainBundle])).andReturn(mockBundle);
- OCMStub([mockBundle objectForInfoDictionaryKey:@"CFBundleURLTypes"]).andReturn(@[
- @{@"CFBundleURLSchemes" : @[ URLScheme ]}
- ]);
- OCMStub([mockBundle bundleIdentifier]).andReturn(kFakeBundleID);
- }
- - (void)mockMissingAPNSToken {
- OCMExpect([_mockAppCredentialManager credential]).andReturn(nil);
- OCMExpect([_mockAPNSTokenManager getTokenWithCallback:OCMOCK_ANY])
- .andCallBlock1(^(FIRAuthAPNSTokenCallback callback) {
- NSError *error = [NSError errorWithDomain:FIRAuthErrorDomain
- code:FIRAuthErrorCodeMissingAppToken
- userInfo:nil];
- callback(nil, error);
- });
- }
- - (void)verifyReCAPTCHAVerificationFlowWithUIDelegate:(id<FIRAuthUIDelegate>)UIDelegate
- clientID:(nullable NSString *)clientID
- firebaseAppID:(nullable NSString *)firebaseAppID {
- OCMExpect([_mockBackend getProjectConfig:[OCMArg any] callback:[OCMArg any]])
- .andCallBlock2(
- ^(FIRGetProjectConfigRequest *request, FIRGetProjectConfigResponseCallback callback) {
- XCTAssertNotNil(request);
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- id mockGetProjectConfigResponse = OCMClassMock([FIRGetProjectConfigResponse class]);
- OCMStub([mockGetProjectConfigResponse authorizedDomains]).andReturn(@[
- kFakeAuthorizedDomain
- ]);
- callback(mockGetProjectConfigResponse, nil);
- });
- });
- // Expect view controller presentation by UIDelegate.
- OCMExpect([_mockURLPresenter presentURL:OCMOCK_ANY
- UIDelegate:UIDelegate
- callbackMatcher:OCMOCK_ANY
- completion:OCMOCK_ANY])
- .andDo(^(NSInvocation *invocation) {
- __unsafe_unretained id unretainedArgument;
- // Indices 0 and 1 indicate the hidden arguments self and _cmd.
- // `presentURL` is at index 2.
- [invocation getArgument:&unretainedArgument atIndex:2];
- NSURL *presentURL = unretainedArgument;
- XCTAssertEqualObjects(presentURL.scheme, @"https");
- XCTAssertEqualObjects(presentURL.host, kFakeAuthorizedDomain);
- XCTAssertEqualObjects(presentURL.path, @"/__/auth/handler");
- NSURLComponents *actualURLComponents = [NSURLComponents componentsWithURL:presentURL
- resolvingAgainstBaseURL:NO];
- NSArray<NSURLQueryItem *> *queryItems = [actualURLComponents queryItems];
- XCTAssertEqualObjects([FIRAuthWebUtils queryItemValue:@"ibi" from:queryItems],
- kFakeBundleID);
- if (clientID) {
- XCTAssertEqualObjects([FIRAuthWebUtils queryItemValue:@"clientId" from:queryItems],
- clientID);
- } else if (firebaseAppID) {
- XCTAssertEqualObjects([FIRAuthWebUtils queryItemValue:@"appId" from:queryItems],
- firebaseAppID);
- }
- XCTAssertEqualObjects([FIRAuthWebUtils queryItemValue:@"apiKey" from:queryItems],
- kFakeAPIKey);
- XCTAssertEqualObjects([FIRAuthWebUtils queryItemValue:@"authType" from:queryItems],
- @"verifyApp");
- XCTAssertNotNil([FIRAuthWebUtils queryItemValue:@"v" from:queryItems]);
- // `callbackMatcher` is at index 4
- [invocation getArgument:&unretainedArgument atIndex:4];
- FIRAuthURLCallbackMatcher callbackMatcher = unretainedArgument;
- NSMutableString *redirectURL;
- if (clientID) {
- redirectURL = [NSMutableString
- stringWithString:[kFakeReverseClientID stringByAppendingString:
- kFakeRedirectURLStringWithReCAPTCHAToken]];
- } else if (firebaseAppID) {
- redirectURL = [NSMutableString
- stringWithString:
- [kFakeEncodedFirebaseAppID
- stringByAppendingString:kFakeRedirectURLStringWithReCAPTCHAToken]];
- }
- // Verify that the URL is rejected by the callback matcher without the event ID.
- XCTAssertFalse(callbackMatcher([NSURL URLWithString:redirectURL]));
- [redirectURL appendString:@"%26eventId%3D"];
- [redirectURL appendString:[FIRAuthWebUtils queryItemValue:@"eventId" from:queryItems]];
- NSURLComponents *originalComponents = [[NSURLComponents alloc] initWithString:redirectURL];
- // Verify that the URL is accepted by the callback matcher with the matching event ID.
- XCTAssertTrue(callbackMatcher([originalComponents URL]));
- NSURLComponents *components = [originalComponents copy];
- components.query = @"https";
- XCTAssertFalse(callbackMatcher([components URL]));
- components = [originalComponents copy];
- components.host = @"badhost";
- XCTAssertFalse(callbackMatcher([components URL]));
- components = [originalComponents copy];
- components.path = @"badpath";
- XCTAssertFalse(callbackMatcher([components URL]));
- components = [originalComponents copy];
- components.query = @"badquery";
- XCTAssertFalse(callbackMatcher([components URL]));
- // `completion` is at index 5
- [invocation getArgument:&unretainedArgument atIndex:5];
- FIRAuthURLPresentationCompletion completion = unretainedArgument;
- dispatch_async(FIRAuthGlobalWorkQueue(), ^() {
- completion([NSURL URLWithString:[kFakeEncodedFirebaseAppID
- stringByAppendingString:
- kFakeRedirectURLStringWithReCAPTCHAToken]],
- nil);
- });
- });
- }
- #pragma clang diagnostic pop // ignored "-Wdeprecated-declarations"
- @end
- NS_ASSUME_NONNULL_END
- #endif
|