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
|
require File.join(File.dirname(__FILE__), 'spec_helper')
describe Sequel::Schema::Generator do
before :all do
@generator = Sequel::Schema::Generator.new(SchemaDummyDatabase.new) do
string :title
column :body, :text
foreign_key :parent_id
primary_key :id
check 'price > 100'
constraint(:xxx) {:yyy == :zzz}
index :title
index [:title, :body]
end
@columns, @indexes = @generator.create_info
end
{:name => :id, :primary_key => true}.each do |column, expected|
it "uses default primary key #{column}" do
@columns.first[column].should == expected
end
end
it "counts primary key, column and constraint definitions as columns" do
@columns.size.should == 6
end
it "places primary key first" do
@columns[0][:primary_key].should be_true
@columns[1][:primary_key].should_not be_true
@columns[2][:primary_key].should_not be_true
end
it "retrieves primary key name" do
@generator.primary_key_name.should == :id
end
it "keeps columns in order" do
@columns[1][:name].should == :title
@columns[1][:type].should == :string
@columns[2][:name].should == :body
@columns[2][:type].should == :text
end
it "creates foreign key column" do
@columns[3][:name].should == :parent_id
@columns[3][:type].should == :integer
end
it "finds columns" do
[:title, :body, :parent_id, :id].each do |col|
@generator.has_column?(col).should be_true
end
@generator.has_column?(:foo).should_not be_true
end
it "creates constraints" do
@columns[4][:name].should == nil
@columns[4][:type].should == :check
@columns[4][:check].should == ['price > 100']
@columns[5][:name].should == :xxx
@columns[5][:type].should == :check
@columns[5][:check].should be_a_kind_of(Proc)
end
it "creates indexes" do
@indexes[0][:columns].should include(:title)
@indexes[1][:columns].should include(:title)
@indexes[1][:columns].should include(:body)
end
end
describe Sequel::Schema::AlterTableGenerator do
before :all do
@generator = Sequel::Schema::AlterTableGenerator.new(SchemaDummyDatabase.new) do
add_column :aaa, :text
drop_column :bbb
rename_column :ccc, :ho
set_column_type :ddd, :float
set_column_default :eee, 1
add_index [:fff, :ggg]
drop_index :hhh
add_full_text_index :blah
add_spatial_index :geom
add_index :blah, :type => :hash
add_index :blah, :where => {:something => true}
add_constraint :con1, ':fred > 100'
drop_constraint :con2
end
end
specify "should generate operation records" do
@generator.operations.should == [
{:op => :add_column, :name => :aaa, :type => :text},
{:op => :drop_column, :name => :bbb},
{:op => :rename_column, :name => :ccc, :new_name => :ho},
{:op => :set_column_type, :name => :ddd, :type => :float},
{:op => :set_column_default, :name => :eee, :default => 1},
{:op => :add_index, :columns => [:fff, :ggg]},
{:op => :drop_index, :columns => [:hhh]},
{:op => :add_index, :columns => [:blah], :type => :full_text},
{:op => :add_index, :columns => [:geom], :type => :spatial},
{:op => :add_index, :columns => [:blah], :type => :hash},
{:op => :add_index, :columns => [:blah], :where => {:something => true}},
{:op => :add_constraint, :type => :check, :name => :con1, :check => [':fred > 100']},
{:op => :drop_constraint, :name => :con2}
]
end
end
|
