test/prefs_test.rb in cliutils-2.1.4 vs test/prefs_test.rb in cliutils-2.2.0

@@ -1,13 +1,13 @@
 require 'test_helper'
+require 'tempfile'
 require 'yaml'
 
 require File.join(File.dirname(__FILE__), '..', 'lib/cliutils/ext/hash_extensions')
 require File.join(File.dirname(__FILE__), '..', 'lib/cliutils/prefs')
 require File.join(File.dirname(__FILE__), '..', 'lib/cliutils/prefs/pref')
 
-# Tests for the Prefs class
 class TestPrefs < Test::Unit::TestCase
   def setup
     @prefs_arr = [
       {
         'prompt_text' => 'Batman or Superman?',
@@ -54,10 +54,15 @@
     p = CLIUtils::Prefs.new(@prefs_filepath)
     prefs = YAML::load_file(@prefs_filepath).deep_symbolize_keys
     assert_equal(prefs[:prompts].map { |p| CLIUtils::Pref.new(p) }, p.prompts)
   end
 
+  def test_bad_file_creation
+    exception = assert_raise(RuntimeError) { p = CLIUtils::Prefs.new('asd') }
+    assert_equal('Invalid configuration file: asd', exception.message)
+  end
+
   def test_array_creation
     p = CLIUtils::Prefs.new(@prefs_arr)
     prefs = @prefs_hash.deep_symbolize_keys
     assert_equal(prefs[:prompts].map { |p| CLIUtils::Pref.new(p) }, p.prompts)
   end
@@ -66,10 +71,15 @@
     p = CLIUtils::Prefs.new(@prefs_hash)
     prefs = @prefs_hash.deep_symbolize_keys
     assert_equal(prefs[:prompts].map { |p| CLIUtils::Pref.new(p) }, p.prompts)
   end
 
+  def test_invalid_type_creation
+    exception = assert_raise(RuntimeError) { p = CLIUtils::Prefs.new(123) }
+    assert_equal('Invalid configuration data', exception.message)
+  end
+
   def test_register
     CLIUtils::Prefs.register_action(File.join(File.dirname(__FILE__), 'test_files/test_action.rb'))
     assert_equal(CLIUtils::Prefs.registered_actions.key?(:Test), true)
     assert_equal(CLIUtils::Prefs.registered_actions[:Test][:class], 'TestAction')
     assert_equal(CLIUtils::Prefs.registered_actions[:Test][:path], File.join(File.dirname(__FILE__), 'test_files/test_action.rb'))
@@ -83,10 +93,23 @@
     assert_equal(CLIUtils::Prefs.registered_validators.key?(:Test), true)
     assert_equal(CLIUtils::Prefs.registered_validators[:Test][:class], 'TestValidator')
     assert_equal(CLIUtils::Prefs.registered_validators[:Test][:path], File.join(File.dirname(__FILE__), 'test_files/test_validator.rb'))
   end
 
+  def test_bad_registration
+    m = 'Registration failed because of unknown filepath: bachya.rb'
+
+    exception = assert_raise(RuntimeError) { CLIUtils::Prefs.register_action('bachya.rb') }
+    assert_equal(m, exception.message)
+
+    exception = assert_raise(RuntimeError) { CLIUtils::Prefs.register_behavior('bachya.rb') }
+    assert_equal(m, exception.message)
+
+    exception = assert_raise(RuntimeError) { CLIUtils::Prefs.register_validator('bachya.rb') }
+    assert_equal(m, exception.message)
+  end
+
   def test_deregister
     CLIUtils::Prefs.register_action(File.join(File.dirname(__FILE__), 'test_files/test_action.rb'))
     CLIUtils::Prefs.deregister_action(:Test)
     assert_equal(CLIUtils::Prefs.registered_actions.key?(:Test), false)
 
@@ -95,7 +118,47 @@
     assert_equal(CLIUtils::Prefs.registered_behaviors.key?(:Test), false)
 
     CLIUtils::Prefs.register_validator(File.join(File.dirname(__FILE__), 'test_files/test_validator.rb'))
     CLIUtils::Prefs.deregister_validator(:Test)
     assert_equal(CLIUtils::Prefs.registered_validators.key?(:Test), false)
+  end
+
+  def test_ask
+    p = CLIUtils::Prefs.new(@prefs_filepath)
+    stdin = Tempfile.new("test_readline_stdin")
+    stdout = Tempfile.new("test_readline_stdout")
+    begin
+      stdin.write("\n")
+      stdin.close
+      stdout.close
+      line = replace_stdio(stdin.path, stdout.path) {
+        p.ask
+      }
+    ensure
+      stdin.close(true)
+      stdout.close(true)
+    end
+  end
+
+  private
+
+  def replace_stdio(stdin_path, stdout_path)
+    open(stdin_path, "r"){|stdin|
+      open(stdout_path, "w"){|stdout|
+        orig_stdin = STDIN.dup
+        orig_stdout = STDOUT.dup
+        STDIN.reopen(stdin)
+        STDOUT.reopen(stdout)
+        begin
+          Readline.input = STDIN
+          Readline.output = STDOUT
+          yield
+        ensure
+          STDIN.reopen(orig_stdin)
+          STDOUT.reopen(orig_stdout)
+          orig_stdin.close
+          orig_stdout.close
+        end
+      }
+    }
   end
 end