mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 16:34:47 +01:00
68 lines
1.6 KiB
Python
68 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.yield_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'}
|