require_relative '../../spec_helper'

describe :io_set_encoding_write, shared: true do
  it "sets the encodings to nil" do
    @io = new_io @name, "#{@object}:ibm437:ibm866"
    @io.set_encoding nil, nil

    @io.external_encoding.should be_nil
    @io.internal_encoding.should be_nil
  end

  it "prevents the encodings from changing when Encoding defaults are changed" do
    @io = new_io @name, "#{@object}:utf-8:us-ascii"
    @io.set_encoding nil, nil

    Encoding.default_external = Encoding::IBM437
    Encoding.default_internal = Encoding::IBM866

    @io.external_encoding.should be_nil
    @io.internal_encoding.should be_nil
  end

  it "sets the encodings to the current Encoding defaults" do
    @io = new_io @name, @object

    Encoding.default_external = Encoding::IBM437
    Encoding.default_internal = Encoding::IBM866

    @io.set_encoding nil, nil

    @io.external_encoding.should == Encoding::IBM437
    @io.internal_encoding.should == Encoding::IBM866
  end
end

describe "IO#set_encoding when passed nil, nil" do
  before :each do
    @external = Encoding.default_external
    @internal = Encoding.default_internal

    Encoding.default_external = Encoding::UTF_8
    Encoding.default_internal = nil

    @name = tmp('io_set_encoding.txt')
    touch(@name)
  end

  after :each do
    Encoding.default_external = @external
    Encoding.default_internal = @internal

    @io.close if @io and not @io.closed?
    rm_r @name
  end

  describe "with 'r' mode" do
    it "sets the encodings to the current Encoding defaults" do
      @io = new_io @name, "r"

      Encoding.default_external = Encoding::IBM437
      Encoding.default_internal = Encoding::IBM866

      @io.set_encoding nil, nil
      @io.external_encoding.should equal(Encoding::IBM437)
      @io.internal_encoding.should equal(Encoding::IBM866)
    end

    it "prevents the #internal_encoding from changing when Encoding.default_internal is changed" do
      @io = new_io @name, "r"
      @io.set_encoding nil, nil

      Encoding.default_internal = Encoding::IBM437

      @io.internal_encoding.should be_nil
    end

    it "allows the #external_encoding to change when Encoding.default_external is changed" do
      @io = new_io @name, "r"
      @io.set_encoding nil, nil

      Encoding.default_external = Encoding::IBM437

      @io.external_encoding.should equal(Encoding::IBM437)
    end
  end

  describe "with 'rb' mode" do
    it "returns Encoding.default_external" do
      @io = new_io @name, "rb"
      @io.external_encoding.should equal(Encoding::BINARY)

      @io.set_encoding nil, nil
      @io.external_encoding.should equal(Encoding.default_external)
    end
  end

  describe "with 'r+' mode" do
    it_behaves_like :io_set_encoding_write, nil, "r+"
  end

  describe "with 'w' mode" do
    it_behaves_like :io_set_encoding_write, nil, "w"
  end

  describe "with 'w+' mode" do
    it_behaves_like :io_set_encoding_write, nil, "w+"
  end

  describe "with 'a' mode" do
    it_behaves_like :io_set_encoding_write, nil, "a"
  end

  describe "with 'a+' mode" do
    it_behaves_like :io_set_encoding_write, nil, "a+"
  end
end

describe "IO#set_encoding" do
  before :each do
    @name = tmp('io_set_encoding.txt')
    touch(@name)
    @io = new_io @name
  end

  after :each do
    @io.close unless @io.closed?
    rm_r @name
  end

  it "returns self" do
    @io.set_encoding(Encoding::UTF_8).should equal(@io)
  end

  it "sets the external encoding when passed an Encoding argument" do
    @io.set_encoding(Encoding::UTF_8)
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should be_nil
  end

  it "sets the external and internal encoding when passed two Encoding arguments" do
    @io.set_encoding(Encoding::UTF_8, Encoding::UTF_16BE)
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should == Encoding::UTF_16BE
  end

  it "sets the external encoding when passed the name of an Encoding" do
    @io.set_encoding("utf-8")
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should be_nil
  end

  it "ignores the internal encoding if the same as external when passed Encoding objects" do
    @io.set_encoding(Encoding::UTF_8, Encoding::UTF_8)
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should be_nil
  end

  it "ignores the internal encoding if the same as external when passed encoding names separated by ':'" do
    @io.set_encoding("utf-8:utf-8")
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should be_nil
  end

  it "sets the external and internal encoding when passed the names of Encodings separated by ':'" do
    @io.set_encoding("utf-8:utf-16be")
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should == Encoding::UTF_16BE
  end

  it "sets the external and internal encoding when passed two String arguments" do
    @io.set_encoding("utf-8", "utf-16be")
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should == Encoding::UTF_16BE
  end

  it "calls #to_str to convert an abject to a String" do
    obj = mock("io_set_encoding")
    obj.should_receive(:to_str).and_return("utf-8:utf-16be")
    @io.set_encoding(obj)
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should == Encoding::UTF_16BE
  end

  it "calls #to_str to convert the second argument to a String" do
    obj = mock("io_set_encoding")
    obj.should_receive(:to_str).at_least(1).times.and_return("utf-16be")
    @io.set_encoding(Encoding::UTF_8, obj)
    @io.external_encoding.should == Encoding::UTF_8
    @io.internal_encoding.should == Encoding::UTF_16BE
  end
end