Lines Matching full:self
37 def NoOp(self):
44 def setUp(self):
49 self.my_mox = mox.Mox()
58 self.files = {'file1': file1, 'file2': file2}
60 def tearDown(self):
62 self.my_mox.UnsetStubs()
64 def testArchiveIsValid(self):
70 self.my_mox.StubOutWithMock(os, 'stat')
72 self.my_mox.StubOutWithMock(stat, 'ST_SIZE')
79 self.assertEqual(False, test_target.ArchiveIsValid(),
85 self.assertEqual(True, test_target.ArchiveIsValid(),
89 def testRemoveLastFile(self):
95 source = self.CreateZipSource()
96 dest = self.CreateZipDestination()
101 self.my_mox.StubOutWithMock(test_target, 'OpenZipFileAtPath')
106 self.my_mox.StubOutWithMock(os, 'rename')
108 self.my_mox.StubOutWithMock(os, 'unlink')
111 self.my_mox.ReplayAll()
113 self.my_mox.VerifyAll()
115 def CreateZipSource(self):
125 source_zip = self.my_mox.CreateMock(zipfile.ZipFile)
126 source_zip.infolist().AndReturn([self.files['file1'], self.files['file1']])
127 source_zip.infolist().AndReturn([self.files['file1'], self.files['file1']])
128 source_zip.read(self.files['file1'].filename).AndReturn(
129 self.files['file1'].contents)
133 def CreateZipDestination(self):
144 dest_zip.writestr(self.files['file1'].filename,
145 self.files['file1'].contents)
153 def setUp(self):
155 self.my_mox = mox.Mox()
156 self.file1 = BagOfParts()
157 self.file1.filename = 'file1.txt'
158 self.file1.contents = 'This is a test file'
160 def tearDown(self):
162 self.my_mox.UnsetStubs()
164 def _InitMultiFileData(self):
169 self.multi_file_dir = []
174 self.multi_file_dir.append(file1)
184 self.multi_file_dir.append(file2)
189 self.multi_file_dir.append(file3)
191 def testSingleFileArchive(self):
199 self.my_mox.StubOutWithMock(test_target, 'OpenZipFileAtPath')
202 self.CreateSingleFileMock())
203 self.my_mox.StubOutWithMock(os, 'unlink')
205 self.my_mox.ReplayAll()
206 self.assertEqual(False, test_target.FixArchive('SIZE'))
207 self.my_mox.VerifyAll()
209 def CreateSingleFileMock(self):
217 mock_zip = self.my_mox.CreateMock(zipfile.ZipFile)
218 mock_zip.infolist().AndReturn([self.file1])
219 mock_zip.infolist().AndReturn([self.file1])
223 def testMultiFileArchive(self):
232 self.my_mox.StubOutWithMock(test_target, 'OpenZipFileAtPath')
235 self.CreateMultiFileMock())
236 self.my_mox.StubOutWithMock(test_target, 'RemoveLastFile')
238 self.my_mox.StubOutWithMock(os, 'stat')
240 self.my_mox.StubOutWithMock(stat, 'ST_SIZE')
242 self.my_mox.ReplayAll()
243 self.assertEqual(True, test_target.FixArchive('SIZE'))
244 self.my_mox.VerifyAll()
246 def CreateMultiFileMock(self):
255 self._InitMultiFileData()
256 mock_zip = self.my_mox.CreateMock(zipfile.ZipFile)
257 mock_zip.infolist().AndReturn(self.multi_file_dir)
265 def setUp(self):
267 self.my_mox = mox.Mox()
268 self.output_dir = '%s/' % os.getcwd()
269 self.file_to_add = 'file.txt'
270 self.input_dir = '/foo/bar/baz/'
272 def tearDown(self):
273 self.my_mox.UnsetStubs()
275 def testAddFileToArchive(self):
281 test_target = divide_and_compress.DirectoryZipper(self.output_dir,
282 self.input_dir,
284 self.my_mox.StubOutWithMock(test_target, 'OpenZipFileAtPath')
285 archive_mock = self.CreateArchiveMock()
287 ''.join([self.output_dir, '0.zip']),
289 self.StubOutOsModule()
290 self.my_mox.ReplayAll()
291 test_target.AddFileToArchive(''.join([self.input_dir, self.file_to_add]),
293 self.my_mox.VerifyAll()
295 def StubOutOsModule(self):
301 self.my_mox.StubOutWithMock(os.path, 'isfile')
302 os.path.isfile(''.join([self.input_dir, self.file_to_add])).AndReturn(True)
303 self.my_mox.StubOutWithMock(os, 'stat')
304 os.stat(''.join([self.input_dir, self.file_to_add])).AndReturn([39480])
305 self.my_mox.StubOutWithMock(stat, 'ST_SIZE')
308 def CreateArchiveMock(self):
317 archive_mock = self.my_mox.CreateMock(zipfile.ZipFile)
318 archive_mock.write(''.join([self.input_dir, self.file_to_add]),
319 self.file_to_add)
350 def setUp(self):
352 self.my_mox = mox.Mox()
354 self.base_dir = '/dir1'
355 self.output_path = '/out_dir/'
356 self.test_target = divide_and_compress.DirectoryZipper(
357 self.output_path, self.base_dir, 1024*1024, True)
359 self.InitArgLists()
360 self.InitOsDotPath()
361 self.InitArchiveIsValid()
362 self.InitWriteIndexRecord()
363 self.InitAddFileToArchive()
365 def tearDown(self):
366 self.my_mox.UnsetStubs()
368 def testCompressDirectory(self):
370 self.my_mox.ReplayAll()
371 for arguments in self.argument_lists:
372 self.test_target.CompressDirectory(None, arguments[0], arguments[1])
373 self.my_mox.VerifyAll()
375 def InitAddFileToArchive(self):
380 self.my_mox.StubOutWithMock(self.test_target, 'AddFileToArchive')
381 self.test_target.AddFileToArchive('/dir1/file1.txt', zipfile.ZIP_DEFLATED)
382 self.test_target.AddFileToArchive('/dir1/file2.txt', zipfile.ZIP_DEFLATED)
383 self.test_target.AddFileToArchive('/dir1/dir2/dir4/file3.txt',
385 self.test_target.AddFileToArchive('/dir1/dir5/file4.txt',
387 self.test_target.AddFileToArchive('/dir1/dir5/file4.txt',
389 self.test_target.AddFileToArchive('/dir1/dir5/file5.txt',
391 self.test_target.AddFileToArchive('/dir1/dir5/file5.txt',
393 self.test_target.AddFileToArchive('/dir1/dir5/file6.txt',
395 self.test_target.AddFileToArchive('/dir1/dir5/file7.txt',
397 self.test_target.AddFileToArchive('/dir1/dir5/file7.txt',
399 self.test_target.AddFileToArchive('/dir1/dir6/file8.txt',
402 def InitWriteIndexRecord(self):
404 self.my_mox.StubOutWithMock(self.test_target, 'WriteIndexRecord')
408 self.test_target.WriteIndexRecord().AndReturn(True)
409 self.test_target.WriteIndexRecord().AndReturn(False)
410 self.test_target.WriteIndexRecord().AndReturn(False)
411 self.test_target.WriteIndexRecord().AndReturn(True)
412 self.test_target.WriteIndexRecord().AndReturn(True)
413 self.test_target.WriteIndexRecord().AndReturn(False)
414 self.test_target.WriteIndexRecord().AndReturn(True)
416 def InitArchiveIsValid(self):
424 self.my_mox.StubOutWithMock(self.test_target, 'ArchiveIsValid')
425 self.my_mox.StubOutWithMock(self.test_target, 'FixArchive')
426 self.test_target.ArchiveIsValid().AndReturn(True)
427 self.test_target.ArchiveIsValid().AndReturn(True)
428 self.test_target.ArchiveIsValid().AndReturn(True)
431 self.test_target.ArchiveIsValid().AndReturn(False)
432 self.test_target.FixArchive('SIZE').AndReturn(True)
433 self.test_target.ArchiveIsValid().AndReturn(True)
436 self.test_target.ArchiveIsValid().AndReturn(False)
437 self.test_target.FixArchive('SIZE').AndReturn(True)
438 self.test_target.ArchiveIsValid().AndReturn(True)
439 self.test_target.ArchiveIsValid().AndReturn(True)
442 self.test_target.ArchiveIsValid().AndReturn(False)
443 self.test_target.FixArchive('SIZE').AndReturn(True)
444 self.test_target.ArchiveIsValid().AndReturn(False)
445 self
446 self.test_target.ArchiveIsValid().AndReturn(True)
448 def InitOsDotPath(self):
456 self.my_mox.StubOutWithMock(os.path, 'isfile')
474 def InitArgLists(self):
476 self.argument_lists = []
477 self.argument_lists.append(['/dir1',
480 self.argument_lists.append(['/dir1/dir2', ['dir3', 'dir4']])
481 self.argument_lists.append(['/dir1/dir2/dir3', []])
482 self.argument_lists.append(['/dir1/dir2/dir4', ['file3.txt']])
483 self.argument_lists.append(['/dir1/dir5',
486 self.argument_lists.append(['/dir1/dir6', ['file8.txt']])
