Qubes/core-admin
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

tests/file: run_until_complete(); coro_maybe() for single arg

Browse Source
This commit is contained in:
Rusty Bird 2020-04-07 21:04:04 +00:00
parent 9122a14f94
commit f9538a578d
No known key found for this signature in database
GPG Key ID: 469D78F47AAF2ADF
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
qubes/tests/storage_file.py
View File

@ -395,9 +395,11 @@ class TC_01_FileVolumes(qubes.tests.QubesTestCase):
} }
vm = qubes.tests.storage.TestVM(self) vm = qubes.tests.storage.TestVM(self)
volume = self.app.get_pool(self.POOL_NAME).init_volume(vm, config) volume = self.app.get_pool(self.POOL_NAME).init_volume(vm, config)
qubes.utils.void_coros_maybe([volume.create()]) self.loop.run_until_complete(
qubes.utils.coro_maybe(volume.create()))
new_size = 64 * 1024 ** 2 new_size = 64 * 1024 ** 2
qubes.utils.void_coros_maybe([volume.resize(new_size)]) self.loop.run_until_complete(
qubes.utils.coro_maybe(volume.resize(new_size)))
self.assertEqual(os.path.getsize(volume.path), new_size) self.assertEqual(os.path.getsize(volume.path), new_size)
self.assertEqual(volume.size, new_size) self.assertEqual(volume.size, new_size)
@ -459,8 +461,8 @@ class TC_01_FileVolumes(qubes.tests.QubesTestCase):
} }
vm = qubes.tests.storage.TestVM(self) vm = qubes.tests.storage.TestVM(self)
volume = self.app.get_pool(self.POOL_NAME).init_volume(vm, config) volume = self.app.get_pool(self.POOL_NAME).init_volume(vm, config)
qubes.utils.void_coros_maybe([volume.create()]) self.loop.run_until_complete(qubes.utils.coro_maybe(volume.create()))
qubes.utils.void_coros_maybe([volume.start()]) self.loop.run_until_complete(qubes.utils.coro_maybe(volume.start()))
self._setup_loop(volume.path) self._setup_loop(volume.path)
new_size = 64 * 1024 ** 2 new_size = 64 * 1024 ** 2
orig_check_call = subprocess.check_call orig_check_call = subprocess.check_call
@ -468,11 +470,12 @@ class TC_01_FileVolumes(qubes.tests.QubesTestCase):
sudo = [] if os.getuid() == 0 else ['sudo'] sudo = [] if os.getuid() == 0 else ['sudo']
mock_subprocess.side_effect = (lambda *args, **kwargs: mock_subprocess.side_effect = (lambda *args, **kwargs:
orig_check_call(sudo + args[0], *args[1:], **kwargs)) orig_check_call(sudo + args[0], *args[1:], **kwargs))
qubes.utils.void_coros_maybe([volume.resize(new_size)]) self.loop.run_until_complete(
qubes.utils.coro_maybe(volume.resize(new_size)))
self.assertEqual(os.path.getsize(volume.path), new_size) self.assertEqual(os.path.getsize(volume.path), new_size)
self.assertEqual(self._get_loop_size(volume.path), new_size) self.assertEqual(self._get_loop_size(volume.path), new_size)
self.assertEqual(volume.size, new_size) self.assertEqual(volume.size, new_size)
qubes.utils.void_coros_maybe([volume.stop()]) self.loop.run_until_complete(qubes.utils.coro_maybe(volume.stop()))
self.assertEqual(os.path.getsize(volume.path), new_size) self.assertEqual(os.path.getsize(volume.path), new_size)
self.assertEqual(volume.size, new_size) self.assertEqual(volume.size, new_size)