1 files changed, 67 insertions, 0 deletions
diff --git a/tests/test_document.py b/tests/test_document.py
new file mode 100644
index 0000000..ba65ffc
--- /dev/null
+++ b/tests/test_document.py
@@ -0,0 +1,67 @@
+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"
+    assert document.char_before_cursor == "n"
+
+
+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)
+
+
+def test_is_cursor_at_the_end(document):
+    assert Document("hello", 5).is_cursor_at_the_end
+    assert not Document("hello", 4).is_cursor_at_the_end