finder_test.rb 44.0 KB
Newer Older
1
require "cases/helper"
2
require 'models/post'
J
Jeremy Kemper 已提交
3
require 'models/author'
4
require 'models/categorization'
J
Jeremy Kemper 已提交
5 6 7 8 9
require 'models/comment'
require 'models/company'
require 'models/topic'
require 'models/reply'
require 'models/entrant'
10
require 'models/project'
J
Jeremy Kemper 已提交
11
require 'models/developer'
12
require 'models/customer'
13
require 'models/job'
D
David Heinemeier Hansson 已提交
14

15 16 17 18 19 20 21 22 23
class DynamicFinderMatchTest < ActiveRecord::TestCase
  def test_find_no_match
    assert_nil ActiveRecord::DynamicFinderMatch.match("not_a_finder")
  end

  def test_find_by
    match = ActiveRecord::DynamicFinderMatch.match("find_by_age_and_sex_and_location")
    assert_not_nil match
    assert match.finder?
24
    assert_equal :first, match.finder
25 26 27
    assert_equal %w(age sex location), match.attribute_names
  end

28 29 30 31 32
  def find_by_bang
    match = ActiveRecord::DynamicFinderMatch.match("find_by_age_and_sex_and_location!")
    assert_not_nil match
    assert match.finder?
    assert match.bang?
33
    assert_equal :first, match.finder
34 35 36
    assert_equal %w(age sex location), match.attribute_names
  end

37 38 39 40
  def test_find_all_by
    match = ActiveRecord::DynamicFinderMatch.match("find_all_by_age_and_sex_and_location")
    assert_not_nil match
    assert match.finder?
41
    assert_equal :all, match.finder
42 43 44 45 46 47 48 49
    assert_equal %w(age sex location), match.attribute_names
  end

  def test_find_or_initialize_by
    match = ActiveRecord::DynamicFinderMatch.match("find_or_initialize_by_age_and_sex_and_location")
    assert_not_nil match
    assert !match.finder?
    assert match.instantiator?
50
    assert_equal :first, match.finder
51 52 53 54 55 56 57 58 59
    assert_equal :new, match.instantiator
    assert_equal %w(age sex location), match.attribute_names
  end

  def test_find_or_create_by
    match = ActiveRecord::DynamicFinderMatch.match("find_or_create_by_age_and_sex_and_location")
    assert_not_nil match
    assert !match.finder?
    assert match.instantiator?
60
    assert_equal :first, match.finder
61 62 63 64 65
    assert_equal :create, match.instantiator
    assert_equal %w(age sex location), match.attribute_names
  end
end

66
class FinderTest < ActiveRecord::TestCase
67
  fixtures :companies, :topics, :entrants, :developers, :developers_projects, :posts, :comments, :accounts, :authors, :customers, :categories, :categorizations
D
David Heinemeier Hansson 已提交
68 69

  def test_find
70
    assert_equal(topics(:first).title, Topic.find(1).title)
D
David Heinemeier Hansson 已提交
71
  end
72

73 74 75 76 77
  # find should handle strings that come from URLs
  # (example: Category.find(params[:id]))
  def test_find_with_string
    assert_equal(Topic.find(1).title,Topic.find("1").title)
  end
78

D
David Heinemeier Hansson 已提交
79
  def test_exists
80 81 82 83 84 85
    assert Topic.exists?(1)
    assert Topic.exists?("1")
    assert Topic.exists?(:author_name => "David")
    assert Topic.exists?(:author_name => "Mary", :approved => true)
    assert Topic.exists?(["parent_id = ?", 1])
    assert !Topic.exists?(45)
86 87 88 89 90 91 92 93 94

    begin
      assert !Topic.exists?("foo")
    rescue ActiveRecord::StatementInvalid
      # PostgreSQL complains about string comparison with integer field
    rescue Exception
      flunk
    end

95
    assert_raise(NoMethodError) { Topic.exists?([1,2]) }
D
David Heinemeier Hansson 已提交
96
  end
97

98 99 100
  def test_exists_returns_true_with_one_record_and_no_args
    assert Topic.exists?
  end
101

102 103 104 105
  def test_does_not_exist_with_empty_table_and_no_args_given
    Topic.delete_all
    assert !Topic.exists?
  end
106

107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
  def test_exists_with_aggregate_having_three_mappings
    existing_address = customers(:david).address
    assert Customer.exists?(:address => existing_address)
  end

  def test_exists_with_aggregate_having_three_mappings_with_one_difference
    existing_address = customers(:david).address
    assert !Customer.exists?(:address =>
      Address.new(existing_address.street, existing_address.city, existing_address.country + "1"))
    assert !Customer.exists?(:address =>
      Address.new(existing_address.street, existing_address.city + "1", existing_address.country))
    assert !Customer.exists?(:address =>
      Address.new(existing_address.street + "1", existing_address.city, existing_address.country))
  end

122 123 124 125 126 127
  def test_exists_with_scoped_include
    Developer.with_scope(:find => { :include => :projects, :order => "projects.name" }) do
      assert Developer.exists?
    end
  end

D
David Heinemeier Hansson 已提交
128 129 130 131
  def test_find_by_array_of_one_id
    assert_kind_of(Array, Topic.find([ 1 ]))
    assert_equal(1, Topic.find([ 1 ]).length)
  end
132

D
David Heinemeier Hansson 已提交
133
  def test_find_by_ids
134 135 136 137 138
    assert_equal 2, Topic.find(1, 2).size
    assert_equal topics(:second).title, Topic.find([2]).first.title
  end

  def test_find_by_ids_with_limit_and_offset
139
    assert_equal 2, Entrant.find([1,3,2], :limit => 2).size
140
    assert_equal 1, Entrant.find([1,3,2], :limit => 3, :offset => 2).size
141 142 143 144 145 146 147

    # Also test an edge case: If you have 11 results, and you set a
    #   limit of 3 and offset of 9, then you should find that there
    #   will be only 2 results, regardless of the limit.
    devs = Developer.find :all
    last_devs = Developer.find devs.map(&:id), :limit => 3, :offset => 9
    assert_equal 2, last_devs.size
D
David Heinemeier Hansson 已提交
148 149
  end

150 151 152 153
  def test_find_an_empty_array
    assert_equal [], Topic.find([])
  end

D
David Heinemeier Hansson 已提交
154
  def test_find_by_ids_missing_one
155
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, 2, 45) }
D
David Heinemeier Hansson 已提交
156
  end
157

D
David Heinemeier Hansson 已提交
158 159
  def test_find_all_with_limit
    entrants = Entrant.find(:all, :order => "id ASC", :limit => 2)
160

