spec/tests/policy_spec.rb in attestor-2.2.0 vs spec/tests/policy_spec.rb in attestor-2.2.1

@@ -35,11 +35,17 @@
     it "builds the policy" do
       expect(subject.new(:baz)).to be_kind_of described_class
     end
 
     it "yields the block in class scope" do
-      subject = described_class.new(:foo) { attr_reader :bar }
+      subject = described_class.new(:foo) do
+        attr_reader :bar
+        def foobar; end
+        def self.barfoo; end
+      end
       expect(subject.new(:baz)).to respond_to :bar
+      expect(subject.new(:baz)).to respond_to :foobar
+      expect(subject.new(:baz).class).to respond_to :barfoo
     end
 
   end # describe .new
 
   describe ".included" do