1 files changed, 49 insertions, 0 deletions
diff --git a/spec/services/ml/find_or_create_model_version_service_spec.rb b/spec/services/ml/find_or_create_model_version_service_spec.rb
new file mode 100644
index 00000000000..1211a9b1165
--- /dev/null
+++ b/spec/services/ml/find_or_create_model_version_service_spec.rb
@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe ::Ml::FindOrCreateModelVersionService, feature_category: :mlops do
+  let_it_be(:existing_version) { create(:ml_model_versions) }
+  let_it_be(:another_project) { create(:project) }
+
+  let(:package) { nil }
+
+  let(:params) do
+    {
+      model_name: name,
+      version: version,
+      package: package
+    }
+  end
+
+  subject(:model_version) { described_class.new(project, params).execute }
+
+  describe '#execute' do
+    context 'when model version exists' do
+      let(:name) { existing_version.name }
+      let(:version) { existing_version.version }
+      let(:project) { existing_version.project }
+
+      it 'returns existing model version', :aggregate_failures do
+        expect { model_version }.to change { Ml::ModelVersion.count }.by(0)
+        expect(model_version).to eq(existing_version)
+      end
+    end
+
+    context 'when model version does not exist' do
+      let(:project) { existing_version.project }
+      let(:name) { 'a_new_model' }
+      let(:version) { '2.0.0' }
+
+      let(:package) { create(:ml_model_package, project: project, name: name, version: version) }
+
+      it 'creates a new model version', :aggregate_failures do
+        expect { model_version }.to change { Ml::ModelVersion.count }
+
+        expect(model_version.name).to eq(name)
+        expect(model_version.version).to eq(version)
+        expect(model_version.package).to eq(package)
+      end
+    end
+  end
+end