test_readline.py

import os
import pty
import sys

import pytest
from pyrepl.readline import _ReadlineWrapper


pytest_plugins = ["pytester"]


@pytest.fixture
def readline_wrapper():
    master, slave = pty.openpty()
    return _ReadlineWrapper(slave, slave)


if sys.version_info < (3, ):
    bytes_type = str
    unicode_type = unicode  # noqa: F821
else:
    bytes_type = bytes
    unicode_type = str


def test_readline():
    master, slave = pty.openpty()
    readline_wrapper = _ReadlineWrapper(slave, slave)
    os.write(master, b'input\n')

    result = readline_wrapper.get_reader().readline()
    assert result == b'input'
    assert isinstance(result, bytes_type)


def test_readline_returns_unicode():
    master, slave = pty.openpty()
    readline_wrapper = _ReadlineWrapper(slave, slave)
    os.write(master, b'input\n')

    result = readline_wrapper.get_reader().readline(returns_unicode=True)
    assert result == 'input'
    assert isinstance(result, unicode_type)


def test_raw_input():
    master, slave = pty.openpty()
    readline_wrapper = _ReadlineWrapper(slave, slave)
    os.write(master, b'input\n')

    result = readline_wrapper.raw_input('prompt:')
    if sys.version_info < (3, ):
        assert result == b'input'
        assert isinstance(result, bytes_type)
    else:
        assert result == 'input'
        assert isinstance(result, unicode_type)


def test_read_history_file(readline_wrapper, tmp_path):
    histfile = tmp_path / "history"
    histfile.touch()

    assert readline_wrapper.reader is None

    readline_wrapper.read_history_file(str(histfile))
    assert readline_wrapper.reader.history == []

    histfile.write_bytes(b"foo\nbar\n")
    readline_wrapper.read_history_file(str(histfile))
    assert readline_wrapper.reader.history == ["foo", "bar"]


def test_write_history_file(readline_wrapper, tmp_path):
    histfile = tmp_path / "history"

    reader = readline_wrapper.get_reader()
    history = reader.history
    assert history == []
    history.extend(["foo", "bar"])

    readline_wrapper.write_history_file(str(histfile))

    assert open(str(histfile), "r").readlines() == ["foo\n", "bar\n"]


def test_write_history_file_with_exception(readline_wrapper, tmp_path):
    """The history file should not get nuked on inner exceptions.

    This was the case with unicode decoding previously."""
    histfile = tmp_path / "history"
    histfile.write_bytes(b"foo\nbar\n")

    class BadEntryException(Exception):
        pass

    class BadEntry(object):
        @classmethod
        def replace(cls, *args):
            raise BadEntryException

    history = readline_wrapper.get_reader().history
    history.extend([BadEntry])

    with pytest.raises(BadEntryException):
        readline_wrapper.write_history_file(str(histfile))

    assert open(str(histfile), "r").readlines() == ["foo\n", "bar\n"]


@pytest.mark.parametrize("use_pyrepl", (True, False))
def test_uses_current_stdin(use_pyrepl, testdir):
    p1 = testdir.makepyfile("""
        import io, os, sys

        assert os.isatty(0)
        dup_stdin = io.FileIO(os.dup(0))
        assert os.isatty(dup_stdin.fileno())

        old_stdin = sys.stdin
        sys.stdin = dup_stdin

        if {use_pyrepl!r}:
            import pyrepl.readline

        print("input1:", input())
        dup_stdin.close()

        sys.stdin = old_stdin
        print("input2:", input())
    """.format(use_pyrepl=use_pyrepl))

    child = testdir.spawn(sys.executable + " " + str(p1), expect_timeout=1)
    child.sendline("line1")
    if use_pyrepl:
        child.expect_exact("input1: line1\r\n")
    else:
        child.expect_exact("input1: b'line1'\r\n")
    child.sendline("line2")
    if use_pyrepl:
        child.expect_exact("input2: line2\r\n")
    else:
        child.expect_exact(b'line2\r\ninput2: line2\r\n')