More Tests Fixes

Author: wimbledon

FirebaseOAuthUI/FirebaseOAuthUITests/FirebaseOAuthUITests.m

@@ -40,13 +40,13 @@ - (void)setUp {
 
   id appClass = OCMClassMock([FIRApp class]);
   OCMStub([authClass app]).andReturn(appClass);
+  
+  FIRAuth *auth = [FIRAuth auth];
+  self.authUI = [FUIAuth authUIWithAuth:auth];
 
   self.mockOAuthProvider = OCMClassMock([FIROAuthProvider class]);
-  OCMStub(ClassMethod([_mockOAuthProvider providerWithProviderID:OCMOCK_ANY auth:OCMOCK_ANY])).
+  OCMStub(ClassMethod([_mockOAuthProvider providerWithProviderID:OCMOCK_ANY auth:self.authUI.auth])).
       andReturn(_mockOAuthProvider);
-
-  FIRAuth *auth = [FIRAuth auth];
-  self.authUI = [FUIAuth authUIWithAuth:auth];
 }
 
 - (void)tearDown {
@@ -78,7 +78,7 @@ - (void)testProviderValidity {
   XCTAssertNil(self.provider.accessToken);
   XCTAssertNil(self.provider.idToken);
 
-  OCMVerify([self.mockOAuthProvider providerWithProviderID:@"dummy"]);
+  OCMVerify([self.mockOAuthProvider providerWithProviderID:@"dummy" auth:self.authUI.auth]);
 }
 
 - (void)testAppleUsesEmulatorCreatesOAuthProvider {
@@ -93,7 +93,7 @@ - (void)testAppleUsesEmulatorCreatesOAuthProvider {
                                             scopes:@[]
                                   customParameters:@{}
                                       loginHintKey:nil];
-  OCMVerify([self.mockOAuthProvider providerWithProviderID:@"apple.com"]);
+  OCMVerify([self.mockOAuthProvider providerWithProviderID:@"apple.com" auth:self.authUI.auth]);
 }
 
 - (void)testAppleNoUseEmulatorNoOAuthProvider {
@@ -106,7 +106,7 @@ - (void)testAppleNoUseEmulatorNoOAuthProvider {
                                             scopes:@[]
                                   customParameters:@{}
                                       loginHintKey:nil];
-  OCMVerify(never(), [self.mockOAuthProvider providerWithProviderID:@"apple.com"]);
+  OCMVerify(never(), [self.mockOAuthProvider providerWithProviderID:@"apple.com" auth:self.authUI.auth]);
 }
 
 @end