SAL-3248 | FEATURE | Add ids method AF query objects

Beyond-Finance · May 25, 2024 · a7d5756 · a7d5756
1 parent be44e31
commit a7d5756
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lib/active_force/query.rb b/lib/active_force/query.rb
@@ -107,6 +107,10 @@ def sum field
       clone_and_set_instance_variables(query_fields: ["sum(#{field})"])
     end
 
+    def ids
+      clone_and_set_instance_variables(query_fields: ["Id"])
+    end
+
     protected
       def and_conditions
         "(#{@conditions.join(') AND (')})" unless @conditions.empty?

diff --git a/spec/active_force/query_spec.rb b/spec/active_force/query_spec.rb
@@ -236,4 +236,11 @@
       expect(query.where("name = 'cool'").sum(:field1).to_s).to eq "SELECT sum(field1) FROM table_name WHERE (name = 'cool')"
     end
   end
+
+  describe ".ids" do
+    it "should return the query for plucking the ids" do
+      query_with_ids = query.where("name = 'cool'").ids
+      expect(query_with_ids.to_s).to eq "SELECT Id FROM table_name WHERE (name = 'cool')"
+    end
+  end
 end