upstream/ipython Files · IPython/frontend/tests/test_process.py

Move crash handling to the application level and simplify class structure....

Move crash handling to the application level and simplify class structure. Starting to try to take real advantage of the refactoring, to have generic crash handling. This also lets us initialize the app without needing all the self.attempt() wrappers, since now there's a good system-wide crash handler at the app level (not inside the shell instance). I didn't yet remove the attempt() method because we may have occasional uses for it (we still do, but in one place only). I also removed some extra class layers that weren't quite needed. Creating classes solely for the purpose of passing parameters makes the code (IMO) harder to understand, I kept getting lost in parts of the class hierarchy. I think these changes provide the same flexibility but with easier to follow code (less things to remember, basically). What I tried to do was to use argument passing instead of inheritance for all cases I saw where the inheritance wasn't really adding new functionality. In some cases, this actually allowed me to remove methods that were effectively duplicated in the subclasses.

Brian Granger - - Load All Authors

File last commit:

r1960:51f38f50


                r2403:08d6ac78

Download file

             test_process.py
        
                    67 lines
            
             | 1.7 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / IPython / frontend / tests / test_process.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # encoding: utf-8

      """

      Test process execution and IO redirection.

      """

      __docformat__ = "restructuredtext en"

      #-----------------------------------------------------------------------------

      #  Copyright (C) 2008-2009  The IPython Development Team

      #

      #  Distributed under the terms of the BSD License.  The full license is

      #  in the file COPYING, distributed as part of this software.

      #-----------------------------------------------------------------------------

      from cStringIO import StringIO

      from time import sleep

      import sys

      from IPython.frontend.process import PipedProcess

      from IPython.testing import decorators as testdec

      def test_capture_out():

          """ A simple test to see if we can execute a process and get the output.

          """

          s = StringIO()

          p = PipedProcess('echo 1', out_callback=s.write, )

          p.start()

          p.join()

          result = s.getvalue().rstrip()

          assert result == '1'

      def test_io():

          """ Checks that we can send characters on stdin to the process.

          """

          s = StringIO()

          p = PipedProcess(sys.executable + ' -c "a = raw_input(); print a"', 

                                  out_callback=s.write, )

          p.start()

          test_string = '12345\n'

          while not hasattr(p, 'process'):

              sleep(0.1)

          p.process.stdin.write(test_string)

          p.join()

          result = s.getvalue()

          assert result == test_string

      def test_kill():

          """ Check that we can kill a process, and its subprocess.

          """

          s = StringIO()

          p = PipedProcess(sys.executable + ' -c "a = raw_input();"', 

                                  out_callback=s.write, )

          p.start()

          while not hasattr(p, 'process'):

              sleep(0.1)

          p.process.kill() 

          assert p.process.poll() is not None

      if __name__ == '__main__':

          test_capture_out()

          test_io()

          test_kill()

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# encoding: utf-8
				"""
				Test process execution and IO redirection.
				"""

				__docformat__ = "restructuredtext en"

				#-----------------------------------------------------------------------------
				# Copyright (C) 2008-2009 The IPython Development Team
				#
				# Distributed under the terms of the BSD License. The full license is
				# in the file COPYING, distributed as part of this software.
				#-----------------------------------------------------------------------------

				from cStringIO import StringIO
				from time import sleep
				import sys

				from IPython.frontend.process import PipedProcess
				from IPython.testing import decorators as testdec


				def test_capture_out():
				""" A simple test to see if we can execute a process and get the output.
				"""
				s = StringIO()
				p = PipedProcess('echo 1', out_callback=s.write, )
				p.start()
				p.join()
				result = s.getvalue().rstrip()
				assert result == '1'


				def test_io():
				""" Checks that we can send characters on stdin to the process.
				"""
				s = StringIO()
				p = PipedProcess(sys.executable + ' -c "a = raw_input(); print a"',
				out_callback=s.write, )
				p.start()
				test_string = '12345\n'
				while not hasattr(p, 'process'):
				sleep(0.1)
				p.process.stdin.write(test_string)
				p.join()
				result = s.getvalue()
				assert result == test_string


				def test_kill():
				""" Check that we can kill a process, and its subprocess.
				"""
				s = StringIO()
				p = PipedProcess(sys.executable + ' -c "a = raw_input();"',
				out_callback=s.write, )
				p.start()
				while not hasattr(p, 'process'):
				sleep(0.1)
				p.process.kill()
				assert p.process.poll() is not None


				if __name__ == '__main__':
				test_capture_out()
				test_io()
				test_kill()