rails--rails/activerecord/test/pk_test.rb

require 'abstract_unit'
require 'fixtures/topic'
require 'fixtures/subscriber'
require 'fixtures/movie'

class PrimaryKeysTest < Test::Unit::TestCase
  def setup
    @topics      = create_fixtures "topics"
    @subscribers = create_fixtures "subscribers"
    @movies      = create_fixtures "movies"
  end

  def test_integer_key
    topic = Topic.find(1)
    assert_equal(@topics["first"]["author_name"], topic.author_name)
    topic = Topic.find(2)
    assert_equal(@topics["second"]["author_name"], topic.author_name)

    topic = Topic.new
    topic.title = "New Topic"
    assert_equal(nil, topic.id)
    assert_nothing_raised{ topic.save }
    id = topic.id

    topicReloaded = Topic.find(id)
    assert_equal("New Topic", topicReloaded.title)
  end

  def test_string_key
    subscriber = Subscriber.find(@subscribers["first"]["nick"])
    assert_equal(@subscribers["first"]["name"], subscriber.name)
    subscriber = Subscriber.find(@subscribers["second"]["nick"])
    assert_equal(@subscribers["second"]["name"], subscriber.name)

    subscriber = Subscriber.new
    subscriber.id = "jdoe"
    assert_equal("jdoe", subscriber.id)
    subscriber.name = "John Doe"
    assert_nothing_raised{ subscriber.save }

    subscriberReloaded = Subscriber.find("jdoe")
    assert_equal("John Doe", subscriberReloaded.name)
  end

  def test_find_with_more_than_one_string_key
    assert_equal 2, Subscriber.find(@subscribers["first"]["nick"], @subscribers["second"]["nick"]).length
  end
  
  def test_primary_key_prefix
    ActiveRecord::Base.primary_key_prefix_type = :table_name
    assert_equal "topicid", Topic.primary_key

    ActiveRecord::Base.primary_key_prefix_type = :table_name_with_underscore
    assert_equal "topic_id", Topic.primary_key

    ActiveRecord::Base.primary_key_prefix_type = nil
    assert_equal "id", Topic.primary_key
  end
end