forked from SchemaPlus/schema_plus
-
Notifications
You must be signed in to change notification settings - Fork 0
/
schema_spec.rb
80 lines (63 loc) · 2.06 KB
/
schema_spec.rb
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
require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
describe ActiveRecord::Schema do
let(:schema) { ActiveRecord::Schema }
let(:connection) { ActiveRecord::Base.connection }
context "defining with auto_index and auto_create" do
around(:each) do |example|
with_auto_index do
with_auto_create do
example.run
end
end
end
it "should pass" do
expect { define_schema }.should_not raise_error
end
it "should create only explicity added indexes" do
define_schema
expected = SchemaPlusHelpers.mysql? ? 2 : 1
connection.tables.collect { |table| connection.indexes(table) }.flatten.should have(expected).items
end
it "should create only explicity added foriegn keys" do
define_schema
connection.tables.collect { |table| connection.foreign_keys(table) }.flatten.should have(2).items
end
end
protected
def define_schema
ActiveRecord::Migration.suppress_messages do
schema.define do
connection.tables.each do |table| drop_table table end
create_table :users, :force => true do
end
create_table :colors, :force => true do
end
create_table :shoes, :force => true do
end
create_table :posts, :force => true do |t|
t.integer :user_id, :references => :users, :index => true
t.integer :shoe_id, :references => :shoes # should not have an index (except mysql)
t.integer :color_id # should not have a foreign key nor index
end
end
end
end
def with_auto_index(value = true)
old_value = SchemaPlus.config.foreign_keys.auto_index
SchemaPlus.config.foreign_keys.auto_index = value
begin
yield
ensure
SchemaPlus.config.foreign_keys.auto_index = old_value
end
end
def with_auto_create(value = true)
old_value = SchemaPlus.config.foreign_keys.auto_create
SchemaPlus.config.foreign_keys.auto_create = value
begin
yield
ensure
SchemaPlus.config.foreign_keys.auto_create = old_value
end
end
end