Fix study group policy

2020-04-16 12:27:59 +02:00
parent a2bb0940eb
commit f4d6533c84
1 changed files with 3 additions and 2 deletions
--- a/app/policies/application_policy.rb
+++ b/app/policies/application_policy.rb
@ -31,10 +31,11 @@ class ApplicationPolicy
      return false if study_group.blank?

      users_in_same_study_group = study_group.users
+    elsif @record.respond_to? :users # e.g. study_group
+      users_in_same_study_group = @record.users
    else # e.g. exercise
      study_groups = @record.user.study_groups
-      users_in_same_study_group = study_groups.collect{ |study_group|
-        study_group.users}.flatten
+      users_in_same_study_group = study_groups.collect(&:users).flatten
    end

    users_in_same_study_group.include? @user