Store back message_key on Grape::Exceptions::Validation

Author: Marieke GUEYE

CHANGELOG.md

@@ -3,6 +3,9 @@
 
 * Your contribution here.
 
+#### Features
+* [#1366](https://github.com/ruby-grape/grape/pull/1366): Store back `message_key` on `Grape::Exceptions::Validation` - [@mkou](https://github.com/mkou).
+
 0.16.2 (4/12/2016)
 ==================
 

lib/grape/exceptions/validation.rb

@@ -9,6 +9,7 @@ class Validation < Grape::Exceptions::Base
       def initialize(args = {})
         fail 'Params are missing:' unless args.key? :params
         @params = args[:params]
+        @message_key = args[:message] if args.key?(:message) && args[:message].is_a?(Symbol)
         args[:message] = translate_message(args[:message]) if args.key? :message
         super
       end

spec/grape/exceptions/validation_spec.rb

@@ -4,4 +4,14 @@
   it 'fails when params are missing' do
     expect { Grape::Exceptions::Validation.new(message: 'presence') }.to raise_error(RuntimeError, 'Params are missing:')
   end
+  context 'when message is a symbol' do
+    it 'stores message_key' do
+      expect(Grape::Exceptions::Validation.new(params: ['id'], message: :presence).message_key).to eq(:presence)
+    end
+  end
+  context 'when message is a String' do
+    it 'does not store the message_key' do
+      expect(Grape::Exceptions::Validation.new(params: ['id'], message: 'presence').message_key).to eq(nil)
+    end
+  end
 end