D
David Heinemeier Hansson 已提交
161
    assert_equal(2, entrants.size)
162
    assert_equal(entrants(:first).name, entrants.first.name)
D
David Heinemeier Hansson 已提交
163 164 165
  end

  def test_find_all_with_prepared_limit_and_offset
166
    entrants = Entrant.find(:all, :order => "id ASC", :limit => 2, :offset => 1)
D
David Heinemeier Hansson 已提交
167

168 169
    assert_equal(2, entrants.size)
    assert_equal(entrants(:second).name, entrants.first.name)
170

171 172 173
    entrants = Entrant.find(:all, :order => "id ASC", :limit => 2, :offset => 2)
    assert_equal(1, entrants.size)
    assert_equal(entrants(:third).name, entrants.first.name)
D
David Heinemeier Hansson 已提交
174
  end
175

176 177 178 179
  def test_find_all_with_limit_and_offset_and_multiple_orderings
    developers = Developer.find(:all, :order => "salary ASC, id DESC", :limit => 3, :offset => 1)
    assert_equal ["David", "fixture_10", "fixture_9"], developers.collect {|d| d.name}
  end
180

181 182 183 184 185
  def test_find_with_limit_and_condition
    developers = Developer.find(:all, :order => "id DESC", :conditions => "salary = 100000", :limit => 3, :offset =>7)
    assert_equal(1, developers.size)
    assert_equal("fixture_3", developers.first.name)
  end
D
David Heinemeier Hansson 已提交
186

187
  def test_find_with_group
188
    developers =  Developer.find(:all, :group => "salary", :select => "salary")
189
    assert_equal 4, developers.size
190
    assert_equal 4, developers.map(&:salary).uniq.size
191 192
  end

193 194 195 196 197 198 199
  def test_find_with_group_and_having
    developers =  Developer.find(:all, :group => "salary", :having => "sum(salary) >  10000", :select => "salary")
    assert_equal 3, developers.size
    assert_equal 3, developers.map(&:salary).uniq.size
    assert developers.all? { |developer|  developer.salary > 10000 }
  end

200 201 202 203 204 205 206
  def test_find_with_group_and_sanitized_having
    developers =  Developer.find(:all, :group => "salary", :having => ["sum(salary) > ?", 10000], :select => "salary")
    assert_equal 3, developers.size
    assert_equal 3, developers.map(&:salary).uniq.size
    assert developers.all? { |developer|  developer.salary > 10000 }
  end

D
David Heinemeier Hansson 已提交
207 208
  def test_find_with_entire_select_statement
    topics = Topic.find_by_sql "SELECT * FROM topics WHERE author_name = 'Mary'"
209

D
David Heinemeier Hansson 已提交
210
    assert_equal(1, topics.size)
211
    assert_equal(topics(:second).title, topics.first.title)
D
David Heinemeier Hansson 已提交
212
  end
213

D
David Heinemeier Hansson 已提交
214 215
  def test_find_with_prepared_select_statement
    topics = Topic.find_by_sql ["SELECT * FROM topics WHERE author_name = ?", "Mary"]
216

D
David Heinemeier Hansson 已提交
217
    assert_equal(1, topics.size)
218
    assert_equal(topics(:second).title, topics.first.title)
D
David Heinemeier Hansson 已提交
219
  end
220

221 222 223 224
  def test_find_by_sql_with_sti_on_joined_table
    accounts = Account.find_by_sql("SELECT * FROM accounts INNER JOIN companies ON companies.id = accounts.firm_id")
    assert_equal [Account], accounts.collect(&:class).uniq
  end
225

D
David Heinemeier Hansson 已提交
226 227
  def test_find_first
    first = Topic.find(:first, :conditions => "title = 'The First Topic'")
228
    assert_equal(topics(:first).title, first.title)
D
David Heinemeier Hansson 已提交
229
  end
230

D
David Heinemeier Hansson 已提交
231 232 233 234
  def test_find_first_failing
    first = Topic.find(:first, :conditions => "title = 'The First Topic!'")
    assert_nil(first)
  end
235

236 237 238
  def test_first
    assert_equal topics(:second).title, Topic.first(:conditions => "title = 'The Second Topic of the day'").title
  end
239

240 241 242
  def test_first_failing
    assert_nil Topic.first(:conditions => "title = 'The Second Topic of the day!'")
  end
D
David Heinemeier Hansson 已提交
243 244

  def test_unexisting_record_exception_handling
245
    assert_raise(ActiveRecord::RecordNotFound) {
D
David Heinemeier Hansson 已提交
246 247
      Topic.find(1).parent
    }
248

249
    Topic.find(2).topic
D
David Heinemeier Hansson 已提交
250
  end
251

252
  def test_find_only_some_columns
253
    topic = Topic.find(1, :select => "author_name")
254
    assert_raise(ActiveRecord::MissingAttributeError) {topic.title}
255 256
    assert_equal "David", topic.author_name
    assert !topic.attribute_present?("title")
257
    #assert !topic.respond_to?("title")
258 259
    assert topic.attribute_present?("author_name")
    assert topic.respond_to?("author_name")
260
  end
J
Jeremy Kemper 已提交
261

262 263 264 265 266
  def test_find_on_blank_conditions
    [nil, " ", [], {}].each do |blank|
      assert_nothing_raised { Topic.find(:first, :conditions => blank) }
    end
  end
J
Jeremy Kemper 已提交
267

268 269 270 271 272
  def test_find_on_blank_bind_conditions
    [ [""], ["",{}] ].each do |blank|
      assert_nothing_raised { Topic.find(:first, :conditions => blank) }
    end
  end
D
David Heinemeier Hansson 已提交
273

274
  def test_find_on_array_conditions
275
    assert Topic.find(1, :conditions => ["approved = ?", false])
276
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => ["approved = ?", true]) }
D
David Heinemeier Hansson 已提交
277
  end
278

279 280
  def test_find_on_hash_conditions
    assert Topic.find(1, :conditions => { :approved => false })
281
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :approved => true }) }
282
  end
283

284 285
  def test_find_on_hash_conditions_with_explicit_table_name
    assert Topic.find(1, :conditions => { 'topics.approved' => false })
286
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { 'topics.approved' => true }) }
287 288
  end

289 290
  def test_find_on_hash_conditions_with_hashed_table_name
    assert Topic.find(1, :conditions => {:topics => { :approved => false }})
