73 lines
2.2 KiB
Ruby
73 lines
2.2 KiB
Ruby
require 'minitest/autorun'
|
|
require 'prism'
|
|
require 'tmpdir'
|
|
|
|
class OrmQueryScopesTest < Minitest::Test
|
|
def setup
|
|
@dir = Dir.mktmpdir('prism-query-scopes')
|
|
@store = Prism::Orm::Store.new(File.join(@dir, 'test.pstore'))
|
|
@store.create_table(:articles, %w[id title published views])
|
|
@article_class = Class.new(Prism::Orm::Model) do
|
|
table_name :articles
|
|
scope :published, -> { where(published: true) }
|
|
scope :popular, ->(threshold) { where(views: { gte: threshold }) }
|
|
end
|
|
Object.const_set(:Article, @article_class)
|
|
Article.store = @store
|
|
|
|
Article.create!(title: 'a', published: true, views: 100)
|
|
Article.create!(title: 'b', published: false, views: 500)
|
|
Article.create!(title: 'c', published: true, views: 5)
|
|
end
|
|
|
|
def teardown
|
|
FileUtils.remove_entry(@dir)
|
|
Object.send(:remove_const, :Article) if Object.const_defined?(:Article)
|
|
end
|
|
|
|
def test_scope_without_arguments
|
|
titles = Article.published.order(:title).pluck(:title)
|
|
assert_equal %w[a c], titles
|
|
end
|
|
|
|
def test_scope_with_argument
|
|
titles = Article.popular(50).order(:title).pluck(:title)
|
|
assert_equal %w[a b], titles
|
|
end
|
|
|
|
def test_scopes_are_chainable_with_relation_methods
|
|
titles = Article.published.popular(10).pluck(:title)
|
|
assert_equal ['a'], titles
|
|
end
|
|
|
|
def test_scopes_return_relation_not_array
|
|
assert_kind_of Prism::Orm::Relation, Article.published
|
|
end
|
|
|
|
def test_default_scope_applies_to_all_and_where
|
|
Object.send(:remove_const, :Article)
|
|
scoped_class = Class.new(Prism::Orm::Model) do
|
|
table_name :articles
|
|
default_scope { where(published: true) }
|
|
end
|
|
Object.const_set(:Article, scoped_class)
|
|
Article.store = @store
|
|
|
|
assert_equal %w[a c], Article.all.order(:title).pluck(:title)
|
|
assert_equal 2, Article.count
|
|
assert_equal %w[a], Article.where(title: 'a').pluck(:title)
|
|
end
|
|
|
|
def test_unscoped_bypasses_default_scope
|
|
Object.send(:remove_const, :Article)
|
|
scoped_class = Class.new(Prism::Orm::Model) do
|
|
table_name :articles
|
|
default_scope { where(published: true) }
|
|
end
|
|
Object.const_set(:Article, scoped_class)
|
|
Article.store = @store
|
|
|
|
assert_equal 3, Article.unscoped.count
|
|
end
|
|
end
|