From 4f1a3b5f9ad05aa7b08715d48909a2b06ee2fcb1 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Mon, 15 Apr 2024 18:35:31 +0200
Subject: Adding upstream version 3.0.43.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 tests/test_yank_nth_arg.py | 86 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 86 insertions(+)
 create mode 100644 tests/test_yank_nth_arg.py

(limited to 'tests/test_yank_nth_arg.py')

diff --git a/tests/test_yank_nth_arg.py b/tests/test_yank_nth_arg.py
new file mode 100644
index 0000000..7167a26
--- /dev/null
+++ b/tests/test_yank_nth_arg.py
@@ -0,0 +1,86 @@
+from __future__ import annotations
+
+import pytest
+
+from prompt_toolkit.buffer import Buffer
+from prompt_toolkit.history import InMemoryHistory
+
+
+@pytest.fixture
+def _history():
+    "Prefilled history."
+    history = InMemoryHistory()
+    history.append_string("alpha beta gamma delta")
+    history.append_string("one two three four")
+    return history
+
+
+# Test yank_last_arg.
+
+
+def test_empty_history():
+    buf = Buffer()
+    buf.yank_last_arg()
+    assert buf.document.current_line == ""
+
+
+def test_simple_search(_history):
+    buff = Buffer(history=_history)
+    buff.yank_last_arg()
+    assert buff.document.current_line == "four"
+
+
+def test_simple_search_with_quotes(_history):
+    _history.append_string("""one two "three 'x' four"\n""")
+    buff = Buffer(history=_history)
+    buff.yank_last_arg()
+    assert buff.document.current_line == '''"three 'x' four"'''
+
+
+def test_simple_search_with_arg(_history):
+    buff = Buffer(history=_history)
+    buff.yank_last_arg(n=2)
+    assert buff.document.current_line == "three"
+
+
+def test_simple_search_with_arg_out_of_bounds(_history):
+    buff = Buffer(history=_history)
+    buff.yank_last_arg(n=8)
+    assert buff.document.current_line == ""
+
+
+def test_repeated_search(_history):
+    buff = Buffer(history=_history)
+    buff.yank_last_arg()
+    buff.yank_last_arg()
+    assert buff.document.current_line == "delta"
+
+
+def test_repeated_search_with_wraparound(_history):
+    buff = Buffer(history=_history)
+    buff.yank_last_arg()
+    buff.yank_last_arg()
+    buff.yank_last_arg()
+    assert buff.document.current_line == "four"
+
+
+# Test yank_last_arg.
+
+
+def test_yank_nth_arg(_history):
+    buff = Buffer(history=_history)
+    buff.yank_nth_arg()
+    assert buff.document.current_line == "two"
+
+
+def test_repeated_yank_nth_arg(_history):
+    buff = Buffer(history=_history)
+    buff.yank_nth_arg()
+    buff.yank_nth_arg()
+    assert buff.document.current_line == "beta"
+
+
+def test_yank_nth_arg_with_arg(_history):
+    buff = Buffer(history=_history)
+    buff.yank_nth_arg(n=2)
+    assert buff.document.current_line == "three"
-- 
cgit v1.2.3