1 files changed, 23 insertions, 0 deletions
diff --git a/spec/models/ci/runner_namespace_spec.rb b/spec/models/ci/runner_namespace_spec.rb
index 2d1fe11147c..6cbb151d703 100644
--- a/spec/models/ci/runner_namespace_spec.rb
+++ b/spec/models/ci/runner_namespace_spec.rb
@@ -12,4 +12,27 @@ RSpec.describe Ci::RunnerNamespace do
 
     let!(:parent) { model.namespace }
   end
+
+  describe '.for_runner' do
+    subject(:for_runner) { described_class.for_runner(runner_ids) }
+
+    let_it_be(:group) { create(:group) }
+    let_it_be(:runners) { create_list(:ci_runner, 3, :group, groups: [group]) }
+
+    context 'with runner ids' do
+      let(:runner_ids) { runners[1..2].map(&:id) }
+
+      it 'returns requested runner namespaces' do
+        is_expected.to eq(runners[1..2].flat_map(&:runner_namespaces))
+      end
+    end
+
+    context 'with runners' do
+      let(:runner_ids) { runners.first }
+
+      it 'returns requested runner namespaces' do
+        is_expected.to eq(runners.first.runner_namespaces)
+      end
+    end
+  end
 end