aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/migration_mysql.rb
blob: 7777bb2c035124b0e8a3a5655712287ddfdc236c (plain) (blame)
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
require 'abstract_unit'
require 'fixtures/person'
require File.dirname(__FILE__) + '/fixtures/migrations/1_people_have_last_names'
require File.dirname(__FILE__) + '/fixtures/migrations/2_we_need_reminders'

class Reminder < ActiveRecord::Base; end

class MigrationTest < Test::Unit::TestCase
  def setup
  end

  def teardown
    ActiveRecord::Base.connection.initialize_schema_information
    ActiveRecord::Base.connection.update "UPDATE schema_info SET version = 0"

    Reminder.connection.drop_table("reminders") rescue nil
    Reminder.reset_column_information

    Person.connection.remove_column("people", "last_name") rescue nil
    Person.reset_column_information
  end

  def test_add_remove_single_field
    assert !Person.column_methods_hash.include?(:last_name)

    PeopleHaveLastNames.up

    Person.reset_column_information
    assert Person.column_methods_hash.include?(:last_name)
    
    PeopleHaveLastNames.down

    Person.reset_column_information
    assert !Person.column_methods_hash.include?(:last_name)
  end

  def test_add_table
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.column_methods_hash }
    
    WeNeedReminders.up
    
    assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
    assert "hello world", Reminder.find_first
    
    WeNeedReminders.down
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.find_first }
  end

  def test_migrator
    assert !Person.column_methods_hash.include?(:last_name)
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.column_methods_hash }

    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/')

    assert_equal 2, ActiveRecord::Migrator.current_version
    Person.reset_column_information
    assert Person.column_methods_hash.include?(:last_name)
    assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
    assert "hello world", Reminder.find_first


    ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/')

    assert_equal 0, ActiveRecord::Migrator.current_version
    Person.reset_column_information
    assert !Person.column_methods_hash.include?(:last_name)
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.find_first }
  end

  def test_migrator_one_up
    assert !Person.column_methods_hash.include?(:last_name)
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.column_methods_hash }

    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 1)

    Person.reset_column_information
    assert Person.column_methods_hash.include?(:last_name)
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.column_methods_hash }


    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 2)

    assert Reminder.create("content" => "hello world", "remind_at" => Time.now)
    assert "hello world", Reminder.find_first
  end
  
  def test_migrator_one_down
    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/')
    
    ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/', 1)

    Person.reset_column_information
    assert Person.column_methods_hash.include?(:last_name)
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.column_methods_hash }
  end
  
  def test_migrator_one_up_one_down
    ActiveRecord::Migrator.up(File.dirname(__FILE__) + '/fixtures/migrations/', 1)
    ActiveRecord::Migrator.down(File.dirname(__FILE__) + '/fixtures/migrations/', 0)

    assert !Person.column_methods_hash.include?(:last_name)
    assert_raises(ActiveRecord::StatementInvalid) { Reminder.column_methods_hash }
  end
end