spec/presenters/ml/model_version_presenter_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Ml::ModelVersionPresenter, feature_category: :mlops do
  let_it_be(:project) { build_stubbed(:project) }
  let_it_be(:model) { build_stubbed(:ml_models, name: 'a_model', project: project) }
  let_it_be(:model_version) { build_stubbed(:ml_model_versions, model: model, version: '1.1.1') }
  let_it_be(:presenter) { model_version.present }

  describe '.display_name' do
    subject { presenter.display_name }

    it { is_expected.to eq('a_model / 1.1.1') }
  end

  describe '#path' do
    subject { model_version.present.path }

    it { is_expected.to eq("/#{project.full_path}/-/ml/models/#{model.id}/versions/#{model_version.id}") }
  end
end