1 files changed, 62 insertions, 0 deletions
diff --git a/spec/controllers/projects/runner_projects_controller_spec.rb b/spec/controllers/projects/runner_projects_controller_spec.rb
new file mode 100644
index 00000000000..beedaad0fa9
--- /dev/null
+++ b/spec/controllers/projects/runner_projects_controller_spec.rb
@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Projects::RunnerProjectsController, feature_category: :runner_fleet do
+  let_it_be(:user) { create(:user) }
+  let_it_be(:group) { create(:group) }
+  let_it_be(:project) { create(:project, group: group) }
+  let_it_be(:source_project) { create(:project) }
+
+  before do
+    sign_in(user)
+    project.add_maintainer(user)
+  end
+
+  describe '#create' do
+    subject(:send_create) do
+      post :create, params: {
+        namespace_id: group.path,
+        project_id: project.path,
+        runner_project: { runner_id: project_runner.id }
+      }
+    end
+
+    context 'when assigning runner to another project' do
+      let(:project_runner) { create(:ci_runner, :project, projects: [source_project]) }
+
+      it 'redirects to the project runners page' do
+        source_project.add_maintainer(user)
+
+        send_create
+
+        expect(flash[:success]).to be_present
+        expect(response).to have_gitlab_http_status(:redirect)
+        expect(response).to redirect_to project_runners_path(project)
+      end
+    end
+  end
+
+  describe '#destroy' do
+    subject(:send_destroy) do
+      delete :destroy, params: {
+        namespace_id: group.path,
+        project_id: project.path,
+        id: runner_project_id
+      }
+    end
+
+    context 'when unassigning runner from project' do
+      let(:project_runner) { create(:ci_runner, :project, projects: [project]) }
+      let(:runner_project_id) { project_runner.runner_projects.last.id }
+
+      it 'redirects to the project runners page' do
+        send_destroy
+
+        expect(flash[:success]).to be_present
+        expect(response).to have_gitlab_http_status(:redirect)
+        expect(response).to redirect_to project_runners_path(project)
+      end
+    end
+  end
+end