added array_contains scopes

Author: Michael Crismali

lib/jsonb_accessor/macro.rb

@@ -73,9 +73,11 @@ def __create_jsonb_typed_scopes(jsonb_attribute, fields_map)
           when :boolean
             ___create_jsonb_boolean_scopes(field)
           when :integer, :float, :decimal, :big_integer
-            ___create_jsonb_numeric_scopes(field, type, jsonb_attribute)
+            ___create_jsonb_numeric_scopes(field, jsonb_attribute, type)
           when :date_time, :date, :time
             ___create_jsonb_date_time_scopes(field, jsonb_attribute, type)
+          when /array/
+            ___create_jsonb_array_scopes(field)
           end
         end
       end
@@ -85,9 +87,9 @@ def ___create_jsonb_boolean_scopes(field)
         scope "not_#{field}", -> { send("with_#{field}", false) }
       end
 
-      def ___create_jsonb_numeric_scopes(field, type, jsonb_attribute)
+      def ___create_jsonb_numeric_scopes(field, jsonb_attribute, type)
         safe_type = type.to_s.gsub("big_", "")
-        scope "__numeric_#{field}_comparator", -> (value, operator) { where("((#{table_name}.#{jsonb_attribute}) ->> ?)::#{safe_type} #{operator} ?", field, value) }
+        scope "__numeric_#{field}_comparator", -> (value, operator) { where("((#{table_name}.#{jsonb_attribute}) -  >> ?)::#{safe_type} #{operator} ?", field, value) }
         scope "#{field}_lt", -> (value) { send("__numeric_#{field}_comparator", value, "<") }
         scope "#{field}_lte", -> (value) { send("__numeric_#{field}_comparator", value, "<=") }
         scope "#{field}_gte", -> (value) { send("__numeric_#{field}_comparator", value, ">=") }
@@ -98,6 +100,10 @@ def ___create_jsonb_date_time_scopes(field, jsonb_attribute, type)
         scope "#{field}_before", -> (value) { where("((#{table_name}.#{jsonb_attribute}) ->> ?)::timestamp < ?::timestamp", field, value.to_json) }
       end
 
+      def ___create_jsonb_array_scopes(field)
+        scope "#{field}_contains", -> (value) { send("with_#{field}", [value]) }
+      end
+
       def _create_jsonb_accessor_methods(jsonb_attribute, jsonb_attribute_initialization_method_name, fields_map)
         jsonb_accessor_methods = Module.new do
           define_method("#{jsonb_attribute}=") do |value|

spec/jsonb_accessor_spec.rb

@@ -790,9 +790,28 @@
   context "scopes" do
     let(:title) { "foo" }
     let(:right_now) { Time.now }
-    let!(:ignored_product) { Product.create!(title: "bar") }
-    let!(:matching_product) { Product.create!(title: title, external_id: 3, admin: true, a_big_number: 23, reviewed_at: right_now, reset_at: right_now, approved_on: right_now) }
-    let!(:other_matching_product) { Product.create!(title: title, admin: false, document: { nested: { are: "0" } }) }
+    let!(:ignored_product) { Product.create!(title: "bar", rankings: [2]) }
+    let!(:matching_product) do
+      Product.create!(
+        title: title,
+        external_id: 3,
+        admin: true,
+        a_big_number: 23,
+        reviewed_at: right_now,
+        reset_at: right_now,
+        approved_on: right_now,
+        rankings: [1, 3]
+      )
+    end
+
+    let!(:other_matching_product) do
+      Product.create!(
+        title: title,
+        admin: false,
+        rankings: [3],
+        document: { nested: { are: "0" } }
+      )
+    end
 
     describe "#<jsonb_attribute_name>_contains" do
       it "is a collection of records that match the query" do
@@ -1007,6 +1026,13 @@
         end
       end
     end
+
+    describe "#<array_field>_contains" do
+      it "is all records that contain the argument" do
+        expect(Product.rankings_contains(3)).to match_array([matching_product, other_matching_product])
+        expect(Product.rankings_contains(1)).to match_array([matching_product])
+      end
+    end
   end
 
   context "ActionDispatch clean up" do