blob: 697b57a51c1a3dcdf749899fc02febc72e3efbbf (
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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Ml::ModelsIndexPresenter, feature_category: :mlops do
let_it_be(:project) { build_stubbed(:project) }
let_it_be(:model1) { build_stubbed(:ml_model_package, project: project) }
let_it_be(:model2) { build_stubbed(:ml_model_package, project: project) }
let_it_be(:models) do
[model1, model2]
end
describe '#execute' do
subject { Gitlab::Json.parse(described_class.new(models).present)['models'] }
it 'presents models correctly' do
expected_models = [
{
'name' => model1.name,
'version' => model1.version,
'path' => "/#{project.full_path}/-/packages/#{model1.id}"
},
{
'name' => model2.name,
'version' => model2.version,
'path' => "/#{project.full_path}/-/packages/#{model2.id}"
}
]
is_expected.to match_array(expected_models)
end
end
end
|