Fix file locking specs

868bb096 · Sanad Liaquat (Personal) · 639adaf7 · 868bb096
Commit 868bb096 authored Mar 11, 2021 by Sanad Liaquat (Personal)
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 4 deletions

qa/qa/ee/page/file/show.rb qa/qa/ee/page/file/show.rb +8 -4

No files found.
--- a/qa/qa/ee/page/file/show.rb
+++ b/qa/qa/ee/page/file/show.rb
@@ -24,22 +24,26 @@ module QA
          end

          def lock
-            click_element :lock_button
+            accept_confirm do
+              click_element :lock_button
+            end

            begin
              has_element? :lock_button, text: 'Unlock'
            rescue
-              raise ElementNotFound, %q(Button did not show expected state)
+              raise QA::Page::Base::ElementNotFound, %q(Button did not show expected state)
            end
          end

          def unlock
-            click_element :lock_button
+            accept_confirm do
+              click_element :lock_button
+            end

            begin
              has_element? :lock_button, text: 'Lock'
            rescue
-              raise ElementNotFound, %q(Button did not show expected state)
+              raise QA::Page::Base::ElementNotFound, %q(Button did not show expected state)
            end
          end