tests: add initial backup test
This commit is contained in:
Marek Marczykowski-Górecki
parent
a12cf158da
commit
603384b4c6
@ -76,8 +76,8 @@ ln -sf . %{name}-%{version}
|
||||
%setup -T -D
|
||||
|
||||
%build
|
||||
python -m compileall core core-modules qmemman
|
||||
python -O -m compileall core dom/core-modules qmemman
|
||||
python -m compileall core core-modules qmemman tests
|
||||
python -O -m compileall core dom/core-modules qmemman tests
|
||||
for dir in dispvm qmemman; do
|
||||
(cd $dir; make)
|
||||
done
|
||||
@ -122,6 +122,9 @@ cp core-modules/0*.py $RPM_BUILD_ROOT%{python_sitearch}/qubes/modules
|
||||
cp core-modules/0*.py[co] $RPM_BUILD_ROOT%{python_sitearch}/qubes/modules
|
||||
cp core-modules/__init__.py $RPM_BUILD_ROOT%{python_sitearch}/qubes/modules
|
||||
cp core-modules/__init__.py[co] $RPM_BUILD_ROOT%{python_sitearch}/qubes/modules
|
||||
mkdir -p $RPM_BUILD_ROOT%{python_sitearch}/qubes/tests
|
||||
cp tests/*.py $RPM_BUILD_ROOT%{python_sitearch}/qubes/tests/
|
||||
cp tests/*.py[co] $RPM_BUILD_ROOT%{python_sitearch}/qubes/tests/
|
||||
|
||||
mkdir -p $RPM_BUILD_ROOT%{_sysconfdir}/qubes
|
||||
cp qmemman/qmemman.conf $RPM_BUILD_ROOT%{_sysconfdir}/qubes/
|
||||
@ -286,6 +289,7 @@ fi
|
||||
%{python_sitearch}/qubes/qmemman*.py*
|
||||
%{python_sitearch}/qubes/modules/0*.py*
|
||||
%{python_sitearch}/qubes/modules/__init__.py*
|
||||
%{python_sitearch}/qubes/tests
|
||||
/usr/lib/qubes/unbind-pci-device.sh
|
||||
/usr/lib/qubes/cleanup-dispvms
|
||||
/usr/lib/qubes/qmemman_daemon.py*
|
||||
|
||||
0
tests/__init__.py
Normal file
0
tests/__init__.py
Normal file
213
tests/backup.py
Normal file
213
tests/backup.py
Normal file
@ -0,0 +1,213 @@
|
||||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
#
|
||||
# The Qubes OS Project, http://www.qubes-os.org
|
||||
#
|
||||
# Copyright (C) 2014 Marek Marczykowski-Górecki <marmarek@invisiblethingslab.com>
|
||||
#
|
||||
# This program is free software; you can redistribute it and/or
|
||||
# modify it under the terms of the GNU General Public License
|
||||
# as published by the Free Software Foundation; either version 2
|
||||
# of the License, or (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program; if not, write to the Free Software
|
||||
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
||||
#
|
||||
#
|
||||
from multiprocessing import Event
|
||||
|
||||
import os
|
||||
import shutil
|
||||
|
||||
import unittest
|
||||
import sys
|
||||
from qubes.qubes import QubesVmCollection, QubesException
|
||||
from qubes import backup
|
||||
|
||||
VM_PREFIX = "test-"
|
||||
|
||||
class BackupTests(unittest.TestCase):
|
||||
def setUp(self):
|
||||
self.error_detected = Event()
|
||||
self.verbose = False
|
||||
self.qc = QubesVmCollection()
|
||||
self.qc.lock_db_for_writing()
|
||||
self.qc.load()
|
||||
|
||||
if self.verbose:
|
||||
print >>sys.stderr, "-> Creating backupvm"
|
||||
backupvm = self.qc.add_new_vm("QubesAppVm",
|
||||
name="%sbackupvm" % VM_PREFIX,
|
||||
template=self.qc.get_default_template())
|
||||
backupvm.create_on_disk(verbose=self.verbose)
|
||||
self.qc.save()
|
||||
self.qc.unlock_db()
|
||||
|
||||
self.backupdir = os.path.join(os.environ["HOME"], "test-backup")
|
||||
os.mkdir(self.backupdir)
|
||||
|
||||
|
||||
def tearDown(self):
|
||||
vmlist = [vm for vm in self.qc.values() if vm.name.startswith(
|
||||
VM_PREFIX)]
|
||||
self.remove_vms(vmlist)
|
||||
shutil.rmtree(self.backupdir)
|
||||
|
||||
def print_progress(self, progress):
|
||||
if self.verbose:
|
||||
print >> sys.stderr, "\r-> Backing up files: {0}%...".format(progress)
|
||||
|
||||
def error_callback(self, message):
|
||||
self.error_detected.set()
|
||||
if self.verbose:
|
||||
print >> sys.stderr, "ERROR: {0}".format(message)
|
||||
|
||||
def print_callback(self, msg):
|
||||
if self.verbose:
|
||||
print msg
|
||||
|
||||
def fill_image(self, path, size=None, sparse=False):
|
||||
block_size = 4096
|
||||
|
||||
if self.verbose:
|
||||
print >>sys.stderr, "-> Filling %s" % path
|
||||
f = open(path, 'w+')
|
||||
if size is None:
|
||||
f.seek(0, 2)
|
||||
size = f.tell()
|
||||
f.seek(0)
|
||||
|
||||
for block_num in xrange(size/block_size):
|
||||
f.write('a' * block_size)
|
||||
if sparse:
|
||||
f.seek(block_size, 1)
|
||||
|
||||
f.close()
|
||||
|
||||
def create_basic_vms(self, leave_locked=False):
|
||||
template=self.qc.get_default_template()
|
||||
|
||||
vms = []
|
||||
self.qc.lock_db_for_writing()
|
||||
self.qc.load()
|
||||
vmname = "%stest1" % VM_PREFIX
|
||||
if self.verbose:
|
||||
print >>sys.stderr, "-> Creating %s" % vmname
|
||||
testvm1 = self.qc.add_new_vm("QubesAppVm",
|
||||
name=vmname,
|
||||
template=template)
|
||||
testvm1.create_on_disk(verbose=self.verbose)
|
||||
vms.append(testvm1)
|
||||
self.fill_image(testvm1.private_img, 100*1024*1024)
|
||||
|
||||
vmname = "%stesthvm1" % VM_PREFIX
|
||||
if self.verbose:
|
||||
print >>sys.stderr, "-> Creating %s" % vmname
|
||||
testvm2 = self.qc.add_new_vm("QubesHVm",
|
||||
name=vmname)
|
||||
testvm2.create_on_disk(verbose=self.verbose)
|
||||
self.fill_image(testvm2.root_img, 1024*1024*1024, True)
|
||||
vms.append(testvm2)
|
||||
|
||||
if not leave_locked:
|
||||
self.qc.save()
|
||||
self.qc.unlock_db()
|
||||
|
||||
return vms
|
||||
|
||||
def remove_vms(self, vms):
|
||||
self.qc.lock_db_for_writing()
|
||||
self.qc.load()
|
||||
|
||||
for vm in vms:
|
||||
if isinstance(vm, str):
|
||||
vm = self.qc.get_vm_by_name(vm)
|
||||
else:
|
||||
vm = self.qc[vm.qid]
|
||||
if self.verbose:
|
||||
print >>sys.stderr, "-> Removing %s" % vm.name
|
||||
vm.remove_from_disk()
|
||||
self.qc.pop(vm.qid)
|
||||
self.qc.save()
|
||||
self.qc.unlock_db()
|
||||
|
||||
def make_backup(self, vms):
|
||||
try:
|
||||
files_to_backup = backup.backup_prepare(vms,
|
||||
print_callback=self.print_callback)
|
||||
except QubesException as e:
|
||||
self.fail("QubesException during backup_prepare: %s" % str(e))
|
||||
|
||||
try:
|
||||
backup.backup_do(self.backupdir, files_to_backup, "qubes",
|
||||
progress_callback=self.print_progress)
|
||||
except QubesException as e:
|
||||
self.fail("QubesException during backup_do: %s" % str(e))
|
||||
|
||||
def restore_backup(self):
|
||||
backupfile = os.path.join(self.backupdir,
|
||||
sorted(os.listdir(self.backupdir))[-1])
|
||||
|
||||
try:
|
||||
backup_info = backup.backup_restore_prepare(
|
||||
backupfile, "qubes", print_callback=self.print_callback)
|
||||
except QubesException as e: self.fail(
|
||||
"QubesException during backup_restore_prepare: %s" % str(e))
|
||||
if self.verbose:
|
||||
backup.backup_restore_print_summary(backup_info)
|
||||
|
||||
try:
|
||||
backup.backup_restore_do(
|
||||
backup_info,
|
||||
print_callback=self.print_callback,
|
||||
error_callback=self.error_callback)
|
||||
# TODO: print_callback=self.print_callback if self.verbose else None,
|
||||
except QubesException as e:
|
||||
self.fail("QubesException during backup_restore_do: %s" % str(e))
|
||||
self.assertFalse(self.error_detected.is_set(),
|
||||
"Error detected during backup_restore_do")
|
||||
os.unlink(backupfile)
|
||||
|
||||
def test_basic_backup(self):
|
||||
vms = self.create_basic_vms()
|
||||
self.make_backup(vms)
|
||||
self.remove_vms(vms)
|
||||
self.restore_backup()
|
||||
self.remove_vms(vms)
|
||||
|
||||
@unittest.expectedFailure
|
||||
def test_sparse_multipart(self):
|
||||
vms = []
|
||||
self.qc.lock_db_for_writing()
|
||||
self.qc.load()
|
||||
|
||||
vmname = "%stesthvm2" % VM_PREFIX
|
||||
if self.verbose:
|
||||
print >>sys.stderr, "-> Creating %s" % vmname
|
||||
|
||||
hvmtemplate = self.qc.add_new_vm("QubesTemplateHVm",
|
||||
name=vmname)
|
||||
hvmtemplate.create_on_disk(verbose=self.verbose)
|
||||
self.fill_image(os.path.join(hvmtemplate.dir_path, '00file'),
|
||||
195*1024*1024-4096*3)
|
||||
self.fill_image(hvmtemplate.private_img, 195*1024*1024-4096*3)
|
||||
self.fill_image(hvmtemplate.root_img, 1024*1024*1024, sparse=True)
|
||||
vms.append(hvmtemplate)
|
||||
|
||||
self.qc.save()
|
||||
self.qc.unlock_db()
|
||||
|
||||
self.make_backup(vms)
|
||||
self.remove_vms(vms)
|
||||
self.restore_backup()
|
||||
self.remove_vms(vms)
|
||||
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user