blob: 79914ff8b22ce5b95a539fbefa271fd38bb520ff (
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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe ::Ml::DestroyModelService, feature_category: :mlops do
let_it_be(:user) { create(:user) }
let_it_be(:model) { create(:ml_models, :with_latest_version_and_package) }
let(:service) { described_class.new(model, user) }
describe '#execute' do
context 'when model name does not exist in the project' do
it 'returns nil' do
allow(model).to receive(:destroy).and_return(false)
expect(service.execute).to be nil
end
end
context 'when a model exists' do
it 'destroys the model' do
expect(Packages::MarkPackagesForDestructionService).to receive(:new).with(packages: model.all_packages,
current_user: user).and_return(instance_double('Packages::MarkPackagesForDestructionService').tap do |service|
expect(service).to receive(:execute)
end)
expect { service.execute }.to change { Ml::Model.count }.by(-1).and change { Ml::ModelVersion.count }.by(-1)
end
end
end
end
|