qubes-installer-qubes-os/anaconda/tests/logpicker_test/archiving_test/archivebaseclass_test.py

36 lines
1.0 KiB
Python
Raw Normal View History

import mock
class ArchiveBaseClassTest(mock.TestCase):
def setUp(self):
self.setupModules([])
self.fs = mock.DiskIO()
def tearDown(self):
self.tearDownModules()
def create_tmp_tar_test(self):
import log_picker.archiving as archiving
FILE1 = "/tmp/abcd/file1"
FILE2 = "/tmp/abcd/file2"
FILELIST = [FILE1, FILE2]
TMPFILE = "/tmp/tmpfile.tar"
archiving.tempfile = mock.Mock()
archiving.tempfile.mkstemp.return_value = "", TMPFILE
archiving.tarfile = mock.Mock()
self.fs.open(FILE1, 'w').write('1\n')
self.fs.open(FILE2, 'w').write('2\n')
self.fs.open(TMPFILE, 'w')
obj = archiving.ArchiveBaseClass()
ret = obj._create_tmp_tar(FILELIST)
self.assertEqual(TMPFILE, ret)
self.assertEqual(archiving.tarfile.open().add.call_args_list,
[((FILE1,), {'arcname': 'abcd/file1'}),
((FILE2,), {'arcname': 'abcd/file2'})]
)