Merge pull request #17 from SlicingDice/feature/correct-data-extraction-validator

Correct data extraction validator

Author: joaosimbiose

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Change Log
 
+## [2.0.2]
+### Updated
+- Correct data extraction validator to accept columns: all
+
 ## [2.0.1]
 ### Updated
 - Update data extraction validator to support order parameter

lib/rbslicer/exceptions.rb

@@ -29,4 +29,6 @@ class IndexEntitiesLimitException < SlicingDiceException; end
 
   class IndexColumnsLimitException < SlicingDiceException; end
 
+  class InvalidQueryException < SlicingDiceException; end
+
 end

lib/rbslicer/utils/validators.rb

@@ -181,8 +181,10 @@ def valid_keys?
           end
         elsif key == "columns"
           if !value.is_a?(Array)
-            raise Exceptions::InvalidQueryException, 'The key \'columns\' in '\
-                                                    'query has a invalid value.'
+            if value != "all"
+              raise Exceptions::InvalidQueryException, 'The key \'columns\' in '\
+                                                      'query has a invalid value.'
+            end
           else
             if value.length > 10
               raise Exceptions::MaxLimitException, 'The key \'columns\' in '\
@@ -201,7 +203,7 @@ def valid_keys?
           end
         else
           raise Exceptions::InvalidQueryException, "This query have the invalid"\
-                                                  "key #{key}"
+                                                  " key #{key}"
         end
       end
       true