aboutsummaryrefslogtreecommitdiffstats
path: root/actionpack/test/template/capture_helper_test.rb
blob: f7c42c7f22dca8d35299e11f355c76abd8cc6e3d (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
109
110
111
112
113
114
115
116
117
118
119
120
121
require 'abstract_unit'

class CaptureHelperTest < ActionView::TestCase
  def setup
    super
    @av = ActionView::Base.new
    @_content_for = Hash.new {|h,k| h[k] = "" }
  end

  def test_capture_captures_the_temporary_output_buffer_in_its_block
    assert_nil @av.output_buffer
    string = @av.capture do
      @av.output_buffer << 'foo'
      @av.output_buffer << 'bar'
    end
    assert_nil @av.output_buffer
    assert_equal 'foobar', string
    assert_kind_of ActionView::NonConcattingString, string
  end

  def test_capture_captures_the_value_returned_by_the_block_if_the_temporary_buffer_is_blank
    string = @av.capture('foo', 'bar') do |a, b|
      a + b
    end
    assert_equal 'foobar', string
    assert_kind_of ActionView::NonConcattingString, string
  end

  def test_capture_returns_nil_if_the_returned_value_is_not_a_string
    assert_nil @av.capture { 1 }
  end

  def test_content_for
    assert ! content_for?(:title)
    content_for :title, 'title'
    assert content_for?(:title)
    assert ! content_for?(:something_else)
  end

  def test_with_output_buffer_swaps_the_output_buffer_given_no_argument
    assert_nil @av.output_buffer
    buffer = @av.with_output_buffer do
      @av.output_buffer << '.'
    end
    assert_equal '.', buffer
    assert_nil @av.output_buffer
  end

  def test_with_output_buffer_swaps_the_output_buffer_with_an_argument
    assert_nil @av.output_buffer
    buffer = ActionView::OutputBuffer.new('.')
    @av.with_output_buffer(buffer) do
      @av.output_buffer << '.'
    end
    assert_equal '..', buffer
    assert_nil @av.output_buffer
  end

  def test_with_output_buffer_restores_the_output_buffer
    buffer = ActionView::OutputBuffer.new
    @av.output_buffer = buffer
    @av.with_output_buffer do
      @av.output_buffer << '.'
    end
    assert buffer.equal?(@av.output_buffer)
  end

  unless RUBY_VERSION < '1.9'
    def test_with_output_buffer_sets_proper_encoding
      @av.output_buffer = ActionView::OutputBuffer.new

      # Ensure we set the output buffer to an encoding different than the default one.
      alt_encoding = alt_encoding(@av.output_buffer)
      @av.output_buffer.force_encoding(alt_encoding)

      @av.with_output_buffer do
        assert_equal alt_encoding, @av.output_buffer.encoding
      end
    end
  end

  def test_with_output_buffer_does_not_assume_there_is_an_output_buffer
    assert_nil @av.output_buffer
    assert_equal "", @av.with_output_buffer {}
  end

  def test_flush_output_buffer_concats_output_buffer_to_response
    view = view_with_controller
    assert_equal [], view.response.body_parts

    view.output_buffer << 'OMG'
    view.flush_output_buffer
    assert_equal ['OMG'], view.response.body_parts
    assert_equal '', view.output_buffer

    view.output_buffer << 'foobar'
    view.flush_output_buffer
    assert_equal ['OMG', 'foobar'], view.response.body_parts
    assert_equal '', view.output_buffer
  end

  unless RUBY_VERSION < '1.9'
    def test_flush_output_buffer_preserves_the_encoding_of_the_output_buffer
      view = view_with_controller
      alt_encoding = alt_encoding(view.output_buffer)
      view.output_buffer.force_encoding(alt_encoding)
      flush_output_buffer
      assert_equal alt_encoding, view.output_buffer.encoding
    end
  end

  def alt_encoding(output_buffer)
    output_buffer.encoding == Encoding::US_ASCII ? Encoding::UTF_8 : Encoding::US_ASCII
  end

  def view_with_controller
    TestController.new.view_context.tap do |view|
      view.output_buffer = ActionView::OutputBuffer.new
    end
  end
end