diff --git a/lib/graphql/preload/loader.rb b/lib/graphql/preload/loader.rb index bd0554b..08182e3 100644 --- a/lib/graphql/preload/loader.rb +++ b/lib/graphql/preload/loader.rb @@ -39,7 +39,10 @@ def perform(records) end private def preload_scope - scope if scope.try(:klass) == model.reflect_on_association(association).klass + return nil unless scope + reflection = model.reflect_on_association(association) + raise ArgumentError, 'Cannot specify preload_scope for polymorphic associations' if reflection.polymorphic? + scope if scope.try(:klass) == reflection.klass end private def validate_association