# -*- encoding: utf-8 -*-
require_relative '../../spec_helper'
require_relative 'fixtures/classes'

# TODO: Add missing String#[]= specs:
#   String#[re, idx] = obj

describe "String#[]= with Integer index" do
  it "replaces the char at idx with other_str" do
    a = "hello"
    a[0] = "bam"
    a.should == "bamello"
    a[-2] = ""
    a.should == "bamelo"
  end

  ruby_version_is ''...'2.7' do
    it "taints self if other_str is tainted" do
      a = "hello"
      a[0] = "".taint
      a.should.tainted?

      a = "hello"
      a[0] = "x".taint
      a.should.tainted?
    end
  end

  it "raises an IndexError without changing self if idx is outside of self" do
    str = "hello"

    -> { str[20] = "bam" }.should raise_error(IndexError)
    str.should == "hello"

    -> { str[-20] = "bam" }.should raise_error(IndexError)
    str.should == "hello"

    -> { ""[-1] = "bam" }.should raise_error(IndexError)
  end

  # Behaviour is verified by matz in
  # http://redmine.ruby-lang.org/issues/show/1750
  it "allows assignment to the zero'th element of an empty String" do
    str = ""
    str[0] = "bam"
    str.should == "bam"
  end

  it "raises IndexError if the string index doesn't match a position in the string" do
    str = "hello"
    -> { str['y'] = "bam" }.should raise_error(IndexError)
    str.should == "hello"
  end

  it "raises a FrozenError when self is frozen" do
    a = "hello"
    a.freeze

    -> { a[0] = "bam" }.should raise_error(FrozenError)
  end

  it "calls to_int on index" do
    str = "hello"
    str[0.5] = "hi "
    str.should == "hi ello"

    obj = mock('-1')
    obj.should_receive(:to_int).and_return(-1)
    str[obj] = "!"
    str.should == "hi ell!"
  end

  it "calls #to_str to convert other to a String" do
    other_str = mock('-test-')
    other_str.should_receive(:to_str).and_return("-test-")

    a = "abc"
    a[1] = other_str
    a.should == "a-test-c"
  end

  it "raises a TypeError if other_str can't be converted to a String" do
    -> { "test"[1] = []        }.should raise_error(TypeError)
    -> { "test"[1] = mock('x') }.should raise_error(TypeError)
    -> { "test"[1] = nil       }.should raise_error(TypeError)
  end

  it "raises a TypeError if passed an Integer replacement" do
    -> { "abc"[1] = 65 }.should raise_error(TypeError)
  end

  it "raises an IndexError if the index is greater than character size" do
    -> { "あれ"[4] = "a" }.should raise_error(IndexError)
  end

  it "calls #to_int to convert the index" do
    index = mock("string element set")
    index.should_receive(:to_int).and_return(1)

    str = "あれ"
    str[index] = "a"
    str.should == "あa"
  end

  it "raises a TypeError if #to_int does not return an Integer" do
    index = mock("string element set")
    index.should_receive(:to_int).and_return('1')

    -> { "abc"[index] = "d" }.should raise_error(TypeError)
  end

  it "raises an IndexError if #to_int returns a value out of range" do
    index = mock("string element set")
    index.should_receive(:to_int).and_return(4)

    -> { "ab"[index] = "c" }.should raise_error(IndexError)
  end

  it "replaces a character with a multibyte character" do
    str = "ありがとu"
    str[4] = "う"
    str.should == "ありがとう"
  end

  it "replaces a multibyte character with a character" do
    str = "ありがとう"
    str[4] = "u"
    str.should == "ありがとu"
  end

  it "replaces a multibyte character with a multibyte character" do
    str = "ありがとお"
    str[4] = "う"
    str.should == "ありがとう"
  end

  it "encodes the String in an encoding compatible with the replacement" do
    str = " ".force_encoding Encoding::US_ASCII
    rep = [160].pack('C').force_encoding Encoding::BINARY
    str[0] = rep
    str.encoding.should equal(Encoding::BINARY)
  end

  it "raises an Encoding::CompatibilityError if the replacement encoding is incompatible" do
    str = "あれ"
    rep = "が".encode Encoding::EUC_JP
    -> { str[0] = rep }.should raise_error(Encoding::CompatibilityError)
  end
end

