Update and tidy up encryption tests.

hainesr · hainesr · commit 8071290ce679 · 2021-06-27T15:56:39.000+01:00
diff --git a/test/encryption_test.rb b/test/encryption_test.rb
@@ -5,6 +5,7 @@
 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
@@ -33,7 +34,7 @@ def test_encrypt
     ) do |zis|
       entry = zis.get_next_entry
       assert_equal test_filename, entry.name
-      assert_equal 1327, entry.size
+      assert_equal 1_327, entry.size
       assert_equal content, zis.read
     end
 
@@ -55,8 +56,12 @@ def test_decrypt
     ) do |zis|
       entry = zis.get_next_entry
       assert_equal 'file1.txt', entry.name
-      assert_equal 1327, entry.size
+      assert_equal 1_327, entry.size
       assert_equal ::File.open(INPUT_FILE1, 'r').read, zis.read
+      entry = zis.get_next_entry
+      assert_equal 'file2.txt', entry.name
+      assert_equal 41_234, entry.size
+      assert_equal ::File.open(INPUT_FILE2, 'r').read, zis.read
     end
   end
 end
diff --git a/test/stored_support_test.rb b/test/stored_support_test.rb
@@ -4,7 +4,8 @@
 
 class StoredSupportTest < MiniTest::Test
   STORED_ZIP_TEST_FILE = 'test/data/zipWithStoredCompression.zip'
-  ENCRYPTED_STORED_ZIP_TEST_FILE = 'test/data/zipWithStoredCompressionAndEncryption.zip'
+  ENCRYPTED_STORED_ZIP_TEST_FILE =
+    'test/data/zipWithStoredCompressionAndEncryption.zip'
   INPUT_FILE1 = 'test/data/file1.txt'
   INPUT_FILE2 = 'test/data/file2.txt'
 
