Merge pull request #91 from etehtsea/fix-arg-error

Fix OpenStruct serializing

Author: dblock

CHANGELOG.md

@@ -2,6 +2,7 @@ Next Release
 ============
 
 * Your contribution here.
+* [#91](https://github.com/intridea/grape-entity/pull/91): Fix OpenStruct serializing - [@etehtsea](https://github.com/etehtsea).
 
 0.4.4 (2014-08-17)
 ==================

lib/grape_entity/entity.rb

@@ -554,7 +554,7 @@ def delegate_attribute(attribute)
         elsif object.respond_to?(:fetch, true)
           object.fetch(name)
         else
-          raise ArgumentError
+          object.send(name)
         end
       end
     end

spec/grape_entity/entity_spec.rb

@@ -1,4 +1,5 @@
 require 'spec_helper'
+require 'ostruct'
 
 describe Grape::Entity do
 
@@ -442,6 +443,19 @@ class Parent < Person
         representation = subject.represent({ awesome: true }, serializable: true)
         expect(representation).to eq(awesome: true)
       end
+
+      it 'returns a serialized hash of an OpenStruct' do
+        subject.expose(:awesome)
+        representation = subject.represent(OpenStruct.new, serializable: true)
+        expect(representation).to eq(awesome: nil)
+      end
+
+      it 'raises error if field not found' do
+        subject.expose(:awesome)
+        expect {
+          subject.represent(Object.new, serializable: true)
+        }.to raise_error(NoMethodError)
+      end
     end
 
     describe '.present_collection' do