Export the current nth value as $FZF_NTH

1 files changed, 26 insertions, 7 deletions

diff --git a/test/test_go.rb b/test/test_go.rb
index 2b5e0a24..f7c5b962 100755
--- a/test/test_go.rb
+++ b/test/test_go.rb
@@ -3729,19 +3729,38 @@ class TestGoFZF < TestBase
       *[''] * 1000
     ]
     writelines(input)
-    tmux.send_keys %(#{FZF} -qfoo -n1 --bind 'space:change-nth:2|3|4|5|' < #{tempname}), :Enter
+    nths = '1,2..4,-1,-3..,..2'
+    tmux.send_keys %(#{FZF} -qfoo -n#{nths} --bind 'space:change-nth(2|3|4|5|),result:transform-prompt:echo "[$FZF_NTH] "' < #{tempname}), :Enter
 
-    tmux.until { |lines| assert_equal 4, lines.match_count }
+    tmux.until do |lines|
+      assert lines.any_include?("[#{nths}] foo")
+      assert_equal 4, lines.match_count
+    end
     tmux.send_keys :Space
-    tmux.until { |lines| assert_equal 3, lines.match_count }
+    tmux.until do |lines|
+      assert lines.any_include?('[2] foo')
+      assert_equal 3, lines.match_count
+    end
     tmux.send_keys :Space
-    tmux.until { |lines| assert_equal 2, lines.match_count }
+    tmux.until do |lines|
+      assert lines.any_include?('[3] foo')
+      assert_equal 2, lines.match_count
+    end
     tmux.send_keys :Space
-    tmux.until { |lines| assert_equal 1, lines.match_count }
+    tmux.until do |lines|
+      assert lines.any_include?('[4] foo')
+      assert_equal 1, lines.match_count
+    end
     tmux.send_keys :Space
-    tmux.until { |lines| assert_equal 0, lines.match_count }
+    tmux.until do |lines|
+      assert lines.any_include?('[5] foo')
+      assert_equal 0, lines.match_count
+    end
     tmux.send_keys :Space
-    tmux.until { |lines| assert_equal 4, lines.match_count }
+    tmux.until do |lines|
+      assert lines.any_include?("[#{nths}] foo")
+      assert_equal 4, lines.match_count
+    end
   end
 end