mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
554597d657
git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@78 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
130 lines
4.4 KiB
Ruby
Executable file
130 lines
4.4 KiB
Ruby
Executable file
require 'abstract_unit'
|
|
require 'fixtures/company'
|
|
require 'fixtures/topic'
|
|
require 'fixtures/entrant'
|
|
|
|
class FinderTest < Test::Unit::TestCase
|
|
def setup
|
|
@company_fixtures = create_fixtures("companies")
|
|
@topic_fixtures = create_fixtures("topics")
|
|
@entrant_fixtures = create_fixtures("entrants")
|
|
end
|
|
|
|
def test_find
|
|
assert_equal(@topic_fixtures["first"]["title"], Topic.find(1).title)
|
|
end
|
|
|
|
def test_find_by_ids
|
|
assert_equal(2, Topic.find(1, 2).length)
|
|
assert_equal(@topic_fixtures["second"]["title"], Topic.find([ 2 ]).title)
|
|
end
|
|
|
|
def test_find_by_ids_missing_one
|
|
assert_raises(ActiveRecord::RecordNotFound) {
|
|
Topic.find(1, 2, 45)
|
|
}
|
|
end
|
|
|
|
def test_find_all_with_limit
|
|
entrants = Entrant.find_all nil, "id ASC", 2
|
|
|
|
assert_equal(2, entrants.size)
|
|
assert_equal(@entrant_fixtures["first"]["name"], entrants.first.name)
|
|
end
|
|
|
|
def test_find_all_with_prepared_limit_and_offset
|
|
entrants = Entrant.find_all nil, "id ASC", ["? OFFSET ?", 2, 1]
|
|
|
|
assert_equal(2, entrants.size)
|
|
assert_equal(@entrant_fixtures["second"]["name"], entrants.first.name)
|
|
end
|
|
|
|
def test_find_with_entire_select_statement
|
|
topics = Topic.find_by_sql "SELECT * FROM topics WHERE author_name = 'Mary'"
|
|
|
|
assert_equal(1, topics.size)
|
|
assert_equal(@topic_fixtures["second"]["title"], topics.first.title)
|
|
end
|
|
|
|
def test_find_with_prepared_select_statement
|
|
topics = Topic.find_by_sql ["SELECT * FROM topics WHERE author_name = ?", "Mary"]
|
|
|
|
assert_equal(1, topics.size)
|
|
assert_equal(@topic_fixtures["second"]["title"], topics.first.title)
|
|
end
|
|
|
|
def test_find_first
|
|
first = Topic.find_first "title = 'The First Topic'"
|
|
assert_equal(@topic_fixtures["first"]["title"], first.title)
|
|
end
|
|
|
|
def test_find_first_failing
|
|
first = Topic.find_first "title = 'The First Topic!'"
|
|
assert_nil(first)
|
|
end
|
|
|
|
def test_unexisting_record_exception_handling
|
|
assert_raises(ActiveRecord::RecordNotFound) {
|
|
Topic.find(1).parent
|
|
}
|
|
|
|
Topic.find(2).parent
|
|
end
|
|
|
|
def test_find_on_conditions
|
|
assert Topic.find_on_conditions(1, "approved = 0")
|
|
assert_raises(ActiveRecord::RecordNotFound) { Topic.find_on_conditions(1, "approved = 1") }
|
|
end
|
|
|
|
def test_condition_interpolation
|
|
assert_kind_of Firm, Company.find_first(["name = '%s'", "37signals"])
|
|
assert_nil Company.find_first(["name = '%s'", "37signals!"])
|
|
assert_nil Company.find_first(["name = '%s'", "37signals!' OR 1=1"])
|
|
assert_kind_of Time, Topic.find_first(["id = %d", 1]).written_on
|
|
end
|
|
|
|
def test_bind_variables
|
|
assert_kind_of Firm, Company.find_first(["name = ?", "37signals"])
|
|
assert_nil Company.find_first(["name = ?", "37signals!"])
|
|
assert_nil Company.find_first(["name = ?", "37signals!' OR 1=1"])
|
|
assert_kind_of Time, Topic.find_first(["id = ?", 1]).written_on
|
|
assert_raises(ActiveRecord::PreparedStatementInvalid) {
|
|
Company.find_first(["id=? AND name = ?", 2])
|
|
}
|
|
assert_raises(ActiveRecord::PreparedStatementInvalid) {
|
|
Company.find_first(["id=?", 2, 3, 4])
|
|
}
|
|
end
|
|
|
|
def test_named_bind_variables
|
|
assert_kind_of Firm, Company.find_first(["name = :name", { :name => "37signals" }])
|
|
assert_nil Company.find_first(["name = :name", { :name => "37signals!" }])
|
|
assert_nil Company.find_first(["name = :name", { :name => "37signals!' OR 1=1" }])
|
|
assert_kind_of Time, Topic.find_first(["id = :id", { :id => 1 }]).written_on
|
|
assert_raises(ActiveRecord::PreparedStatementInvalid) {
|
|
Company.find_first(["id=:id and name=:name", { :id=>3 }])
|
|
}
|
|
assert_raises(ActiveRecord::PreparedStatementInvalid) {
|
|
Company.find_first(["id=:id", { :id=>3, :name=>"37signals!" }])
|
|
}
|
|
end
|
|
|
|
|
|
|
|
def test_string_sanitation
|
|
assert_not_equal "'something ' 1=1'", ActiveRecord::Base.sanitize("something ' 1=1")
|
|
assert_equal "'something; select table'", ActiveRecord::Base.sanitize("something; select table")
|
|
end
|
|
|
|
def test_count
|
|
assert_equal(0, Entrant.count("id > 3"))
|
|
assert_equal(1, Entrant.count(["id > ?", 2]))
|
|
assert_equal(2, Entrant.count(["id > ?", 1]))
|
|
end
|
|
|
|
def test_count_by_sql
|
|
assert_equal(0, Entrant.count_by_sql("SELECT COUNT(*) FROM entrants WHERE id > 3"))
|
|
assert_equal(1, Entrant.count_by_sql(["SELECT COUNT(*) FROM entrants WHERE id > ?", 2]))
|
|
assert_equal(2, Entrant.count_by_sql(["SELECT COUNT(*) FROM entrants WHERE id > ?", 1]))
|
|
end
|
|
end
|