diff --git a/app/api/entities.rb b/app/api/entities.rb index f945629..75ee4d5 100644 --- a/app/api/entities.rb +++ b/app/api/entities.rb @@ -39,7 +39,7 @@ module Api end def self.expose_optional_field?(field, options = {}) - return false if options[:params].nil? + return false if options[:params].nil? optional_fields = options[:params][:optional_fields] || [] optional_fields.include?(field.to_s) end @@ -160,6 +160,9 @@ module Api expose :articles_count do |person, options| person.articles.count end + expose :friends_count do |person, options| + person.friends.size + end end class Enterprise < Profile @@ -316,7 +319,7 @@ module Api end expose :params, :if => lambda { |activity, options| activity.kind_of?(ActionTracker::Record)} expose :content, :if => lambda { |activity, options| activity.kind_of?(Scrap)} - expose :verb do |activity, options| + expose :verb do |activity, options| activity.kind_of?(Scrap) ? 'leave_scrap' : activity.verb end diff --git a/test/api/people_test.rb b/test/api/people_test.rb index 7ac1cb2..b4866b8 100644 --- a/test/api/people_test.rb +++ b/test/api/people_test.rb @@ -369,7 +369,7 @@ class PeopleTest < ActiveSupport::TestCase assert_equal "www.blog.org", json['person']['additional_data']['Custom Blog'] end - PERSON_ATTRIBUTES = %w(vote_count comments_count articles_count following_articles_count) + PERSON_ATTRIBUTES = %w(vote_count comments_count articles_count following_articles_count friends_count) PERSON_ATTRIBUTES.map do |attribute| define_method "test_should_not_expose_#{attribute}_attribute_in_person_enpoint_if_field_parameter_does_not_contain_the_attribute" do -- libgit2 0.21.2