123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680 |
- # -*- encoding: utf8 -*-
- #
- # The Qubes OS Project, http://www.qubes-os.org
- #
- # Copyright (C) 2017 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 Lesser General Public License as published by
- # the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
- #
- # You should have received a copy of the GNU Lesser General Public License along
- # with this program; if not, see <http://www.gnu.org/licenses/>.
- import functools
- import os
- import signal
- import tempfile
- import unittest.mock
- import asyncio
- import qubesadmin.tests
- import qubesadmin.tools.qvm_start_gui
- import qubesadmin.vm
- class TC_00_qvm_start_gui(qubesadmin.tests.QubesTestCase):
- def setUp(self):
- super(TC_00_qvm_start_gui, self).setUp()
- self.launcher = qubesadmin.tools.qvm_start_gui.GUILauncher(self.app)
- @unittest.mock.patch('subprocess.check_output')
- def test_000_kde_args(self, proc_mock):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'label', None)] = \
- b'0\x00default=False type=label red'
- proc_mock.side_effect = [
- b'KWIN_RUNNING = 0x1\n',
- b'access control enabled, only authorized clients can connect\n'
- b'SI:localuser:root\n'
- b'SI:localuser:' + os.environ['USER'].encode() + b'\n',
- ]
- args = self.launcher.kde_guid_args(self.app.domains['test-vm'])
- self.assertEqual(args, ['-T', '-p',
- '_KDE_NET_WM_COLOR_SCHEME=s:' +
- os.path.expanduser('~/.local/share/qubes-kde/red.colors')])
- self.assertAllCalled()
- @unittest.mock.patch('subprocess.check_output')
- def test_001_kde_args_none(self, proc_mock):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- proc_mock.side_effect = [b'']
- args = self.launcher.kde_guid_args(self.app.domains['test-vm'])
- self.assertEqual(args, [])
- self.assertAllCalled()
- def test_010_common_args(self):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'label', None)] = \
- b'0\x00default=False type=label red'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'debug', None)] = \
- b'0\x00default=False type=bool False'
- self.app.expected_calls[
- ('dom0', 'admin.label.Get', 'red', None)] = \
- b'0\x000xff0000'
- self.app.expected_calls[
- ('dom0', 'admin.label.Index', 'red', None)] = \
- b'0\x001'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'rpc-clipboard', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- with unittest.mock.patch.object(self.launcher, 'kde_guid_args') as \
- kde_mock:
- kde_mock.return_value = []
- args = self.launcher.common_guid_args(self.app.domains['test-vm'])
- self.assertEqual(args, [
- '/usr/bin/qubes-guid', '-N', 'test-vm',
- '-c', '0xff0000',
- '-i', '/usr/share/icons/hicolor/128x128/devices/appvm-red.png',
- '-l', '1', '-q'])
- self.assertAllCalled()
- def test_011_common_args_debug(self):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'label', None)] = \
- b'0\x00default=False type=label red'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'debug', None)] = \
- b'0\x00default=False type=bool True'
- self.app.expected_calls[
- ('dom0', 'admin.label.Get', 'red', None)] = \
- b'0\x000xff0000'
- self.app.expected_calls[
- ('dom0', 'admin.label.Index', 'red', None)] = \
- b'0\x001'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'rpc-clipboard', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- with unittest.mock.patch.object(self.launcher, 'kde_guid_args') as \
- kde_mock:
- kde_mock.return_value = []
- args = self.launcher.common_guid_args(self.app.domains['test-vm'])
- self.assertEqual(args, [
- '/usr/bin/qubes-guid', '-N', 'test-vm',
- '-c', '0xff0000',
- '-i', '/usr/share/icons/hicolor/128x128/devices/appvm-red.png',
- '-l', '1', '-v', '-v'])
- self.assertAllCalled()
- def test_012_common_args_rpc_clipboard(self):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'label', None)] = \
- b'0\x00default=False type=label red'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'debug', None)] = \
- b'0\x00default=False type=bool False'
- self.app.expected_calls[
- ('dom0', 'admin.label.Get', 'red', None)] = \
- b'0\x000xff0000'
- self.app.expected_calls[
- ('dom0', 'admin.label.Index', 'red', None)] = \
- b'0\x001'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'rpc-clipboard', None)] = \
- b'0\x001'
- with unittest.mock.patch.object(self.launcher, 'kde_guid_args') as \
- kde_mock:
- kde_mock.return_value = []
- args = self.launcher.common_guid_args(self.app.domains['test-vm'])
- self.assertEqual(args, [
- '/usr/bin/qubes-guid', '-N', 'test-vm',
- '-c', '0xff0000',
- '-i', '/usr/share/icons/hicolor/128x128/devices/appvm-red.png',
- '-l', '1', '-q', '-Q'])
- self.assertAllCalled()
- @unittest.mock.patch('asyncio.create_subprocess_exec')
- def test_020_start_gui_for_vm(self, proc_mock):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 3000'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'virt_mode', None)] = \
- b'0\x00default=False type=str pv'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- with unittest.mock.patch.object(self.launcher,
- 'common_guid_args', lambda vm: []):
- loop.run_until_complete(self.launcher.start_gui_for_vm(
- self.app.domains['test-vm']))
- # common arguments dropped for simplicity
- proc_mock.assert_called_once_with('-d', '3000')
- self.assertAllCalled()
- @unittest.mock.patch('asyncio.create_subprocess_exec')
- def test_021_start_gui_for_vm_hvm(self, proc_mock):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 3000'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'stubdom_xid', None)] = \
- b'0\x00default=False type=int 3001'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'virt_mode', None)] = \
- b'0\x00default=False type=str hvm'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'debug', None)] = \
- b'0\x00default=False type=bool False'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- with unittest.mock.patch.object(self.launcher,
- 'common_guid_args', lambda vm: []):
- loop.run_until_complete(self.launcher.start_gui_for_vm(
- self.app.domains['test-vm']))
- # common arguments dropped for simplicity
- proc_mock.assert_called_once_with('-d', '3000', '-n')
- self.assertAllCalled()
- def test_022_start_gui_for_vm_hvm_stubdom(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 3000'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'stubdom_xid', None)] = \
- b'0\x00default=False type=int 3001'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'virt_mode', None)] = \
- b'0\x00default=False type=str hvm'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'debug', None)] = \
- b'0\x00default=False type=bool False'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- pidfile = tempfile.NamedTemporaryFile()
- pidfile.write(b'1234\n')
- pidfile.flush()
- self.addCleanup(pidfile.close)
- patch_proc = unittest.mock.patch('asyncio.create_subprocess_exec')
- patch_args = unittest.mock.patch.object(self.launcher,
- 'common_guid_args', lambda vm: [])
- patch_pidfile = unittest.mock.patch.object(self.launcher,
- 'guid_pidfile', lambda vm: pidfile.name)
- try:
- mock_proc = patch_proc.start()
- patch_args.start()
- patch_pidfile.start()
- loop.run_until_complete(self.launcher.start_gui_for_vm(
- self.app.domains['test-vm']))
- # common arguments dropped for simplicity
- mock_proc.assert_called_once_with(
- '-d', '3000', '-n', '-K', '1234')
- finally:
- unittest.mock.patch.stopall()
- self.assertAllCalled()
- def test_030_start_gui_for_stubdomain(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 3000'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'stubdom_xid', None)] = \
- b'0\x00default=False type=int 3001'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate', 'gui', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate', 'gui-emulated',
- None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- proc_mock = unittest.mock.Mock()
- with unittest.mock.patch('asyncio.create_subprocess_exec',
- lambda *args: self.mock_coroutine(proc_mock, *args)):
- with unittest.mock.patch.object(self.launcher,
- 'common_guid_args', lambda vm: []):
- loop.run_until_complete(self.launcher.start_gui_for_stubdomain(
- self.app.domains['test-vm']))
- # common arguments dropped for simplicity
- proc_mock.assert_called_once_with('-d', '3001', '-t', '3000')
- self.assertAllCalled()
- def test_031_start_gui_for_stubdomain_forced(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 3000'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'stubdom_xid', None)] = \
- b'0\x00default=False type=int 3001'
- # self.app.expected_calls[
- # ('test-vm', 'admin.vm.feature.CheckWithTemplate', 'gui', None)] = \
- # b'0\x00'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate', 'gui-emulated',
- None)] = \
- b'0\x001'
- proc_mock = unittest.mock.Mock()
- with unittest.mock.patch('asyncio.create_subprocess_exec',
- lambda *args: self.mock_coroutine(proc_mock, *args)):
- with unittest.mock.patch.object(self.launcher,
- 'common_guid_args', lambda vm: []):
- loop.run_until_complete(self.launcher.start_gui_for_stubdomain(
- self.app.domains['test-vm']))
- # common arguments dropped for simplicity
- proc_mock.assert_called_once_with('-d', '3001', '-t', '3000')
- self.assertAllCalled()
- @asyncio.coroutine
- def mock_coroutine(self, mock, *args, **kwargs):
- mock(*args, **kwargs)
- def test_040_start_gui(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate', 'gui', None)] = \
- b'0\x00True'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'virt_mode', None)] = \
- b'0\x00default=False type=str hvm'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 3000'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'stubdom_xid', None)] = \
- b'0\x00default=False type=int 3001'
- vm = self.app.domains['test-vm']
- mock_start_vm = unittest.mock.Mock()
- mock_start_stubdomain = unittest.mock.Mock()
- patch_start_vm = unittest.mock.patch.object(
- self.launcher, 'start_gui_for_vm', functools.partial(
- self.mock_coroutine, mock_start_vm))
- patch_start_stubdomain = unittest.mock.patch.object(
- self.launcher, 'start_gui_for_stubdomain', lambda vm_, force:
- self.mock_coroutine(mock_start_stubdomain, vm_))
- try:
- patch_start_vm.start()
- patch_start_stubdomain.start()
- loop.run_until_complete(self.launcher.start_gui(vm))
- mock_start_vm.assert_called_once_with(vm, monitor_layout=None)
- mock_start_stubdomain.assert_called_once_with(vm)
- finally:
- unittest.mock.patch.stopall()
- def test_041_start_gui_running(self):
- # simulate existing pidfiles, should not start processes
- self.skipTest('todo')
- def test_042_start_gui_pvh(self):
- # PVH - no stubdomain
- self.skipTest('todo')
- @unittest.mock.patch('subprocess.Popen')
- def test_050_get_monitor_layout1(self, proc_mock):
- proc_mock().stdout = b'''Screen 0: minimum 8 x 8, current 1920 x 1200, maximum 32767 x 32767
- HDMI1 connected 1920x1200+0+0 (normal left inverted right x axis y axis) 518mm x 324mm
- 1920x1200 59.95*+
- 1920x1080 60.00 50.00 59.94
- 1920x1080i 60.00 50.00 59.94
- 1600x1200 60.00
- 1680x1050 59.88
- 1280x1024 60.02
- 1440x900 59.90
- 1280x960 60.00
- 1280x720 60.00 50.00 59.94
- 1024x768 60.00
- 800x600 60.32
- 720x576 50.00
- 720x480 60.00 59.94
- 720x480i 60.00 59.94
- 640x480 60.00 59.94
- HDMI2 disconnected (normal left inverted right x axis y axis)
- VGA1 disconnected (normal left inverted right x axis y axis)
- VIRTUAL1 disconnected (normal left inverted right x axis y axis)
- '''.splitlines()
- self.assertEqual(qubesadmin.tools.qvm_start_gui.get_monitor_layout(),
- ['1920 1200 0 0\n'])
- @unittest.mock.patch('subprocess.Popen')
- def test_051_get_monitor_layout_multiple(self, proc_mock):
- proc_mock().stdout = b'''Screen 0: minimum 8 x 8, current 2880 x 1024, maximum 32767 x 32767
- LVDS1 connected 1600x900+0+0 (normal left inverted right x axis y axis)
- VGA1 connected 1280x1024+1600+0 (normal left inverted right x axis y axis)
- '''.splitlines()
- self.assertEqual(qubesadmin.tools.qvm_start_gui.get_monitor_layout(),
- ['1600 900 0 0\n', '1280 1024 1600 0\n'])
- @unittest.mock.patch('subprocess.Popen')
- def test_052_get_monitor_layout_hidpi1(self, proc_mock):
- proc_mock().stdout = b'''Screen 0: minimum 8 x 8, current 1920 x 1200, maximum 32767 x 32767
- HDMI1 connected 2560x1920+0+0 (normal left inverted right x axis y axis) 372mm x 208mm
- 1920x1200 60.00*+
- '''.splitlines()
- dpi = 150
- self.assertEqual(qubesadmin.tools.qvm_start_gui.get_monitor_layout(),
- ['2560 1920 0 0 {} {}\n'.format(
- int(2560/dpi*254/10), int(1920/dpi*254/10))])
- @unittest.mock.patch('subprocess.Popen')
- def test_052_get_monitor_layout_hidpi2(self, proc_mock):
- proc_mock().stdout = b'''Screen 0: minimum 8 x 8, current 1920 x 1200, maximum 32767 x 32767
- HDMI1 connected 2560x1920+0+0 (normal left inverted right x axis y axis) 310mm x 174mm
- 1920x1200 60.00*+
- '''.splitlines()
- dpi = 200
- self.assertEqual(qubesadmin.tools.qvm_start_gui.get_monitor_layout(),
- ['2560 1920 0 0 {} {}\n'.format(
- int(2560/dpi*254/10), int(1920/dpi*254/10))])
- @unittest.mock.patch('subprocess.Popen')
- def test_052_get_monitor_layout_hidpi3(self, proc_mock):
- proc_mock().stdout = b'''Screen 0: minimum 8 x 8, current 1920 x 1200, maximum 32767 x 32767
- HDMI1 connected 2560x1920+0+0 (normal left inverted right x axis y axis) 206mm x 116mm
- 1920x1200 60.00*+
- '''.splitlines()
- dpi = 300
- self.assertEqual(qubesadmin.tools.qvm_start_gui.get_monitor_layout(),
- ['2560 1920 0 0 {} {}\n'.format(
- int(2560/dpi*254/10), int(1920/dpi*254/10))])
- def test_060_send_monitor_layout(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- vm = self.app.domains['test-vm']
- mock_run_service = unittest.mock.Mock(spec={})
- patch_run_service = unittest.mock.patch.object(
- qubesadmin.vm.QubesVM, 'run_service_for_stdio',
- mock_run_service)
- patch_run_service.start()
- self.addCleanup(patch_run_service.stop)
- monitor_layout = ['1920 1080 0 0\n']
- loop.run_until_complete(self.launcher.send_monitor_layout(
- vm, layout=monitor_layout, startup=True))
- mock_run_service.assert_called_once_with(
- 'qubes.SetMonitorLayout', b'1920 1080 0 0\n')
- self.assertAllCalled()
- def test_061_send_monitor_layout_exclude(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'0\x00True'
- vm = self.app.domains['test-vm']
- mock_run_service = unittest.mock.Mock()
- patch_run_service = unittest.mock.patch.object(
- qubesadmin.vm.QubesVM, 'run_service_for_stdio',
- mock_run_service)
- patch_run_service.start()
- self.addCleanup(patch_run_service.stop)
- monitor_layout = ['1920 1080 0 0\n']
- loop.run_until_complete(self.launcher.send_monitor_layout(
- vm, layout=monitor_layout, startup=True))
- self.assertFalse(mock_run_service.called)
- self.assertAllCalled()
- def test_062_send_monitor_layout_not_running(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Halted\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Halted\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- vm = self.app.domains['test-vm']
- mock_run_service = unittest.mock.Mock()
- patch_run_service = unittest.mock.patch.object(
- qubesadmin.vm.QubesVM, 'run_service_for_stdio',
- mock_run_service)
- patch_run_service.start()
- self.addCleanup(patch_run_service.stop)
- monitor_layout = ['1920 1080 0 0\n']
- loop.run_until_complete(self.launcher.send_monitor_layout(
- vm, layout=monitor_layout, startup=True))
- self.assertFalse(mock_run_service.called)
- self.assertAllCalled()
- def test_063_send_monitor_layout_signal_existing(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'xid', None)] = \
- b'0\x00default=False type=int 123'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.property.Get', 'stubdom_xid', None)] = \
- b'0\x00default=False type=int 124'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'no-monitor-layout', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- vm = self.app.domains['test-vm']
- self.addCleanup(unittest.mock.patch.stopall)
- with tempfile.NamedTemporaryFile() as pidfile:
- pidfile.write(b'1234\n')
- pidfile.flush()
- patch_guid_pidfile = unittest.mock.patch.object(
- self.launcher, 'guid_pidfile')
- mock_guid_pidfile = patch_guid_pidfile.start()
- mock_guid_pidfile.return_value = pidfile.name
- mock_kill = unittest.mock.patch('os.kill').start()
- monitor_layout = ['1920 1080 0 0\n']
- loop.run_until_complete(self.launcher.send_monitor_layout(
- vm, layout=monitor_layout, startup=False))
- self.assertEqual(mock_guid_pidfile.mock_calls,
- [unittest.mock.call(123),
- unittest.mock.call(124)])
- self.assertEqual(mock_kill.mock_calls,
- [unittest.mock.call(1234, signal.SIGHUP),
- unittest.mock.call(1234, signal.SIGHUP)])
- self.assertAllCalled()
- def test_070_send_monitor_layout_all(self):
- loop = asyncio.new_event_loop()
- asyncio.set_event_loop(loop)
- self.addCleanup(loop.close)
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n' \
- b'test-vm2 class=AppVM state=Running\n' \
- b'test-vm3 class=AppVM state=Runnig\n' \
- b'test-vm4 class=AppVM state=Halted\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm2', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm2 class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm3', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm3 class=AppVM state=Running\n'
- self.app.expected_calls[
- ('test-vm4', 'admin.vm.List', None, None)] = \
- b'0\x00test-vm4 class=AppVM state=Halted\n'
- self.app.expected_calls[
- ('test-vm', 'admin.vm.feature.CheckWithTemplate',
- 'gui', None)] = \
- b'2\x00QubesFeatureNotFoundError\x00\x00Feature not set\x00'
- self.app.expected_calls[
- ('test-vm2', 'admin.vm.feature.CheckWithTemplate',
- 'gui', None)] = \
- b'0\x00True'
- self.app.expected_calls[
- ('test-vm3', 'admin.vm.feature.CheckWithTemplate',
- 'gui', None)] = \
- b'0\x00'
- vm = self.app.domains['test-vm']
- vm2 = self.app.domains['test-vm2']
- self.addCleanup(unittest.mock.patch.stopall)
- mock_send_monitor_layout = unittest.mock.Mock()
- patch_send_monitor_layout = unittest.mock.patch.object(
- self.launcher, 'send_monitor_layout',
- functools.partial(self.mock_coroutine, mock_send_monitor_layout))
- patch_send_monitor_layout.start()
- monitor_layout = ['1920 1080 0 0\n']
- mock_get_monior_layout = unittest.mock.patch(
- 'qubesadmin.tools.qvm_start_gui.get_monitor_layout').start()
- mock_get_monior_layout.return_value = monitor_layout
- self.launcher.send_monitor_layout_all()
- loop.stop()
- loop.run_forever()
- # test-vm3 not called b/c feature 'gui' set to false
- # test-vm4 not called b/c not running
- self.assertCountEqual(mock_send_monitor_layout.mock_calls,
- [unittest.mock.call(vm, monitor_layout),
- unittest.mock.call(vm2, monitor_layout)])
- mock_get_monior_layout.assert_called_once_with()
- self.assertAllCalled()
|