64 lines
2.5 KiB
Python
64 lines
2.5 KiB
Python
|
# The Qubes OS Project, https://www.qubes-os.org/
|
||
|
|
#
|
||
|
|
# Copyright (C) 2015 Bahtiar `kalkin-` Gadimov <bahtiar@gadimov.de>
|
||
|
|
#
|
||
|
|
# 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 qubes.tests import QubesTestCase, SystemTestsMixin
|
||
|
|
from qubes.qubes import defaults
|
||
|
|
|
||
|
|
import qubes.storage
|
||
|
|
from qubes.storage.xen import QubesXenVmStorage, XenPool
|
||
|
|
|
||
|
|
|
||
|
|
class TC_00_Storage(SystemTestsMixin, QubesTestCase):
|
||
|
|
|
||
|
|
def test_000_dump(self):
|
||
|
|
""" Dumps storage instance to a storage string """
|
||
|
|
vmname = self.make_vm_name('appvm')
|
||
|
|
template = self.qc.get_default_template()
|
||
|
|
storage = self.qc.add_new_vm('QubesAppVm', name=vmname, pool='default',
|
||
|
|
template=template).storage
|
||
|
|
result = qubes.storage.dump(storage)
|
||
|
|
expected = 'qubes.storage.xen.QubesXenVmStorage'
|
||
|
|
self.assertEquals(result, expected)
|
||
|
|
|
||
|
|
def test_001_load(self):
|
||
|
|
""" Loads storage type from a storage string """
|
||
|
|
result = qubes.storage.load('qubes.storage.xen.QubesXenVmStorage')
|
||
|
|
self.assertTrue(result is QubesXenVmStorage)
|
||
|
|
|
||
|
|
def test_002_default_pool_types(self):
|
||
|
|
""" The only predifined pool type is xen """
|
||
|
|
result = defaults['pool_types'].keys()
|
||
|
|
expected = ["xen"]
|
||
|
|
self.assertEquals(result, expected)
|
||
|
|
|
||
|
|
def test_003_get_pool_klass(self):
|
||
|
|
""" Expect the default pool to be `XenPool` """
|
||
|
|
result = qubes.storage._get_pool_klass('default')
|
||
|
|
self.assertTrue(result is XenPool)
|
||
|
|
|
||
|
|
|
||
|
|
class TC_01_Storage(SystemTestsMixin, QubesTestCase):
|
||
|
|
|
||
|
|
def test_000_vm_use_default_pool(self):
|
||
|
|
vmname = self.make_vm_name('appvm')
|
||
|
|
template = self.qc.get_default_template()
|
||
|
|
vm = self.qc.add_new_vm('QubesAppVm', name=vmname, template=template,
|
||
|
|
pool='default')
|
||
|
|
self.assertIsInstance(vm.storage, QubesXenVmStorage)
|