add menu default test

1 files changed, 12 insertions, 0 deletions

diff --git a/test/test_menu.rb b/test/test_menu.rb
index b654ef3..402b47d 100644
--- a/test/test_menu.rb
+++ b/test/test_menu.rb
@@ -31,6 +31,18 @@ class TestMenu < Minitest::Test
     assert_equal("Sample2", output)
   end
 
+  def test_default
+    @input << "\n"
+    @input.rewind
+
+    output = @terminal.choose do |menu|
+      menu.choices("Sample1", "Sample2", "Sample3")
+      menu.default = "Sample1"
+    end
+
+    assert_equal("Sample1", output)
+  end
+
   def test_flow
     @input << "Sample1\n"
     @input.rewind