activesupport/test/core_ext/module/setup_test.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

require 'abstract_unit'
require 'active_support/core/time'
require 'active_support/core_ext/module/setup'

class SetupTest < Test::Unit::TestCase
  module Baz
    module ClassMethods
      def baz
        "baz"
      end

      def setup=(value)
        @@setup = value
      end

      def setup
        @@setup
      end
    end

    setup do
      self.setup = true
    end

    def baz
      "baz"
    end
  end

  module Bar
    depends_on Baz

    def bar
      "bar"
    end

    def baz
      "bar+" + super
    end
  end

  def setup
    @klass = Class.new
  end

  def test_module_is_included_normally
    @klass.use(Baz)
    assert_equal "baz", @klass.new.baz
    assert_equal SetupTest::Baz, @klass.included_modules[0]

    @klass.use(Baz)
    assert_equal "baz", @klass.new.baz
    assert_equal SetupTest::Baz, @klass.included_modules[0]
  end

  def test_class_methods_are_extended
    @klass.use(Baz)
    assert_equal "baz", @klass.baz
    assert_equal SetupTest::Baz::ClassMethods, (class << @klass; self.included_modules; end)[0]
  end

  def test_setup_block_is_ran
    @klass.use(Baz)
    assert_equal true, @klass.setup
  end

  def test_modules_dependencies_are_met
    @klass.use(Bar)
    assert_equal "bar", @klass.new.bar
    assert_equal "bar+baz", @klass.new.baz
    assert_equal "baz", @klass.baz
    assert_equal [SetupTest::Bar, SetupTest::Baz], @klass.included_modules[0..1]
  end
end