Calling #could without a block raises an error

isaacsanders · Feb 3, 2012 · 096c78a · 096c78a
1 parent f897275
commit 096c78a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/lib/reuser/role.rb b/lib/reuser/role.rb
@@ -23,6 +23,7 @@ def can? permission
     end
 
     def could permission, &block
+      raise "#could requires a block" unless block_given?
       @permissions[permission] = block
     end
 

diff --git a/spec/reuser/role_spec.rb b/spec/reuser/role_spec.rb
@@ -33,6 +33,10 @@
     subject.can?(:write).should be_true
   end
 
+  it 'you need to supply #could with a test block' do
+    lambda { subject.could(:read) }.should raise_error
+  end
+
   it 'conditional permissions are added and checked with #could and #could?' do
     subject.can?(:write).should be_false
     subject.could :write do |language|