Skip to content

Commit

Permalink
Add account_id param to GET /api/v1/notifications (mastodon#10796)
Browse files Browse the repository at this point in the history
* Add `from_account` to notifications API

this adds the ability to filter notifications by the account they
originated from

* passing a non-existent user should cause none to be returned

* Fix codeclimate warnings

* fix more codeclimate warnings

* make requested changes:

* use account id instead of user@domain
* name the param `account_id` instead of `from_account`

* Don't use `return` in a lambda
  • Loading branch information
Paul Woolcock authored and hiyuki2578 committed Oct 2, 2019
1 parent df62781 commit 3ddebc6
Show file tree
Hide file tree
Showing 3 changed files with 77 additions and 3 deletions.
6 changes: 5 additions & 1 deletion app/controllers/api/v1/notifications_controller.rb
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ def paginated_notifications
end

def browserable_account_notifications
current_account.notifications.browserable(exclude_types)
current_account.notifications.browserable(exclude_types, from_account)
end

def target_statuses_from_notifications
Expand Down Expand Up @@ -81,6 +81,10 @@ def exclude_types
val
end

def from_account
params[:account_id]
end

def pagination_params(core_params)
params.slice(:limit, :exclude_types).permit(:limit, exclude_types: []).merge(core_params)
end
Expand Down
8 changes: 6 additions & 2 deletions app/models/notification.rb
Original file line number Diff line number Diff line change
Expand Up @@ -41,9 +41,13 @@ class Notification < ApplicationRecord
validates :account_id, uniqueness: { scope: [:activity_type, :activity_id] }
validates :activity_type, inclusion: { in: TYPE_CLASS_MAP.values }

scope :browserable, ->(exclude_types = []) {
scope :browserable, ->(exclude_types = [], account_id = nil) {
types = TYPE_CLASS_MAP.values - activity_types_from_types(exclude_types + [:follow_request])
where(activity_type: types)
if account_id.nil?
where(activity_type: types)
else
where(activity_type: types, from_account_id: account_id)
end
}

cache_associated :from_account, status: STATUS_INCLUDES, mention: [status: STATUS_INCLUDES], favourite: [:account, status: STATUS_INCLUDES], follow: :account, poll: [status: STATUS_INCLUDES]
Expand Down
66 changes: 66 additions & 0 deletions spec/controllers/api/v1/notifications_controller_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
let(:user) { Fabricate(:user, account: Fabricate(:account, username: 'alice')) }
let(:token) { Fabricate(:accessible_access_token, resource_owner_id: user.id, scopes: scopes) }
let(:other) { Fabricate(:user, account: Fabricate(:account, username: 'bob')) }
let(:third) { Fabricate(:user, account: Fabricate(:account, username: 'carol')) }

before do
allow(controller).to receive(:doorkeeper_token) { token }
Expand Down Expand Up @@ -55,6 +56,7 @@
mentioning_status = PostStatusService.new.call(other.account, text: 'Hello @alice')
@mention_from_status = mentioning_status.mentions.first
@favourite = FavouriteService.new.call(other.account, first_status)
@second_favourite = FavouriteService.new.call(third.account, first_status)
@follow = FollowService.new.call(other.account, 'alice')
end

Expand Down Expand Up @@ -84,6 +86,66 @@
end
end

describe 'from specified user' do
before do
get :index, params: { account_id: third.account.id }
end

it 'returns http success' do
expect(response).to have_http_status(200)
end

it 'includes favourite' do
expect(assigns(:notifications).map(&:activity)).to include(@second_favourite)
end

it 'excludes favourite' do
expect(assigns(:notifications).map(&:activity)).to_not include(@favourite)
end

it 'excludes mention' do
expect(assigns(:notifications).map(&:activity)).to_not include(@mention_from_status)
end

it 'excludes reblog' do
expect(assigns(:notifications).map(&:activity)).to_not include(@reblog_of_first_status)
end

it 'excludes follow' do
expect(assigns(:notifications).map(&:activity)).to_not include(@follow)
end
end

describe 'from nonexistent user' do
before do
get :index, params: { account_id: 'foo' }
end

it 'returns http success' do
expect(response).to have_http_status(200)
end

it 'excludes favourite' do
expect(assigns(:notifications).map(&:activity)).to_not include(@favourite)
end

it 'excludes second favourite' do
expect(assigns(:notifications).map(&:activity)).to_not include(@second_favourite)
end

it 'excludes mention' do
expect(assigns(:notifications).map(&:activity)).to_not include(@mention_from_status)
end

it 'excludes reblog' do
expect(assigns(:notifications).map(&:activity)).to_not include(@reblog_of_first_status)
end

it 'excludes follow' do
expect(assigns(:notifications).map(&:activity)).to_not include(@follow)
end
end

describe 'with excluded mentions' do
before do
get :index, params: { exclude_types: ['mention'] }
Expand All @@ -105,6 +167,10 @@
expect(assigns(:notifications).map(&:activity)).to include(@favourite)
end

it 'includes third favourite' do
expect(assigns(:notifications).map(&:activity)).to include(@second_favourite)
end

it 'includes follow' do
expect(assigns(:notifications).map(&:activity)).to include(@follow)
end
Expand Down

0 comments on commit 3ddebc6

Please sign in to comment.