36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
|
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'})]
|
||
|
)
|
||
|
|