More integration tests

Author: sam-gc

packages-exp/auth-exp/test/integration/flows/custom.local.test.ts

@@ -19,6 +19,7 @@ import {
   Auth,
   createUserWithEmailAndPassword,
   EmailAuthProvider,
+  getAdditionalUserInfo,
   linkWithCredential,
   OperationType,
   reload,
@@ -73,6 +74,9 @@ describe('Integration test: custom auth', () => {
       'some-claim'
     );
     expect(user.providerId).to.eq('firebase');
+    const additionalUserInfo = await getAdditionalUserInfo(cred)!;
+    expect(additionalUserInfo.providerId).to.be.null;
+    expect(additionalUserInfo.isNewUser).to.be.true;
   });
 
   it('uid will overwrite existing user, joining accounts', async () => {
@@ -128,6 +132,20 @@ describe('Integration test: custom auth', () => {
     expect(user.photoURL).to.eq('photo-url');
   });
 
+  it('token can be refreshed', async () => {
+    const { user } = await signInWithCustomToken(auth, customToken);
+    const origToken = await user.getIdToken();
+    await new Promise(resolve => setTimeout(resolve, 1000));
+    expect(await user.getIdToken(true)).not.to.eq(origToken);
+  });
+
+  it('signing in will not override anonymous user', async () => {
+    const { user: anonUser } = await signInAnonymously(auth);
+    const { user: customUser } = await signInWithCustomToken(auth, customToken);
+    expect(auth.currentUser).to.eql(customUser);
+    expect(customUser.uid).not.to.eql(anonUser.uid);
+  });
+
   context('email/password interaction', () => {
     let email: string;
     let customToken: string;

packages-exp/auth-exp/test/integration/flows/email.test.ts

@@ -27,7 +27,8 @@ import {
   updateProfile,
   Auth,
   OperationType,
-  UserCredential
+  UserCredential,
+  getAdditionalUserInfo
   // eslint-disable-next-line import/no-extraneous-dependencies
 } from '@firebase/auth-exp';
 import { FirebaseError } from '@firebase/util';
@@ -64,6 +65,13 @@ describe('Integration test: email/password auth', () => {
     expect(user.uid).to.be.a('string');
     expect(user.email).to.eq(email);
     expect(user.emailVerified).to.be.false;
+    expect(user.providerData.length).to.eq(1);
+    expect(user.providerData[0].providerId).to.eq('password');
+    expect(user.providerData[0].email).to.eq(email);
+
+    const additionalUserInfo = getAdditionalUserInfo(userCred)!;
+    expect(additionalUserInfo.isNewUser).to.be.true;
+    expect(additionalUserInfo.providerId).to.eq('password');
   });
 
   it('errors when createUser called twice', async () => {
@@ -95,6 +103,9 @@ describe('Integration test: email/password auth', () => {
 
       expect(signInCred.operationType).to.eq(OperationType.SIGN_IN);
       expect(signInCred.user.uid).to.eq(signUpCred.user.uid);
+      const additionalUserInfo = getAdditionalUserInfo(signInCred)!;
+      expect(additionalUserInfo.isNewUser).to.be.false;
+      expect(additionalUserInfo.providerId).to.eq('password');
     });
 
     it('allows the user to sign in with signInWithCredential', async () => {
@@ -104,6 +115,9 @@ describe('Integration test: email/password auth', () => {
 
       expect(signInCred.operationType).to.eq(OperationType.SIGN_IN);
       expect(signInCred.user.uid).to.eq(signUpCred.user.uid);
+      const additionalUserInfo = getAdditionalUserInfo(signInCred)!;
+      expect(additionalUserInfo.isNewUser).to.be.false;
+      expect(additionalUserInfo.providerId).to.eq('password');
     });
 
     it('allows the user to update profile', async () => {

packages-exp/auth-exp/test/integration/flows/phone.test.ts

@@ -82,6 +82,15 @@ describe('Integration test: phone auth', () => {
     document.body.removeChild(fakeRecaptchaContainer);
   });
 
+  function resetVerifier(): void {
+    verifier.clear();
+    verifier = new RecaptchaVerifier(
+      fakeRecaptchaContainer,
+      undefined as any,
+      auth
+    );
+  }
+
   /** If in the emulator, search for the code in the API */
   async function code(
     crOrId: ConfirmationResult | string,
@@ -126,17 +135,40 @@ describe('Integration test: phone auth', () => {
     expect(auth.currentUser!.phoneNumber).to.be.null;
   });
 
-  context('with already-created user', () => {
-    let signUpCred: UserCredential;
+  it('anonymous users can upgrade using phone number', async () => {
+    const { user } = await signInAnonymously(auth);
+    const { uid: anonId } = user;
 
-    function resetVerifier(): void {
-      verifier.clear();
-      verifier = new RecaptchaVerifier(
-        fakeRecaptchaContainer,
-        undefined as any,
-        auth
+    const provider = new PhoneAuthProvider(auth);
+      const verificationId = await provider.verifyPhoneNumber(
+        PHONE_B.phoneNumber,
+        verifier
       );
-    }
+
+      await updatePhoneNumber(
+        user,
+        PhoneAuthProvider.credential(
+          verificationId,
+          await code(verificationId, PHONE_B.code)
+        )
+      );
+      expect(user.phoneNumber).to.eq(PHONE_B.phoneNumber);
+
+      await auth.signOut();
+      resetVerifier();
+
+      const cr = await signInWithPhoneNumber(auth, PHONE_B.phoneNumber, verifier);
+      const { user: secondSignIn } = await cr.confirm(
+        await code(cr, PHONE_B.code)
+      );
+      expect(secondSignIn.uid).to.eq(anonId);
+      expect(secondSignIn.isAnonymous).to.be.false;
+      expect(secondSignIn.providerData[0].phoneNumber).to.eq(PHONE_B.phoneNumber);
+      expect(secondSignIn.providerData[0].providerId).to.eq('phone');
+  });
+
+  context('with already-created user', () => {
+    let signUpCred: UserCredential;
 
     beforeEach(async () => {
       const cr = await signInWithPhoneNumber(

packages-exp/auth-exp/test/integration/webdriver/popup.test.ts

@@ -26,7 +26,7 @@ import { expect, use } from 'chai';
 import { IdPPage } from './util/idp_page';
 import * as chaiAsPromised from 'chai-as-promised';
 import { browserDescribe } from './util/test_runner';
-import { AnonFunction, CoreFunction, PopupFunction } from './util/functions';
+import { AnonFunction, CoreFunction, EmailFunction, PopupFunction } from './util/functions';
 
 use(chaiAsPromised);
 
@@ -146,6 +146,79 @@ browserDescribe('Popup IdP tests', driver => {
     ]);
   });
 
+  it('does not auto-upgrade anon accounts', async () => {
+    const {user: anonUser}: UserCredential = await driver.call(AnonFunction.SIGN_IN_ANONYMOUSLY);
+    await driver.callNoWait(PopupFunction.IDP_POPUP);
+    await driver.selectPopupWindow();
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // On redirect, check that the signed in user is different
+    await driver.selectMainWindow();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).not.to.eq(anonUser.uid);
+  });
+
+  it('linking with anonymous user upgrades account', async () => {
+    const {user: anonUser}: UserCredential = await driver.call(AnonFunction.SIGN_IN_ANONYMOUSLY);
+    await driver.callNoWait(PopupFunction.IDP_LINK_POPUP);
+    await driver.selectPopupWindow();
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // On redirect, check that the signed in user is upgraded
+    await driver.selectMainWindow();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).to.eq(anonUser.uid);
+    expect(curUser.isAnonymous).to.be.false;
+  });
+
+  it('is possible to link with different email', async () => {
+    const {user: emailUser}: UserCredential = await driver.call(EmailFunction.CREATE_USER, '[email protected]');
+  
+    // Link using pre-poulated user
+    await driver.callNoWait(PopupFunction.IDP_LINK_POPUP);
+    await driver.selectPopupWindow();
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // Check the linked account
+    await driver.selectMainWindow();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).to.eq(emailUser.uid);
+    expect(curUser.emailVerified).to.be.false;
+    expect(curUser.providerData.length).to.eq(2);
+  });
+
+  it('is possible to link with the same email', async () => {
+    const {user: emailUser}: UserCredential = await driver.call(EmailFunction.CREATE_USER, '[email protected]');
+  
+    // Link using pre-poulated user
+    await driver.callNoWait(PopupFunction.IDP_LINK_POPUP);
+    await driver.selectPopupWindow();
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // Check the linked account
+    await driver.selectMainWindow();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).to.eq(emailUser.uid);
+    expect(curUser.emailVerified).to.be.true;
+    expect(curUser.providerData.length).to.eq(2);
+  });
+
   context('with existing user', () => {
     let user1: User;
     let user2: User;

packages-exp/auth-exp/test/integration/webdriver/redirect.test.ts

@@ -26,7 +26,7 @@ import { expect, use } from 'chai';
 import { IdPPage } from './util/idp_page';
 import * as chaiAsPromised from 'chai-as-promised';
 import { browserDescribe } from './util/test_runner';
-import { AnonFunction, CoreFunction, RedirectFunction } from './util/functions';
+import { AnonFunction, CoreFunction, EmailFunction, RedirectFunction } from './util/functions';
 
 use(chaiAsPromised);
 
@@ -146,6 +146,77 @@ browserDescribe('WebDriver redirect IdP test', driver => {
     ]);
   });
 
+  it('does not auto-upgrade anon accounts', async () => {
+    const {user: anonUser}: UserCredential = await driver.call(AnonFunction.SIGN_IN_ANONYMOUSLY);
+    await driver.callNoWait(RedirectFunction.IDP_REDIRECT);
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // On redirect, check that the signed in user is different
+    await driver.reinitOnRedirect();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).not.to.eq(anonUser.uid);
+  });
+
+  it('linking with anonymous user upgrades account', async () => {
+    const {user: anonUser}: UserCredential = await driver.call(AnonFunction.SIGN_IN_ANONYMOUSLY);
+    await driver.callNoWait(RedirectFunction.IDP_LINK_REDIRECT);
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // On redirect, check that the signed in user is upgraded
+    await driver.reinitOnRedirect();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).to.eq(anonUser.uid);
+    expect(curUser.isAnonymous).to.be.false;
+  });
+
+  it('is possible to link with different email', async () => {
+    const {user: emailUser}: UserCredential = await driver.call(EmailFunction.CREATE_USER, '[email protected]');
+  
+    // Link using pre-poulated user
+    await driver.callNoWait(RedirectFunction.IDP_LINK_REDIRECT);
+
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // Check the linked account
+    await driver.reinitOnRedirect();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).to.eq(emailUser.uid);
+    expect(curUser.emailVerified).to.be.false;
+    expect(curUser.providerData.length).to.eq(2);
+  });
+
+  it('is possible to link with the same email', async () => {
+    const {user: emailUser}: UserCredential = await driver.call(EmailFunction.CREATE_USER, '[email protected]');
+  
+    // Link using pre-poulated user
+    await driver.callNoWait(RedirectFunction.IDP_LINK_REDIRECT);
+
+    const widget = new IdPPage(driver.webDriver);
+    await widget.pageLoad();
+    await widget.clickAddAccount();
+    await widget.fillEmail('[email protected]');
+    await widget.clickSignIn();
+
+    // Check the linked account
+    await driver.reinitOnRedirect();
+    const curUser = await driver.getUserSnapshot();
+    expect(curUser.uid).to.eq(emailUser.uid);
+    expect(curUser.emailVerified).to.be.true;
+    expect(curUser.providerData.length).to.eq(2);
+  });
+
   context('with existing user', () => {
     let user1: User;
     let user2: User;

packages-exp/auth-exp/test/integration/webdriver/static/email.js

@@ -0,0 +1,7 @@
+import { createUserWithEmailAndPassword } from '@firebase/auth-exp';
+
+const TEST_PASSWORD = 'password';
+
+export function createUser(email) {
+  return createUserWithEmailAndPassword(auth, email, TEST_PASSWORD);
+}

packages-exp/auth-exp/test/integration/webdriver/static/index.js

@@ -19,13 +19,15 @@ import * as redirect from './redirect';
 import * as anonymous from './anonymous';
 import * as core from './core';
 import * as popup from './popup';
+import * as email from './email';
 import { initializeApp } from '@firebase/app-exp';
 import { getAuth, useAuthEmulator } from '@firebase/auth-exp';
 
 window.core = { ...core };
 window.anonymous = { ...anonymous };
 window.redirect = { ...redirect };
 window.popup = { ...popup };
+window.email = { ...email };
 
 // The config and emulator URL are injected by the test. The test framework
 // calls this function after that injection.

packages-exp/auth-exp/test/integration/webdriver/util/functions.ts

@@ -46,6 +46,11 @@ export enum PopupFunction {
   TRY_TO_SIGN_IN_UNVERIFIED = 'popup.tryToSignInUnverified'
 }
 
+/** Available email functions. See static/email.js */
+export enum EmailFunction {
+  CREATE_USER = 'email.createUser',
+}
+
 /** Available core functions within the browser. See static/core.js */
 export enum CoreFunction {
   RESET = 'core.reset',