Add latest changes from gitlab-org/gitlab@master

1 files changed, 27 insertions, 0 deletions

diff --git a/qa/qa/service/praefect_manager.rb b/qa/qa/service/praefect_manager.rb
new file mode 100644
index 00000000000..d8fa72456ad
--- /dev/null
+++ b/qa/qa/service/praefect_manager.rb
@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module QA
+  module Service
+    class PraefectManager
+      include Service::Shellout
+
+      def initialize
+        @praefect = 'praefect'
+        @first_node = 'gitaly1'
+        @second_node = 'gitaly2'
+        @primary_node = @first_node
+        @secondary_node = @second_node
+      end
+
+      def stop_primary_node
+        shell "docker stop #{@primary_node}"
+        @secondary_node, @primary_node = @primary_node, @secondary_node
+      end
+
+      def reset
+        shell "docker start #{@primary_node}"
+        shell "docker start #{@secondary_node}"
+      end
+    end
+  end
+end