"""Test embedding of IPython"""

# -----------------------------------------------------------------------------
#  Copyright (C) 2013 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.
# -----------------------------------------------------------------------------

# -----------------------------------------------------------------------------
# Imports
# -----------------------------------------------------------------------------

import os
import sys
from IPython.testing.decorators import skip_win32
from IPython.testing import IPYTHON_TESTING_TIMEOUT_SCALE

# -----------------------------------------------------------------------------
# Tests
# -----------------------------------------------------------------------------


@skip_win32
def test_debug_magic_passes_through_generators():
    """
    This test that we can correctly pass through frames of a generator post-mortem.
    """
    import pexpect
    import re

    in_prompt = re.compile(rb"In ?\[\d+\]:")
    ipdb_prompt = "ipdb>"
    env = os.environ.copy()
    child = pexpect.spawn(
        sys.executable,
        ["-m", "IPython", "--colors=nocolor", "--simple-prompt"],
        env=env,
    )
    child.timeout = 15 * IPYTHON_TESTING_TIMEOUT_SCALE

    child.expect(in_prompt)

    child.timeout = 2 * IPYTHON_TESTING_TIMEOUT_SCALE

    child.sendline("def f(x):")
    child.sendline("    raise Exception")
    child.sendline("")

    child.expect(in_prompt)
    child.sendline("gen = (f(x) for x in [0])")
    child.sendline("")

    child.expect(in_prompt)
    child.sendline("for x in gen:")
    child.sendline("    pass")
    child.sendline("")

    child.timeout = 10 * IPYTHON_TESTING_TIMEOUT_SCALE

    child.expect("Exception:")

    child.expect(in_prompt)
    child.sendline(r"%debug")
    child.expect("----> 2     raise Exception")

    child.expect(ipdb_prompt)
    child.sendline("u")
    child.expect_exact(r"----> 1 gen = (f(x) for x in [0])")

    child.expect(ipdb_prompt)
    child.sendline("u")
    child.expect_exact("----> 1 for x in gen:")

    child.expect(ipdb_prompt)
    child.sendline("u")
    child.expect_exact(
        "*** all frames above skipped (hidden frames and ignored modules). Use `skip_hidden False` for hidden frames or unignore_module for ignored modules."
    )

    child.expect(ipdb_prompt)
    child.sendline("exit")

    child.expect(in_prompt)
    child.sendline("exit")

    child.close()