add menu default test

JEG2 · Jul 11, 2016 · 813b510 · 813b510
1 parent d789f43
commit 813b510
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/test_menu.rb b/test/test_menu.rb
@@ -31,6 +31,18 @@ def test_choices
     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