require 'test_helper'

class ZipFileSplitTest < MiniTest::Test
  TEST_ZIP = TestZipFile::TEST_ZIP2.clone
  TEST_ZIP.zip_name = 'large_zip_file.zip'
  EXTRACTED_FILENAME = 'test/data/generated/extEntrySplit'
  UNSPLITTED_FILENAME = 'test/data/generated/unsplitted.zip'
  ENTRY_TO_EXTRACT = TEST_ZIP.entry_names.first

  def setup
    FileUtils.cp(TestZipFile::TEST_ZIP2.zip_name, TEST_ZIP.zip_name)
  end

  def teardown
    File.delete(TEST_ZIP.zip_name)
    File.delete(UNSPLITTED_FILENAME) if File.exist?(UNSPLITTED_FILENAME)

    Dir["#{TEST_ZIP.zip_name}.*"].each do |zip_file_name|
      File.delete(zip_file_name) if File.exist?(zip_file_name)
    end
  end

  def test_split_method_respond
    assert_respond_to ::Zip::File, :split, 'Does not have split class method'
  end

  def test_split
    result = ::Zip::File.split(TEST_ZIP.zip_name, 65_536, false)

    return if result.nil?
    Dir["#{TEST_ZIP.zip_name}.*"].sort.each_with_index do |zip_file_name, index|
      File.open(zip_file_name, 'rb') do |zip_file|
        zip_file.read([::Zip::File::SPLIT_SIGNATURE].pack('V').size) if index == 0
        File.open(UNSPLITTED_FILENAME, 'ab') do |file|
          file << zip_file.read
        end
      end
    end

    ::Zip::File.open(UNSPLITTED_FILENAME) do |zf|
      zf.extract(ENTRY_TO_EXTRACT, EXTRACTED_FILENAME)

      assert(File.exist?(EXTRACTED_FILENAME))
      AssertEntry.assert_contents(EXTRACTED_FILENAME,
                                  zf.get_input_stream(ENTRY_TO_EXTRACT) { |is| is.read })

      File.unlink(EXTRACTED_FILENAME)

      entry = zf.get_entry(ENTRY_TO_EXTRACT)
      entry.extract(EXTRACTED_FILENAME)

      assert(File.exist?(EXTRACTED_FILENAME))
      AssertEntry.assert_contents(EXTRACTED_FILENAME,
                                  entry.get_input_stream { |is| is.read })
    end
  end
end