1 files changed, 17 insertions, 6 deletions
diff --git a/spec/requests/projects/ml/candidates_controller_spec.rb b/spec/requests/projects/ml/candidates_controller_spec.rb
index eec7af99063..4c7491970e1 100644
--- a/spec/requests/projects/ml/candidates_controller_spec.rb
+++ b/spec/requests/projects/ml/candidates_controller_spec.rb
@@ -10,13 +10,17 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
 
   let(:ff_value) { true }
   let(:candidate_iid) { candidate.iid }
-  let(:model_experiments_enabled) { true }
+  let(:read_model_experiments) { true }
+  let(:write_model_experiments) { true }
 
   before do
     allow(Ability).to receive(:allowed?).and_call_original
     allow(Ability).to receive(:allowed?)
                         .with(user, :read_model_experiments, project)
-                        .and_return(model_experiments_enabled)
+                        .and_return(read_model_experiments)
+    allow(Ability).to receive(:allowed?)
+                        .with(user, :write_model_experiments, project)
+                        .and_return(write_model_experiments)
     sign_in(user)
   end
 
@@ -34,9 +38,9 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
     end
   end
 
-  shared_examples '404 when model experiments is unavailable' do
+  shared_examples 'requires read_model_experiments' do
     context 'when user does not have access' do
-      let(:model_experiments_enabled) { false }
+      let(:read_model_experiments) { false }
 
       it_behaves_like 'renders 404'
     end
@@ -61,7 +65,7 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
     end
 
     it_behaves_like '404 if candidate does not exist'
-    it_behaves_like '404 when model experiments is unavailable'
+    it_behaves_like 'requires read_model_experiments'
   end
 
   describe 'DELETE #destroy' do
@@ -83,7 +87,14 @@ RSpec.describe Projects::Ml::CandidatesController, feature_category: :mlops do
     end
 
     it_behaves_like '404 if candidate does not exist'
-    it_behaves_like '404 when model experiments is unavailable'
+
+    describe 'requires write_model_experiments' do
+      let(:write_model_experiments) { false }
+
+      it 'is 404' do
+        expect(response).to have_gitlab_http_status(:not_found)
+      end
+    end
   end
 
   private