291
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => {:topics => { :approved => true }}) }
292 293 294 295 296 297 298 299 300 301 302 303 304 305
  end

  def test_find_with_hash_conditions_on_joined_table
    firms = Firm.all :joins => :account, :conditions => {:accounts => { :credit_limit => 50 }}
    assert_equal 1, firms.size
    assert_equal companies(:first_firm), firms.first
  end

  def test_find_with_hash_conditions_on_joined_table_and_with_range
    firms = DependentFirm.all :joins => :account, :conditions => {:name => 'RailsCore', :accounts => { :credit_limit => 55..60 }}
    assert_equal 1, firms.size
    assert_equal companies(:rails_core), firms.first
  end

306 307 308
  def test_find_on_hash_conditions_with_explicit_table_name_and_aggregate
    david = customers(:david)
    assert Customer.find(david.id, :conditions => { 'customers.name' => david.name, :address => david.address })
309
    assert_raise(ActiveRecord::RecordNotFound) {
310 311 312 313
      Customer.find(david.id, :conditions => { 'customers.name' => david.name + "1", :address => david.address })
    }
  end

314 315 316 317
  def test_find_on_association_proxy_conditions
    assert_equal [1, 2, 3, 5, 6, 7, 8, 9, 10], Comment.find_all_by_post_id(authors(:david).posts).map(&:id).sort
  end

318 319
  def test_find_on_hash_conditions_with_range
    assert_equal [1,2], Topic.find(:all, :conditions => { :id => 1..2 }).map(&:id).sort
320
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :id => 2..3 }) }
321
  end
322

323 324 325
  def test_find_on_hash_conditions_with_end_exclusive_range
    assert_equal [1,2,3], Topic.find(:all, :conditions => { :id => 1..3 }).map(&:id).sort
    assert_equal [1,2], Topic.find(:all, :conditions => { :id => 1...3 }).map(&:id).sort
326
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(3, :conditions => { :id => 2...3 }) }
327 328
  end

329 330 331 332
  def test_find_on_hash_conditions_with_multiple_ranges
    assert_equal [1,2,3], Comment.find(:all, :conditions => { :id => 1..3, :post_id => 1..2 }).map(&:id).sort
    assert_equal [1], Comment.find(:all, :conditions => { :id => 1..1, :post_id => 1..10 }).map(&:id).sort
  end
333

334 335
  def test_find_on_multiple_hash_conditions
    assert Topic.find(1, :conditions => { :author_name => "David", :title => "The First Topic", :replies_count => 1, :approved => false })
336 337 338
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :author_name => "David", :title => "The First Topic", :replies_count => 1, :approved => true }) }
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :author_name => "David", :title => "HHC", :replies_count => 1, :approved => false }) }
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find(1, :conditions => { :author_name => "David", :title => "The First Topic", :replies_count => 1, :approved => true }) }
339
  end
340

341 342 343 344 345 346 347
  def test_condition_interpolation
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = '%s'", "37signals"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!' OR 1=1"])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = %d", 1]).written_on
  end

348
  def test_condition_array_interpolation
D
David Heinemeier Hansson 已提交
349 350 351 352 353
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = '%s'", "37signals"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!"])
    assert_nil Company.find(:first, :conditions => ["name = '%s'", "37signals!' OR 1=1"])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = %d", 1]).written_on
  end
354

355 356 357 358 359
  def test_condition_hash_interpolation
    assert_kind_of Firm, Company.find(:first, :conditions => { :name => "37signals"})
    assert_nil Company.find(:first, :conditions => { :name => "37signals!"})
    assert_kind_of Time, Topic.find(:first, :conditions => {:id => 1}).written_on
  end
360

361
  def test_hash_condition_find_malformed
362
    assert_raise(ActiveRecord::StatementInvalid) {
363 364 365
      Company.find(:first, :conditions => { :id => 2, :dhh => true })
    }
  end
366

367 368
  def test_hash_condition_find_with_escaped_characters
    Company.create("name" => "Ain't noth'n like' \#stuff")
369 370 371 372 373 374 375 376 377 378 379 380 381
    assert Company.find(:first, :conditions => { :name => "Ain't noth'n like' \#stuff" })
  end

  def test_hash_condition_find_with_array
    p1, p2 = Post.find(:all, :limit => 2, :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => { :id => [p1, p2] }, :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => { :id => [p1, p2.id] }, :order => 'id asc')
  end

  def test_hash_condition_find_with_nil
    topic = Topic.find(:first, :conditions => { :last_read => nil } )
    assert_not_nil topic
    assert_nil topic.last_read
382
  end
D
David Heinemeier Hansson 已提交
383

384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425
  def test_hash_condition_find_with_aggregate_having_one_mapping
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customer = Customer.find(:first, :conditions => {:balance => balance})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_attribute_having_same_name_as_field_and_key_value_being_aggregate
    gps_location = customers(:david).gps_location
    assert_kind_of GpsLocation, gps_location
    found_customer = Customer.find(:first, :conditions => {:gps_location => gps_location})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_having_one_mapping_and_key_value_being_attribute_value
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customer = Customer.find(:first, :conditions => {:balance => balance.amount})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_attribute_having_same_name_as_field_and_key_value_being_attribute_value
    gps_location = customers(:david).gps_location
    assert_kind_of GpsLocation, gps_location
    found_customer = Customer.find(:first, :conditions => {:gps_location => gps_location.gps_location})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_aggregate_having_three_mappings
    address = customers(:david).address
    assert_kind_of Address, address
    found_customer = Customer.find(:first, :conditions => {:address => address})
    assert_equal customers(:david), found_customer
  end

  def test_hash_condition_find_with_one_condition_being_aggregate_and_another_not
    address = customers(:david).address
    assert_kind_of Address, address
    found_customer = Customer.find(:first, :conditions => {:address => address, :name => customers(:david).name})
    assert_equal customers(:david), found_customer
  end

D
David Heinemeier Hansson 已提交
426 427 428 429 430
  def test_bind_variables
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = ?", "37signals"])
    assert_nil Company.find(:first, :conditions => ["name = ?", "37signals!"])
    assert_nil Company.find(:first, :conditions => ["name = ?", "37signals!' OR 1=1"])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = ?", 1]).written_on
431
    assert_raise(ActiveRecord::PreparedStatementInvalid) {
D
David Heinemeier Hansson 已提交
432 433
      Company.find(:first, :conditions => ["id=? AND name = ?", 2])
    }
434
    assert_raise(ActiveRecord::PreparedStatementInvalid) {
435
     Company.find(:first, :conditions => ["id=?", 2, 3, 4])
D
David Heinemeier Hansson 已提交
436 437
    }
  end
438

D
David Heinemeier Hansson 已提交
439 440 441 442 443 444 445 446 447 448 449 450
  def test_bind_variables_with_quotes
    Company.create("name" => "37signals' go'es agains")
    assert Company.find(:first, :conditions => ["name = ?", "37signals' go'es agains"])
  end

  def test_named_bind_variables_with_quotes
    Company.create("name" => "37signals' go'es agains")
    assert Company.find(:first, :conditions => ["name = :name", {:name => "37signals' go'es agains"}])
  end

  def test_bind_arity
    assert_nothing_raised                                 { bind '' }
