From 2ac55c898f58e2e941bea4864315f2e6ab2665a6 Mon Sep 17 00:00:00 2001 From: Malaber <32635600+Malaber@users.noreply.github.com> Date: Tue, 7 May 2019 11:36:28 +0200 Subject: [PATCH] Refactors Stage.teams to return empty list if no teams are found --- app/models/stage.rb | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/app/models/stage.rb b/app/models/stage.rb index 48ad20d..0beb43b 100644 --- a/app/models/stage.rb +++ b/app/models/stage.rb @@ -6,8 +6,12 @@ class Stage < ApplicationRecord has_many :groups, dependent: :destroy def teams - return matches.map(&:teams).flatten.uniq unless matches.size.zero? + if groups.size.zero? + return matches.map(&:teams).flatten.uniq + elsif matches.size.zero? + return groups.map(&:teams).flatten.uniq + end - groups.map(&:teams).flatten.uniq unless groups.size.zero? + [] end end