blob: 2f4f0e60309cf8a246ce997534e499c0b5bd0aed (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
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'})]
)
|
