Check if variant array contains only symbols

This commit is contained in:
Łukasz Strzałkowski 2014-02-13 18:05:55 +01:00
parent f2dfa83fac
commit 00a4af9ab7
2 changed files with 10 additions and 2 deletions

@ -70,10 +70,10 @@ def formats
def variant=(variant)
if variant.is_a?(Symbol)
@variant = [variant]
elsif variant.is_a?(Array)
elsif variant.is_a?(Array) && variant.any? && variant.all?{ |v| v.is_a?(Symbol) }
@variant = variant
else
raise ArgumentError, "request.variant must be set to a Symbol or Array, not a #{variant.class}. " \
raise ArgumentError, "request.variant must be set to a Symbol or an Array of Symbols, not a #{variant.class}. " \
"For security reasons, never directly set the variant to a user-provided value, " \
"like params[:variant].to_sym. Check user-provided value against a whitelist first, " \
"then set the variant: request.variant = :tablet if params[:variant] == 'tablet'"

@ -852,6 +852,14 @@ def url_for(options = {})
request.variant = [:phone, :tablet]
assert_equal [:phone, :tablet], request.variant
assert_raise ArgumentError do
request.variant = [:phone, "tablet"]
end
assert_raise ArgumentError do
request.variant = "yolo"
end
end
test "setting variant with non symbol value" do