diff --git a/spec/factories/groups.rb b/spec/factories/groups.rb index 839c230..a91ce1b 100644 --- a/spec/factories/groups.rb +++ b/spec/factories/groups.rb @@ -3,6 +3,7 @@ FactoryBot.define do factory :group do transient do + matches { nil } match_count { 4 } end @@ -10,7 +11,11 @@ FactoryBot.define do stage after(:create) do |group, evaluator| - create_list(:group_match, evaluator.match_count, group: group) + if evaluator.matches.nil? + create_list(:group_match, evaluator.match_count, group: group) + else + evaluator.matches + end end end end diff --git a/spec/factories/stages.rb b/spec/factories/stages.rb index e8dce2d..22a0ccd 100644 --- a/spec/factories/stages.rb +++ b/spec/factories/stages.rb @@ -6,10 +6,15 @@ FactoryBot.define do factory :group_stage do level { -1 } transient do + groups { nil } group_count { 4 } end after(:create) do |stage, evaluator| - stage.groups = create_list(:group, evaluator.group_count) + stage.groups = if evaluator.groups.nil? + create_list(:group, evaluator.group_count) + else + evaluator.groups + end end end