1 files changed, 40 insertions, 0 deletions
diff --git a/spec/workers/gitlab/github_import/import_protected_branch_worker_spec.rb b/spec/workers/gitlab/github_import/import_protected_branch_worker_spec.rb
new file mode 100644
index 00000000000..4a3ef2bf560
--- /dev/null
+++ b/spec/workers/gitlab/github_import/import_protected_branch_worker_spec.rb
@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::GithubImport::ImportProtectedBranchWorker do
+  let(:worker) { described_class.new }
+
+  let(:import_state) { build_stubbed(:import_state, :started) }
+  let(:project) { instance_double('Project', full_path: 'foo/bar', id: 1, import_state: import_state) }
+  let(:client) { instance_double('Gitlab::GithubImport::Client') }
+  let(:importer) { instance_double('Gitlab::GithubImport::Importer::ProtectedBranchImporter') }
+
+  describe '#import' do
+    let(:json_hash) do
+      {
+        id: 'main',
+        allow_force_pushes: true
+      }
+    end
+
+    it 'imports protected branch rule' do
+      expect(Gitlab::GithubImport::Importer::ProtectedBranchImporter)
+        .to receive(:new)
+        .with(
+          an_instance_of(Gitlab::GithubImport::Representation::ProtectedBranch),
+          project,
+          client
+        )
+        .and_return(importer)
+
+      expect(importer).to receive(:execute)
+
+      expect(Gitlab::GithubImport::ObjectCounter)
+        .to receive(:increment)
+        .with(project, :protected_branch, :imported)
+
+      worker.import(project, client, json_hash)
+    end
+  end
+end