451
    assert_raise(ActiveRecord::PreparedStatementInvalid) { bind '', 1 }
452

453
    assert_raise(ActiveRecord::PreparedStatementInvalid) { bind '?' }
D
David Heinemeier Hansson 已提交
454
    assert_nothing_raised                                 { bind '?', 1 }
455
    assert_raise(ActiveRecord::PreparedStatementInvalid) { bind '?', 1, 1  }
D
David Heinemeier Hansson 已提交
456
  end
457

D
David Heinemeier Hansson 已提交
458 459 460
  def test_named_bind_variables
    assert_equal '1', bind(':a', :a => 1) # ' ruby-mode
    assert_equal '1 1', bind(':a :a', :a => 1)  # ' ruby-mode
461

462
    assert_nothing_raised { bind("'+00:00'", :foo => "bar") }
463

D
David Heinemeier Hansson 已提交
464 465 466 467 468 469
    assert_kind_of Firm, Company.find(:first, :conditions => ["name = :name", { :name => "37signals" }])
    assert_nil Company.find(:first, :conditions => ["name = :name", { :name => "37signals!" }])
    assert_nil Company.find(:first, :conditions => ["name = :name", { :name => "37signals!' OR 1=1" }])
    assert_kind_of Time, Topic.find(:first, :conditions => ["id = :id", { :id => 1 }]).written_on
  end

470
  def test_bind_enumerable
471 472
    quoted_abc = %(#{ActiveRecord::Base.connection.quote('a')},#{ActiveRecord::Base.connection.quote('b')},#{ActiveRecord::Base.connection.quote('c')})

D
David Heinemeier Hansson 已提交
473
    assert_equal '1,2,3', bind('?', [1, 2, 3])
474
    assert_equal quoted_abc, bind('?', %w(a b c))
D
David Heinemeier Hansson 已提交
475 476

    assert_equal '1,2,3', bind(':a', :a => [1, 2, 3])
477
    assert_equal quoted_abc, bind(':a', :a => %w(a b c)) # '
478 479 480

    require 'set'
    assert_equal '1,2,3', bind('?', Set.new([1, 2, 3]))
481
    assert_equal quoted_abc, bind('?', Set.new(%w(a b c)))
482 483

    assert_equal '1,2,3', bind(':a', :a => Set.new([1, 2, 3]))
484
    assert_equal quoted_abc, bind(':a', :a => Set.new(%w(a b c))) # '
D
David Heinemeier Hansson 已提交
485 486
  end

487 488 489 490 491 492 493
  def test_bind_empty_enumerable
    quoted_nil = ActiveRecord::Base.connection.quote(nil)
    assert_equal quoted_nil, bind('?', [])
    assert_equal " in (#{quoted_nil})", bind(' in (?)', [])
    assert_equal "foo in (#{quoted_nil})", bind('foo in (?)', [])
  end

J
Jeremy Kemper 已提交
494 495 496
  def test_bind_empty_string
    quoted_empty = ActiveRecord::Base.connection.quote('')
    assert_equal quoted_empty, bind('?', '')
497 498
  end

499 500 501 502 503
  def test_bind_chars
    quoted_bambi = ActiveRecord::Base.connection.quote("Bambi")
    quoted_bambi_and_thumper = ActiveRecord::Base.connection.quote("Bambi\nand\nThumper")
    assert_equal "name=#{quoted_bambi}", bind('name=?', "Bambi")
    assert_equal "name=#{quoted_bambi_and_thumper}", bind('name=?', "Bambi\nand\nThumper")
504 505
    assert_equal "name=#{quoted_bambi}", bind('name=?', "Bambi".mb_chars)
    assert_equal "name=#{quoted_bambi_and_thumper}", bind('name=?', "Bambi\nand\nThumper".mb_chars)
506 507
  end

508 509 510 511 512 513 514 515
  def test_bind_record
    o = Struct.new(:quoted_id).new(1)
    assert_equal '1', bind('?', o)

    os = [o] * 3
    assert_equal '1,1,1', bind('?', os)
  end

516 517 518 519 520 521
  def test_named_bind_with_postgresql_type_casts
    l = Proc.new { bind(":a::integer '2009-01-01'::date", :a => '10') }
    assert_nothing_raised(&l)
    assert_equal "#{ActiveRecord::Base.quote_value('10')}::integer '2009-01-01'::date", l.call
  end

D
David Heinemeier Hansson 已提交
522
  def test_string_sanitation
523 524
    assert_not_equal "#{ActiveRecord::Base.connection.quoted_string_prefix}'something ' 1=1'", ActiveRecord::Base.sanitize("something ' 1=1")
    assert_equal "#{ActiveRecord::Base.connection.quoted_string_prefix}'something; select table'", ActiveRecord::Base.sanitize("something; select table")
D
David Heinemeier Hansson 已提交
525 526 527
  end

  def test_count
528 529 530
    assert_equal(0, Entrant.count(:conditions => "id > 3"))
    assert_equal(1, Entrant.count(:conditions => ["id > ?", 2]))
    assert_equal(2, Entrant.count(:conditions => ["id > ?", 1]))
D
David Heinemeier Hansson 已提交
531 532 533 534 535 536 537 538
  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

539 540 541
  def test_dynamic_finders_should_go_through_the_find_class_method
    Topic.expects(:find).with(:first, :conditions => { :title => 'The First Topic!' })
    Topic.find_by_title("The First Topic!")
542

543 544
    Topic.expects(:find).with(:last, :conditions => { :title => 'The Last Topic!' })
    Topic.find_last_by_title("The Last Topic!")
545

546 547
    Topic.expects(:find).with(:all, :conditions => { :title => 'A Topic.' })
    Topic.find_all_by_title("A Topic.")
548

549 550 551
    Topic.expects(:find).with(:first, :conditions => { :title => 'Does not exist yet for sure!' }).times(2)
    Topic.find_or_initialize_by_title('Does not exist yet for sure!')
    Topic.find_or_create_by_title('Does not exist yet for sure!')
552 553
  end

D
David Heinemeier Hansson 已提交
554
  def test_find_by_one_attribute
555
    assert_equal topics(:first), Topic.find_by_title("The First Topic")
D
David Heinemeier Hansson 已提交
556 557
    assert_nil Topic.find_by_title("The First Topic!")
  end
J
Jeremy Kemper 已提交
558

559 560
  def test_find_by_one_attribute_bang
    assert_equal topics(:first), Topic.find_by_title!("The First Topic")
561
    assert_raise(ActiveRecord::RecordNotFound) { Topic.find_by_title!("The First Topic!") }
562 563
  end

564 565
  def test_find_by_one_attribute_caches_dynamic_finder
    # ensure this test can run independently of order
566 567
    class << Topic; self; end.send(:remove_method, :find_by_title) if Topic.public_methods.any? { |m| m.to_s == 'find_by_title' }
    assert !Topic.public_methods.any? { |m| m.to_s == 'find_by_title' }
568
    t = Topic.find_by_title("The First Topic")
569
    assert Topic.public_methods.any? { |m| m.to_s == 'find_by_title' }
570 571 572 573
  end

  def test_dynamic_finder_returns_same_results_after_caching
    # ensure this test can run independently of order
574
    class << Topic; self; end.send(:remove_method, :find_by_title) if Topic.public_method_defined?(:find_by_title)
575 576 577
    t = Topic.find_by_title("The First Topic")
    assert_equal t, Topic.find_by_title("The First Topic") # find_by_title has been cached
  end
D
David Heinemeier Hansson 已提交
578

579
  def test_find_by_one_attribute_with_order_option
580
    assert_equal accounts(:signals37), Account.find_by_credit_limit(50, :order => 'id')
581 582 583 584 585 586 587
    assert_equal accounts(:rails_core_account), Account.find_by_credit_limit(50, :order => 'id DESC')
  end

  def test_find_by_one_attribute_with_conditions
    assert_equal accounts(:rails_core_account), Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6])
  end

