xonsh/tests/test_replay.py
Konstantinos Tsakiltzidis aa163fbc0a 'replay/wizard'
2016-06-22 23:20:37 +03:00

59 lines
1.3 KiB
Python

# -*- coding: utf-8 -*-
"""Tests the xonsh replay functionality."""
from __future__ import unicode_literals, print_function
import os
import builtins
from contextlib import contextmanager
import pytest
from xonsh.tools import swap
from xonsh.shell import Shell
from xonsh.replay import Replayer
from tools import ON_MAC, skip_if
SHELL = Shell({'PATH': []})
HISTDIR = os.path.join(os.path.dirname(__file__), 'histories')
def run_replay(re_file):
with swap(builtins, '__xonsh_shell__', SHELL):
r = Replayer(re_file)
hist = r.replay()
return hist
def cleanup_replay(hist):
fname = hist.filename
del hist
if os.path.isfile(fname):
os.remove(fname)
@contextmanager
def a_replay(re_file):
hist = run_replay(re_file)
yield hist
cleanup_replay(hist)
@skip_if(ON_MAC)
def test_echo():
f = os.path.join(HISTDIR, 'echo.json')
with a_replay(f) as hist:
assert 2 == len(hist)
@skip_if(ON_MAC)
def test_reecho():
f = os.path.join(HISTDIR, 'echo.json')
with a_replay(f) as hist:
assert 2 == len(hist)
@skip_if(ON_MAC)
def test_simple_python():
f = os.path.join(HISTDIR, 'simple-python.json')
with a_replay(f) as hist:
assert 4 == len(hist)
assert "print('The Turtles')" == hist.inps[0].strip()