1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
require 'cases/helper'
if ActiveRecord::Base.connection.supports_foreign_keys?
module ActiveRecord
class Migration
class ReferencesForeignKeyTest < ActiveRecord::TestCase
setup do
@connection = ActiveRecord::Base.connection
@connection.create_table(:testing_parents, force: true)
end
teardown do
@connection.drop_table "testings", if_exists: true
@connection.drop_table "testing_parents", if_exists: true
end
test "foreign keys can be created with the table" do
@connection.create_table :testings do |t|
t.references :testing_parent, foreign_key: true
end
fk = @connection.foreign_keys("testings").first
assert_equal "testings", fk.from_table
assert_equal "testing_parents", fk.to_table
end
test "no foreign key is created by default" do
@connection.create_table :testings do |t|
t.references :testing_parent
end
assert_equal [], @connection.foreign_keys("testings")
end
test "options hash can be passed" do
@connection.change_table :testing_parents do |t|
t.integer :other_id
t.index :other_id, unique: true
end
@connection.create_table :testings do |t|
t.references :testing_parent, foreign_key: { primary_key: :other_id }
end
fk = @connection.foreign_keys("testings").find { |k| k.to_table == "testing_parents" }
assert_equal "other_id", fk.primary_key
end
test "foreign keys cannot be added to polymorphic relations when creating the table" do
@connection.create_table :testings do |t|
assert_raises(ArgumentError) do
t.references :testing_parent, polymorphic: true, foreign_key: true
end
end
end
test "foreign keys can be created while changing the table" do
@connection.create_table :testings
@connection.change_table :testings do |t|
t.references :testing_parent, foreign_key: true
end
fk = @connection.foreign_keys("testings").first
assert_equal "testings", fk.from_table
assert_equal "testing_parents", fk.to_table
end
test "foreign keys are not added by default when changing the table" do
@connection.create_table :testings
@connection.change_table :testings do |t|
t.references :testing_parent
end
assert_equal [], @connection.foreign_keys("testings")
end
test "foreign keys accept options when changing the table" do
@connection.change_table :testing_parents do |t|
t.integer :other_id
t.index :other_id, unique: true
end
@connection.create_table :testings
@connection.change_table :testings do |t|
t.references :testing_parent, foreign_key: { primary_key: :other_id }
end
fk = @connection.foreign_keys("testings").find { |k| k.to_table == "testing_parents" }
assert_equal "other_id", fk.primary_key
end
test "foreign keys cannot be added to polymorphic relations when changing the table" do
@connection.create_table :testings
@connection.change_table :testings do |t|
assert_raises(ArgumentError) do
t.references :testing_parent, polymorphic: true, foreign_key: true
end
end
end
test "foreign key column can be removed" do
@connection.create_table :testings do |t|
t.references :testing_parent, index: true, foreign_key: true
end
assert_difference "@connection.foreign_keys('testings').size", -1 do
@connection.remove_reference :testings, :testing_parent, foreign_key: true
end
end
test "foreign key methods respect pluralize_table_names" do
begin
original_pluralize_table_names = ActiveRecord::Base.pluralize_table_names
ActiveRecord::Base.pluralize_table_names = false
@connection.create_table :testing
@connection.change_table :testing_parents do |t|
t.references :testing, foreign_key: true
end
fk = @connection.foreign_keys("testing_parents").first
assert_equal "testing_parents", fk.from_table
assert_equal "testing", fk.to_table
assert_difference "@connection.foreign_keys('testing_parents').size", -1 do
@connection.remove_reference :testing_parents, :testing, foreign_key: true
end
ensure
ActiveRecord::Base.pluralize_table_names = original_pluralize_table_names
@connection.drop_table "testing", if_exists: true
end
end
end
end
end
end
|
