123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303 |
- # pylint: disable=protected-access,pointless-statement
- #
- # The Qubes OS Project, https://www.qubesmgmt-os.org/
- #
- # Copyright (C) 2015 Joanna Rutkowska <joanna@invisiblethingslab.com>
- # Copyright (C) 2015 Wojtek Porczyk <woju@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 unittest
- import qubesadmin
- import qubesadmin.vm
- import qubesadmin.tools.qvm_ls
- import qubesadmin.tests
- import qubesadmin.tests.tools
- from qubesadmin.tests import TestVM, TestVMCollection
- class TestApp(object):
- def __init__(self):
- self.domains = TestVMCollection(
- [
- ('dom0', TestVM('dom0')),
- ('test-vm', TestVM('test-vm')),
- ]
- )
- class TC_00_Column(qubesadmin.tests.QubesTestCase):
- def test_100_init(self):
- try:
- testcolumn = qubesadmin.tools.qvm_ls.Column('TESTCOLUMN')
- self.assertEqual(testcolumn.ls_head, 'TESTCOLUMN')
- finally:
- try:
- qubesadmin.tools.qvm_ls.Column.columns['TESTCOLUMN']
- except KeyError:
- pass
- class TC_10_globals(qubesadmin.tests.QubesTestCase):
- def test_100_simple_flag(self):
- flag = qubesadmin.tools.qvm_ls.simple_flag(1, 'T', 'internal')
- # TODO after serious testing of QubesVM and Qubes app, this should be
- # using normal components
- vm = TestVM('test-vm', internal=False)
- self.assertFalse(flag(None, vm))
- vm.internal = True
- self.assertTrue(flag(None, vm))
- @unittest.skip('column list generated dynamically')
- def test_900_formats_columns(self):
- for fmt in qubesadmin.tools.qvm_ls.formats:
- for col in qubesadmin.tools.qvm_ls.formats[fmt]:
- self.assertIn(col.upper(), qubesadmin.tools.qvm_ls.Column.columns)
- class TC_50_List(qubesadmin.tests.QubesTestCase):
- def test_100_list_with_status(self):
- app = TestApp()
- app.domains['test-vm'].internal = False
- app.domains['test-vm'].updateable = False
- app.domains['test-vm'].template = TestVM('template')
- app.domains['test-vm'].netvm = TestVM('sys-net')
- app.domains['test-vm'].label = 'green'
- app.domains['dom0'].label = 'black'
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main([], app=app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'dom0 Running TestVM black - -\n'
- 'test-vm Running TestVM green template sys-net\n')
- def test_101_list_with_underscore(self):
- app = TestApp()
- app.domains['test-vm'].virt_mode = 'pv'
- app.domains['test-vm'].label = 'green'
- app.domains['dom0'].label = 'black'
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['-O', 'name,virt_mode,class'], app=app)
- self.assertEqual(stdout.getvalue(),
- 'NAME VIRT-MODE CLASS\n'
- 'dom0 - TestVM\n'
- 'test-vm pv TestVM\n')
- def test_102_list_selected(self):
- app = TestApp()
- app.domains['test-vm'].internal = False
- app.domains['test-vm'].updateable = False
- app.domains['test-vm'].template = TestVM('template')
- app.domains['test-vm'].netvm = TestVM('sys-net')
- app.domains['test-vm'].label = 'green'
- app.domains['dom0'].label = 'black'
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['test-vm'], app=app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'test-vm Running TestVM green template sys-net\n')
- def test_102_raw_list(self):
- app = TestApp()
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--raw-list'], app=app)
- self.assertEqual(stdout.getvalue(),
- 'dom0\n'
- 'test-vm\n')
- class TC_70_Tags(qubesadmin.tests.QubesTestCase):
- def setUp(self):
- self.app = TestApp()
- self.app.domains = TestVMCollection(
- [
- (
- 'dom0',
- TestVM(
- 'dom0',
- tags=['my'],
- label='black'
- )
- ),
- (
- 'test-vm',
- TestVM(
- 'test-vm',
- tags=['not-my', 'other'],
- label='red',
- netvm=TestVM('sys-firewall'),
- template=TestVM('template')
- )
- ),
- ]
- )
- def test_100_tag(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--tags', 'my'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'dom0 Running TestVM black - -\n')
- def test_100_tag_nomatch(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--tags', 'nx'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n')
- def test_100_tags(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--tags', 'my', 'other'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'dom0 Running TestVM black - -\n'
- 'test-vm Running TestVM red template sys-firewall\n')
- def test_100_tags_nomatch(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--tags', 'nx1', 'nx2'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n')
- class TC_80_Power_state_filters(qubesadmin.tests.QubesTestCase):
- def setUp(self):
- self.app = TestApp()
- self.app.domains = TestVMCollection(
- [
- ('a', TestVM('a', power_state='Halted')),
- ('b', TestVM('b', power_state='Transient')),
- ('c', TestVM('c', power_state='Running'))
- ]
- )
- def test_100_nofilter(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main([], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'a Halted TestVM - - -\n'
- 'b Transient TestVM - - -\n'
- 'c Running TestVM - - -\n')
- def test_100_running(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--running'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'c Running TestVM - - -\n')
- def test_100_running_or_halted(self):
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['--running', '--halted'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'a Halted TestVM - - -\n'
- 'c Running TestVM - - -\n')
- class TC_90_List_with_qubesd_calls(qubesadmin.tests.QubesTestCase):
- def test_100_list_with_status(self):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00vm1 class=AppVM state=Running\n' \
- b'template1 class=TemplateVM state=Halted\n' \
- b'sys-net class=AppVM state=Running\n'
- self.app.expected_calls[
- ('vm1', 'admin.vm.List', None, None)] = \
- b'0\x00vm1 class=AppVM state=Running\n'
- self.app.expected_calls[
- ('sys-net', 'admin.vm.List', None, None)] = \
- b'0\x00sys-net class=AppVM state=Running\n'
- self.app.expected_calls[
- ('template1', 'admin.vm.List', None, None)] = \
- b'0\x00template1 class=TemplateVM state=Halted\n'
- props = {
- 'label': b'type=label green',
- 'template': b'type=vm template1',
- 'netvm': b'type=vm sys-net',
- # 'virt_mode': b'type=str pv',
- }
- for key, value in props.items():
- self.app.expected_calls[
- ('vm1', 'admin.vm.property.Get', key, None)] = \
- b'0\x00default=True ' + value
- # setup template1
- props['label'] = b'type=label black'
- for key, value in props.items():
- self.app.expected_calls[
- ('template1', 'admin.vm.property.Get', key, None)] = \
- b'0\x00default=True ' + value
- self.app.expected_calls[
- ('template1', 'admin.vm.property.Get', 'template', None)] = \
- b'' # request refused - no such property
- # setup sys-net
- props['label'] = b'type=label red'
- for key, value in props.items():
- self.app.expected_calls[
- ('sys-net', 'admin.vm.property.Get', key, None)] = \
- b'0\x00default=True ' + value
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main([], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'sys-net Running AppVM red template1 sys-net\n'
- 'template1 Halted TemplateVM black - sys-net\n'
- 'vm1 Running AppVM green template1 sys-net\n')
- self.assertAllCalled()
- def test_101_list_selected(self):
- self.app.expected_calls[
- ('dom0', 'admin.vm.List', None, None)] = \
- b'0\x00vm1 class=AppVM state=Running\n' \
- b'template1 class=TemplateVM state=Halted\n' \
- b'sys-net class=AppVM state=Running\n'
- self.app.expected_calls[
- ('vm1', 'admin.vm.List', None, None)] = \
- b'0\x00vm1 class=AppVM state=Running\n'
- self.app.expected_calls[
- ('sys-net', 'admin.vm.List', None, None)] = \
- b'0\x00sys-net class=AppVM state=Running\n'
- props = {
- 'label': b'type=label green',
- 'template': b'type=vm template1',
- 'netvm': b'type=vm sys-net',
- # 'virt_mode': b'type=str pv',
- }
- for key, value in props.items():
- self.app.expected_calls[
- ('vm1', 'admin.vm.property.Get', key, None)] = \
- b'0\x00default=True ' + value
- # setup sys-net
- props['label'] = b'type=label red'
- for key, value in props.items():
- self.app.expected_calls[
- ('sys-net', 'admin.vm.property.Get', key, None)] = \
- b'0\x00default=True ' + value
- with qubesadmin.tests.tools.StdoutBuffer() as stdout:
- qubesadmin.tools.qvm_ls.main(['vm1', 'sys-net'], app=self.app)
- self.assertEqual(stdout.getvalue(),
- 'NAME STATE CLASS LABEL TEMPLATE NETVM\n'
- 'sys-net Running AppVM red template1 sys-net\n'
- 'vm1 Running AppVM green template1 sys-net\n')
- self.assertAllCalled()
|