106 lines
2.8 KiB
Ruby
106 lines
2.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Gitlab::Graphql::Authorize::AuthorizeResource do
|
|
let(:fake_class) do
|
|
Class.new do
|
|
include Gitlab::Graphql::Authorize::AuthorizeResource
|
|
|
|
attr_reader :user, :found_object
|
|
|
|
authorize :read_the_thing
|
|
|
|
def initialize(user, found_object)
|
|
@user = user
|
|
@found_object = found_object
|
|
end
|
|
|
|
def find_object
|
|
found_object
|
|
end
|
|
|
|
def current_user
|
|
user
|
|
end
|
|
|
|
def context
|
|
{ current_user: user }
|
|
end
|
|
|
|
def self.authorization
|
|
@authorization ||= ::Gitlab::Graphql::Authorize::ObjectAuthorization.new(required_permissions)
|
|
end
|
|
end
|
|
end
|
|
|
|
let(:user) { build(:user) }
|
|
let(:project) { build(:project) }
|
|
|
|
subject(:loading_resource) { fake_class.new(user, project) }
|
|
|
|
before do
|
|
# don't allow anything by default
|
|
allow(Ability).to receive(:allowed?).and_return(false)
|
|
end
|
|
|
|
context 'when the user is allowed to perform the action' do
|
|
before do
|
|
allow(Ability).to receive(:allowed?).with(user, :read_the_thing, project).and_return(true)
|
|
end
|
|
|
|
describe '#authorized_find!' do
|
|
it 'returns the object' do
|
|
expect(loading_resource.authorized_find!).to eq(project)
|
|
end
|
|
end
|
|
|
|
describe '#authorize!' do
|
|
it 'does not raise an error' do
|
|
expect { loading_resource.authorize!(project) }.not_to raise_error
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when the user is not allowed to perform the action' do
|
|
describe '#authorized_find!' do
|
|
it 'raises an error' do
|
|
expect { loading_resource.authorized_find! }.to raise_error(Gitlab::Graphql::Errors::ResourceNotAvailable)
|
|
end
|
|
end
|
|
|
|
describe '#authorize!' do
|
|
it 'raises an error' do
|
|
expect { loading_resource.authorize!(project) }.to raise_error(Gitlab::Graphql::Errors::ResourceNotAvailable)
|
|
end
|
|
end
|
|
end
|
|
|
|
context 'when the class does not define #find_object' do
|
|
let(:fake_class) do
|
|
Class.new { include Gitlab::Graphql::Authorize::AuthorizeResource }
|
|
end
|
|
|
|
it 'raises a comprehensive error message' do
|
|
expect { fake_class.new.find_object }.to raise_error(/Implement #find_object in #{fake_class.name}/)
|
|
end
|
|
end
|
|
|
|
describe '#authorize' do
|
|
it 'adds permissions from subclasses to those of superclasses when used on classes' do
|
|
base_class = Class.new do
|
|
include Gitlab::Graphql::Authorize::AuthorizeResource
|
|
|
|
authorize :base_authorization
|
|
end
|
|
|
|
sub_class = Class.new(base_class) do
|
|
authorize :sub_authorization
|
|
end
|
|
|
|
expect(base_class.required_permissions).to contain_exactly(:base_authorization)
|
|
expect(sub_class.required_permissions)
|
|
.to contain_exactly(:base_authorization, :sub_authorization)
|
|
end
|
|
end
|
|
end
|