mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
373 lines
11 KiB
Ruby
373 lines
11 KiB
Ruby
require "cases/helper"
|
|
require "support/schema_dumping_helper"
|
|
require "models/topic"
|
|
require "models/reply"
|
|
require "models/subscriber"
|
|
require "models/movie"
|
|
require "models/keyboard"
|
|
require "models/mixed_case_monkey"
|
|
require "models/dashboard"
|
|
|
|
class PrimaryKeysTest < ActiveRecord::TestCase
|
|
fixtures :topics, :subscribers, :movies, :mixed_case_monkeys
|
|
|
|
def test_to_key_with_default_primary_key
|
|
topic = Topic.new
|
|
assert_nil topic.to_key
|
|
topic = Topic.find(1)
|
|
assert_equal [1], topic.to_key
|
|
end
|
|
|
|
def test_to_key_with_customized_primary_key
|
|
keyboard = Keyboard.new
|
|
assert_nil keyboard.to_key
|
|
keyboard.save
|
|
assert_equal keyboard.to_key, [keyboard.id]
|
|
end
|
|
|
|
def test_read_attribute_with_custom_primary_key
|
|
keyboard = Keyboard.create!
|
|
assert_equal keyboard.key_number, keyboard.read_attribute(:id)
|
|
end
|
|
|
|
def test_to_key_with_primary_key_after_destroy
|
|
topic = Topic.find(1)
|
|
topic.destroy
|
|
assert_equal [1], topic.to_key
|
|
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_nil topic.id
|
|
assert_nothing_raised { topic.save! }
|
|
id = topic.id
|
|
|
|
topicReloaded = Topic.find(id)
|
|
assert_equal("New Topic", topicReloaded.title)
|
|
end
|
|
|
|
def test_customized_primary_key_auto_assigns_on_save
|
|
Keyboard.delete_all
|
|
keyboard = Keyboard.new(name: "HHKB")
|
|
assert_nothing_raised { keyboard.save! }
|
|
assert_equal keyboard.id, Keyboard.find_by_name("HHKB").id
|
|
end
|
|
|
|
def test_customized_primary_key_can_be_get_before_saving
|
|
keyboard = Keyboard.new
|
|
assert_nil keyboard.id
|
|
assert_nothing_raised { assert_nil keyboard.key_number }
|
|
end
|
|
|
|
def test_customized_string_primary_key_settable_before_save
|
|
subscriber = Subscriber.new
|
|
assert_nothing_raised { subscriber.id = "webster123" }
|
|
assert_equal "webster123", subscriber.id
|
|
assert_equal "webster123", subscriber.nick
|
|
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! }
|
|
assert_equal("jdoe", subscriber.id)
|
|
|
|
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
|
|
old_primary_key_prefix_type = ActiveRecord::Base.primary_key_prefix_type
|
|
ActiveRecord::Base.primary_key_prefix_type = :table_name
|
|
Topic.reset_primary_key
|
|
assert_equal "topicid", Topic.primary_key
|
|
|
|
ActiveRecord::Base.primary_key_prefix_type = :table_name_with_underscore
|
|
Topic.reset_primary_key
|
|
assert_equal "topic_id", Topic.primary_key
|
|
|
|
ActiveRecord::Base.primary_key_prefix_type = nil
|
|
Topic.reset_primary_key
|
|
assert_equal "id", Topic.primary_key
|
|
ensure
|
|
ActiveRecord::Base.primary_key_prefix_type = old_primary_key_prefix_type
|
|
end
|
|
|
|
def test_delete_should_quote_pkey
|
|
assert_nothing_raised { MixedCaseMonkey.delete(1) }
|
|
end
|
|
def test_update_counters_should_quote_pkey_and_quote_counter_columns
|
|
assert_nothing_raised { MixedCaseMonkey.update_counters(1, fleaCount: 99) }
|
|
end
|
|
def test_find_with_one_id_should_quote_pkey
|
|
assert_nothing_raised { MixedCaseMonkey.find(1) }
|
|
end
|
|
def test_find_with_multiple_ids_should_quote_pkey
|
|
assert_nothing_raised { MixedCaseMonkey.find([1,2]) }
|
|
end
|
|
def test_instance_update_should_quote_pkey
|
|
assert_nothing_raised { MixedCaseMonkey.find(1).save }
|
|
end
|
|
def test_instance_destroy_should_quote_pkey
|
|
assert_nothing_raised { MixedCaseMonkey.find(1).destroy }
|
|
end
|
|
|
|
def test_supports_primary_key
|
|
assert_nothing_raised do
|
|
ActiveRecord::Base.connection.supports_primary_key?
|
|
end
|
|
end
|
|
|
|
if ActiveRecord::Base.connection.supports_primary_key?
|
|
def test_primary_key_returns_value_if_it_exists
|
|
klass = Class.new(ActiveRecord::Base) do
|
|
self.table_name = "developers"
|
|
end
|
|
|
|
assert_equal "id", klass.primary_key
|
|
end
|
|
|
|
def test_primary_key_returns_nil_if_it_does_not_exist
|
|
klass = Class.new(ActiveRecord::Base) do
|
|
self.table_name = "developers_projects"
|
|
end
|
|
|
|
assert_nil klass.primary_key
|
|
end
|
|
end
|
|
|
|
def test_quoted_primary_key_after_set_primary_key
|
|
k = Class.new( ActiveRecord::Base )
|
|
assert_equal k.connection.quote_column_name("id"), k.quoted_primary_key
|
|
k.primary_key = "foo"
|
|
assert_equal k.connection.quote_column_name("foo"), k.quoted_primary_key
|
|
end
|
|
|
|
def test_auto_detect_primary_key_from_schema
|
|
MixedCaseMonkey.reset_primary_key
|
|
assert_equal "monkeyID", MixedCaseMonkey.primary_key
|
|
end
|
|
|
|
def test_primary_key_update_with_custom_key_name
|
|
dashboard = Dashboard.create!(dashboard_id: "1")
|
|
dashboard.id = "2"
|
|
dashboard.save!
|
|
|
|
dashboard = Dashboard.first
|
|
assert_equal "2", dashboard.id
|
|
end
|
|
|
|
def test_create_without_primary_key_no_extra_query
|
|
klass = Class.new(ActiveRecord::Base) do
|
|
self.table_name = "dashboards"
|
|
end
|
|
klass.create! # warmup schema cache
|
|
assert_queries(3, ignore_none: true) { klass.create! }
|
|
end
|
|
|
|
if current_adapter?(:PostgreSQLAdapter)
|
|
def test_serial_with_quoted_sequence_name
|
|
column = MixedCaseMonkey.columns_hash[MixedCaseMonkey.primary_key]
|
|
assert_equal "nextval('\"mixed_case_monkeys_monkeyID_seq\"'::regclass)", column.default_function
|
|
assert column.serial?
|
|
end
|
|
|
|
def test_serial_with_unquoted_sequence_name
|
|
column = Topic.columns_hash[Topic.primary_key]
|
|
assert_equal "nextval('topics_id_seq'::regclass)", column.default_function
|
|
assert column.serial?
|
|
end
|
|
end
|
|
end
|
|
|
|
class PrimaryKeyWithNoConnectionTest < ActiveRecord::TestCase
|
|
self.use_transactional_tests = false
|
|
|
|
unless in_memory_db?
|
|
def test_set_primary_key_with_no_connection
|
|
connection = ActiveRecord::Base.remove_connection
|
|
|
|
model = Class.new(ActiveRecord::Base)
|
|
model.primary_key = "foo"
|
|
|
|
assert_equal "foo", model.primary_key
|
|
|
|
ActiveRecord::Base.establish_connection(connection)
|
|
|
|
assert_equal "foo", model.primary_key
|
|
end
|
|
end
|
|
end
|
|
|
|
class PrimaryKeyAnyTypeTest < ActiveRecord::TestCase
|
|
include SchemaDumpingHelper
|
|
|
|
self.use_transactional_tests = false
|
|
|
|
class Barcode < ActiveRecord::Base
|
|
end
|
|
|
|
setup do
|
|
@connection = ActiveRecord::Base.connection
|
|
@connection.create_table(:barcodes, primary_key: "code", id: :string, limit: 42, force: true)
|
|
end
|
|
|
|
teardown do
|
|
@connection.drop_table(:barcodes, if_exists: true)
|
|
end
|
|
|
|
def test_any_type_primary_key
|
|
assert_equal "code", Barcode.primary_key
|
|
|
|
column = Barcode.column_for_attribute(Barcode.primary_key)
|
|
assert_not column.null
|
|
assert_equal :string, column.type
|
|
assert_equal 42, column.limit
|
|
end
|
|
|
|
test "schema dump primary key includes type and options" do
|
|
schema = dump_table_schema "barcodes"
|
|
assert_match %r{create_table "barcodes", primary_key: "code", id: :string, limit: 42}, schema
|
|
end
|
|
end
|
|
|
|
class CompositePrimaryKeyTest < ActiveRecord::TestCase
|
|
include SchemaDumpingHelper
|
|
|
|
self.use_transactional_tests = false
|
|
|
|
def setup
|
|
@connection = ActiveRecord::Base.connection
|
|
@connection.schema_cache.clear!
|
|
@connection.create_table(:barcodes, primary_key: ["region", "code"], force: true) do |t|
|
|
t.string :region
|
|
t.integer :code
|
|
end
|
|
end
|
|
|
|
def teardown
|
|
@connection.drop_table(:barcodes, if_exists: true)
|
|
end
|
|
|
|
def test_composite_primary_key
|
|
assert_equal ["region", "code"], @connection.primary_keys("barcodes")
|
|
end
|
|
|
|
def test_primary_key_issues_warning
|
|
model = Class.new(ActiveRecord::Base) do
|
|
def self.table_name
|
|
"barcodes"
|
|
end
|
|
end
|
|
warning = capture(:stderr) do
|
|
assert_nil model.primary_key
|
|
end
|
|
assert_match(/WARNING: Active Record does not support composite primary key\./, warning)
|
|
end
|
|
|
|
def test_collectly_dump_composite_primary_key
|
|
schema = dump_table_schema "barcodes"
|
|
assert_match %r{create_table "barcodes", primary_key: \["region", "code"\]}, schema
|
|
end
|
|
end
|
|
|
|
if current_adapter?(:Mysql2Adapter)
|
|
class PrimaryKeyBigintNilDefaultTest < ActiveRecord::TestCase
|
|
include SchemaDumpingHelper
|
|
|
|
self.use_transactional_tests = false
|
|
|
|
def setup
|
|
@connection = ActiveRecord::Base.connection
|
|
@connection.create_table(:bigint_defaults, id: :bigint, default: nil, force: true)
|
|
end
|
|
|
|
def teardown
|
|
@connection.drop_table :bigint_defaults, if_exists: true
|
|
end
|
|
|
|
test "primary key with bigint allows default override via nil" do
|
|
column = @connection.columns(:bigint_defaults).find { |c| c.name == "id" }
|
|
assert column.bigint?
|
|
assert_not column.auto_increment?
|
|
end
|
|
|
|
test "schema dump primary key with bigint default nil" do
|
|
schema = dump_table_schema "bigint_defaults"
|
|
assert_match %r{create_table "bigint_defaults", id: :bigint, default: nil}, schema
|
|
end
|
|
end
|
|
end
|
|
|
|
if current_adapter?(:PostgreSQLAdapter, :Mysql2Adapter)
|
|
class PrimaryKeyBigSerialTest < ActiveRecord::TestCase
|
|
include SchemaDumpingHelper
|
|
|
|
self.use_transactional_tests = false
|
|
|
|
class Widget < ActiveRecord::Base
|
|
end
|
|
|
|
setup do
|
|
@connection = ActiveRecord::Base.connection
|
|
if current_adapter?(:PostgreSQLAdapter)
|
|
@connection.create_table(:widgets, id: :bigserial, force: true)
|
|
else
|
|
@connection.create_table(:widgets, id: :bigint, force: true)
|
|
end
|
|
end
|
|
|
|
teardown do
|
|
@connection.drop_table :widgets, if_exists: true
|
|
Widget.reset_column_information
|
|
end
|
|
|
|
test "primary key column type with bigserial" do
|
|
column_type = Widget.type_for_attribute(Widget.primary_key)
|
|
assert_equal :integer, column_type.type
|
|
assert_equal 8, column_type.limit
|
|
end
|
|
|
|
test "primary key with bigserial are automatically numbered" do
|
|
widget = Widget.create!
|
|
assert_not_nil widget.id
|
|
end
|
|
|
|
test "schema dump primary key with bigserial" do
|
|
schema = dump_table_schema "widgets"
|
|
if current_adapter?(:PostgreSQLAdapter)
|
|
assert_match %r{create_table "widgets", id: :bigserial, force: :cascade}, schema
|
|
else
|
|
assert_match %r{create_table "widgets", id: :bigint, force: :cascade}, schema
|
|
end
|
|
end
|
|
|
|
if current_adapter?(:Mysql2Adapter)
|
|
test "primary key column type with options" do
|
|
@connection.create_table(:widgets, id: :primary_key, limit: 8, unsigned: true, force: true)
|
|
column = @connection.columns(:widgets).find { |c| c.name == "id" }
|
|
assert column.auto_increment?
|
|
assert_equal :integer, column.type
|
|
assert_equal 8, column.limit
|
|
assert column.unsigned?
|
|
end
|
|
end
|
|
end
|
|
end
|