Handle Tracks as resource_type in User#get_roles

2017-07-28 02:05:08 +03:00 · 2017-07-28 02:05:08 +03:00 · 3756eed9ca
commit 3756eed9ca
parent cdd7f614f5
1 changed files with 5 additions and 1 deletions
--- a/app/models/user.rb
+++ b/app/models/user.rb
@ -165,7 +165,11 @@ class User < ActiveRecord::Base
  def get_roles
    result = {}
    roles.each do |role|
-      resource = Conference.find(role.resource_id).short_title
+      resource = if role.resource_type == 'Conference'
+                   Conference.find(role.resource_id).short_title
+                 elsif role.resource_type == 'Track'
+                   Track.find(role.resource_id).name
+                 end
      if result[role.name].nil?
        result[role.name] = [resource]
      else