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()
