require 'test_helper'

class ZipFileSplitTest < MiniTest::Unit::TestCase
  TEST_ZIP = TestZipFile::TEST_ZIP2.clone
  TEST_ZIP.zip_name = "large_zip_file.zip"
  EXTRACTED_FILENAME = "test/data/generated/extEntry"
  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, 65536, false)

    unless 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
end