Add `Client.protect_branch`

lib/gitlab_client.rb

@@ -12,8 +12,9 @@ class GitlabClient
     end
   end
-  # Hard-code the CE project ID to save a request
+  # Hard-code the project IDs to save a request
   CE_PROJECT_ID = 13_083
+  EE_PROJECT_ID = 278_964
   def self.current_user
     @current_user ||= Gitlab.user
@@ -70,4 +71,10 @@ class GitlabClient
   def self.issue_url(issue)
     "https://gitlab.com/gitlab-org/gitlab-ce/issues/#{issue.iid}"
   end
+
+  def self.protect_branch(branch_name)
+    project_id = branch_name.end_with?('-ee') ? EE_PROJECT_ID : CE_PROJECT_ID
+
+    Gitlab.protect_branch(project_id, branch_name)
+  end
 end

spec/lib/gitlab_client_spec.rb

@@ -40,4 +40,20 @@ describe GitlabClient do
         to eq "https://gitlab.com/gitlab-org/gitlab-ce/issues/1234"
     end
   end
+
+  describe '.protect_branch' do
+    it 'uses EE ID when given an EE branch' do
+      expect(Gitlab).to receive(:protect_branch).
+        with(described_class::EE_PROJECT_ID, '8-4-stable-ee')
+
+      described_class.protect_branch('8-4-stable-ee')
+    end
+
+    it 'uses CE ID when given a CE branch' do
+      expect(Gitlab).to receive(:protect_branch).
+        with(described_class::CE_PROJECT_ID, '8-4-stable')
+
+      described_class.protect_branch('8-4-stable')
+    end
+  end
 end