69 lines
1.8 KiB
Ruby
69 lines
1.8 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'test_helper'
|
|
|
|
class EncryptionTest < MiniTest::Test
|
|
ENCRYPT_ZIP_TEST_FILE = 'test/data/zipWithEncryption.zip'
|
|
INPUT_FILE1 = 'test/data/file1.txt'
|
|
INPUT_FILE2 = 'test/data/file2.txt'
|
|
|
|
def setup
|
|
Zip.default_compression = ::Zlib::DEFAULT_COMPRESSION
|
|
end
|
|
|
|
def teardown
|
|
Zip.reset!
|
|
end
|
|
|
|
def test_encrypt
|
|
content = File.read(INPUT_FILE1)
|
|
test_filename = 'top_secret_file.txt'
|
|
|
|
password = 'swordfish'
|
|
|
|
encrypted_zip = Zip::OutputStream.write_buffer(
|
|
::StringIO.new,
|
|
encrypter: Zip::TraditionalEncrypter.new(password)
|
|
) do |out|
|
|
out.put_next_entry(test_filename)
|
|
out.write content
|
|
end
|
|
|
|
Zip::InputStream.open(
|
|
encrypted_zip, decrypter: Zip::TraditionalDecrypter.new(password)
|
|
) do |zis|
|
|
entry = zis.get_next_entry
|
|
assert_equal test_filename, entry.name
|
|
assert_equal 1_327, entry.size
|
|
assert_equal content, zis.read
|
|
end
|
|
|
|
error = assert_raises(Zip::DecompressionError) do
|
|
Zip::InputStream.open(
|
|
encrypted_zip,
|
|
decrypter: Zip::TraditionalDecrypter.new("#{password}wrong")
|
|
) do |zis|
|
|
zis.get_next_entry
|
|
assert_equal content, zis.read
|
|
end
|
|
end
|
|
assert_match(/Zlib error \('.+'\) while inflating\./, error.message)
|
|
end
|
|
|
|
def test_decrypt
|
|
Zip::InputStream.open(
|
|
ENCRYPT_ZIP_TEST_FILE,
|
|
decrypter: Zip::TraditionalDecrypter.new('password')
|
|
) do |zis|
|
|
entry = zis.get_next_entry
|
|
assert_equal 'file1.txt', entry.name
|
|
assert_equal 1_327, entry.size
|
|
assert_equal ::File.read(INPUT_FILE1), zis.read
|
|
entry = zis.get_next_entry
|
|
assert_equal 'file2.txt', entry.name
|
|
assert_equal 41_234, entry.size
|
|
assert_equal ::File.read(INPUT_FILE2), zis.read
|
|
end
|
|
end
|
|
end
|