OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import doctest | 6 import doctest |
7 import os | 7 import os |
8 import oshelpers | 8 import oshelpers |
9 import shutil | 9 import shutil |
10 import subprocess | 10 import subprocess |
(...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
88 file2 = self.MakeFile('file2', 3354) | 88 file2 = self.MakeFile('file2', 3354) |
89 self.RunZip([self.zipname, file1, file2]) | 89 self.RunZip([self.zipname, file1, file2]) |
90 self.OpenZipFile() | 90 self.OpenZipFile() |
91 self.assertEqual(len(self.zipfile.namelist()), 2) | 91 self.assertEqual(len(self.zipfile.namelist()), 2) |
92 self.assertEqual(self.GetZipInfo(file1).file_size, 1024) | 92 self.assertEqual(self.GetZipInfo(file1).file_size, 1024) |
93 self.assertEqual(self.GetZipInfo(file2).file_size, 3354) | 93 self.assertEqual(self.GetZipInfo(file2).file_size, 3354) |
94 # make sure files are added in order | 94 # make sure files are added in order |
95 self.assertEqual(self.zipfile.namelist()[0], file1) | 95 self.assertEqual(self.zipfile.namelist()[0], file1) |
96 | 96 |
97 def testAddFilesWithGlob(self): | 97 def testAddFilesWithGlob(self): |
98 file1 = self.MakeFile('file1', 1024) | 98 self.MakeFile('file1', 1024) |
99 file2 = self.MakeFile('file2', 3354) | 99 self.MakeFile('file2', 3354) |
100 self.RunZip([self.zipname, 'file*']) | 100 self.RunZip([self.zipname, 'file*']) |
101 self.OpenZipFile() | 101 self.OpenZipFile() |
102 self.assertEqual(len(self.zipfile.namelist()), 2) | 102 self.assertEqual(len(self.zipfile.namelist()), 2) |
103 | 103 |
104 def testAddDir(self): | 104 def testAddDir(self): |
105 os.mkdir(self.GetTempPath('dir1')) | 105 os.mkdir(self.GetTempPath('dir1')) |
106 self.RunZip([self.zipname, 'dir1']) | 106 self.RunZip([self.zipname, 'dir1']) |
107 self.OpenZipFile() | 107 self.OpenZipFile() |
108 self.assertEqual(len(self.zipfile.namelist()), 1) | 108 self.assertEqual(len(self.zipfile.namelist()), 1) |
109 | 109 |
(...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
148 self.assertEqual(self.GetZipInfo(file2).file_size, 256) | 148 self.assertEqual(self.GetZipInfo(file2).file_size, 256) |
149 self.assertEqual(self.zipfile.namelist()[2], file3) | 149 self.assertEqual(self.zipfile.namelist()[2], file3) |
150 self.assertEqual(self.GetZipInfo(file3).file_size, 768) | 150 self.assertEqual(self.GetZipInfo(file3).file_size, 768) |
151 self.assertEqual(self.zipfile.namelist()[3], file4) | 151 self.assertEqual(self.zipfile.namelist()[3], file4) |
152 self.assertEqual(self.GetZipInfo(file4).file_size, 1024) | 152 self.assertEqual(self.GetZipInfo(file4).file_size, 1024) |
153 | 153 |
154 def testUpdateSubdirectory(self): | 154 def testUpdateSubdirectory(self): |
155 os.mkdir(self.GetTempPath('dir1')) | 155 os.mkdir(self.GetTempPath('dir1')) |
156 file1 = self.MakeFile(os.path.join('dir1', 'file1'), 256) | 156 file1 = self.MakeFile(os.path.join('dir1', 'file1'), 256) |
157 os.mkdir(self.GetTempPath(os.path.join('dir1', 'dir2'))) | 157 os.mkdir(self.GetTempPath(os.path.join('dir1', 'dir2'))) |
158 file2 = self.MakeFile(os.path.join('dir1', 'dir2', 'file2'), 1234) | 158 self.MakeFile(os.path.join('dir1', 'dir2', 'file2'), 1234) |
159 self.RunZip([self.zipname, '-r', 'dir1']) | 159 self.RunZip([self.zipname, '-r', 'dir1']) |
160 self.OpenZipFile() | 160 self.OpenZipFile() |
161 self.assertEqual(len(self.zipfile.namelist()), 4) | 161 self.assertEqual(len(self.zipfile.namelist()), 4) |
162 self.assertEqual(self.GetZipInfo(file1).file_size, 256) | 162 self.assertEqual(self.GetZipInfo(file1).file_size, 256) |
163 self.CloseZipFile() | 163 self.CloseZipFile() |
164 | 164 |
165 self.MakeFile(file1, 2560) | 165 self.MakeFile(file1, 2560) |
166 self.RunZip([self.zipname, file1]) | 166 self.RunZip([self.zipname, file1]) |
167 self.OpenZipFile() | 167 self.OpenZipFile() |
168 self.assertEqual(len(self.zipfile.namelist()), 4) | 168 self.assertEqual(len(self.zipfile.namelist()), 4) |
(...skipping 13 matching lines...) Expand all Loading... |
182 self.assertEqual(len(self.zipfile.namelist()), 3) | 182 self.assertEqual(len(self.zipfile.namelist()), 3) |
183 | 183 |
184 | 184 |
185 def main(): | 185 def main(): |
186 suite = unittest.TestLoader().loadTestsFromTestCase(TestZip) | 186 suite = unittest.TestLoader().loadTestsFromTestCase(TestZip) |
187 suite.addTests(doctest.DocTestSuite(oshelpers)) | 187 suite.addTests(doctest.DocTestSuite(oshelpers)) |
188 result = unittest.TextTestRunner(verbosity=2).run(suite) | 188 result = unittest.TextTestRunner(verbosity=2).run(suite) |
189 return int(not result.wasSuccessful()) | 189 return int(not result.wasSuccessful()) |
190 | 190 |
191 | 191 |
192 if __name__=='__main__': | 192 if __name__ == '__main__': |
193 sys.exit(main()) | 193 sys.exit(main()) |
OLD | NEW |