describe "String#[]= with String index" do
  it "replaces fewer characters with more characters" do
    str = "abcde"
    str["cd"] = "ghi"
    str.should == "abghie"
  end

  it "replaces more characters with fewer characters" do
    str = "abcde"
    str["bcd"] = "f"
    str.should == "afe"
  end

  it "replaces characters with no characters" do
    str = "abcde"
    str["cd"] = ""
    str.should == "abe"
  end

  it "raises an IndexError if the search String is not found" do
    str = "abcde"
    -> { str["g"] = "h" }.should raise_error(IndexError)
  end

  it "replaces characters with a multibyte character" do
    str = "ありgaとう"
    str["ga"] = "が"
    str.should == "ありがとう"
  end

  it "replaces multibyte characters with characters" do
    str = "ありがとう"
    str["が"] = "ga"
    str.should == "ありgaとう"
  end

  it "replaces multibyte characters with multibyte characters" do
    str = "ありがとう"
    str["が"] = "か"
    str.should == "ありかとう"
  end

  it "encodes the String in an encoding compatible with the replacement" do
    str = " ".force_encoding Encoding::US_ASCII
    rep = [160].pack('C').force_encoding Encoding::BINARY
    str[" "] = rep
    str.encoding.should equal(Encoding::BINARY)
  end

  it "raises an Encoding::CompatibilityError if the replacement encoding is incompatible" do
    str = "あれ"
    rep = "が".encode Encoding::EUC_JP
    -> { str["れ"] = rep }.should raise_error(Encoding::CompatibilityError)
  end
end

describe "String#[]= with a Regexp index" do
  it "replaces the matched text with the rhs" do
    str = "hello"
    str[/lo/] = "x"
    str.should == "helx"
  end

  it "raises IndexError if the regexp index doesn't match a position in the string" do
    str = "hello"
    -> { str[/y/] = "bam" }.should raise_error(IndexError)
    str.should == "hello"
  end

  it "calls #to_str to convert the replacement" do
    rep = mock("string element set regexp")
    rep.should_receive(:to_str).and_return("b")

    str = "abc"
    str[/ab/] = rep
    str.should == "bc"
  end

  it "checks the match before calling #to_str to convert the replacement" do
    rep = mock("string element set regexp")
    rep.should_not_receive(:to_str)

    -> { "abc"[/def/] = rep }.should raise_error(IndexError)
  end

  describe "with 3 arguments" do
    it "calls #to_int to convert the second object" do
      ref = mock("string element set regexp ref")
      ref.should_receive(:to_int).and_return(1)

      str = "abc"
      str[/a(b)/, ref] = "x"
      str.should == "axc"
    end

    it "raises a TypeError if #to_int does not return an Integer" do
      ref = mock("string element set regexp ref")
      ref.should_receive(:to_int).and_return(nil)

      -> { "abc"[/a(b)/, ref] = "x" }.should raise_error(TypeError)
    end

    it "uses the 2nd of 3 arguments as which capture should be replaced" do
      str = "aaa bbb ccc"
      str[/a (bbb) c/, 1] = "ddd"
      str.should == "aaa ddd ccc"
    end

    it "allows the specified capture to be negative and count from the end" do
      str = "abcd"
      str[/(a)(b)(c)(d)/, -2] = "e"
      str.should == "abed"
    end

    it "checks the match index before calling #to_str to convert the replacement" do
      rep = mock("string element set regexp")
      rep.should_not_receive(:to_str)

      -> { "abc"[/a(b)/, 2] = rep }.should raise_error(IndexError)
    end

    it "raises IndexError if the specified capture isn't available" do
      str = "aaa bbb ccc"
      -> { str[/a (bbb) c/,  2] = "ddd" }.should raise_error(IndexError)
      -> { str[/a (bbb) c/, -2] = "ddd" }.should raise_error(IndexError)
    end

    describe "when the optional capture does not match" do
      it "raises an IndexError before setting the replacement" do
        str1 = "a b c"
        str2 = str1.dup
        -> { str2[/a (b) (Z)?/,  2] = "d" }.should raise_error(IndexError)
        str2.should == str1
      end
    end
  end

  it "replaces characters with a multibyte character" do
    str = "ありgaとう"
    str[/ga/] = "が"
    str.should == "ありがとう"
  end

  it "replaces multibyte characters with characters" do
    str = "ありがとう"
    str[/が/] = "ga"
    str.should == "ありgaとう"
  end

  it "replaces multibyte characters with multibyte characters" do
    str = "ありがとう"
    str[/が/] = "か"
    str.should == "ありかとう"
  end

  it "encodes the String in an encoding compatible with the replacement" do
    str = " ".force_encoding Encoding::US_ASCII
    rep = [160].pack('C').force_encoding Encoding::BINARY
    str[/ /] = rep
    str.encoding.should equal(Encoding::BINARY)
  end

  it "raises an Encoding::CompatibilityError if the replacement encoding is incompatible" do
    str = "あれ"
    rep = "が".encode Encoding::EUC_JP
    -> { str[/れ/] = rep }.should raise_error(Encoding::CompatibilityError)
  end
