aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/python/prompt-toolkit/py2/tests/test_document.py
blob: 8c0cf5704c1f4772735dc1b14c527b108e41b749 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
from __future__ import unicode_literals

import pytest

from prompt_toolkit.document import Document


@pytest.fixture
def document():
    return Document(
        'line 1\n' +
        'line 2\n' +
        'line 3\n' +
        'line 4\n',
        len('line 1\n' + 'lin')
    )


def test_current_char(document):
    assert document.current_char == 'e'


def test_text_before_cursor(document):
    assert document.text_before_cursor == 'line 1\nlin'


def test_text_after_cursor(document):
    assert document.text_after_cursor == 'e 2\n' + \
        'line 3\n' + \
        'line 4\n'


def test_lines(document):
    assert document.lines == [
        'line 1',
        'line 2',
        'line 3',
        'line 4', '']


def test_line_count(document):
    assert document.line_count == 5


def test_current_line_before_cursor(document):
    assert document.current_line_before_cursor == 'lin'


def test_current_line_after_cursor(document):
    assert document.current_line_after_cursor == 'e 2'


def test_current_line(document):
    assert document.current_line == 'line 2'


def test_cursor_position(document):
    assert document.cursor_position_row == 1
    assert document.cursor_position_col == 3

    d = Document('', 0)
    assert d.cursor_position_row == 0
    assert d.cursor_position_col == 0


def test_translate_index_to_position(document):
    pos = document.translate_index_to_position(
        len('line 1\nline 2\nlin'))

    assert pos[0] == 2
    assert pos[1] == 3

    pos = document.translate_index_to_position(0)
    assert pos == (0, 0)