From bf811e4d4a8626579fc5187f465cdf1e79a32e10 Mon Sep 17 00:00:00 2001 From: Matt Jankowski Date: Wed, 31 May 2017 14:27:34 -0400 Subject: [PATCH] Improve spec coverage and clean up api/v1/mutes controller (#3481) --- app/controllers/api/v1/mutes_controller.rb | 56 +++++++++++++++---- .../api/v1/mutes_controller_spec.rb | 4 +- 2 files changed, 49 insertions(+), 11 deletions(-) diff --git a/app/controllers/api/v1/mutes_controller.rb b/app/controllers/api/v1/mutes_controller.rb index 53eb2554b6..b9ac741765 100644 --- a/app/controllers/api/v1/mutes_controller.rb +++ b/app/controllers/api/v1/mutes_controller.rb @@ -3,24 +3,60 @@ class Api::V1::MutesController < ApiController before_action -> { doorkeeper_authorize! :follow } before_action :require_user! + after_action :insert_pagination_headers respond_to :json def index - @accounts = Account.includes(:muted_by) - .references(:muted_by) - .merge(Mute.where(account: current_account) - .paginate_by_max_id(limit_param(DEFAULT_ACCOUNTS_LIMIT), params[:max_id], params[:since_id])) - .to_a - - next_path = api_v1_mutes_url(pagination_params(max_id: @accounts.last.muted_by_ids.last)) if @accounts.size == limit_param(DEFAULT_ACCOUNTS_LIMIT) - prev_path = api_v1_mutes_url(pagination_params(since_id: @accounts.first.muted_by_ids.first)) unless @accounts.empty? - - set_pagination_headers(next_path, prev_path) + @accounts = load_accounts end private + def load_accounts + default_accounts.merge(paginated_mutes).to_a + end + + def default_accounts + Account.includes(:muted_by).references(:muted_by) + end + + def paginated_mutes + Mute.where(account: current_account).paginate_by_max_id( + limit_param(DEFAULT_ACCOUNTS_LIMIT), + params[:max_id], + params[:since_id] + ) + end + + def insert_pagination_headers + set_pagination_headers(next_path, prev_path) + end + + def next_path + if records_continue? + api_v1_mutes_url pagination_params(max_id: pagination_max_id) + end + end + + def prev_path + unless @accounts.empty? + api_v1_mutes_url pagination_params(since_id: pagination_since_id) + end + end + + def pagination_max_id + @accounts.last.muted_by_ids.last + end + + def pagination_since_id + @accounts.first.muted_by_ids.first + end + + def records_continue? + @accounts.size == limit_param(DEFAULT_ACCOUNTS_LIMIT) + end + def pagination_params(core_params) params.permit(:limit).merge(core_params) end diff --git a/spec/controllers/api/v1/mutes_controller_spec.rb b/spec/controllers/api/v1/mutes_controller_spec.rb index be8a5e7dd2..85aad43841 100644 --- a/spec/controllers/api/v1/mutes_controller_spec.rb +++ b/spec/controllers/api/v1/mutes_controller_spec.rb @@ -7,12 +7,14 @@ RSpec.describe Api::V1::MutesController, type: :controller do let(:token) { double acceptable?: true, resource_owner_id: user.id } before do + Fabricate(:mute, account: user.account) allow(controller).to receive(:doorkeeper_token) { token } end describe 'GET #index' do it 'returns http success' do - get :index + get :index, params: { limit: 1 } + expect(response).to have_http_status(:success) end end