1 files changed, 37 insertions, 0 deletions
diff --git a/spec/helpers/projects/ml/model_registry_helper_spec.rb b/spec/helpers/projects/ml/model_registry_helper_spec.rb
new file mode 100644
index 00000000000..2180d4388ca
--- /dev/null
+++ b/spec/helpers/projects/ml/model_registry_helper_spec.rb
@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+require 'rspec'
+
+require 'spec_helper'
+require 'mime/types'
+
+RSpec.describe Projects::Ml::ModelRegistryHelper, feature_category: :mlops do
+  let_it_be(:project) { build_stubbed(:project) }
+  let_it_be(:user) { project.owner }
+
+  describe '#index_ml_model_data' do
+    subject(:parsed) { Gitlab::Json.parse(helper.index_ml_model_data(project, user)) }
+
+    it 'generates the correct data' do
+      is_expected.to eq({
+        'projectPath' => project.full_path,
+        'createModelPath' => "/#{project.full_path}/-/ml/models/new",
+        'canWriteModelRegistry' => true,
+        'mlflowTrackingUrl' => "http://localhost/api/v4/projects/#{project.id}/ml/mlflow/api/2.0/mlflow/"
+      })
+    end
+
+    context 'when user does not have write access to model registry' do
+      before do
+        allow(Ability).to receive(:allowed?).and_call_original
+        allow(Ability).to receive(:allowed?)
+                            .with(user, :write_model_registry, project)
+                            .and_return(false)
+      end
+
+      it 'canWriteModelRegistry is false' do
+        expect(parsed['canWriteModelRegistry']).to eq(false)
+      end
+    end
+  end
+end