test: cover OrmQueryBatchesTest, teardown and OrmQueryCombinatorsTest

test/test_orm_query_combinators.rb

@@ -0,0 +1,64 @@
+require 'minitest/autorun'
+require 'prism'
+require 'tmpdir'
+
+class OrmQueryCombinatorsTest < Minitest::Test
+  def setup
+    @dir = Dir.mktmpdir('prism-query-combinators')
+    @store = Prism::Orm::Store.new(File.join(@dir, 'test.pstore'))
+    @store.create_table(:tickets, %w[id title status priority])
+    ticket_class = Class.new(Prism::Orm::Model) do
+      table_name :tickets
+    end
+    ticket_class.store = @store
+    Object.const_set(:Ticket, ticket_class)
+
+    Ticket.create!(title: 'a', status: 'open', priority: 'low')
+    Ticket.create!(title: 'b', status: 'open', priority: 'high')
+    Ticket.create!(title: 'c', status: 'closed', priority: 'high')
+    Ticket.create!(title: 'd', status: 'closed', priority: 'low')
+  end
+
+  def teardown
+    FileUtils.remove_entry(@dir)
+    Object.send(:remove_const, :Ticket) if Object.const_defined?(:Ticket)
+  end
+
+  def test_not_excludes_matching_rows
+    titles = Ticket.where(status: 'open').not(priority: 'low').pluck(:title)
+    assert_equal ['b'], titles
+  end
+
+  def test_model_level_not_shorthand
+    titles = Ticket.not(status: 'open').order(:title).pluck(:title)
+    assert_equal %w[c d], titles
+  end
+
+  def test_or_combines_two_relations
+    titles = Ticket.where(status: 'open').or(Ticket.where(priority: 'high')).order(:title).pluck(:title)
+    assert_equal %w[a b c], titles
+  end
+
+  def test_or_with_plain_hash
+    titles = Ticket.where(status: 'closed').or(priority: 'high').order(:title).pluck(:title)
+    assert_equal %w[b c d], titles
+  end
+
+  def test_chained_and_or_and_not
+    titles = Ticket.where(status: 'open').not(priority: 'low').or(Ticket.where(status: 'closed').not(priority: 'high'))
+                   .order(:title).pluck(:title)
+    assert_equal %w[b d], titles
+  end
+
+  def test_none_returns_empty_relation_without_touching_store
+    relation = Ticket.none
+    assert_equal 0, relation.count
+    assert_equal [], relation.to_a
+    refute relation.exists?
+  end
+
+  def test_where_still_ands_across_chained_calls
+    titles = Ticket.where(status: 'open').where(priority: 'high').pluck(:title)
+    assert_equal ['b'], titles
+  end
+end