Add test_with option to values validator

Author: Joe Faber

CHANGELOG.md

@@ -3,6 +3,7 @@
 #### Features
 
 * [#1555](https://github.com/ruby-grape/grape/pull/1555): Added code coverage w/Coveralls - [@dblock](https://github.com/dblock).
+* [#1568](https://github.com/ruby-grape/grape/pull/1568): Add `test_with` option to `values` validator to allow custom checks - [@jlfaber](https://github.com/jlfaber).
 * Your contribution here.
 
 #### Fixes

README.md

@@ -1152,6 +1152,20 @@ params do
 end
 ```
 
+Finally, for even greater control, an explicit validation Proc may be supplied using ```test_with```.
+It will be called with a single argument (the input value), and should return
+a truthy value if the value passes validation. If the input is an array, the Proc will be called
+multiple times, once for each element in the array.
+
+```ruby
+params do
+  requires :number, type: Integer, values: { test_with: ->(v) { v.even? && v < 25 }, message: 'is odd or greater than 25' }
+end
+```
+
+While ```test_with``` is convenient for single cases, consider using [Custom Validators](#custom-validators) in cases where a validation is used more than once.
+
+
 #### `regexp`
 
 Parameters can be restricted to match a specific regular expression with the `:regexp` option. If the value

lib/grape/validations/validators/values.rb

@@ -5,6 +5,8 @@ def initialize(attrs, options, required, scope, opts = {})
         if options.is_a?(Hash)
           @excepts = options[:except]
           @values = options[:value]
+          @test_with = options[:test_with]
+          raise ArgumentError, 'test_with must be a Proc' if @test_with && !@test_with.is_a?(Proc)
         else
           @values = options
         end
@@ -24,6 +26,9 @@ def validate_param!(attr_name, params)
 
         raise Grape::Exceptions::Validation, params: [@scope.full_name(attr_name)], message: message(:values) \
           if !values.nil? && !param_array.all? { |param| values.include?(param) }
+
+        raise Grape::Exceptions::Validation, params: [@scope.full_name(attr_name)], message: message(:values) \
+          if @test_with && !param_array.all? { |param| @test_with.call(param) }
       end
 
       private

spec/grape/validations/validators/values_spec.rb

@@ -174,6 +174,20 @@ class API < Grape::API
           optional :optional, type: Array[String], values: %w(a b c)
         end
         put '/optional_with_array_of_string_values'
+
+        params do
+          requires :type, values: { test_with: ->(v) { ValuesModel.values.include? v } }
+        end
+        get '/test_with' do
+          { type: params[:type] }
+        end
+
+        params do
+          requires :type, values: { test_with: ->(v) { ValuesModel.values.include? v }, message: 'failed check' }
+        end
+        get '/test_with/message' do
+          { type: params[:type] }
+        end
       end
     end
   end
@@ -505,4 +519,36 @@ def app
       expect(last_response.body).to eq({ error: 'type does not have a valid value' }.to_json)
     end
   end
+
+  context 'custom validation using test_with' do
+    it 'accepts a single valid value' do
+      get '/test_with', type: 'valid-type1'
+      expect(last_response.status).to eq 200
+      expect(last_response.body).to eq({ type: 'valid-type1' }.to_json)
+    end
+
+    it 'accepts multiple valid values' do
+      get '/test_with', type: ['valid-type1', 'valid-type3']
+      expect(last_response.status).to eq 200
+      expect(last_response.body).to eq({ type: ['valid-type1', 'valid-type3'] }.to_json)
+    end
+
+    it 'rejects a single invalid value' do
+      get '/test_with', type: 'invalid-type1'
+      expect(last_response.status).to eq 400
+      expect(last_response.body).to eq({ error: 'type does not have a valid value' }.to_json)
+    end
+
+    it 'rejects an invalid value among valid ones' do
+      get '/test_with', type: ['valid-type1', 'invalid-type1', 'valid-type3']
+      expect(last_response.status).to eq 400
+      expect(last_response.body).to eq({ error: 'type does not have a valid value' }.to_json)
+    end
+
+    it 'uses supplied message' do
+      get '/test_with/message', type: 'invalid-type1'
+      expect(last_response.status).to eq 400
+      expect(last_response.body).to eq({ error: 'type failed check' }.to_json)
+    end
+  end
 end