NotAuthorizedError can be initialized with a string

lib/pundit.rb

@@ -14,11 +14,15 @@ module Pundit
     attr_reader :query, :record, :policy
 
     def initialize(options = {})
-      @query  = options[:query]
+      if options.is_a? String
-      @record = options[:record]
+        message = options
-      @policy = options[:policy]
+      else
+        @query  = options[:query]
+        @record = options[:record]
+        @policy = options[:policy]
 
-      message = options.fetch(:message) { "not allowed to #{query} this #{record.inspect}" }
+        message = options.fetch(:message) { "not allowed to #{query} this #{record.inspect}" }
+      end
 
       super(message)
     end

spec/pundit_spec.rb

@@ -344,4 +344,11 @@ describe Pundit do
       expect(Controller.new(double, params).permitted_attributes(post)).to eq({ 'votes' => 5 })
     end
   end
+
+  describe "Pundit::NotAuthorizedError" do
+    it "can be initialized with a string as message" do
+      error = Pundit::NotAuthorizedError.new("must be logged in")
+      expect(error.message).to eq "must be logged in"
+    end
+  end
 end