Remove overridden :key attribute. And test key method.

Author: James McCarthy

lib/grape_entity/exposure/base.rb

@@ -4,7 +4,7 @@ module Grape
   class Entity
     module Exposure
       class Base
-        attr_reader :attribute, :key, :is_safe, :documentation, :conditions, :for_merge
+        attr_reader :attribute, :is_safe, :documentation, :conditions, :for_merge
 
         def self.new(attribute, options, conditions, *args, &block)
           super(attribute, options, conditions).tap { |e| e.setup(*args, &block) }
@@ -106,7 +106,7 @@ def attr_path(entity, options)
         end
 
         def key(entity = nil)
-          @key.respond_to?(:call) ? @key.call(entity).try(:to_sym) : @key
+          @key.respond_to?(:call) ? @key.call(entity) : @key
         end
 
         def with_attr_path(entity, options)

spec/grape_entity/exposure_spec.rb

@@ -26,12 +26,17 @@
   describe '#key' do
     it 'returns the attribute if no :as is set' do
       fresh_class.expose :name
-      expect(subject.key).to eq :name
+      expect(subject.key(entity)).to eq :name
     end
 
     it 'returns the :as alias if one exists' do
       fresh_class.expose :name, as: :nombre
-      expect(subject.key).to eq :nombre
+      expect(subject.key(entity)).to eq :nombre
+    end
+
+    it 'returns the result if :as is a proc' do
+      fresh_class.expose :name, as: -> (entity) { entity.object.name.reverse }
+      expect(subject.key(entity)).to eq(model.name.reverse)
     end
   end