1 files changed, 51 insertions, 0 deletions
diff --git a/test/reline/key_stroke_test.rb b/test/reline/key_stroke_test.rb
new file mode 100644
index 0000000000..b6d5ce4150
--- /dev/null
+++ b/test/reline/key_stroke_test.rb
@@ -0,0 +1,51 @@
+require_relative 'helper'
+
+class Reline::KeyStroke::Test < Reline::TestCase
+  using Module.new {
+    refine Array do
+      def as_s
+        map(&:chr).join
+      end
+    end
+  }
+
+  def test_input_to!
+    config = {
+      key_mapping: {
+        "a" => "xx",
+        "ab" => "y",
+        "abc" => "z",
+        "x" => "rr"
+      }
+    }
+    stroke = Reline::KeyStroke.new(config)
+    result = ("abzwabk".bytes).map { |char|
+      stroke.input_to!(char)&.then { |result|
+        "#{result.as_s}"
+      }
+    }
+    assert_equal(result, [nil, nil, "yz", "w", nil, nil, "yk"])
+  end
+
+  def test_input_to
+    config = {
+      key_mapping: {
+        "a" => "xx",
+        "ab" => "y",
+        "abc" => "z",
+        "x" => "rr"
+      }
+    }
+    stroke = Reline::KeyStroke.new(config)
+    assert_equal(stroke.input_to("a".bytes)&.as_s, nil)
+    assert_equal(stroke.input_to("ab".bytes)&.as_s, nil)
+    assert_equal(stroke.input_to("abc".bytes)&.as_s, "z")
+    assert_equal(stroke.input_to("abz".bytes)&.as_s, "yz")
+    assert_equal(stroke.input_to("abx".bytes)&.as_s, "yrr")
+    assert_equal(stroke.input_to("ac".bytes)&.as_s, "rrrrc")
+    assert_equal(stroke.input_to("aa".bytes)&.as_s, "rrrrrrrr")
+    assert_equal(stroke.input_to("x".bytes)&.as_s, "rr")
+    assert_equal(stroke.input_to("m".bytes)&.as_s, "m")
+    assert_equal(stroke.input_to("abzwabk".bytes)&.as_s, "yzwabk")
+  end
+end