aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/test/cases/migration/command_recorder_test.rb
blob: ae531ebb4c5ad8fefeafdde46813c8a65e90ba55 (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
105
106
107
108
require "cases/helper"

module ActiveRecord
  class Migration
    class CommandRecorderTest < ActiveRecord::TestCase
      def setup
        @recorder = CommandRecorder.new
      end

      def test_respond_to_delegates
        recorder = CommandRecorder.new(Class.new {
          def america; end
        }.new)
        assert recorder.respond_to?(:america)
      end

      def test_send_calls_super
        assert_raises(NoMethodError) do
          @recorder.send(:non_existing_method, :horses)
        end
      end

      def test_send_delegates_to_record
        recorder = CommandRecorder.new(Class.new {
          def create_table(name); end
        }.new)
        assert recorder.respond_to?(:create_table), 'respond_to? create_table'
        recorder.send(:create_table, :horses)
        assert_equal [[:create_table, [:horses]]], recorder.commands
      end

      def test_unknown_commands_raise_exception
        @recorder.record :execute, ['some sql']
        assert_raises(ActiveRecord::IrreversibleMigration) do
          @recorder.inverse
        end
      end

      def test_record
        @recorder.record :create_table, [:system_settings]
        assert_equal 1, @recorder.commands.length
      end

      def test_inverse
        @recorder.record :create_table, [:system_settings]
        assert_equal 1, @recorder.inverse.length

        @recorder.record :rename_table, [:old, :new]
        assert_equal 2, @recorder.inverse.length
      end

      def test_inverted_commands_are_reveresed
        @recorder.record :create_table, [:hello]
        @recorder.record :create_table, [:world]
        tables = @recorder.inverse.map(&:last)
        assert_equal [[:world], [:hello]], tables
      end

      def test_invert_create_table
        @recorder.record :create_table, [:system_settings]
        drop_table = @recorder.inverse.first
        assert_equal [:drop_table, [:system_settings]], drop_table
      end

      def test_invert_rename_table
        @recorder.record :rename_table, [:old, :new]
        rename = @recorder.inverse.first
        assert_equal [:rename_table, [:new, :old]], rename
      end

      def test_invert_add_column
        @recorder.record :add_column, [:table, :column, :type, {}]
        remove = @recorder.inverse.first
        assert_equal [:remove_column, [:table, :column]], remove
      end

      def test_invert_rename_column
        @recorder.record :rename_column, [:table, :old, :new]
        rename = @recorder.inverse.first
        assert_equal [:rename_column, [:table, :new, :old]], rename
      end

      def test_invert_add_index
        @recorder.record :add_index, [:table, [:one, :two], {:options => true}]
        remove = @recorder.inverse.first
        assert_equal [:remove_index, [:table, {:column => [:one, :two]}]], remove
      end

      def test_invert_rename_index
        @recorder.record :rename_index, [:old, :new]
        rename = @recorder.inverse.first
        assert_equal [:rename_index, [:new, :old]], rename
      end

      def test_invert_add_timestamps
        @recorder.record :add_timestamps, [:table]
        remove = @recorder.inverse.first
        assert_equal [:remove_timestamps, [:table]], remove
      end

      def test_invert_remove_timestamps
        @recorder.record :remove_timestamps, [:table]
        add = @recorder.inverse.first
        assert_equal [:add_timestamps, [:table]], add
      end
    end
  end
end