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

120 lines
3.8 KiB
Python
Raw Normal View History

import os
import mock
import tempfile
from log_picker.archiving import ArchivationError
class Bzip2ArchiveTest(mock.TestCase):
def setUp(self):
self.tmp_files = []
pass
def _get_tmp(self):
_, filename = tempfile.mkstemp()
self.tmp_files.append(filename)
return filename
def tearDown(self):
for file in self.tmp_files:
os.unlink(file)
def mimetype_test(self):
from log_picker.archiving import Bzip2Archive
obj = Bzip2Archive(usetar=True)
ret = obj.mimetype
self.assertEqual(ret, 'application/x-bzip2')
def support_compression_test(self):
from log_picker.archiving import Bzip2Archive
obj = Bzip2Archive(usetar=True)
ret = obj.support_compression
self.assertTrue(ret)
def file_ext_1_test(self):
from log_picker.archiving import Bzip2Archive
obj = Bzip2Archive(usetar=True)
ret = obj.file_ext
self.assertEqual(ret, '.tar.bz2')
def file_ext_2_test(self):
from log_picker.archiving import Bzip2Archive
obj = Bzip2Archive(usetar=False)
ret = obj.file_ext
self.assertEqual(ret, '.bz2')
def create_archive_1_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
infile1 = self._get_tmp()
infile2 = self._get_tmp()
obj = Bzip2Archive(usetar=False)
self.assertRaises(ArchivationError, obj.create_archive, out_filename, \
[infile1, infile2])
def create_archive_2_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
obj = Bzip2Archive(usetar=False)
self.assertRaises(ArchivationError, obj.create_archive, out_filename, [])
def create_archive_3_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
obj = Bzip2Archive(usetar=True)
self.assertRaises(ArchivationError, obj.create_archive, out_filename, [])
def create_archive_4_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
infile1 = self._get_tmp()
infile2 = self._get_tmp()
obj = Bzip2Archive(usetar=True)
# Because both files (infile1 and infile2) are empty
self.assertRaises(ArchivationError, obj.create_archive, out_filename,
[infile1, infile2])
def create_archive_5_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
infile1 = self._get_tmp()
infile2 = self._get_tmp()
open(infile1, 'w').write('abcdefghijklmnopqrstuvwxyz')
open(infile2, 'w').write('1234567890')
obj = Bzip2Archive(usetar=True)
obj.create_archive(out_filename, [infile1, infile2])
self.assertTrue(os.path.getsize(out_filename))
def create_archive_6_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
infile1 = self._get_tmp()
open(infile1, 'w').write('abcdefghijklmnopqrstuvwxyz')
obj = Bzip2Archive(usetar=False)
obj.create_archive(out_filename, [infile1])
self.assertTrue(os.path.getsize(out_filename))
def create_archive_7_test(self):
from log_picker.archiving import Bzip2Archive
out_filename = self._get_tmp()
infile1 = self._get_tmp()
open(infile1, 'w').write('abcdefghijklmnopqrstuvwxyz')
obj = Bzip2Archive(usetar=True)
obj.create_archive(out_filename, [infile1])
self.assertTrue(os.path.getsize(out_filename))