2 files changed, 6 insertions, 6 deletions
diff --git a/spec/support/shared_examples/features/wiki/file_attachments_shared_examples.rb b/spec/support/shared_examples/features/wiki/file_attachments_shared_examples.rb
index 0ef1ccdfe57..8d1502bed84 100644
--- a/spec/support/shared_examples/features/wiki/file_attachments_shared_examples.rb
+++ b/spec/support/shared_examples/features/wiki/file_attachments_shared_examples.rb
@@ -12,8 +12,8 @@ RSpec.shared_examples 'wiki file attachments' do
     end
 
     context 'before uploading' do
-      it 'shows "Attach a file" button' do
-        expect(page).to have_button('Attach a file')
+      it 'shows "Attach a file or image" button' do
+        expect(page).to have_selector('[data-testid="button-attach-file"]')
         expect(page).not_to have_selector('.uploading-progress-container', visible: true)
       end
     end
@@ -26,7 +26,7 @@ RSpec.shared_examples 'wiki file attachments' do
           click_button 'Cancel'
         end
 
-        expect(page).to have_button('Attach a file')
+        expect(page).to have_selector('[data-testid="button-attach-file"]')
         expect(page).not_to have_button('Cancel')
         expect(page).not_to have_selector('.uploading-progress-container', visible: true)
       end
@@ -41,11 +41,11 @@ RSpec.shared_examples 'wiki file attachments' do
     end
 
     context 'uploading is complete' do
-      it 'shows "Attach a file" button on uploading complete' do
+      it 'shows "Attach a file or image" button on uploading complete' do
         attach_with_dropzone
         wait_for_requests
 
-        expect(page).to have_button('Attach a file')
+        expect(page).to have_selector('[data-testid="button-attach-file"]')
         expect(page).not_to have_selector('.uploading-progress-container', visible: true)
       end
 
diff --git a/spec/support/shared_examples/features/wiki/user_updates_wiki_page_shared_examples.rb b/spec/support/shared_examples/features/wiki/user_updates_wiki_page_shared_examples.rb
index 79c7c1891ac..87067336a36 100644
--- a/spec/support/shared_examples/features/wiki/user_updates_wiki_page_shared_examples.rb
+++ b/spec/support/shared_examples/features/wiki/user_updates_wiki_page_shared_examples.rb
@@ -140,7 +140,7 @@ RSpec.shared_examples 'User updates wiki page' do
     context 'when using the content editor' do
       context 'with feature flag on' do
         before do
-          click_button 'Edit rich text'
+          find('[data-testid="toggle-editing-mode-button"] label', text: 'Rich text').click
         end
 
         it_behaves_like 'edits content using the content editor'