borg/darc/test.py

127 lines
4.4 KiB
Python

import doctest
import filecmp
import os
from StringIO import StringIO
import sys
import shutil
import tempfile
import unittest
from xattr import xattr, XATTR_NOFOLLOW
import getpass
getpass.getpass = lambda m: 'abc123'
from . import store, helpers
from .archiver import Archiver
class Test(unittest.TestCase):
def setUp(self):
self.archiver = Archiver()
self.tmpdir = tempfile.mkdtemp()
self.store_path = os.path.join(self.tmpdir, 'store')
self.input_path = os.path.join(self.tmpdir, 'input')
self.output_path = os.path.join(self.tmpdir, 'output')
os.mkdir(self.input_path)
os.mkdir(self.output_path)
os.chdir(self.tmpdir)
self.keychain = '/tmp/_test_dedupstore.keychain'
if not os.path.exists(self.keychain):
self.darc('init-keychain')
self.darc('init', self.store_path)
def tearDown(self):
shutil.rmtree(self.tmpdir)
def darc(self, *args, **kwargs):
exit_code = kwargs.get('exit_code', 0)
args = ['--keychain', self.keychain] + list(args)
try:
stdout, stderr = sys.stdout, sys.stderr
output = StringIO()
sys.stdout = sys.stderr = output
ret = self.archiver.run(args)
sys.stdout, sys.stderr = stdout, stderr
if ret != exit_code:
print output.getvalue()
self.assertEqual(exit_code, ret)
return output.getvalue()
finally:
sys.stdout, sys.stderr = stdout, stderr
def create_src_archive(self, name):
src_dir = os.path.join(os.getcwd(), os.path.dirname(__file__))
self.darc('create', self.store_path + '::' + name, src_dir)
def create_regual_file(self, name, size=0):
filename = os.path.join(self.input_path, name)
if not os.path.exists(os.path.dirname(filename)):
os.makedirs(os.path.dirname(filename))
with open(filename, 'wb') as fd:
fd.write('X' * size)
def get_xattrs(self, path):
try:
return dict(xattr(path, XATTR_NOFOLLOW))
except IOError:
return {}
def diff_dirs(self, dir1, dir2):
diff = filecmp.dircmp(dir1, dir2)
self.assertEqual(diff.left_only, [])
self.assertEqual(diff.right_only, [])
self.assertEqual(diff.diff_files, [])
for filename in diff.common:
path1 = os.path.join(dir1, filename)
path2 = os.path.join(dir2, filename)
s1 = os.lstat(path1)
s2 = os.lstat(path2)
attrs = ['st_mode', 'st_uid', 'st_gid']
# We can't restore symlink atime/mtime right now
if not os.path.islink(path1):
attrs.append('st_mtime')
d1 = [filename] + [getattr(s1, a) for a in attrs]
d2 = [filename] + [getattr(s2, a) for a in attrs]
d1.append(self.get_xattrs(path1))
d2.append(self.get_xattrs(path2))
self.assertEqual(d1, d2)
def test_basic_functionality(self):
self.create_regual_file('file1', size=1024*80)
self.create_regual_file('dir2/file2', size=1024*80)
x = xattr(os.path.join(self.input_path, 'file1'))
x.set('user.foo', 'bar')
os.symlink('somewhere', os.path.join(self.input_path, 'link1'))
os.mkfifo(os.path.join(self.input_path, 'fifo1'))
self.darc('create', self.store_path + '::test', 'input')
self.darc('extract', self.store_path + '::test', 'output')
self.diff_dirs('input', 'output/input')
def test_corrupted_store(self):
self.create_src_archive('test')
self.darc('verify', self.store_path + '::test')
fd = open(os.path.join(self.tmpdir, 'store', 'bands', '0', '0'), 'r+')
fd.seek(100)
fd.write('X')
fd.close()
self.darc('verify', self.store_path + '::test', exit_code=1)
def test_keychain(self):
keychain = os.path.join(self.tmpdir, 'keychain')
keychain2 = os.path.join(self.tmpdir, 'keychain2')
self.darc('-k', keychain, 'init-keychain')
self.darc('-k', keychain, 'change-password')
self.darc('-k', keychain, 'export-restricted', keychain2)
def suite():
suite = unittest.TestSuite()
suite.addTest(unittest.TestLoader().loadTestsFromTestCase(Test))
suite.addTest(store.suite())
suite.addTest(doctest.DocTestSuite(helpers))
return suite
if __name__ == '__main__':
unittest.TextTestRunner(verbosity=2).run(suite())