588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621
  def test_find_by_one_attribute_that_is_an_aggregate
    address = customers(:david).address
    assert_kind_of Address, address
    found_customer = Customer.find_by_address(address)
    assert_equal customers(:david), found_customer
  end

  def test_find_by_one_attribute_that_is_an_aggregate_with_one_attribute_difference
    address = customers(:david).address
    assert_kind_of Address, address
    missing_address = Address.new(address.street, address.city, address.country + "1")
    assert_nil Customer.find_by_address(missing_address)
    missing_address = Address.new(address.street, address.city + "1", address.country)
    assert_nil Customer.find_by_address(missing_address)
    missing_address = Address.new(address.street + "1", address.city, address.country)
    assert_nil Customer.find_by_address(missing_address)
  end

  def test_find_by_two_attributes_that_are_both_aggregates
    balance = customers(:david).balance
    address = customers(:david).address
    assert_kind_of Money, balance
    assert_kind_of Address, address
    found_customer = Customer.find_by_balance_and_address(balance, address)
    assert_equal customers(:david), found_customer
  end

  def test_find_by_two_attributes_with_one_being_an_aggregate
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customer = Customer.find_by_balance_and_name(balance, customers(:david).name)
    assert_equal customers(:david), found_customer
  end

622 623
  def test_dynamic_finder_on_one_attribute_with_conditions_caches_method
    # ensure this test can run independently of order
624 625
    class << Account; self; end.send(:remove_method, :find_by_credit_limit) if Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
    assert !Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
626
    a = Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6])
627
    assert Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
628 629 630 631
  end

  def test_dynamic_finder_on_one_attribute_with_conditions_returns_same_results_after_caching
    # ensure this test can run independently of order
632
    class << Account; self; end.send(:remove_method, :find_by_credit_limit) if Account.public_methods.any? { |m| m.to_s == 'find_by_credit_limit' }
633 634 635 636
    a = Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6])
    assert_equal a, Account.find_by_credit_limit(50, :conditions => ['firm_id = ?', 6]) # find_by_credit_limit has been cached
  end

637 638 639
  def test_find_by_one_attribute_with_several_options
    assert_equal accounts(:unknown), Account.find_by_credit_limit(50, :order => 'id DESC', :conditions => ['id != ?', 3])
  end
640

D
David Heinemeier Hansson 已提交
641
  def test_find_by_one_missing_attribute
642
    assert_raise(NoMethodError) { Topic.find_by_undertitle("The First Topic!") }
D
David Heinemeier Hansson 已提交
643
  end
644

645
  def test_find_by_invalid_method_syntax
646 647 648 649
    assert_raise(NoMethodError) { Topic.fail_to_find_by_title("The First Topic") }
    assert_raise(NoMethodError) { Topic.find_by_title?("The First Topic") }
    assert_raise(NoMethodError) { Topic.fail_to_find_or_create_by_title("Nonexistent Title") }
    assert_raise(NoMethodError) { Topic.find_or_create_by_title?("Nonexistent Title") }
650
  end
D
David Heinemeier Hansson 已提交
651 652

  def test_find_by_two_attributes
653
    assert_equal topics(:first), Topic.find_by_title_and_author_name("The First Topic", "David")
D
David Heinemeier Hansson 已提交
654 655 656
    assert_nil Topic.find_by_title_and_author_name("The First Topic", "Mary")
  end

657 658 659 660 661 662 663 664 665
  def test_find_last_by_one_attribute
    assert_equal Topic.last, Topic.find_last_by_title(Topic.last.title)
    assert_nil Topic.find_last_by_title("A title with no matches")
  end

  def test_find_last_by_one_attribute_caches_dynamic_finder
    # ensure this test can run independently of order
    class << Topic; self; end.send(:remove_method, :find_last_by_title) if Topic.public_methods.any? { |m| m.to_s == 'find_last_by_title' }
    assert !Topic.public_methods.any? { |m| m.to_s == 'find_last_by_title' }
666
    t = Topic.find_last_by_title(Topic.last.title)
667 668 669 670
    assert Topic.public_methods.any? { |m| m.to_s == 'find_last_by_title' }
  end

  def test_find_last_by_invalid_method_syntax
671 672
    assert_raise(NoMethodError) { Topic.fail_to_find_last_by_title("The First Topic") }
    assert_raise(NoMethodError) { Topic.find_last_by_title?("The First Topic") }
673 674 675 676 677 678 679
  end

  def test_find_last_by_one_attribute_with_several_options
    assert_equal accounts(:signals37), Account.find_last_by_credit_limit(50, :order => 'id DESC', :conditions => ['id != ?', 3])
  end

  def test_find_last_by_one_missing_attribute
680
    assert_raise(NoMethodError) { Topic.find_last_by_undertitle("The Last Topic!") }
681 682 683 684 685 686 687 688
  end

  def test_find_last_by_two_attributes
    topic = Topic.last
    assert_equal topic, Topic.find_last_by_title_and_author_name(topic.title, topic.author_name)
    assert_nil Topic.find_last_by_title_and_author_name(topic.title, "Anonymous")
  end