end

describe "String#[]= with a Range index" do
  describe "with an empty replacement" do
    it "does not replace a character with a zero-index, zero exclude-end range" do
      str = "abc"
      str[0...0] = ""
      str.should == "abc"
    end

    it "does not replace a character with a zero exclude-end range" do
      str = "abc"
      str[1...1] = ""
      str.should == "abc"
    end

    it "replaces a character with zero-index, zero non-exclude-end range" do
      str = "abc"
      str[0..0] = ""
      str.should == "bc"
    end

    it "replaces a character with a zero non-exclude-end range" do
      str = "abc"
      str[1..1] = ""
      str.should == "ac"
    end
  end

  it "replaces the contents with a shorter String" do
    str = "abcde"
    str[0..-1] = "hg"
    str.should == "hg"
  end

  it "replaces the contents with a longer String" do
    str = "abc"
    str[0...4] = "uvwxyz"
    str.should == "uvwxyz"
  end

  it "replaces a partial string" do
    str = "abcde"
    str[1..3] = "B"
    str.should == "aBe"
  end

  it "raises a RangeError if negative Range begin is out of range" do
    -> { "abc"[-4..-2] = "x" }.should raise_error(RangeError)
  end

  it "raises a RangeError if positive Range begin is greater than String size" do
    -> { "abc"[4..2] = "x" }.should raise_error(RangeError)
  end

  it "uses the Range end as an index rather than a count" do
    str = "abcdefg"
    str[-5..3] = "xyz"
    str.should == "abxyzefg"
  end

  it "treats a negative out-of-range Range end with a positive Range begin as a zero count" do
    str = "abc"
    str[1..-4] = "x"
    str.should == "axbc"
  end

  it "treats a negative out-of-range Range end with a negative Range begin as a zero count" do
    str = "abcd"
    str[-1..-4] = "x"
    str.should == "abcxd"
  end

  it "replaces characters with a multibyte character" do
    str = "ありgaとう"
    str[2..3] = "が"
    str.should == "ありがとう"
  end

  it "replaces multibyte characters with characters" do
    str = "ありがとう"
    str[2...3] = "ga"
    str.should == "ありgaとう"
  end

  it "replaces multibyte characters by negative indexes" do
    str = "ありがとう"
    str[-3...-2] = "ga"
    str.should == "ありgaとう"
  end

  it "replaces multibyte characters with multibyte characters" do
    str = "ありがとう"
    str[2..2] = "か"
    str.should == "ありかとう"
  end

  it "deletes a multibyte character" do
    str = "ありとう"
    str[2..3] = ""
    str.should == "あり"
  end

  it "inserts a multibyte character" do
    str = "ありとう"
    str[2...2] = "が"
    str.should == "ありがとう"
  end

  it "encodes the String in an encoding compatible with the replacement" do
    str = " ".force_encoding Encoding::US_ASCII
    rep = [160].pack('C').force_encoding Encoding::BINARY
    str[0..1] = rep
    str.encoding.should equal(Encoding::BINARY)
  end

  it "raises an Encoding::CompatibilityError if the replacement encoding is incompatible" do
    str = "あれ"
    rep = "が".encode Encoding::EUC_JP
    -> { str[0..1] = rep }.should raise_error(Encoding::CompatibilityError)
  end
end

