Skip to content

Commit

Permalink
Add more test coverage for admin policies
Browse files Browse the repository at this point in the history
  • Loading branch information
martinemde committed Nov 21, 2024
1 parent b5564c9 commit f8115a3
Show file tree
Hide file tree
Showing 10 changed files with 137 additions and 0 deletions.
9 changes: 9 additions & 0 deletions app/policies/admin/nil_class_policy.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
class Admin::NilClassPolicy < Admin::ApplicationPolicy
class Scope < Admin::ApplicationPolicy::Scope
def resolve
raise Pundit::NotDefinedError, "Cannot scope NilClass"
end
end

# fallback to parent policy which rejects all actions
end
6 changes: 6 additions & 0 deletions test/policies/admin/api_key_policy_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,12 @@ def test_scope
).to_a
end

def test_associations
assert_association @admin, @api_key, :api_key_rubygem_scope, Admin::ApiKeyPolicy
assert_association @admin, @api_key, :ownership, Admin::OwnershipPolicy
assert_association @admin, @api_key, :oidc_id_token, Admin::OIDC::IdTokenPolicy
end

def test_avo_index
refute_authorizes @admin, ApiKey, :avo_index?
refute_authorizes @non_admin, ApiKey, :avo_index?
Expand Down
27 changes: 27 additions & 0 deletions test/policies/admin/application_policy_test.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
require "test_helper"

class Admin::ApplicationPolicyTest < AdminPolicyTestCase
should "onle inherit from Admin::ApplicationPolicy in Admin:: namespace" do
Admin.constants.each do |const|
next if const == :ApplicationPolicy
next unless const.to_s.end_with?("Policy")

klass = Admin.const_get(const)

assert_operator klass, :<, Admin::ApplicationPolicy, "#{const} does not inherit from Admin::ApplicationPolicy"
assert_operator klass::Scope, :<, Admin::ApplicationPolicy::Scope, "#{const}::Scope does not inherit from Admin::ApplicationPolicy::Scope"
end
end

should "not authorize any avo action" do
refute_authorizes nil, nil, :avo_index?
refute_authorizes nil, nil, :avo_show?
refute_authorizes nil, nil, :avo_create?
refute_authorizes nil, nil, :avo_new?
refute_authorizes nil, nil, :avo_update?
refute_authorizes nil, nil, :avo_edit?
refute_authorizes nil, nil, :avo_destroy?
refute_authorizes nil, nil, :act_on?
refute_authorizes nil, nil, :avo_search?
end
end
4 changes: 4 additions & 0 deletions test/policies/admin/deletion_policy_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,10 @@ class Admin::DeletionPolicyTest < AdminPolicyTestCase
@non_admin = create(:admin_github_user)
end

def test_associations
assert_association @admin, @deletion, :version, Admin::VersionPolicy
end

def test_scope
assert_equal [@deletion], policy_scope!(
@admin,
Expand Down
4 changes: 4 additions & 0 deletions test/policies/admin/geoip_info_policy_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,10 @@ class Admin::GeoipInfoPolicyTest < AdminPolicyTestCase
@non_admin = create(:admin_github_user)
end

def test_associations
assert_association @admin, @geoip_info, :ip_addresses, Admin::IpAddressPolicy
end

def test_scope
assert_equal [@geoip_info], policy_scope!(
@admin,
Expand Down
6 changes: 6 additions & 0 deletions test/policies/admin/ip_address_policy_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,12 @@ class Admin::IpAddressPolicyTest < AdminPolicyTestCase
@non_admin = create(:admin_github_user)
end

def test_associations
assert_association @admin, @ip_address, :user_events, Admin::Events::UserEventPolicy
assert_association @admin, @ip_address, :rubygem_events, Admin::Events::RubygemEventPolicy
assert_association @admin, @ip_address, :organization_events, Admin::Events::OrganizationEventPolicy
end

def test_scope
assert_equal [@ip_address], policy_scope!(
@admin,
Expand Down
32 changes: 32 additions & 0 deletions test/policies/admin/nil_class_policy_test.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
require "test_helper"

class Admin::NilClassPolicyTest < AdminPolicyTestCase
def policy!(api_key)
Pundit.policy!(api_key, [:api, nil])
end

should "inherit from Admin::ApplicationPolicy" do
assert_operator Admin::NilClassPolicy, :<, Admin::ApplicationPolicy
assert_operator Admin::NilClassPolicy::Scope, :<, Admin::ApplicationPolicy::Scope
end

context "::Scope.resolve" do
should "raise" do
assert_raises Pundit::NotDefinedError do
Admin::NilClassPolicy::Scope.new(nil, nil).resolve
end
end
end

should "not authorize any avo action" do
refute_authorizes nil, nil, :avo_index?
refute_authorizes nil, nil, :avo_show?
refute_authorizes nil, nil, :avo_create?
refute_authorizes nil, nil, :avo_new?
refute_authorizes nil, nil, :avo_update?
refute_authorizes nil, nil, :avo_edit?
refute_authorizes nil, nil, :avo_destroy?
refute_authorizes nil, nil, :act_on?
refute_authorizes nil, nil, :avo_search?
end
end
5 changes: 5 additions & 0 deletions test/policies/admin/rubygem_policy_test.rb
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,11 @@ def test_scope
@admin,
Rubygem
).to_a

assert_empty policy_scope!(
@non_admin,
Rubygem
).to_a
end

def test_avo_index
Expand Down
38 changes: 38 additions & 0 deletions test/policies/admin/version_policy_test.rb
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
require "test_helper"

class Admin::VersionPolicyTest < AdminPolicyTestCase
setup do
@admin = FactoryBot.create(:admin_github_user, :is_admin)
@non_admin = FactoryBot.create(:admin_github_user)
@version = FactoryBot.create(:version, :yanked)
end

def test_scope
assert_equal [@version], policy_scope!(
@admin,
Version
).to_a
assert_empty policy_scope!(
@non_admin,
Version
).to_a
end

def test_avo_index
assert_authorizes @admin, Version, :avo_index?

refute_authorizes @non_admin, Version, :avo_index?
end

def test_avo_show
assert_authorizes @admin, @version, :avo_show?

refute_authorizes @non_admin, @version, :avo_show?
end

def test_act_on
assert_authorizes @admin, @version, :act_on?

refute_authorizes @non_admin, @version, :act_on?
end
end
6 changes: 6 additions & 0 deletions test/test_helper.rb
Original file line number Diff line number Diff line change
Expand Up @@ -286,6 +286,12 @@ def refute_authorizes(user, record, action)
end
end

def assert_association(user, record, association, _policy_class)
%w[create attach detach destroy edit].each do |action|
refute_authorizes(user, record, :"#{action}_#{association}?")
end
end

def policy_class
nil
end
Expand Down

0 comments on commit f8115a3

Please sign in to comment.