1 files changed, 13 insertions, 0 deletions
diff --git a/web-ui/src/account_recovery/page.spec.js b/web-ui/src/account_recovery/page.spec.js
index 37c7be6f..f5e52c85 100644
--- a/web-ui/src/account_recovery/page.spec.js
+++ b/web-ui/src/account_recovery/page.spec.js
@@ -6,6 +6,7 @@ import { Page } from 'src/account_recovery/page';
 import Header from 'src/common/header/header';
 import AdminRecoveryCodeForm from 'src/account_recovery/forms/admin_recovery_code_form';
 import UserRecoveryCodeForm from 'src/account_recovery/forms/user_recovery_code_form';
+import NewPasswordForm from 'src/account_recovery/forms/new_password_form';
 import Footer from 'src/common/footer/footer';
 
 describe('Account Recovery Page', () => {
@@ -37,6 +38,8 @@ describe('Account Recovery Page', () => {
 
     it('renders admin recovery code form as default form', () => {
       expect(page.find(AdminRecoveryCodeForm).length).toEqual(1);
+      expect(page.find(UserRecoveryCodeForm).length).toEqual(0);
+      expect(page.find(NewPasswordForm).length).toEqual(0);
     });
 
     it('renders user recovery code form when admin code submitted', () => {
@@ -44,6 +47,16 @@ describe('Account Recovery Page', () => {
 
       expect(page.find(AdminRecoveryCodeForm).length).toEqual(0);
       expect(page.find(UserRecoveryCodeForm).length).toEqual(1);
+      expect(page.find(NewPasswordForm).length).toEqual(0);
+    });
+
+    it('renders new password form when user code submitted', () => {
+      pageInstance.nextStep({ preventDefault: () => {} });
+      pageInstance.nextStep({ preventDefault: () => {} });
+
+      expect(page.find(AdminRecoveryCodeForm).length).toEqual(0);
+      expect(page.find(UserRecoveryCodeForm).length).toEqual(0);
+      expect(page.find(NewPasswordForm).length).toEqual(1);
     });
   });
 });