Add query method for id query

1 files changed, 13 insertions, 0 deletions

diff --git a/spec/models/application_record_spec.rb b/spec/models/application_record_spec.rb
new file mode 100644
index 00000000000..68aed387bfc
--- /dev/null
+++ b/spec/models/application_record_spec.rb
@@ -0,0 +1,13 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+describe ApplicationRecord do
+  describe '#id_in' do
+    let(:records) { create_list(:user, 3) }
+
+    it 'returns records of the ids' do
+      expect(User.id_in(records.last(2).map(&:id))).to eq(records.last(2))
+    end
+  end
+end