qa/qa/page/profile/ssh_keys.rb


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

# frozen_string_literal: true

module QA
  module Page
    module Profile
      class SSHKeys < Page::Base
        view 'app/views/profiles/keys/_form.html.haml' do
          element :key_expiry_date_field
          element :key_title_field
          element :key_public_key_field
          element :add_key_button
        end

        view 'app/helpers/ssh_keys_helper.rb' do
          element :delete_ssh_key_button
          element :ssh_key_delete_modal
        end

        view 'app/views/profiles/keys/_key_table.html.haml' do
          element :ssh_keys_list
        end

        def add_key(public_key, title)
          fill_element(:key_public_key_field, public_key)
          fill_element(:key_title_field, title)
          # Expire in 2 days just in case the key is created just before midnight
          fill_expiry_date(Date.today + 2)

          click_element(:add_key_button)
        end

        def fill_expiry_date(date)
          date = date.strftime('%m/%d/%Y') if date.is_a?(Date)
          Date.strptime(date, '%m/%d/%Y') rescue ArgumentError raise "Expiry date must be in mm/dd/yyyy format"

          fill_element(:key_expiry_date_field, date)
        end

        def remove_key(title)
          click_link(title)

          click_element(:delete_ssh_key_button)

          # Retrying due to https://gitlab.com/gitlab-org/gitlab/-/issues/255287
          retry_on_exception do
            wait_for_animated_element(:ssh_key_delete_modal)
            within_element(:ssh_key_delete_modal) do
              click_button('Delete')
            end
          end
        end

        def keys_list
          find_element(:ssh_keys_list).text
        end
      end
    end
  end
end