D
David Heinemeier Hansson 已提交
689 690 691
  def test_find_all_by_one_attribute
    topics = Topic.find_all_by_content("Have a nice day")
    assert_equal 2, topics.size
692
    assert topics.include?(topics(:first))
D
David Heinemeier Hansson 已提交
693 694 695

    assert_equal [], Topic.find_all_by_title("The First Topic!!")
  end
696

697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722
  def test_find_all_by_one_attribute_that_is_an_aggregate
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customers = Customer.find_all_by_balance(balance)
    assert_equal 1, found_customers.size
    assert_equal customers(:david), found_customers.first
  end

  def test_find_all_by_two_attributes_that_are_both_aggregates
    balance = customers(:david).balance
    address = customers(:david).address
    assert_kind_of Money, balance
    assert_kind_of Address, address
    found_customers = Customer.find_all_by_balance_and_address(balance, address)
    assert_equal 1, found_customers.size
    assert_equal customers(:david), found_customers.first
  end

  def test_find_all_by_two_attributes_with_one_being_an_aggregate
    balance = customers(:david).balance
    assert_kind_of Money, balance
    found_customers = Customer.find_all_by_balance_and_name(balance, customers(:david).name)
    assert_equal 1, found_customers.size
    assert_equal customers(:david), found_customers.first
  end

723 724
  def test_find_all_by_one_attribute_with_options
    topics = Topic.find_all_by_content("Have a nice day", :order => "id DESC")
725
    assert topics(:first), topics.last
726 727

    topics = Topic.find_all_by_content("Have a nice day", :order => "id")
728
    assert topics(:first), topics.first
729
  end
D
David Heinemeier Hansson 已提交
730

731
  def test_find_all_by_array_attribute
732
    assert_equal 2, Topic.find_all_by_title(["The First Topic", "The Second Topic of the day"]).size
733 734
  end

D
David Heinemeier Hansson 已提交
735 736 737
  def test_find_all_by_boolean_attribute
    topics = Topic.find_all_by_approved(false)
    assert_equal 1, topics.size
738
    assert topics.include?(topics(:first))
D
David Heinemeier Hansson 已提交
739 740

    topics = Topic.find_all_by_approved(true)
741
    assert_equal 3, topics.size
742
    assert topics.include?(topics(:second))
D
David Heinemeier Hansson 已提交
743
  end
744

D
David Heinemeier Hansson 已提交
745 746 747 748 749
  def test_find_by_nil_attribute
    topic = Topic.find_by_last_read nil
    assert_not_nil topic
    assert_nil topic.last_read
  end
750

D
David Heinemeier Hansson 已提交
751 752
  def test_find_all_by_nil_attribute
    topics = Topic.find_all_by_last_read nil
753 754
    assert_equal 3, topics.size
    assert topics.collect(&:last_read).all?(&:nil?)
D
David Heinemeier Hansson 已提交
755
  end
756

D
David Heinemeier Hansson 已提交
757 758 759 760 761 762 763 764 765 766 767
  def test_find_by_nil_and_not_nil_attributes
    topic = Topic.find_by_last_read_and_author_name nil, "Mary"
    assert_equal "Mary", topic.author_name
  end

  def test_find_all_by_nil_and_not_nil_attributes
    topics = Topic.find_all_by_last_read_and_author_name nil, "Mary"
    assert_equal 1, topics.size
    assert_equal "Mary", topics[0].author_name
  end

768 769 770 771 772
  def test_find_or_create_from_one_attribute
    number_of_companies = Company.count
    sig38 = Company.find_or_create_by_name("38signals")
    assert_equal number_of_companies + 1, Company.count
    assert_equal sig38, Company.find_or_create_by_name("38signals")
773
    assert !sig38.new_record?
774 775 776
  end

  def test_find_or_create_from_two_attributes
777 778 779 780
    number_of_topics = Topic.count
    another = Topic.find_or_create_by_title_and_author_name("Another topic","John")
    assert_equal number_of_topics + 1, Topic.count
    assert_equal another, Topic.find_or_create_by_title_and_author_name("Another topic", "John")
781
    assert !another.new_record?
782
  end
783

784 785 786 787 788 789 790 791
  def test_find_or_create_from_two_attributes_with_one_being_an_aggregate
    number_of_customers = Customer.count
    created_customer = Customer.find_or_create_by_balance_and_name(Money.new(123), "Elizabeth")
    assert_equal number_of_customers + 1, Customer.count
    assert_equal created_customer, Customer.find_or_create_by_balance(Money.new(123), "Elizabeth")
    assert !created_customer.new_record?
  end

792 793 794 795 796 797 798 799 800 801 802
  def test_find_or_create_from_one_attribute_and_hash
    number_of_companies = Company.count
    sig38 = Company.find_or_create_by_name({:name => "38signals", :firm_id => 17, :client_of => 23})
    assert_equal number_of_companies + 1, Company.count
    assert_equal sig38, Company.find_or_create_by_name({:name => "38signals", :firm_id => 17, :client_of => 23})
    assert !sig38.new_record?
    assert_equal "38signals", sig38.name
    assert_equal 17, sig38.firm_id
    assert_equal 23, sig38.client_of
  end

803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822
  def test_find_or_create_from_one_aggregate_attribute
    number_of_customers = Customer.count
    created_customer = Customer.find_or_create_by_balance(Money.new(123))
    assert_equal number_of_customers + 1, Customer.count
    assert_equal created_customer, Customer.find_or_create_by_balance(Money.new(123))
    assert !created_customer.new_record?
  end

  def test_find_or_create_from_one_aggregate_attribute_and_hash
    number_of_customers = Customer.count
    balance = Money.new(123)
    name = "Elizabeth"
    created_customer = Customer.find_or_create_by_balance({:balance => balance, :name => name})
    assert_equal number_of_customers + 1, Customer.count
    assert_equal created_customer, Customer.find_or_create_by_balance({:balance => balance, :name => name})
    assert !created_customer.new_record?
    assert_equal balance, created_customer.balance
    assert_equal name, created_customer.name
  end

823 824 825
  def test_find_or_initialize_from_one_attribute
    sig38 = Company.find_or_initialize_by_name("38signals")
    assert_equal "38signals", sig38.name
826
    assert sig38.new_record?
827
  end
J
Jeremy Kemper 已提交
828

829 830 831 832 833 834
  def test_find_or_initialize_from_one_aggregate_attribute
    new_customer = Customer.find_or_initialize_by_balance(Money.new(123))
    assert_equal 123, new_customer.balance.amount
    assert new_customer.new_record?
  end

