Welcome to mirror list, hosted at ThFree Co, Russian Federation.

two_factor_helpers.rb « features « helpers « support « spec - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: d5f069a40ea3b03b05a3b5bb2cae2477386446d8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
# frozen_string_literal: true
# These helpers allow you to manage and register
# U2F and WebAuthn devices
#
# Usage:
#   describe "..." do
#   include Spec::Support::Helpers::Features::TwoFactorHelpers
#     ...
#
#   manage_two_factor_authentication
#
module Spec
  module Support
    module Helpers
      module Features
        module TwoFactorHelpers
          def copy_recovery_codes
            click_on _('Copy codes')
            click_on _('Proceed')
          end

          def enable_two_factor_authentication
            click_on _('Enable two-factor authentication')
            expect(page).to have_content(_('Set up new device'))
            wait_for_requests
          end

          def manage_two_factor_authentication
            click_on 'Manage two-factor authentication'
            expect(page).to have_content("Set up new device")
            wait_for_requests
          end

          # Registers webauthn device via UI
          # Remove after `webauthn_without_totp` feature flag is deleted.
          def register_webauthn_device(webauthn_device = nil, name: 'My device')
            webauthn_device ||= FakeWebauthnDevice.new(page, name)
            webauthn_device.respond_to_webauthn_registration
            click_on 'Set up new device'
            expect(page).to have_content('Your device was successfully set up')
            fill_in 'Pick a name', with: name
            click_on 'Register device'
            webauthn_device
          end

          def webauthn_device_registration(webauthn_device: nil, name: 'My device', password: 'fake')
            webauthn_device ||= FakeWebauthnDevice.new(page, name)
            webauthn_device.respond_to_webauthn_registration
            click_on _('Set up new device')
            webauthn_fill_form_and_submit(name: name, password: password)
            webauthn_device
          end

          def webauthn_fill_form_and_submit(name: 'My device', password: 'fake')
            expect(page).to have_content(
              _('Your device was successfully set up! Give it a name and register it with the GitLab server.')
            )
            within '[data-testid="create-webauthn"]' do
              fill_in _('Device name'), with: name
              fill_in _('Current password'), with: password
              click_on _('Register device')
            end
          end

          # Adds webauthn device directly via database
          def add_webauthn_device(app_id, user, fake_device = nil, name: 'My device')
            fake_device ||= WebAuthn::FakeClient.new(app_id)

            options_for_create = WebAuthn::Credential.options_for_create(
              user: { id: user.webauthn_xid, name: user.username },
              authenticator_selection: { user_verification: 'discouraged' },
              rp: { name: 'GitLab' }
            )
            challenge = options_for_create.challenge

            device_response = fake_device.create(challenge: challenge).to_json # rubocop:disable Rails/SaveBang
            device_registration_params = { device_response: device_response,
                                           name: name }

            Webauthn::RegisterService.new(
              user, device_registration_params, challenge).execute
            FakeWebauthnDevice.new(page, name, fake_device)
          end

          def assert_fallback_ui(page)
            expect(page).to have_button('Verify code')
            expect(page).to have_css('#user_otp_attempt')
            expect(page).not_to have_link('Sign in via 2FA code')
            expect(page).not_to have_css("#js-authenticate-token-2fa")
          end
        end
      end
    end
  end
end