Write a first test of getoptlong.rb

Author: BurdetteLamar

Rakefile

@@ -1,3 +1,10 @@
-require "bundler/gem_tasks"
+require 'bundler/gem_tasks'
+require 'rake/testtask'
+
+Rake::TestTask.new(:test) do |t|
+  t.libs << 'test'
+  t.libs << 'lib'
+  t.test_files = FileList['test/test_*.rb']
+end
 
 task :default => :test

test/test_getoptlong.rb

@@ -0,0 +1,163 @@
+require 'test/unit'
+require 'getoptlong'
+
+class TestGetoptLong < Test::Unit::TestCase
+
+  def verify(test_argv, expected_remaining_argv, expected_options)
+    # Save ARGV and replace it with a test ARGV.
+    argv_saved = ARGV
+    ARGV.replace(test_argv)
+    # Define options.
+    opts = GetoptLong.new(
+      ['--xxx', '-x', '--aaa', '-a', GetoptLong::REQUIRED_ARGUMENT],
+      ['--yyy', '-y', '--bbb', '-b', GetoptLong::OPTIONAL_ARGUMENT],
+      ['--zzz', '-z', '--ccc', '-c', GetoptLong::NO_ARGUMENT]
+    )
+    opts.quiet = true
+    # Gather options.
+    actual_options = []
+    opts.each do |opt, arg|
+      actual_options << "#{opt}: #{arg}"
+    end
+    # Save remaining test ARGV and restore original ARGV.
+    actual_remaining_argv = ARGV
+    ARGV.replace(argv_saved)
+    # Assert.
+    assert_equal(expected_remaining_argv, actual_remaining_argv, 'ARGV')
+    assert_equal(expected_options, actual_options, 'Options')
+  end
+
+  def test_no_options
+    expected_options = []
+    expected_argv = %w[foo bar]
+    argv = %w[foo bar]
+    verify(argv, expected_argv, expected_options)
+  end
+
+  def test_required_argument
+    expected_options = [
+      '--xxx: arg'
+    ]
+    expected_argv = %w[foo bar]
+    options = %w[--xxx --xx --x -x --aaa --aa --a -a]
+    options.each do |option|
+      argv = ['foo', option, 'arg', 'bar']
+      verify(argv, expected_argv, expected_options)
+    end
+  end
+
+  def test_required_argument_missing
+    options = %w[--xxx --xx --x -x --aaa --aa --a -a]
+    options.each do |option|
+      argv = [option]
+      e = assert_raise(GetoptLong::MissingArgument) do
+        verify(argv, [], [])
+      end
+      assert_match('requires an argument', e.message)
+    end
+  end
+
+  def test_optional_argument
+    expected_options = [
+      '--yyy: arg'
+    ]
+    expected_argv = %w[foo bar]
+    options = %w[--yyy --y --y -y --bbb --bb --b -b]
+    options.each do |option|
+      argv = ['foo', 'bar', option, 'arg']
+      verify(argv, expected_argv, expected_options)
+    end
+  end
+
+  def test_optional_argument_missing
+    expected_options = [
+      '--yyy: '
+    ]
+    expected_argv = %w[foo bar]
+    options = %w[--yyy --y --y -y --bbb --bb --b -b]
+    options.each do |option|
+      argv = ['foo', 'bar', option]
+      verify(argv, expected_argv, expected_options)
+    end
+  end
+
+  def test_no_argument
+    expected_options = [
+      '--zzz: '
+    ]
+    expected_argv = %w[foo bar]
+    options = %w[--zzz --zz --z -z --ccc --cc --c -c]
+    options.each do |option|
+      argv = ['foo', option, 'bar']
+      verify(argv, expected_argv, expected_options)
+    end
+  end
+
+  def test_new_with_empty_array
+    e = assert_raise(ArgumentError) do
+      GetoptLong.new([])
+    end
+    assert_match(/no argument-flag/, e.message)
+  end
+
+  def test_new_with_bad_array
+    e = assert_raise(ArgumentError) do
+      GetoptLong.new('foo')
+    end
+    assert_match(/option list contains non-Array argument/, e.message)
+  end
+
+  def test_new_with_empty_subarray
+    e = assert_raise(ArgumentError) do
+      GetoptLong.new([[]])
+    end
+    assert_match(/no argument-flag/, e.message)
+  end
+
+  def test_new_with_bad_subarray
+    e = assert_raise(ArgumentError) do
+      GetoptLong.new([1])
+    end
+    assert_match(/no option name/, e.message)
+  end
+
+  def test_new_with_invalid_option
+    invalid_options = %w[verbose -verbose -- +]
+    invalid_options.each do |invalid_option|
+      e = assert_raise(ArgumentError, invalid_option.to_s) do
+        arguments = [
+          [invalid_option, '-v', GetoptLong::NO_ARGUMENT]
+        ]
+        GetoptLong.new(*arguments)
+      end
+      assert_match(/invalid option/, e.message)
+    end
+  end
+
+  def test_new_with_invalid_alias
+    invalid_aliases = %w[v - -- +]
+    invalid_aliases.each do |invalid_alias|
+      e = assert_raise(ArgumentError, invalid_alias.to_s) do
+        arguments = [
+          ['--verbose', invalid_alias, GetoptLong::NO_ARGUMENT]
+        ]
+        GetoptLong.new(*arguments)
+      end
+      assert_match(/invalid option/, e.message)
+    end
+  end
+
+  def test_new_with_invalid_flag
+    invalid_flags = ['foo']
+    invalid_flags.each do |invalid_flag|
+      e = assert_raise(ArgumentError, invalid_flag.to_s) do
+        arguments = [
+          ['--verbose', '-v', invalid_flag]
+        ]
+        GetoptLong.new(*arguments)
+      end
+      assert_match(/no argument-flag/, e.message)
+    end
+  end
+
+end