Commit d253440e authored by Mark Chao's avatar Mark Chao

Add query method for id query

parent 2abbfda3
...@@ -2,4 +2,8 @@ ...@@ -2,4 +2,8 @@
class ApplicationRecord < ActiveRecord::Base class ApplicationRecord < ActiveRecord::Base
self.abstract_class = true self.abstract_class = true
def self.id_in(ids)
where(id: ids)
end
end end
# frozen_string_literal: true # frozen_string_literal: true
class Namespace < ActiveRecord::Base class Namespace < ApplicationRecord
include CacheMarkdownField include CacheMarkdownField
include Sortable include Sortable
include Gitlab::VisibilityLevel include Gitlab::VisibilityLevel
......
...@@ -2,7 +2,7 @@ ...@@ -2,7 +2,7 @@
require 'carrierwave/orm/activerecord' require 'carrierwave/orm/activerecord'
class User < ActiveRecord::Base class User < ApplicationRecord
extend Gitlab::ConfigHelper extend Gitlab::ConfigHelper
include Gitlab::ConfigHelper include Gitlab::ConfigHelper
......
# 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
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment