1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activerecord/test/cases/statement_cache_test.rb
Sean Griffin 3af40b71f3 Prepared statements shouldn't share a cache with unprepared statements
When prepared statements are enabled, the statement cache caches the SQL
directly, including the bind parameters. If a similar query is run later
with prepared statements disabled, we need to use a separate cache
instead of trying to share the same one.

Fixes #24351
2016-03-31 13:41:36 -06:00

110 lines
3.1 KiB
Ruby

require 'cases/helper'
require 'models/book'
require 'models/liquid'
require 'models/molecule'
require 'models/electron'
module ActiveRecord
class StatementCacheTest < ActiveRecord::TestCase
def setup
@connection = ActiveRecord::Base.connection
end
#Cache v 1.1 tests
def test_statement_cache
Book.create(name: "my book")
Book.create(name: "my other book")
cache = StatementCache.create(Book.connection) do |params|
Book.where(:name => params.bind)
end
b = cache.execute([ "my book" ], Book, Book.connection)
assert_equal "my book", b[0].name
b = cache.execute([ "my other book" ], Book, Book.connection)
assert_equal "my other book", b[0].name
end
def test_statement_cache_id
b1 = Book.create(name: "my book")
b2 = Book.create(name: "my other book")
cache = StatementCache.create(Book.connection) do |params|
Book.where(id: params.bind)
end
b = cache.execute([ b1.id ], Book, Book.connection)
assert_equal b1.name, b[0].name
b = cache.execute([ b2.id ], Book, Book.connection)
assert_equal b2.name, b[0].name
end
def test_find_or_create_by
Book.create(name: "my book")
a = Book.find_or_create_by(name: "my book")
b = Book.find_or_create_by(name: "my other book")
assert_equal("my book", a.name)
assert_equal("my other book", b.name)
end
#End
def test_statement_cache_with_simple_statement
cache = ActiveRecord::StatementCache.create(Book.connection) do |params|
Book.where(name: "my book").where("author_id > 3")
end
Book.create(name: "my book", author_id: 4)
books = cache.execute([], Book, Book.connection)
assert_equal "my book", books[0].name
end
def test_statement_cache_with_complex_statement
cache = ActiveRecord::StatementCache.create(Book.connection) do |params|
Liquid.joins(:molecules => :electrons).where('molecules.name' => 'dioxane', 'electrons.name' => 'lepton')
end
salty = Liquid.create(name: 'salty')
molecule = salty.molecules.create(name: 'dioxane')
molecule.electrons.create(name: 'lepton')
liquids = cache.execute([], Book, Book.connection)
assert_equal "salty", liquids[0].name
end
def test_statement_cache_values_differ
cache = ActiveRecord::StatementCache.create(Book.connection) do |params|
Book.where(name: "my book")
end
3.times do
Book.create(name: "my book")
end
first_books = cache.execute([], Book, Book.connection)
3.times do
Book.create(name: "my book")
end
additional_books = cache.execute([], Book, Book.connection)
assert first_books != additional_books
end
def test_unprepared_statements_dont_share_a_cache_with_prepared_statements
Book.create(name: "my book")
Book.create(name: "my other book")
book = Book.find_by(name: "my book")
other_book = Book.connection.unprepared_statement do
Book.find_by(name: "my other book")
end
refute_equal book, other_book
end
end
end