Support NE condition operator

Dynamoid · Mar 10, 2018 · a82e881 · a82e881
1 parent 46b1784
commit a82e881
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/lib/dynamoid/adapter_plugin/aws_sdk_v2.rb b/lib/dynamoid/adapter_plugin/aws_sdk_v2.rb
@@ -18,6 +18,7 @@ class AwsSdkV2
       # we declare schema in models
       FIELD_MAP = {
           eq:           'EQ',
+          ne:           'NE',
           gt:           'GT',
           lt:           'LT',
           gte:          'GE',

diff --git a/lib/dynamoid/criteria/chain.rb b/lib/dynamoid/criteria/chain.rb
@@ -196,6 +196,8 @@ def field_hash(key)
         val = type_cast_condition_parameter(name, query[key])
 
         hash = case operation
+        when 'ne'
+          { ne: val }
         when 'gt'
           { gt: val }
         when 'lt'

diff --git a/spec/dynamoid/criteria/chain_spec.rb b/spec/dynamoid/criteria/chain_spec.rb
@@ -247,6 +247,13 @@ def request_params
       expect(chain.where(id: 1, array: [1, 2]).all).to contain_exactly(document1)
     end
 
+    it 'supports ne' do
+      customer1 = model.create(name: 'a', last_name: 'a', age: 5)
+      customer2 = model.create(name: 'a', last_name: 'b', age: 9)
+
+      expect(model.where(name: 'a', 'age.ne' => 9).all).to contain_exactly(customer1)
+    end
+
     it 'supports lt' do
       customer1 = model.create(name: 'a', last_name: 'a', age: 5)
       customer2 = model.create(name: 'a', last_name: 'b', age: 9)
@@ -388,6 +395,13 @@ def request_params
       expect(klass.where(array: ['a', 'b']).all).to contain_exactly(document1)
     end
 
+    it 'supports ne' do
+      customer1 = model.create(age: 5)
+      customer2 = model.create(age: 9)
+
+      expect(model.where('age.ne' => 9).all).to contain_exactly(customer1)
+    end
+
     it 'supports lt' do
       customer1 = model.create(age: 5)
       customer2 = model.create(age: 9)