aboutsummaryrefslogtreecommitdiffstats
path: root/Library/Homebrew/test/test_inreplace.rb
blob: 0e62f9d3f9691ddcc93b8f48138db9bfe283f29e (plain)
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
require "testing_env"
require "extend/string"
require "utils/inreplace"

class InreplaceTest < Homebrew::TestCase
  def test_change_make_var
    # Replace flag
    s1 = "OTHER=def\nFLAG = abc\nFLAG2=abc"
    s1.extend(StringInreplaceExtension)
    s1.change_make_var! "FLAG", "def"
    assert_equal "OTHER=def\nFLAG=def\nFLAG2=abc", s1
  end

  def test_change_make_var_empty
    # Replace empty flag
    s1 = "OTHER=def\nFLAG = \nFLAG2=abc"
    s1.extend(StringInreplaceExtension)
    s1.change_make_var! "FLAG", "def"
    assert_equal "OTHER=def\nFLAG=def\nFLAG2=abc", s1
  end

  def test_change_make_var_empty_2
    # Replace empty flag
    s1 = "FLAG = \nmv file_a file_b"
    s1.extend(StringInreplaceExtension)
    s1.change_make_var! "FLAG", "def"
    assert_equal "FLAG=def\nmv file_a file_b", s1
  end

  def test_change_make_var_append
    # Append to flag
    s1 = "OTHER=def\nFLAG = abc\nFLAG2=abc"
    s1.extend(StringInreplaceExtension)
    s1.change_make_var! "FLAG", "\\1 def"
    assert_equal "OTHER=def\nFLAG=abc def\nFLAG2=abc", s1
  end

  def test_change_make_var_shell_style
    # Shell variables have no spaces around =
    s1 = "OTHER=def\nFLAG=abc\nFLAG2=abc"
    s1.extend(StringInreplaceExtension)
    s1.change_make_var! "FLAG", "def"
    assert_equal "OTHER=def\nFLAG=def\nFLAG2=abc", s1
  end

  def test_remove_make_var
    # Replace flag
    s1 = "OTHER=def\nFLAG = abc\nFLAG2 = def"
    s1.extend(StringInreplaceExtension)
    s1.remove_make_var! "FLAG"
    assert_equal "OTHER=def\nFLAG2 = def", s1
  end

  def test_remove_make_vars
    # Replace flag
    s1 = "OTHER=def\nFLAG = abc\nFLAG2 = def\nOTHER2=def"
    s1.extend(StringInreplaceExtension)
    s1.remove_make_var! ["FLAG", "FLAG2"]
    assert_equal "OTHER=def\nOTHER2=def", s1
  end

  def test_get_make_var
    s = "CFLAGS = -Wall -O2\nLDFLAGS = -lcrypto -lssl"
    s.extend(StringInreplaceExtension)
    assert_equal "-Wall -O2", s.get_make_var("CFLAGS")
  end

  def test_change_make_var_with_tabs
    s = "CFLAGS\t=\t-Wall -O2\nLDFLAGS\t=\t-lcrypto -lssl"
    s.extend(StringInreplaceExtension)

    assert_equal "-Wall -O2", s.get_make_var("CFLAGS")

    s.change_make_var! "CFLAGS", "-O3"
    assert_equal "CFLAGS=-O3\nLDFLAGS\t=\t-lcrypto -lssl", s

    s.remove_make_var! "LDFLAGS"
    assert_equal "CFLAGS=-O3\n", s
  end

  def test_sub_gsub
    s = "foo"
    s.extend(StringInreplaceExtension)

    s.sub!("f", "b")
    assert_equal "boo", s

    # Under current context, we are testing `String#gsub!`, so let's disable rubocop temporarily.
    s.gsub!("o", "e") # rubocop:disable Performance/StringReplacement
    assert_equal "bee", s
  end

  def test_inreplace_errors
    require "tempfile"
    extend(Utils::Inreplace)

    file = Tempfile.new("test")

    file.write "a\nb\nc\n"

    assert_raises(Utils::InreplaceError) do
      inreplace file.path, "d", "f"
    end

    assert_raises(Utils::InreplaceError) do
      # Under current context, we are testing `String#gsub!`, so let's disable rubocop temporarily.
      inreplace(file.path) { |s| s.gsub!("d", "f") } # rubocop:disable Performance/StringReplacement
    end

    assert_raises(Utils::InreplaceError) do
      inreplace(file.path) do |s|
        s.change_make_var! "VAR", "value"
        s.remove_make_var! "VAR2"
      end
    end
  ensure
    file.unlink
  end
end