1 files changed, 37 insertions, 0 deletions
diff --git a/spec/lib/gitlab/sidekiq_middleware/pause_control/workers_map_spec.rb b/spec/lib/gitlab/sidekiq_middleware/pause_control/workers_map_spec.rb
new file mode 100644
index 00000000000..1aa4b470db0
--- /dev/null
+++ b/spec/lib/gitlab/sidekiq_middleware/pause_control/workers_map_spec.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::SidekiqMiddleware::PauseControl::WorkersMap, feature_category: :global_search do
+  let(:worker_class) do
+    Class.new do
+      def self.name
+        'TestPauseWorker'
+      end
+
+      include ApplicationWorker
+
+      pause_control :zoekt
+
+      def perform(*); end
+    end
+  end
+
+  before do
+    stub_const('TestPauseWorker', worker_class)
+  end
+
+  describe '.strategy_for' do
+    it 'accepts classname' do
+      expect(described_class.strategy_for(worker: worker_class)).to eq(:zoekt)
+    end
+
+    it 'accepts worker instance' do
+      expect(described_class.strategy_for(worker: worker_class.new)).to eq(:zoekt)
+    end
+
+    it 'returns nil for unknown worker' do
+      expect(described_class.strategy_for(worker: described_class)).to be_nil
+    end
+  end
+end