mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-04 08:24:40 +01:00
97 lines
2.3 KiB
Python
97 lines
2.3 KiB
Python
"""Tests the xonsh lexer."""
|
|
import os
|
|
|
|
import pytest # noqa F401
|
|
|
|
from xonsh.wizard import (
|
|
FileInserter,
|
|
Message,
|
|
Pass,
|
|
PrettyFormatter,
|
|
Question,
|
|
StateVisitor,
|
|
Wizard,
|
|
)
|
|
|
|
TREE0 = Wizard(children=[Pass(), Message(message="yo")])
|
|
TREE1 = Question("wakka?", {"jawaka": Pass()})
|
|
|
|
|
|
def test_pretty_format_tree0():
|
|
exp = "Wizard(children=[\n" " Pass(),\n" " Message('yo')\n" "])"
|
|
obs = PrettyFormatter(TREE0).visit()
|
|
assert exp == obs
|
|
assert exp == str(TREE0)
|
|
assert exp.replace("\n", "") == repr(TREE0)
|
|
|
|
|
|
def test_pretty_format_tree1():
|
|
exp = (
|
|
"Question(\n"
|
|
" question='wakka?',\n"
|
|
" responses={\n"
|
|
" 'jawaka': Pass()\n"
|
|
" }\n"
|
|
")"
|
|
)
|
|
obs = PrettyFormatter(TREE1).visit()
|
|
assert exp == obs
|
|
assert exp == str(TREE1)
|
|
assert exp.replace("\n", "") == repr(TREE1)
|
|
|
|
|
|
def test_state_visitor_store():
|
|
exp = {"rick": [{}, {}, {"and": "morty"}]}
|
|
sv = StateVisitor()
|
|
sv.store("/rick/2/and", "morty")
|
|
obs = sv.state
|
|
assert exp == obs
|
|
|
|
exp["rick"][1]["mr"] = "meeseeks"
|
|
sv.store("/rick/-2/mr", "meeseeks")
|
|
assert exp == obs
|
|
|
|
flat_exp = {
|
|
"/": {"rick": [{}, {"mr": "meeseeks"}, {"and": "morty"}]},
|
|
"/rick/": [{}, {"mr": "meeseeks"}, {"and": "morty"}],
|
|
"/rick/0/": {},
|
|
"/rick/1/": {"mr": "meeseeks"},
|
|
"/rick/1/mr": "meeseeks",
|
|
"/rick/2/": {"and": "morty"},
|
|
"/rick/2/and": "morty",
|
|
}
|
|
flat_obs = sv.flatten()
|
|
assert flat_exp == flat_obs
|
|
|
|
|
|
def dump_xonfig_env_mock(path, value):
|
|
name = os.path.basename(path.rstrip("/"))
|
|
return f"${name} = {value!r}"
|
|
|
|
|
|
def test_tuple_store_and_write():
|
|
# setup
|
|
sv = StateVisitor()
|
|
sv.store("/env/XONSH_HISTORY_SIZE", (1073741824, "b"))
|
|
dump_rules = {
|
|
"/": None,
|
|
"/env/": None,
|
|
"/env/*": dump_xonfig_env_mock,
|
|
"/env/*/[0-9]*": None,
|
|
}
|
|
fi = FileInserter(
|
|
prefix="# XONSH WIZARD START",
|
|
suffix="# XONSH WIZARD END",
|
|
dump_rules=dump_rules,
|
|
default_file=None,
|
|
check=False,
|
|
ask_filename=False,
|
|
)
|
|
# run test
|
|
exp = (
|
|
"# XONSH WIZARD START\n"
|
|
"$XONSH_HISTORY_SIZE = (1073741824, 'b')\n"
|
|
"# XONSH WIZARD END\n"
|
|
)
|
|
obs = fi.dumps(sv.flatten())
|
|
assert exp == obs
|