Mercurial > treepkg
view test/test_run.py @ 164:a68a4e22549c
Extend treepkg.run.call so that data can be fed to stdin of the subprocess.
author | Bernhard Herzog <bh@intevation.de> |
---|---|
date | Fri, 20 Jun 2008 14:06:27 +0000 |
parents | e08b4b09d35f |
children | c7ac67366492 |
line wrap: on
line source
# Copyright (C) 2008 by Intevation GmbH # Authors: # Bernhard Herzog <bh@intevation.de> # # This program is free software under the GPL (>=v2) # Read the file COPYING coming with the software for details. """Tests for treepkg.run""" import sys import os import unittest from treepkg.run import call, SubprocessError class TestCall(unittest.TestCase): def test_call_defaults(self): call([sys.executable, "-c", "pass"]) def test_call_error(self): """Test call with a subprocess that exits with a non-zero exit code""" try: call([sys.executable, "-c", "import sys; sys.exit(1)"]) except SubprocessError, exc: self.assertEquals(exc.returncode, 1) else: self.fail("call did not raise an exception") def test_call_extra_env(self): """Test call with the extra_env parameter""" subprocess_cmd = [sys.executable, "-c", "import sys, os;" "value = os.environ.get('TREEPKG_TEST');" "sys.exit(int(value != 'xyzzy'))" ] # sanity check that the extra environment variable we use for # the test is not set yet self.assertRaises(SubprocessError, call, subprocess_cmd) # the actual test try: call(subprocess_cmd, extra_env=dict(TREEPKG_TEST="xyzzy")) except SubprocessError, exc: self.fail("the extra_env variables were not set properly") else: # test OK pass def test_call_inputdata(self): """Test call with inputdata passed to the subprocesses stdin""" # FIXME: If the feature being tested is not implemented # properly, it's likely that the subprocess will wait # indefinitely waiting for input. data = "1" subprocess_cmd = [sys.executable, "-c", "import sys; sys.exit(int(raw_input("")))"] try: call(subprocess_cmd, inputdata=data) except SubprocessError, exc: self.assertEquals(exc.returncode, 1)