Welcome to mirror list, hosted at ThFree Co, Russian Federation.

package_spec.rb « ml_model « packages « models « spec - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 6a3271976720a17d1e27bf8229a56ba29b3f9922 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# frozen_string_literal: true

require 'spec_helper'

RSpec.describe Packages::MlModel::Package, feature_category: :mlops do
  let_it_be(:project) { create(:project) }
  let_it_be(:ml_model) { create(:ml_model_package, project: project) }
  let_it_be(:generic_package) { create(:generic_package, project: project) }

  describe 'associations' do
    it { is_expected.to have_one(:model_version) }
  end

  describe 'all' do
    it 'fetches only ml_model packages' do
      expect(described_class.all).to eq([ml_model])
    end
  end

  describe '#valid?' do
    using RSpec::Parameterized::TableSyntax

    let_it_be(:valid_version) { '1.0.0' }
    let_it_be(:valid_name) { 'some_model' }

    let(:version) { valid_version }
    let(:name) { valid_name }

    let(:ml_model) { described_class.new(version: version, name: name, project: project) }

    subject(:errors) do
      ml_model.validate
      ml_model.errors
    end

    it { expect(ml_model).to validate_presence_of(:name) }
    it { expect(ml_model).to validate_presence_of(:version) }

    it 'validates a valid ml_model package' do
      expect(errors).to be_empty
    end

    describe 'name' do
      where(:ctx, :name) do
        'name is blank'                     | ''
        'name is nil'                       | nil
        'name is not valid package name'    | '!!()()'
        'name is too large'                 | ('a' * 256)
      end
      with_them do
        it { expect(errors).to include(:name) }
      end
    end

    describe 'version' do
      where(:ctx, :version) do
        'version is blank'            | ''
        'version is nil'              | nil
        'version is not valid semver' | 'v1.0.0'
        'version is too large'        | ('a' * 256)
      end
      with_them do
        it { expect(errors).to include(:version) }
      end
    end
  end
end