spec/unit/ability_spec.rb in hydra-access-controls-5.1.0 vs spec/unit/ability_spec.rb in hydra-access-controls-5.2.0

@@ -243,9 +243,33 @@
       subject.can?(:create, Hydra::AdminPolicy).should be_true
     end
 
   end
 
+  describe "custom method" do
+    before do
+      class MyAbility
+        include Hydra::Ability
+        self.ability_logic +=[:setup_my_permissions]
+
+        def setup_my_permissions
+          can :accept, ActiveFedora::Base
+        end
+      end
+    end
+
+    after do
+      Object.send(:remove_const, :MyAbility)
+    end
+
+    subject { MyAbility.new(FactoryGirl.create(:staff)) }
+
+    it "should be set the custom permission" do
+      subject.can?(:accept, ActiveFedora::Base).should be_true
+    end
+
+  end
+
   #
   # Policy-based Access Controls
   #
   describe "When accessing assets with Policies associated" do
     before do