diff options
Diffstat (limited to 'spec/models/ml/experiment_spec.rb')
-rw-r--r-- | spec/models/ml/experiment_spec.rb | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/spec/models/ml/experiment_spec.rb b/spec/models/ml/experiment_spec.rb index 52e9f9217f5..c75331a2ab5 100644 --- a/spec/models/ml/experiment_spec.rb +++ b/spec/models/ml/experiment_spec.rb @@ -57,4 +57,21 @@ RSpec.describe Ml::Experiment do it { is_expected.to be_empty } end end + + describe '#with_candidate_count' do + let_it_be(:exp3) do + create(:ml_experiments, project: exp.project).tap do |e| + create_list(:ml_candidates, 3, experiment: e, user: nil) + create(:ml_candidates, experiment: exp2, user: nil) + end + end + + subject { described_class.with_candidate_count.to_h { |e| [e.id, e.candidate_count] } } + + it 'fetches the candidate count', :aggregate_failures do + expect(subject[exp.id]).to eq(0) + expect(subject[exp2.id]).to eq(1) + expect(subject[exp3.id]).to eq(3) + end + end end |