1 files changed, 80 insertions, 0 deletions
diff --git a/spec/frontend/authentication/two_factor_auth/index_spec.js b/spec/frontend/authentication/two_factor_auth/index_spec.js
new file mode 100644
index 00000000000..b181170b0a1
--- /dev/null
+++ b/spec/frontend/authentication/two_factor_auth/index_spec.js
@@ -0,0 +1,80 @@
+import { createWrapper } from '@vue/test-utils';
+import { getByTestId, fireEvent } from '@testing-library/dom';
+import * as urlUtils from '~/lib/utils/url_utility';
+import { initRecoveryCodes, initClose2faSuccessMessage } from '~/authentication/two_factor_auth';
+import RecoveryCodes from '~/authentication/two_factor_auth/components/recovery_codes.vue';
+import { codesJsonString, codes, profileAccountPath } from './mock_data';
+
+describe('initRecoveryCodes', () => {
+  let el;
+  let wrapper;
+
+  const findRecoveryCodesComponent = () => wrapper.find(RecoveryCodes);
+
+  beforeEach(() => {
+    el = document.createElement('div');
+    el.setAttribute('class', 'js-2fa-recovery-codes');
+    el.setAttribute('data-codes', codesJsonString);
+    el.setAttribute('data-profile-account-path', profileAccountPath);
+    document.body.appendChild(el);
+
+    wrapper = createWrapper(initRecoveryCodes());
+  });
+
+  afterEach(() => {
+    document.body.innerHTML = '';
+  });
+
+  it('parses `data-codes` and passes to `RecoveryCodes` as `codes` prop', () => {
+    expect(findRecoveryCodesComponent().props('codes')).toEqual(codes);
+  });
+
+  it('parses `data-profile-account-path` and passes to `RecoveryCodes` as `profileAccountPath` prop', () => {
+    expect(findRecoveryCodesComponent().props('profileAccountPath')).toEqual(profileAccountPath);
+  });
+});
+
+describe('initClose2faSuccessMessage', () => {
+  beforeEach(() => {
+    document.body.innerHTML = `
+      <button
+        data-testid="close-2fa-enabled-success-alert"
+        class="js-close-2fa-enabled-success-alert"
+      >
+      </button>
+    `;
+
+    initClose2faSuccessMessage();
+  });
+
+  afterEach(() => {
+    document.body.innerHTML = '';
+  });
+
+  describe('when alert is closed', () => {
+    beforeEach(() => {
+      delete window.location;
+      window.location = new URL(
+        'https://localhost/-/profile/account?two_factor_auth_enabled_successfully=true',
+      );
+
+      document.title = 'foo bar';
+
+      urlUtils.updateHistory = jest.fn();
+    });
+
+    afterEach(() => {
+      document.title = '';
+    });
+
+    it('removes `two_factor_auth_enabled_successfully` query param', () => {
+      fireEvent.click(getByTestId(document.body, 'close-2fa-enabled-success-alert'));
+
+      expect(urlUtils.updateHistory).toHaveBeenCalledWith({
+        url: 'https://localhost/-/profile/account',
+        title: 'foo bar',
+        replace: true,
+      });
+    });
+  });
+});