File: file_options_test.rb

package info (click to toggle)
ruby-zip 3.2.2-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 11,120 kB
  • sloc: ruby: 9,958; makefile: 23
file content (177 lines) | stat: -rw-r--r-- 6,339 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
# frozen_string_literal: true

require 'fileutils'

require_relative 'test_helper'

class FileOptionsTest < Minitest::Test
  ZIPPATH = ::File.join(Dir.tmpdir, 'options.zip').freeze
  TXTPATH = ::File.expand_path(::File.join('data', 'file1.txt'), __dir__).freeze
  TXTPATH_600 = ::File.join(Dir.tmpdir, 'file1.600.txt').freeze
  TXTPATH_755 = ::File.join(Dir.tmpdir, 'file1.755.txt').freeze
  ENTRY_1 = 'entry_1.txt'
  ENTRY_2 = 'entry_2.txt'
  ENTRY_3 = 'entry_3.txt'
  EXTRACT_1 = 'extracted_1.txt'
  EXTRACT_2 = 'extracted_2.txt'
  EXTRACT_3 = 'extracted_3.txt'
  EXTPATH_1 = ::File.join(Dir.tmpdir, EXTRACT_1).freeze
  EXTPATH_2 = ::File.join(Dir.tmpdir, EXTRACT_2).freeze
  EXTPATH_3 = ::File.join(Dir.tmpdir, EXTRACT_3).freeze

  def teardown
    FileUtils.rm_f(ZIPPATH)
    FileUtils.rm_f(EXTPATH_1)
    FileUtils.rm_f(EXTPATH_2)
    FileUtils.rm_f(EXTPATH_3)
    FileUtils.rm_f(TXTPATH_600)
    FileUtils.rm_f(TXTPATH_755)
  end

  def test_restore_permissions_true
    # Copy and set up files with different permissions.
    ::FileUtils.cp(TXTPATH, TXTPATH_600)
    ::File.chmod(0o600, TXTPATH_600)
    ::FileUtils.cp(TXTPATH, TXTPATH_755)
    ::File.chmod(0o755, TXTPATH_755)

    ::Zip::File.open(ZIPPATH, create: true) do |zip|
      zip.add(ENTRY_1, TXTPATH)
      zip.add(ENTRY_2, TXTPATH_600)
      zip.add(ENTRY_3, TXTPATH_755)
    end

    ::Zip::File.open(ZIPPATH, restore_permissions: true) do |zip|
      zip.extract(ENTRY_1, EXTRACT_1, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_2, EXTRACT_2, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_3, EXTRACT_3, destination_directory: Dir.tmpdir)
    end

    assert_equal(::File.stat(TXTPATH).mode, ::File.stat(EXTPATH_1).mode)
    assert_equal(::File.stat(TXTPATH_600).mode, ::File.stat(EXTPATH_2).mode)
    assert_equal(::File.stat(TXTPATH_755).mode, ::File.stat(EXTPATH_3).mode)
  end

  def test_restore_permissions_false
    # Copy and set up files with different permissions.
    ::FileUtils.cp(TXTPATH, TXTPATH_600)
    ::File.chmod(0o600, TXTPATH_600)
    ::FileUtils.cp(TXTPATH, TXTPATH_755)
    ::File.chmod(0o755, TXTPATH_755)

    ::Zip::File.open(ZIPPATH, create: true) do |zip|
      zip.add(ENTRY_1, TXTPATH)
      zip.add(ENTRY_2, TXTPATH_600)
      zip.add(ENTRY_3, TXTPATH_755)
    end

    ::Zip::File.open(ZIPPATH, restore_permissions: false) do |zip|
      zip.extract(ENTRY_1, EXTRACT_1, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_2, EXTRACT_2, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_3, EXTRACT_3, destination_directory: Dir.tmpdir)
    end

    default_perms = (Zip::RUNNING_ON_WINDOWS ? 0o100_644 : 0o100_666) - ::File.umask
    assert_equal(default_perms, ::File.stat(EXTPATH_1).mode)
    assert_equal(default_perms, ::File.stat(EXTPATH_2).mode)
    assert_equal(default_perms, ::File.stat(EXTPATH_3).mode)
  end

  def test_restore_permissions_as_default
    # Copy and set up files with different permissions.
    ::FileUtils.cp(TXTPATH, TXTPATH_600)
    ::File.chmod(0o600, TXTPATH_600)
    ::FileUtils.cp(TXTPATH, TXTPATH_755)
    ::File.chmod(0o755, TXTPATH_755)

    ::Zip::File.open(ZIPPATH, create: true) do |zip|
      zip.add(ENTRY_1, TXTPATH)
      zip.add(ENTRY_2, TXTPATH_600)
      zip.add(ENTRY_3, TXTPATH_755)
    end

    ::Zip::File.open(ZIPPATH) do |zip|
      zip.extract(ENTRY_1, EXTRACT_1, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_2, EXTRACT_2, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_3, EXTRACT_3, destination_directory: Dir.tmpdir)
    end

    assert_equal(::File.stat(TXTPATH).mode, ::File.stat(EXTPATH_1).mode)
    assert_equal(::File.stat(TXTPATH_600).mode, ::File.stat(EXTPATH_2).mode)
    assert_equal(::File.stat(TXTPATH_755).mode, ::File.stat(EXTPATH_3).mode)
  end

  def test_restore_times_true
    ::Zip::File.open(ZIPPATH, create: true) do |zip|
      zip.add(ENTRY_1, TXTPATH)
      zip.add_stored(ENTRY_2, TXTPATH)
    end

    ::Zip::File.open(ZIPPATH, restore_times: true) do |zip|
      zip.extract(ENTRY_1, EXTRACT_1, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_2, EXTRACT_2, destination_directory: Dir.tmpdir)
    end

    assert_time_equal(::File.mtime(TXTPATH), ::File.mtime(EXTPATH_1))
    assert_time_equal(::File.mtime(TXTPATH), ::File.mtime(EXTPATH_2))
  end

  def test_restore_times_false
    ::Zip::File.open(ZIPPATH, create: true) do |zip|
      zip.add(ENTRY_1, TXTPATH)
      zip.add_stored(ENTRY_2, TXTPATH)
    end

    ::Zip::File.open(ZIPPATH, restore_times: false) do |zip|
      zip.extract(ENTRY_1, EXTRACT_1, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_2, EXTRACT_2, destination_directory: Dir.tmpdir)
    end

    assert_time_equal(::Time.now, ::File.mtime(EXTPATH_1))
    assert_time_equal(::Time.now, ::File.mtime(EXTPATH_2))
  end

  def test_restore_times_true_as_default
    ::Zip::File.open(ZIPPATH, create: true) do |zip|
      zip.add(ENTRY_1, TXTPATH)
      zip.add_stored(ENTRY_2, TXTPATH)
    end

    ::Zip::File.open(ZIPPATH) do |zip|
      zip.extract(ENTRY_1, EXTRACT_1, destination_directory: Dir.tmpdir)
      zip.extract(ENTRY_2, EXTRACT_2, destination_directory: Dir.tmpdir)
    end

    assert_time_equal(::File.mtime(TXTPATH), ::File.mtime(EXTPATH_1))
    assert_time_equal(::File.mtime(TXTPATH), ::File.mtime(EXTPATH_2))
  end

  def test_get_find_consistency
    testzip = ::File.expand_path(::File.join('data', 'globTest.zip'), __dir__)
    Dir.mktmpdir do |tmp|
      file_f = ::File.expand_path('f_test.txt', tmp)
      file_g = ::File.expand_path('g_test.txt', tmp)

      ::Zip::File.open(testzip) do |zip|
        e1 = zip.find_entry('globTest/food.txt')
        e1.extract('f_test.txt', destination_directory: tmp)
        e2 = zip.get_entry('globTest/food.txt')
        e2.extract('g_test.txt', destination_directory: tmp)
      end

      assert_time_equal(::File.mtime(file_f), ::File.mtime(file_g))
    end
  end

  private

  # Method to compare file times. DOS times only have 2 second accuracy.
  def assert_time_equal(expected, actual)
    assert_equal(expected.year, actual.year)
    assert_equal(expected.month, actual.month)
    assert_equal(expected.day, actual.day)
    assert_equal(expected.hour, actual.hour)
    assert_equal(expected.min, actual.min)
    assert_in_delta(expected.sec, actual.sec, 1)
  end
end