mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 16:34:47 +01:00
79 lines
1.6 KiB
Python
79 lines
1.6 KiB
Python
"""xontrib tests, such as they are"""
|
|
import sys
|
|
import pytest
|
|
from xonsh.xontribs import xontrib_metadata, xontrib_context
|
|
|
|
|
|
def test_load_xontrib_metadata():
|
|
# Simply tests that the xontribs JSON files isn't malformed.
|
|
xontrib_metadata()
|
|
|
|
|
|
@pytest.fixture
|
|
def tmpmod(tmpdir):
|
|
"""
|
|
Same as tmpdir but also adds/removes it to the front of sys.path.
|
|
|
|
Also cleans out any modules loaded as part of the test.
|
|
"""
|
|
sys.path.insert(0, str(tmpdir))
|
|
loadedmods = set(sys.modules.keys())
|
|
try:
|
|
yield tmpdir
|
|
finally:
|
|
del sys.path[0]
|
|
newmods = set(sys.modules.keys()) - loadedmods
|
|
for m in newmods:
|
|
del sys.modules[m]
|
|
|
|
|
|
def test_noall(tmpmod):
|
|
"""
|
|
Tests what get's exported from a module without __all__
|
|
"""
|
|
|
|
with tmpmod.mkdir("xontrib").join("spameggs.py").open("w") as x:
|
|
x.write(
|
|
"""
|
|
spam = 1
|
|
eggs = 2
|
|
_foobar = 3
|
|
"""
|
|
)
|
|
|
|
ctx = xontrib_context("spameggs")
|
|
assert ctx == {"spam": 1, "eggs": 2}
|
|
|
|
|
|
def test_withall(tmpmod):
|
|
"""
|
|
Tests what get's exported from a module with __all__
|
|
"""
|
|
|
|
with tmpmod.mkdir("xontrib").join("spameggs.py").open("w") as x:
|
|
x.write(
|
|
"""
|
|
__all__ = 'spam', '_foobar'
|
|
spam = 1
|
|
eggs = 2
|
|
_foobar = 3
|
|
"""
|
|
)
|
|
|
|
ctx = xontrib_context("spameggs")
|
|
assert ctx == {"spam": 1, "_foobar": 3}
|
|
|
|
|
|
def test_xshxontrib(tmpmod):
|
|
"""
|
|
Test that .xsh xontribs are loadable
|
|
"""
|
|
with tmpmod.mkdir("xontrib").join("script.xsh").open("w") as x:
|
|
x.write(
|
|
"""
|
|
hello = 'world'
|
|
"""
|
|
)
|
|
|
|
ctx = xontrib_context("script")
|
|
assert ctx == {"hello": "world"}
|