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