108 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			108 lines
		
	
	
		
			2.8 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
# frozen_string_literal: true
 | 
						|
 | 
						|
require 'rails_helper'
 | 
						|
 | 
						|
describe Api::BaseController do
 | 
						|
  before do
 | 
						|
    stub_const('FakeService', Class.new)
 | 
						|
  end
 | 
						|
 | 
						|
  controller do
 | 
						|
    def success
 | 
						|
      head 200
 | 
						|
    end
 | 
						|
 | 
						|
    def failure
 | 
						|
      FakeService.new
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  it 'returns private cache control headers by default' do
 | 
						|
    routes.draw { get 'success' => 'api/base#success' }
 | 
						|
    get :success
 | 
						|
    expect(response.headers['Cache-Control']).to include('private, no-store')
 | 
						|
  end
 | 
						|
 | 
						|
  describe 'forgery protection' do
 | 
						|
    before do
 | 
						|
      routes.draw { post 'success' => 'api/base#success' }
 | 
						|
    end
 | 
						|
 | 
						|
    it 'does not protect from forgery' do
 | 
						|
      ActionController::Base.allow_forgery_protection = true
 | 
						|
      post :success
 | 
						|
      expect(response).to have_http_status(200)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe 'non-functional accounts handling' do
 | 
						|
    let(:user)  { Fabricate(:user) }
 | 
						|
    let(:token) { Fabricate(:accessible_access_token, resource_owner_id: user.id, scopes: 'read') }
 | 
						|
 | 
						|
    controller do
 | 
						|
      before_action :require_user!
 | 
						|
    end
 | 
						|
 | 
						|
    before do
 | 
						|
      routes.draw { post 'success' => 'api/base#success' }
 | 
						|
      allow(controller).to receive(:doorkeeper_token) { token }
 | 
						|
    end
 | 
						|
 | 
						|
    it 'returns http forbidden for unconfirmed accounts' do
 | 
						|
      user.update(confirmed_at: nil)
 | 
						|
      post :success
 | 
						|
      expect(response).to have_http_status(403)
 | 
						|
    end
 | 
						|
 | 
						|
    it 'returns http forbidden for pending accounts' do
 | 
						|
      user.update(approved: false)
 | 
						|
      post :success
 | 
						|
      expect(response).to have_http_status(403)
 | 
						|
    end
 | 
						|
 | 
						|
    it 'returns http forbidden for disabled accounts' do
 | 
						|
      user.update(disabled: true)
 | 
						|
      post :success
 | 
						|
      expect(response).to have_http_status(403)
 | 
						|
    end
 | 
						|
 | 
						|
    it 'returns http forbidden for suspended accounts' do
 | 
						|
      user.account.suspend!
 | 
						|
      post :success
 | 
						|
      expect(response).to have_http_status(403)
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  describe 'error handling' do
 | 
						|
    before do
 | 
						|
      routes.draw { get 'failure' => 'api/base#failure' }
 | 
						|
    end
 | 
						|
 | 
						|
    {
 | 
						|
      ActiveRecord::RecordInvalid => 422,
 | 
						|
      ActiveRecord::RecordNotFound => 404,
 | 
						|
      ActiveRecord::RecordNotUnique => 422,
 | 
						|
      Date::Error => 422,
 | 
						|
      HTTP::Error => 503,
 | 
						|
      Mastodon::InvalidParameterError => 400,
 | 
						|
      Mastodon::NotPermittedError => 403,
 | 
						|
      Mastodon::RaceConditionError => 503,
 | 
						|
      Mastodon::RateLimitExceededError => 429,
 | 
						|
      Mastodon::UnexpectedResponseError => 503,
 | 
						|
      Mastodon::ValidationError => 422,
 | 
						|
      OpenSSL::SSL::SSLError => 503,
 | 
						|
      Seahorse::Client::NetworkingError => 503,
 | 
						|
      Stoplight::Error::RedLight => 503,
 | 
						|
    }.each do |error, code|
 | 
						|
      it "Handles error class of #{error}" do
 | 
						|
        allow(FakeService).to receive(:new).and_raise(error)
 | 
						|
 | 
						|
        get :failure
 | 
						|
 | 
						|
        expect(response).to have_http_status(code)
 | 
						|
        expect(FakeService).to have_received(:new)
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
end
 |