describe "String#[]= with Integer index, count" do
  it "starts at idx and overwrites count characters before inserting the rest of other_str" do
    a = "hello"
    a[0, 2] = "xx"
    a.should == "xxllo"
    a = "hello"
    a[0, 2] = "jello"
    a.should == "jellollo"
  end

  it "counts negative idx values from end of the string" do
    a = "hello"
    a[-1, 0] = "bob"
    a.should == "hellbobo"
    a = "hello"
    a[-5, 0] = "bob"
    a.should == "bobhello"
  end

  it "overwrites and deletes characters if count is more than the length of other_str" do
    a = "hello"
    a[0, 4] = "x"
    a.should == "xo"
    a = "hello"
    a[0, 5] = "x"
    a.should == "x"
  end

  it "deletes characters if other_str is an empty string" do
    a = "hello"
    a[0, 2] = ""
    a.should == "llo"
  end

  it "deletes characters up to the maximum length of the existing string" do
    a = "hello"
    a[0, 6] = "x"
    a.should == "x"
    a = "hello"
    a[0, 100] = ""
    a.should == ""
  end

  it "appends other_str to the end of the string if idx == the length of the string" do
    a = "hello"
    a[5, 0] = "bob"
    a.should == "hellobob"
  end

  ruby_version_is ''...'2.7' do
    it "taints self if other_str is tainted" do
      a = "hello"
      a[0, 0] = "".taint
      a.should.tainted?

      a = "hello"
      a[1, 4] = "x".taint
      a.should.tainted?
    end
  end

  it "calls #to_int to convert the index and count objects" do
    index = mock("string element set index")
    index.should_receive(:to_int).and_return(-4)

    count = mock("string element set count")
    count.should_receive(:to_int).and_return(2)

    str = "abcde"
    str[index, count] = "xyz"
    str.should == "axyzde"
  end

  it "raises a TypeError if #to_int for index does not return an Integer" do
    index = mock("string element set index")
    index.should_receive(:to_int).and_return("1")

    -> { "abc"[index, 2] = "xyz" }.should raise_error(TypeError)
  end

  it "raises a TypeError if #to_int for count does not return an Integer" do
    count = mock("string element set count")
    count.should_receive(:to_int).and_return("1")

    -> { "abc"[1, count] = "xyz" }.should raise_error(TypeError)
  end

  it "calls #to_str to convert the replacement object" do
    r = mock("string element set replacement")
    r.should_receive(:to_str).and_return("xyz")

    str = "abcde"
    str[2, 2] = r
    str.should == "abxyze"
  end

  it "raises a TypeError of #to_str does not return a String" do
    r = mock("string element set replacement")
    r.should_receive(:to_str).and_return(nil)

    -> { "abc"[1, 1] = r }.should raise_error(TypeError)
  end

  it "raises an IndexError if |idx| is greater than the length of the string" do
    -> { "hello"[6, 0] = "bob"  }.should raise_error(IndexError)
    -> { "hello"[-6, 0] = "bob" }.should raise_error(IndexError)
  end

  it "raises an IndexError if count < 0" do
    -> { "hello"[0, -1] = "bob" }.should raise_error(IndexError)
    -> { "hello"[1, -1] = "bob" }.should raise_error(IndexError)
  end

  it "raises a TypeError if other_str is a type other than String" do
    -> { "hello"[0, 2] = nil  }.should raise_error(TypeError)
    -> { "hello"[0, 2] = []   }.should raise_error(TypeError)
    -> { "hello"[0, 2] = 33   }.should raise_error(TypeError)
  end

  it "replaces characters with a multibyte character" do
    str = "ありgaとう"
    str[2, 2] = "が"
    str.should == "ありがとう"
  end

  it "replaces multibyte characters with characters" do
    str = "ありがとう"
    str[2, 1] = "ga"
    str.should == "ありgaとう"
  end

  it "replaces multibyte characters with multibyte characters" do
    str = "ありがとう"
    str[2, 1] = "か"
    str.should == "ありかとう"
  end

  it "deletes a multibyte character" do
    str = "ありとう"
    str[2, 2] = ""
    str.should == "あり"
  end

  it "inserts a multibyte character" do
    str = "ありとう"
    str[2, 0] = "が"
    str.should == "ありがとう"
  end

  it "raises an IndexError if the character index is out of range of a multibyte String" do
    -> { "あれ"[3, 0] = "り" }.should raise_error(IndexError)
  end

  it "encodes the String in an encoding compatible with the replacement" do
    str = " ".force_encoding Encoding::US_ASCII
    rep = [160].pack('C').force_encoding Encoding::BINARY
    str[0, 1] = rep
    str.encoding.should equal(Encoding::BINARY)
  end

  it "raises an Encoding::CompatibilityError if the replacement encoding is incompatible" do
    str = "あれ"
    rep = "が".encode Encoding::EUC_JP
    -> { str[0, 1] = rep }.should raise_error(Encoding::CompatibilityError)
  end
end