diff options
Diffstat (limited to 'spec/factories/ml/candidates.rb')
-rw-r--r-- | spec/factories/ml/candidates.rb | 17 |
1 files changed, 6 insertions, 11 deletions
diff --git a/spec/factories/ml/candidates.rb b/spec/factories/ml/candidates.rb index 9bfb78066bd..27b31524b1f 100644 --- a/spec/factories/ml/candidates.rb +++ b/spec/factories/ml/candidates.rb @@ -1,10 +1,9 @@ # frozen_string_literal: true FactoryBot.define do factory :ml_candidates, class: '::Ml::Candidate' do - association :project, factory: :project - association :user - - experiment { association :ml_experiments, project_id: project.id } + project { association :project } + user { project.owner } + experiment { association :ml_experiments, project_id: project.id, user: project.owner } trait :with_metrics_and_params do metrics { Array.new(2) { association(:ml_candidate_metrics, candidate: instance) } } @@ -16,13 +15,9 @@ FactoryBot.define do end trait :with_artifact do - after(:create) do |candidate| - candidate.package = FactoryBot.create( - :generic_package, - name: candidate.package_name, - version: candidate.package_version, - project: candidate.project - ) + artifact do + association(:generic_package, name: instance.package_name, version: instance.package_version || '1', + project: project) end end end |