1 files changed, 25 insertions, 1 deletions
diff --git a/spec/requests/projects/ml/candidates_controller_spec.rb b/spec/requests/projects/ml/candidates_controller_spec.rb
index 4c7491970e1..78f31be26d1 100644
--- a/spec/requests/projects/ml/candidates_controller_spec.rb
+++ b/spec/requests/projects/ml/candidates_controller_spec.rb
@@ -6,7 +6,11 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
   let_it_be(:project) { create(:project, :repository) }
   let_it_be(:user) { project.first_owner }
   let_it_be(:experiment) { create(:ml_experiments, project: project, user: user) }
-  let_it_be(:candidate) { create(:ml_candidates, experiment: experiment, user: user, project: project) }
+  let_it_be(:candidate) do
+    create(:ml_candidates, experiment: experiment, user: user, project: project).tap do |c|
+      c.update!(ci_build: create(:ci_build))
+    end
+  end
 
   let(:ff_value) { true }
   let(:candidate_iid) { candidate.iid }
@@ -47,7 +51,13 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
   end
 
   describe 'GET show' do
+    let(:can_read_build) { true }
+
     before do
+      allow(Ability).to receive(:allowed?)
+                          .with(user, :read_build, candidate.ci_build)
+                          .and_return(can_read_build)
+
       show_candidate
     end
 
@@ -64,6 +74,20 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
       expect { show_candidate }.not_to exceed_all_query_limit(control_count)
     end
 
+    context 'when user has permission to read the build' do
+      it 'includes ci build info' do
+        expect(assigns[:include_ci_info]).to eq(true)
+      end
+    end
+
+    context 'when user has no permission to read the build' do
+      let(:can_read_build) { false }
+
+      it 'sets include_ci_job to false' do
+        expect(assigns[:include_ci_info]).to eq(false)
+      end
+    end
+
     it_behaves_like '404 if candidate does not exist'
     it_behaves_like 'requires read_model_experiments'
   end