1 files changed, 71 insertions, 0 deletions
diff --git a/spec/lib/gitlab/seeders/ci/runner/runner_fleet_seeder_spec.rb b/spec/lib/gitlab/seeders/ci/runner/runner_fleet_seeder_spec.rb
new file mode 100644
index 00000000000..fe52b586d49
--- /dev/null
+++ b/spec/lib/gitlab/seeders/ci/runner/runner_fleet_seeder_spec.rb
@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+NULL_LOGGER = Gitlab::JsonLogger.new('/dev/null')
+
+RSpec.describe ::Gitlab::Seeders::Ci::Runner::RunnerFleetSeeder, feature_category: :runner_fleet do
+  let_it_be(:user) { create(:user, :admin, username: 'test-admin') }
+
+  subject(:seeder) do
+    described_class.new(NULL_LOGGER,
+                        username: user.username,
+                        registration_prefix: registration_prefix,
+                        runner_count: runner_count)
+  end
+
+  describe '#seed', :enable_admin_mode do
+    subject(:seed) { seeder.seed }
+
+    let(:runner_count) { 20 }
+    let(:registration_prefix) { 'prefix-' }
+    let(:runner_releases_url) do
+      ::Gitlab::CurrentSettings.current_application_settings.public_runner_releases_url
+    end
+
+    before do
+      WebMock.stub_request(:get, runner_releases_url).to_return(
+        body: '[]',
+        status: 200,
+        headers: { 'Content-Type' => 'application/json' }
+      )
+    end
+
+    it 'creates expected hierarchy', :aggregate_failures do
+      expect { seed }.to change { Ci::Runner.count }.by(runner_count)
+        .and change { Ci::Runner.instance_type.count }.by(1)
+        .and change { Project.count }.by(3)
+        .and change { Group.count }.by(6)
+
+      expect(Group.search(registration_prefix)).to contain_exactly(
+        an_object_having_attributes(name: "#{registration_prefix}top-level group 1"),
+        an_object_having_attributes(name: "#{registration_prefix}top-level group 2"),
+        an_object_having_attributes(name: "#{registration_prefix}group 1.1"),
+        an_object_having_attributes(name: "#{registration_prefix}group 1.1.1"),
+        an_object_having_attributes(name: "#{registration_prefix}group 1.1.2"),
+        an_object_having_attributes(name: "#{registration_prefix}group 2.1")
+      )
+
+      expect(Project.search(registration_prefix)).to contain_exactly(
+        an_object_having_attributes(name: "#{registration_prefix}project 1.1.1.1"),
+        an_object_having_attributes(name: "#{registration_prefix}project 1.1.2.1"),
+        an_object_having_attributes(name: "#{registration_prefix}project 2.1.1")
+      )
+
+      project_1_1_1_1 = Project.find_by_name("#{registration_prefix}project 1.1.1.1")
+      project_1_1_2_1 = Project.find_by_name("#{registration_prefix}project 1.1.2.1")
+      project_2_1_1 = Project.find_by_name("#{registration_prefix}project 2.1.1")
+      expect(seed).to contain_exactly(
+        { project_id: project_1_1_1_1.id, runner_ids: an_instance_of(Array) },
+        { project_id: project_1_1_2_1.id, runner_ids: an_instance_of(Array) },
+        { project_id: project_2_1_1.id, runner_ids: an_instance_of(Array) }
+      )
+      seed.each do |project|
+        expect(project[:runner_ids].length).to be_between(0, 5)
+        expect(Project.find(project[:project_id]).all_available_runners.ids).to include(*project[:runner_ids])
+        expect(::Ci::Pipeline.for_project(project[:runner_ids])).to be_empty
+        expect(::Ci::Build.where(runner_id: project[:runner_ids])).to be_empty
+      end
+    end
+  end
+end