123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238 |
- # -*- 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 io
- import unittest.mock
- import subprocess
- import sys
- import qubesadmin.tests
- import qubesadmin.tools.qvm_run
- class TC_00_qvm_run(qubesadmin.tests.QubesTestCase):
- def setUp(self):
- if sys.stdout is not sys.__stdout__ or \
- sys.stderr is not sys.__stderr__:
- self.skipTest('qvm-run change behavior on redirected stdout/stderr')
- super(TC_00_qvm_run, self).setUp()
- def test_000_run_single(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- ret = qubesadmin.tools.qvm_run.main(['test-vm', 'command'], app=self.app)
- self.assertEqual(ret, 0)
- # make sure we have the same instance below
- null = self.app.service_calls[0][2]['stdout']
- self.assertIsInstance(null, io.BufferedWriter)
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': None,
- 'stdout': null,
- 'stderr': null,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\n')
- ])
- self.assertAllCalled()
- def test_001_run_multiple(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n' \
- b'test-vm2 class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- ret = qubesadmin.tools.qvm_run.main(['test-vm', 'test-vm2', 'command'],
- app=self.app)
- self.assertEqual(ret, 0)
- for i in range(0, len(self.app.service_calls), 2):
- self.assertIsInstance(self.app.service_calls[i][2]['stdout'],
- io.BufferedWriter)
- self.assertIsInstance(self.app.service_calls[i][2]['stderr'],
- io.BufferedWriter)
- # make sure we have the same instance below
- null = self.app.service_calls[0][2]['stdout']
- null2 = self.app.service_calls[2][2]['stdout']
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': None,
- 'stdout': null,
- 'stderr': null,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\n'),
- ('test-vm2', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': None,
- 'stdout': null2,
- 'stderr': null2,
- 'user': None,
- }),
- ('test-vm2', 'qubes.VMShell', b'command\n')
- ])
- self.assertAllCalled()
- def test_002_passio(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- echo = subprocess.Popen(['echo', 'some-data'], stdout=subprocess.PIPE)
- with unittest.mock.patch('sys.stdin', echo.stdout):
- ret = qubesadmin.tools.qvm_run.main(
- ['--pass-io', 'test-vm', 'command'],
- app=self.app)
- self.assertEqual(ret, 0)
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': None,
- 'stdout': None,
- 'stderr': None,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\nsome-data\n')
- ])
- self.assertAllCalled()
- def test_002_color_output(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- stdout = io.StringIO()
- echo = subprocess.Popen(['echo', 'some-data'], stdout=subprocess.PIPE)
- with unittest.mock.patch('sys.stdin', echo.stdout):
- with unittest.mock.patch('sys.stdout', stdout):
- ret = qubesadmin.tools.qvm_run.main(
- ['--pass-io', 'test-vm', 'command'],
- app=self.app)
- self.assertEqual(ret, 0)
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': None,
- 'stdout': None,
- 'stderr': None,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\nsome-data\n')
- ])
- self.assertEqual(stdout.getvalue(), '\033[0;31m\033[0m')
- stdout.close()
- self.assertAllCalled()
- def test_003_no_color_output(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- stdout = io.StringIO()
- echo = subprocess.Popen(['echo', 'some-data'], stdout=subprocess.PIPE)
- with unittest.mock.patch('sys.stdin', echo.stdout):
- with unittest.mock.patch('sys.stdout', stdout):
- ret = qubesadmin.tools.qvm_run.main(
- ['--pass-io', '--no-color-output', 'test-vm', 'command'],
- app=self.app)
- self.assertEqual(ret, 0)
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': None,
- 'stdout': None,
- 'stderr': None,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\nsome-data\n')
- ])
- self.assertEqual(stdout.getvalue(), '')
- stdout.close()
- self.assertAllCalled()
- def test_004_no_filter_esc(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- stdout = io.StringIO()
- echo = subprocess.Popen(['echo', 'some-data'], stdout=subprocess.PIPE)
- with unittest.mock.patch('sys.stdin', echo.stdout):
- with unittest.mock.patch('sys.stdout', stdout):
- ret = qubesadmin.tools.qvm_run.main(
- ['--pass-io', '--no-filter-esc', 'test-vm', 'command'],
- app=self.app)
- self.assertEqual(ret, 0)
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': False,
- 'localcmd': None,
- 'stdout': None,
- 'stderr': None,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\nsome-data\n')
- ])
- self.assertEqual(stdout.getvalue(), '')
- stdout.close()
- self.assertAllCalled()
- def test_005_localcmd(self):
- self.app.expected_calls[
- ('dom0', 'mgmt.vm.List', None, None)] = \
- b'0\x00test-vm class=AppVM state=Running\n'
- # self.app.expected_calls[
- # ('test-vm', 'mgmt.vm.List', None, None)] = \
- # b'0\x00test-vm class=AppVM state=Running\n'
- ret = qubesadmin.tools.qvm_run.main(
- ['--pass-io', '--localcmd', 'local-command',
- 'test-vm', 'command'],
- app=self.app)
- self.assertEqual(ret, 0)
- self.assertEqual(self.app.service_calls, [
- ('test-vm', 'qubes.VMShell', {
- 'filter_esc': True,
- 'localcmd': 'local-command',
- 'stdout': None,
- 'stderr': None,
- 'user': None,
- }),
- ('test-vm', 'qubes.VMShell', b'command\n')
- ])
- self.assertAllCalled()
|