From 658e769a7f235c74c0af38b6e49b709aa260de2b Mon Sep 17 00:00:00 2001
From: Malaber <32635600+Malaber@users.noreply.github.com>
Date: Sun, 19 May 2019 16:46:45 +0200
Subject: [PATCH] Update groups and stages factory to be used in production

---
 spec/factories/groups.rb | 7 ++++++-
 spec/factories/stages.rb | 7 ++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

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