835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850
  def test_find_or_initialize_from_one_attribute_should_not_set_attribute_even_when_protected
    c = Company.find_or_initialize_by_name({:name => "Fortune 1000", :rating => 1000})
    assert_equal "Fortune 1000", c.name
    assert_not_equal 1000, c.rating
    assert c.valid?
    assert c.new_record?
  end

  def test_find_or_create_from_one_attribute_should_set_not_attribute_even_when_protected
    c = Company.find_or_create_by_name({:name => "Fortune 1000", :rating => 1000})
    assert_equal "Fortune 1000", c.name
    assert_not_equal 1000, c.rating
    assert c.valid?
    assert !c.new_record?
  end

851 852 853 854 855
  def test_find_or_initialize_from_one_attribute_should_set_attribute_even_when_protected
    c = Company.find_or_initialize_by_name_and_rating("Fortune 1000", 1000)
    assert_equal "Fortune 1000", c.name
    assert_equal 1000, c.rating
    assert c.valid?
J
Jeremy Kemper 已提交
856
    assert c.new_record?
857 858 859 860 861 862 863
  end

  def test_find_or_create_from_one_attribute_should_set_attribute_even_when_protected
    c = Company.find_or_create_by_name_and_rating("Fortune 1000", 1000)
    assert_equal "Fortune 1000", c.name
    assert_equal 1000, c.rating
    assert c.valid?
J
Jeremy Kemper 已提交
864
    assert !c.new_record?
865
  end
866

867 868 869 870 871 872 873 874 875 876 877 878 879 880 881
  def test_find_or_initialize_should_set_protected_attributes_if_given_as_block
    c = Company.find_or_initialize_by_name(:name => "Fortune 1000") { |f| f.rating = 1000 }
    assert_equal "Fortune 1000", c.name
    assert_equal 1000.to_f, c.rating.to_f
    assert c.valid?
    assert c.new_record?
  end

  def test_find_or_create_should_set_protected_attributes_if_given_as_block
    c = Company.find_or_create_by_name(:name => "Fortune 1000") { |f| f.rating = 1000 }
    assert_equal "Fortune 1000", c.name
    assert_equal 1000.to_f, c.rating.to_f
    assert c.valid?
    assert !c.new_record?
  end
882

883 884 885 886 887 888 889 890 891 892 893
  def test_find_or_create_should_work_with_block_on_first_call
	  class << Company
		undef_method(:find_or_create_by_name) if method_defined?(:find_or_create_by_name)
	  end
    c = Company.find_or_create_by_name(:name => "Fortune 1000") { |f| f.rating = 1000 }
    assert_equal "Fortune 1000", c.name
    assert_equal 1000.to_f, c.rating.to_f
    assert c.valid?
    assert !c.new_record?
  end

894
  def test_dynamic_find_or_initialize_from_one_attribute_caches_method
895 896
    class << Company; self; end.send(:remove_method, :find_or_initialize_by_name) if Company.public_methods.any? { |m| m.to_s == 'find_or_initialize_by_name' }
    assert !Company.public_methods.any? { |m| m.to_s == 'find_or_initialize_by_name' }
897
    sig38 = Company.find_or_initialize_by_name("38signals")
898
    assert Company.public_methods.any? { |m| m.to_s == 'find_or_initialize_by_name' }
899 900
  end

901 902 903 904
  def test_find_or_initialize_from_two_attributes
    another = Topic.find_or_initialize_by_title_and_author_name("Another topic","John")
    assert_equal "Another topic", another.title
    assert_equal "John", another.author_name
905
    assert another.new_record?
906
  end
907

908 909 910 911 912 913 914
  def test_find_or_initialize_from_one_aggregate_attribute_and_one_not
    new_customer = Customer.find_or_initialize_by_balance_and_name(Money.new(123), "Elizabeth")
    assert_equal 123, new_customer.balance.amount
    assert_equal "Elizabeth", new_customer.name
    assert new_customer.new_record?
  end

915 916 917 918 919 920 921
  def test_find_or_initialize_from_one_attribute_and_hash
    sig38 = Company.find_or_initialize_by_name({:name => "38signals", :firm_id => 17, :client_of => 23})
    assert_equal "38signals", sig38.name
    assert_equal 17, sig38.firm_id
    assert_equal 23, sig38.client_of
    assert sig38.new_record?
  end
922

923 924 925 926 927 928 929 930 931
  def test_find_or_initialize_from_one_aggregate_attribute_and_hash
    balance = Money.new(123)
    name = "Elizabeth"
    new_customer = Customer.find_or_initialize_by_balance({:balance => balance, :name => name})
    assert_equal balance, new_customer.balance
    assert_equal name, new_customer.name
    assert new_customer.new_record?
  end

D
David Heinemeier Hansson 已提交
932
  def test_find_with_bad_sql
933
    assert_raise(ActiveRecord::StatementInvalid) { Topic.find_by_sql "select 1 from badtable" }
D
David Heinemeier Hansson 已提交
934 935
  end

936
  def test_find_with_invalid_params
937 938
    assert_raise(ArgumentError) { Topic.find :first, :join => "It should be `joins'" }
    assert_raise(ArgumentError) { Topic.find :first, :conditions => '1 = 1', :join => "It should be `joins'" }
939 940
  end

941
  def test_dynamic_finder_with_invalid_params
942
    assert_raise(ArgumentError) { Topic.find_by_title 'No Title', :join => "It should be `joins'" }
943 944
  end

D
David Heinemeier Hansson 已提交
945
  def test_find_all_with_limit
946
    first_five_developers = Developer.find :all, :order => 'id ASC', :limit =>  5
D
David Heinemeier Hansson 已提交
947 948 949
    assert_equal 5, first_five_developers.length
    assert_equal 'David', first_five_developers.first.name
    assert_equal 'fixture_5', first_five_developers.last.name
950

951
    no_developers = Developer.find :all, :order => 'id ASC', :limit => 0
D
David Heinemeier Hansson 已提交
952 953
    assert_equal 0, no_developers.length
  end
954

D
David Heinemeier Hansson 已提交
955 956 957 958
  def test_find_all_with_limit_and_offset
    first_three_developers = Developer.find :all, :order => 'id ASC', :limit => 3, :offset => 0
    second_three_developers = Developer.find :all, :order => 'id ASC', :limit => 3, :offset => 3
    last_two_developers = Developer.find :all, :order => 'id ASC', :limit => 2, :offset => 8
959

D
David Heinemeier Hansson 已提交
960 961 962
    assert_equal 3, first_three_developers.length
    assert_equal 3, second_three_developers.length
    assert_equal 2, last_two_developers.length
963

D
David Heinemeier Hansson 已提交
964 965 966 967 968
    assert_equal 'David', first_three_developers.first.name
    assert_equal 'fixture_4', second_three_developers.first.name
    assert_equal 'fixture_9', last_two_developers.first.name
  end

