From 856a1b7af5eacd2c17e8bcc87afa3548229f40f9 2014-11-11 04:43:56
From: Jason Newton <nevion@gmail.com>
Date: 2014-11-11 04:43:56
Subject: [PATCH] add test_nest_embed

    -use pexpect since these tests are interactive
    -skip test_nest_embed on win32

---

diff --git a/IPython/terminal/tests/test_embed.py b/IPython/terminal/tests/test_embed.py
index 70a4508..ec51316 100644
--- a/IPython/terminal/tests/test_embed.py
+++ b/IPython/terminal/tests/test_embed.py
@@ -16,11 +16,13 @@ import sys
 import nose.tools as nt
 from IPython.utils.process import process_handler
 from IPython.utils.tempdir import NamedFileInTemporaryDirectory
+from IPython.testing.decorators import skip_win32
 
 #-----------------------------------------------------------------------------
 # Tests
 #-----------------------------------------------------------------------------
 
+
 _sample_embed = b"""
 from __future__ import print_function
 import IPython
@@ -55,3 +57,69 @@ def test_ipython_embed():
             nt.assert_in('IPython', std)
         nt.assert_in('bye!', std)
 
+@skip_win32
+def test_nest_embed():
+    """test that `IPython.embed()` is nestable"""
+    from IPython.external import pexpect
+    ipy_prompt = r']:' #ansi color codes give problems matching beyond this
+
+
+    child = pexpect.spawn('%s -m IPython'%(sys.executable, ))
+    child.expect(ipy_prompt)
+    child.sendline("from __future__ import print_function")
+    child.expect(ipy_prompt)
+    child.sendline("import IPython")
+    child.expect(ipy_prompt)
+    child.sendline("ip0 = get_ipython()")
+    #enter first nested embed
+    child.sendline("IPython.embed()")
+    #skip the banner until we get to a prompt
+    try:
+        prompted = -1
+        while prompted != 0:
+            prompted = child.expect([ipy_prompt, '\r\n'])
+    except pexpect.TIMEOUT as e:
+        print(e)
+        #child.interact()
+    child.sendline("embed1 = get_ipython()"); child.expect(ipy_prompt)
+    child.sendline("print('true' if embed1 is not ip0 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    child.sendline("print('true' if IPython.get_ipython() is embed1 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    #enter second nested embed
+    child.sendline("IPython.embed()")
+    #skip the banner until we get to a prompt
+    try:
+        prompted = -1
+        while prompted != 0:
+            prompted = child.expect([ipy_prompt, '\r\n'])
+    except pexpect.TIMEOUT as e:
+        print(e)
+        #child.interact()
+    child.sendline("embed2 = get_ipython()"); child.expect(ipy_prompt)
+    child.sendline("print('true' if embed2 is not embed1 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    child.sendline("print('true' if embed2 is IPython.get_ipython() else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    child.sendline('exit')
+    #back at first embed
+    child.expect(ipy_prompt)
+    child.sendline("print('true' if get_ipython() is embed1 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    child.sendline("print('true' if IPython.get_ipython() is embed1 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    child.sendline('exit')
+    #back at launching scope
+    child.expect(ipy_prompt)
+    child.sendline("print('true' if get_ipython() is ip0 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)
+    child.sendline("print('true' if IPython.get_ipython() is ip0 else 'false')")
+    assert(child.expect(['true\r\n', 'false\r\n']) == 0)
+    child.expect(ipy_prompt)