def test_archive_logs(self, archivelogs): self.mox.StubOutWithMock(fileutils, 'zip') if archivelogs: fileutils.zip(archivelogs, os.path.join('sym', 'var', 'log'), os.path.join('sym', 'var')) self.mox.ReplayAll() args = self.Args({'archivelogs': archivelogs}) upgrade = self.PartialMockUnixInstall(args, None) upgrade.archive_logs() self.mox.VerifyAll()
def test_archive_logs(self, archivelogs): self.mox.StubOutWithMock(fileutils, 'zip') if archivelogs: fileutils.zip( archivelogs, os.path.join('sym', 'var', 'log'), os.path.join('sym', 'var')) self.mox.ReplayAll() args = self.Args({'archivelogs': archivelogs}) upgrade = self.PartialMockUnixInstall(args, None) upgrade.archive_logs() self.mox.VerifyAll()
def test_zip(self): self.mox.StubOutClassWithMocks(zipfile, 'ZipFile') self.mox.StubOutWithMock(os, 'walk') self.mox.StubOutWithMock(os.path, 'isfile') # files = ['test', 'test/a', 'test/b', 'test/b/c'] os.walk('test').AndReturn([ ('test', ['b'], ['a']), ('test/b', [], ['c'])]) os.path.isfile('test').AndReturn(False) mockzip = zipfile.ZipFile( 'path/to/test.zip', 'w', zipfile.ZIP_DEFLATED) mockzip.write('test/a', 'a') mockzip.write('test/b/c', 'b/c') mockzip.close() self.mox.ReplayAll() fileutils.zip('path/to/test.zip', ['test'], 'test') self.mox.VerifyAll()