969 970 971 972 973 974 975 976 977 978
  def test_find_all_with_limit_and_offset_and_multiple_order_clauses
    first_three_posts = Post.find :all, :order => 'author_id, id', :limit => 3, :offset => 0
    second_three_posts = Post.find :all, :order => ' author_id,id ', :limit => 3, :offset => 3
    last_posts = Post.find :all, :order => ' author_id, id  ', :limit => 3, :offset => 6

    assert_equal [[0,3],[1,1],[1,2]], first_three_posts.map { |p| [p.author_id, p.id] }
    assert_equal [[1,4],[1,5],[1,6]], second_three_posts.map { |p| [p.author_id, p.id] }
    assert_equal [[2,7]], last_posts.map { |p| [p.author_id, p.id] }
  end

979
  def test_find_all_with_join
980
    developers_on_project_one = Developer.find(
981 982
      :all,
      :joins => 'LEFT JOIN developers_projects ON developers.id = developers_projects.developer_id',
983 984
      :conditions => 'project_id=1'
    )
985
    assert_equal 3, developers_on_project_one.length
986 987 988
    developer_names = developers_on_project_one.map { |d| d.name }
    assert developer_names.include?('David')
    assert developer_names.include?('Jamis')
989
  end
990

991 992 993 994 995 996 997 998 999
  def test_joins_dont_clobber_id
    first = Firm.find(
      :first,
      :joins => 'INNER JOIN companies AS clients ON clients.firm_id = companies.id',
      :conditions => 'companies.id = 1'
    )
    assert_equal 1, first.id
  end

1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010
  def test_joins_with_string_array
    person_with_reader_and_post = Post.find(
      :all,
      :joins => [
        "INNER JOIN categorizations ON categorizations.post_id = posts.id",
        "INNER JOIN categories ON categories.id = categorizations.category_id AND categories.type = 'SpecialCategory'"
      ]
    )
    assert_equal 1, person_with_reader_and_post.size
  end

1011 1012 1013
  def test_find_by_id_with_conditions_with_or
    assert_nothing_raised do
      Post.find([1,2,3],
1014
        :conditions => "posts.id <= 3 OR posts.#{QUOTED_TYPE} = 'Post'")
1015 1016
    end
  end
1017

1018 1019
  # http://dev.rubyonrails.org/ticket/6778
  def test_find_ignores_previously_inserted_record
1020
    post = Post.create!(:title => 'test', :body => 'it out')
1021 1022 1023
    assert_equal [], Post.find_all_by_id(nil)
  end

1024 1025 1026 1027 1028 1029 1030 1031 1032
  def test_find_by_empty_ids
    assert_equal [], Post.find([])
  end

  def test_find_by_empty_in_condition
    assert_equal [], Post.find(:all, :conditions => ['id in (?)', []])
  end

  def test_find_by_records
J
Jeremy Kemper 已提交
1033 1034 1035
    p1, p2 = Post.find(:all, :limit => 2, :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => ['id in (?)', [p1, p2]], :order => 'id asc')
    assert_equal [p1, p2], Post.find(:all, :conditions => ['id in (?)', [p1, p2.id]], :order => 'id asc')
1036 1037
  end

1038 1039 1040 1041 1042 1043 1044 1045 1046
  def test_select_value
    assert_equal "37signals", Company.connection.select_value("SELECT name FROM companies WHERE id = 1")
    assert_nil Company.connection.select_value("SELECT name FROM companies WHERE id = -1")
    # make sure we didn't break count...
    assert_equal 0, Company.count_by_sql("SELECT COUNT(*) FROM companies WHERE name = 'Halliburton'")
    assert_equal 1, Company.count_by_sql("SELECT COUNT(*) FROM companies WHERE name = '37signals'")
  end

  def test_select_values
1047 1048
    assert_equal ["1","2","3","4","5","6","7","8","9", "10"], Company.connection.select_values("SELECT id FROM companies ORDER BY id").map! { |i| i.to_s }
    assert_equal ["37signals","Summit","Microsoft", "Flamboyant Software", "Ex Nihilo", "RailsCore", "Leetsoft", "Jadedpixel", "Odegy", "Ex Nihilo Part Deux"], Company.connection.select_values("SELECT name FROM companies ORDER BY id")
1049 1050
  end

1051 1052 1053 1054 1055
  def test_select_rows
    assert_equal(
      [["1", nil, nil, "37signals"],
       ["2", "1", "2", "Summit"],
       ["3", "1", "1", "Microsoft"]],
1056
      Company.connection.select_rows("SELECT id, firm_id, client_of, name FROM companies WHERE id IN (1,2,3) ORDER BY id").map! {|i| i.map! {|j| j.to_s unless j.nil?}})
1057
    assert_equal [["1", "37signals"], ["2", "Summit"], ["3", "Microsoft"]],
1058
      Company.connection.select_rows("SELECT id, name FROM companies WHERE id IN (1,2,3) ORDER BY id").map! {|i| i.map! {|j| j.to_s unless j.nil?}}
1059 1060
  end

1061
  def test_find_with_order_on_included_associations_with_construct_finder_sql_for_association_limiting_and_is_distinct
1062
    assert_equal 2, Post.find(:all, :include => { :authors => :author_address }, :order => ' author_addresses.id DESC ', :limit => 2).size
1063

1064 1065
    assert_equal 3, Post.find(:all, :include => { :author => :author_address, :authors => :author_address},
                              :order => ' author_addresses_authors.id DESC ', :limit => 3).size
1066 1067
  end

1068
  def test_with_limiting_with_custom_select
T
Tarmo Tänav 已提交
1069
    posts = Post.find(:all, :include => :author, :select => ' posts.*, authors.id as "author_id"', :limit => 3, :order => 'posts.id')
1070 1071
    assert_equal 3, posts.size
    assert_equal [0, 1, 1], posts.map(&:author_id).sort
1072
  end
1073

1074 1075 1076 1077 1078 1079 1080 1081
  def test_finder_with_scoped_from
    all_topics = Topic.all

    Topic.with_scope(:find => { :from => 'fake_topics' }) do
      assert_equal all_topics, Topic.all(:from => 'topics')
    end
  end

D
David Heinemeier Hansson 已提交
1082 1083 1084 1085 1086 1087 1088 1089 1090
  protected
    def bind(statement, *vars)
      if vars.first.is_a?(Hash)
        ActiveRecord::Base.send(:replace_named_bind_variables, statement, vars.first)
      else
        ActiveRecord::Base.send(:replace_bind_variables, statement, vars)
      end
    end
end