import os.path

from tempfile import TemporaryDirectory

import IPython.testing.tools as tt

from IPython.utils.syspathcontext import prepended_to_syspath

ext1_content = """

def load_ipython_extension(ip):

print("Running ext1 load")

def unload_ipython_extension(ip):

print("Running ext1 unload")

"""

ext2_content = """

def load_ipython_extension(ip):

print("Running ext2 load")

"""

ext3_content = """

def load_ipython_extension(ip):

ip2 = get_ipython()

print(ip is ip2)

"""

def test_extension_loading():

em = get_ipython().extension_manager

with TemporaryDirectory() as td:

ext1 = os.path.join(td, "ext1.py")

with open(ext1, "w", encoding="utf-8") as f:

f.write(ext1_content)

ext2 = os.path.join(td, "ext2.py")

with open(ext2, "w", encoding="utf-8") as f:

f.write(ext2_content)

with prepended_to_syspath(td):

assert "ext1" not in em.loaded

assert "ext2" not in em.loaded

# Load extension

with tt.AssertPrints("Running ext1 load"):

assert em.load_extension("ext1") is None

assert "ext1" in em.loaded

# Should refuse to load it again

with tt.AssertNotPrints("Running ext1 load"):

assert em.load_extension("ext1") == "already loaded"

# Reload

with tt.AssertPrints("Running ext1 unload"):

with tt.AssertPrints("Running ext1 load", suppress=False):

em.reload_extension("ext1")

# Unload

with tt.AssertPrints("Running ext1 unload"):

assert em.unload_extension("ext1") is None

# Can't unload again

with tt.AssertNotPrints("Running ext1 unload"):

assert em.unload_extension("ext1") == "not loaded"

assert em.unload_extension("ext2") == "not loaded"

# Load extension 2

with tt.AssertPrints("Running ext2 load"):

assert em.load_extension("ext2") is None

# Can't unload this

assert em.unload_extension("ext2") == "no unload function"

# But can reload it

with tt.AssertPrints("Running ext2 load"):

em.reload_extension("ext2")

def test_extension_builtins():

em = get_ipython().extension_manager

with TemporaryDirectory() as td:

ext3 = os.path.join(td, "ext3.py")

with open(ext3, "w", encoding="utf-8") as f:

f.write(ext3_content)

assert "ext3" not in em.loaded

with prepended_to_syspath(td):

# Load extension

with tt.AssertPrints("True"):

assert em.load_extension("ext3") is None

assert "ext3" in em.loaded

def test_non_extension():

em = get_ipython().extension_manager

assert em.load_extension("sys") == "no load function"