win32-process 0.7.2 → 0.7.3

data/test/test_win32_process.rb

@@ -1,319 +1,353 @@
-###############################################################################
-# test_win32_process.rb
-#
-# Test suite for the win32-process library.  This test suite will start
-# at least two instances of Notepad on your system, which will then
-# be killed. Requires the sys-proctable library.
-#
-# I haven't added a lot of test cases for fork/wait because it's difficult
-# to run such tests without causing havoc with Test::Unit itself.  Ideas
-# welcome.
-#
-# You should run this test case via the 'rake test' task.
-###############################################################################
-require 'test-unit'
-require 'win32/process'
-
-class TC_Win32Process < Test::Unit::TestCase
-  def self.startup
-    @@pids  = []
-    @@jruby = RUBY_PLATFORM == 'java'
-  end
-
-  def setup
-    @priority = Process::BELOW_NORMAL_PRIORITY_CLASS
-  end
-
-  test "win32-process version is set to the correct value" do
-    assert_equal('0.7.2', Process::WIN32_PROCESS_VERSION)
-  end
-
-  test "create basic functionality" do
-    assert_respond_to(Process, :create)
-  end
-
-  test "create with common flags works as expected" do
-    assert_nothing_raised{
-      @@pids << Process.create(
-         :app_name         => "notepad.exe",
-         :creation_flags   => Process::DETACHED_PROCESS,
-         :process_inherit  => false,
-         :thread_inherit   => true,
-         :cwd              => "C:\\"
-      ).process_id
-    }
-
-    assert_nothing_raised{ Process.kill(9, @@pids.pop) }
-  end
-
-  test "create requires a hash argument" do
-    assert_raise(TypeError){ Process.create("bogusapp.exe") }
-  end
-
-  test "create does not accept invalid keys" do
-    assert_raise(ArgumentError){ Process.create(:bogus => 'test.exe') }
-    assert_raise_message("invalid key 'bogus'"){
-      Process.create(:bogus => 'test.exe')
-    }
-  end
-
-  test "create does not accept invalid startup_info keys" do
-    assert_raise(ArgumentError){
-      Process.create(:startup_info => {:foo => 'test'})
-    }
-    assert_raise_message("invalid startup_info key 'foo'"){
-      Process.create(:startup_info => {:foo => 'test'})
-    }
-  end
-
-  test "create raises an error if the executable cannot be found" do
-    assert_raise(Errno::ENOENT){ Process.create(:app_name => "bogusapp.exe") }
-  end
-
-  test "create passes local environment when environment is not specified" do
-    omit_if(@@jruby)
-    stdout_read, stdout_write = IO.pipe
-
-    ENV['AARDVARK'] = 'B'
-    assert_nothing_raised {
-      Process.create(
-        :app_name         => 'cmd.exe /c echo %AARDVARK%',
-        :creation_flags   => Process::DETACHED_PROCESS,
-        :startup_info     => { :stdout => stdout_write }
-      )
-    }
-
-    stdout_write.close
-    assert_equal('B', stdout_read.read.chomp)
-  end
-
-  test "create does not pass local environment when environment is specified" do
-    omit_if(@@jruby)
-    stdout_read, stdout_write = IO.pipe
-
-    ENV['AARDVARK'] = 'B'
-    assert_nothing_raised {
-      Process.create(
-        :app_name         => 'cmd.exe /c echo %AARDVARK%',
-        :creation_flags   => Process::DETACHED_PROCESS,
-        :environment      => "",
-        :startup_info     => { :stdout => stdout_write }
-      )
-    }
-
-    stdout_write.close
-    assert_equal('%AARDVARK%', stdout_read.read.chomp)
-  end
-
-  test "create supports :environment as a string" do
-    omit_if(@@jruby)
-    stdout_read, stdout_write = IO.pipe
-
-    assert_nothing_raised {
-      Process.create(
-        :app_name         => 'cmd.exe /c echo %A% %C%',
-        :creation_flags   => Process::DETACHED_PROCESS,
-        :environment      => "A=B;C=D",
-        :startup_info     => { :stdout => stdout_write }
-      )
-    }
-
-    stdout_write.close
-    assert_equal("B D", stdout_read.read.chomp)
-  end
-
-  test "create supports :environment as an array" do
-    omit_if(@@jruby)
-    stdout_read, stdout_write = IO.pipe
-
-    assert_nothing_raised {
-      Process.create(
-        :app_name         => 'cmd.exe /c echo %A% %C%',
-        :creation_flags   => Process::DETACHED_PROCESS,
-        :environment      => [ "A=B;X;", "C=;D;Y" ],
-        :startup_info     => { :stdout => stdout_write }
-      )
-    }
-
-    stdout_write.close
-    assert_equal("B;X; ;D;Y", stdout_read.read.chomp)
-  end
-
-  test "create supports empty :environment string" do
-    assert_nothing_raised {
-      Process.create(
-        :creation_flags => Process::DETACHED_PROCESS,
-        :app_name       => 'cmd.exe',
-        :environment    => ''
-      )
-    }
-  end
-
-  test "create supports empty :environment array" do
-    assert_nothing_raised {
-      Process.create(
-        :creation_flags   => Process::DETACHED_PROCESS,
-        :app_name    => 'cmd.exe',
-        :environment => []
-      )
-    }
-  end
-
-  test "uid basic functionality" do
-    assert_respond_to(Process, :uid)
-    assert_kind_of(Fixnum, Process.uid)
-  end
-
-  test "uid accepts a boolean argument" do
-    assert_nothing_raised{ Process.uid(true) }
-    assert_nothing_raised{ Process.uid(true) }
-  end
-
-  test "uid returns a string if its argument is true" do
-    assert_kind_of(String, Process.uid(true))
-  end
-
-  test "uid accepts a maximum of one argument" do
-    assert_raise(ArgumentError){ Process.uid(true, true) }
-  end
-
-  test "argument to uid must be a boolean" do
-    assert_raise(TypeError){ Process.uid('test') }
-  end
-
-  test "getpriority basic functionality" do
-    assert_respond_to(Process, :getpriority)
-    assert_nothing_raised{ Process.getpriority(Process::PRIO_PROCESS, Process.pid) }
-    assert_kind_of(Fixnum, Process.getpriority(Process::PRIO_PROCESS, Process.pid))
-  end
-
-  test "getpriority treats an int argument of zero as the current process" do
-    assert_nothing_raised{ Process.getpriority(0, 0) }
-  end
-
-  test "getpriority requires both a kind and an int" do
-    assert_raise(ArgumentError){ Process.getpriority }
-    assert_raise(ArgumentError){ Process.getpriority(Process::PRIO_PROCESS) }
-  end
-
-  test "getpriority requires integer arguments" do
-    assert_raise(TypeError){ Process.getpriority('test', 0) }
-    assert_raise(TypeError){ Process.getpriority(Process::PRIO_PROCESS, 'test') }
-  end
-
-  test "setpriority basic functionality" do
-    assert_respond_to(Process, :setpriority)
-    assert_nothing_raised{ Process.setpriority(0, Process.pid, @priority) }
-    assert_equal(@priority, Process.getpriority(0, Process.pid))
-  end
-
-  test "setpriority returns zero on success" do
-    assert_equal(0, Process.setpriority(0, Process.pid, @priority))
-  end
-
-  test "setpriority treats an int argument of zero as the current process" do
-    assert_equal(0, Process.setpriority(0, 0, @priority))
-  end
-
-  test "setpriority requires at least three arguments" do
-    assert_raise(ArgumentError){ Process.setpriority }
-    assert_raise(ArgumentError){ Process.setpriority(0) }
-    assert_raise(ArgumentError){ Process.setpriority(0, 0) }
-  end
-
-  test "arguments to setpriority must be numeric" do
-    assert_raise(TypeError){ Process.setpriority('test', 0, @priority) }
-    assert_raise(TypeError){ Process.setpriority(0, 'test', @priority) }
-    assert_raise(TypeError){ Process.setpriority(0, 0, 'test') }
-  end
-
-  test "custom creation constants are defined" do
-    assert_not_nil(Process::CREATE_DEFAULT_ERROR_MODE)
-    assert_not_nil(Process::CREATE_NEW_CONSOLE)
-    assert_not_nil(Process::CREATE_NEW_PROCESS_GROUP)
-    assert_not_nil(Process::CREATE_NO_WINDOW)
-    assert_not_nil(Process::CREATE_SEPARATE_WOW_VDM)
-    assert_not_nil(Process::CREATE_SHARED_WOW_VDM)
-    assert_not_nil(Process::CREATE_SUSPENDED)
-    assert_not_nil(Process::CREATE_UNICODE_ENVIRONMENT)
-    assert_not_nil(Process::DEBUG_ONLY_THIS_PROCESS)
-    assert_not_nil(Process::DEBUG_PROCESS)
-    assert_not_nil(Process::DETACHED_PROCESS)
-  end
-
-  test "getrlimit basic functionality" do
-    assert_respond_to(Process, :getrlimit)
-    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
-    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_FSIZE) }
-  end
-
-  test "getrlimit returns an array of two numeric elements" do
-    assert_kind_of(Array, Process.getrlimit(Process::RLIMIT_CPU))
-    assert_equal(2, Process.getrlimit(Process::RLIMIT_CPU).length)
-    assert_kind_of(Integer, Process.getrlimit(Process::RLIMIT_CPU).first)
-  end
-
-  test "getrlimit can be called multiple times without issue" do
-    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
-    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
-    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
-  end
-
-  test "getrlimit requires a valid resource value" do
-    assert_raise(ArgumentError){ Process.getrlimit(9999) }
-  end
-
-  test "setrlimit basic functionality" do
-    assert_respond_to(Process, :setrlimit)
-    assert_nothing_raised{ Process.setrlimit(Process::RLIMIT_CPU, 1000000) }
-  end
-
-  test "setrlimit returns nil on success" do
-    assert_nil(Process.setrlimit(Process::RLIMIT_CPU, 1000000))
-  end
-
-  test "setrlimit sets the resource limit as expected" do
-    assert_nothing_raised{ Process.setrlimit(Process::RLIMIT_CPU, 1000000) }
-    assert_equal([1000000, 1000000], Process.getrlimit(Process::RLIMIT_CPU))
-  end
-
-  test "setrlimit raises an error if the resource value is invalid" do
-    assert_raise(ArgumentError){ Process.setrlimit(9999, 100) }
-  end
-
-  test "is_job basic functionality" do
-    assert_respond_to(Process, :job?)
-    assert_nothing_raised{ Process.job? }
-  end
-
-  test "is_job returns a boolean value" do
-    assert_boolean(Process.job?)
-  end
-
-  test "is_job does not accept any arguments" do
-    assert_raise(ArgumentError){ Process.job?(Process.pid) }
-  end
-
-  test "volume_type is a private method" do
-    assert_not_respond_to(Process, :volume_type)
-  end
-
-  test "ffi functions are private" do
-    assert_not_respond_to(Process, :CloseHandle)
-    assert_not_respond_to(Process, :GetCurrentProcess)
-    assert_not_respond_to(Process, :GetProcessAffinityMask)
-    assert_not_respond_to(Process, :GetPriorityClass)
-    assert_not_respond_to(Process, :IsProcessInJob)
-    assert_not_respond_to(Process, :OpenProcess)
-    assert_not_respond_to(Process, :SetHandleInformation)
-    assert_not_respond_to(Process, :SetPriorityClass)
-  end
-
-  def teardown
-    @priority = nil
-  end
-
-  def self.shutdown
-    @@pids = nil
-  end
-end
+###############################################################################
+# test_win32_process.rb
+#
+# Test suite for the win32-process library.  This test suite will start
+# at least two instances of Notepad on your system, which will then
+# be killed. Requires the sys-proctable library.
+#
+# I haven't added a lot of test cases for fork/wait because it's difficult
+# to run such tests without causing havoc with Test::Unit itself.  Ideas
+# welcome.
+#
+# You should run this test case via the 'rake test' task.
+###############################################################################
+require 'test-unit'
+require 'win32/process'
+
+class TC_Win32Process < Test::Unit::TestCase
+  def self.startup
+    @@pids  = []
+    @@jruby = RUBY_PLATFORM == 'java'
+  end
+
+  def setup
+    @priority = Process::BELOW_NORMAL_PRIORITY_CLASS
+  end
+
+  test "win32-process version is set to the correct value" do
+    assert_equal('0.7.3', Process::WIN32_PROCESS_VERSION)
+  end
+
+  test "create basic functionality" do
+    assert_respond_to(Process, :create)
+  end
+
+  test "create with common flags works as expected" do
+    assert_nothing_raised{
+      @@pids << Process.create(
+         :app_name         => "notepad.exe",
+         :creation_flags   => Process::DETACHED_PROCESS,
+         :process_inherit  => false,
+         :thread_inherit   => true,
+         :cwd              => "C:\\"
+      ).process_id
+    }
+
+    assert_nothing_raised{ Process.kill(9, @@pids.pop) }
+  end
+
+  test "create requires a hash argument" do
+    assert_raise(TypeError){ Process.create("bogusapp.exe") }
+  end
+
+  test "create does not accept invalid keys" do
+    assert_raise(ArgumentError){ Process.create(:bogus => 'test.exe') }
+    assert_raise_message("invalid key 'bogus'"){
+      Process.create(:bogus => 'test.exe')
+    }
+  end
+
+  test "create does not accept invalid startup_info keys" do
+    assert_raise(ArgumentError){
+      Process.create(:startup_info => {:foo => 'test'})
+    }
+    assert_raise_message("invalid startup_info key 'foo'"){
+      Process.create(:startup_info => {:foo => 'test'})
+    }
+  end
+
+  test "create raises an error if the executable cannot be found" do
+    assert_raise(Errno::ENOENT){ Process.create(:app_name => "bogusapp.exe") }
+  end
+
+  test "create passes local environment when environment is not specified" do
+    omit_if(@@jruby)
+    stdout_read, stdout_write = IO.pipe
+
+    ENV['AARDVARK'] = 'B'
+    assert_nothing_raised {
+      Process.create(
+        :app_name         => 'cmd.exe /c echo %AARDVARK%',
+        :creation_flags   => Process::DETACHED_PROCESS,
+        :startup_info     => { :stdout => stdout_write }
+      )
+    }
+
+    stdout_write.close
+    assert_equal('B', stdout_read.read.chomp)
+  end
+
+  test "create does not pass local environment when environment is specified" do
+    omit_if(@@jruby)
+    stdout_read, stdout_write = IO.pipe
+
+    ENV['AARDVARK'] = 'B'
+    assert_nothing_raised {
+      Process.create(
+        :app_name         => 'cmd.exe /c echo %AARDVARK%',
+        :creation_flags   => Process::DETACHED_PROCESS,
+        :environment      => "",
+        :startup_info     => { :stdout => stdout_write }
+      )
+    }
+
+    stdout_write.close
+    assert_equal('%AARDVARK%', stdout_read.read.chomp)
+  end
+
+  test "create supports :environment as a string" do
+    omit_if(@@jruby)
+    stdout_read, stdout_write = IO.pipe
+
+    assert_nothing_raised {
+      Process.create(
+        :app_name         => 'cmd.exe /c echo %A% %C%',
+        :creation_flags   => Process::DETACHED_PROCESS,
+        :environment      => "A=B;C=D",
+        :startup_info     => { :stdout => stdout_write }
+      )
+    }
+
+    stdout_write.close
+    assert_equal("B D", stdout_read.read.chomp)
+  end
+
+  test "create supports :environment as an array" do
+    omit_if(@@jruby)
+    stdout_read, stdout_write = IO.pipe
+
+    assert_nothing_raised {
+      Process.create(
+        :app_name         => 'cmd.exe /c echo %A% %C%',
+        :creation_flags   => Process::DETACHED_PROCESS,
+        :environment      => [ "A=B;X;", "C=;D;Y" ],
+        :startup_info     => { :stdout => stdout_write }
+      )
+    }
+
+    stdout_write.close
+    assert_equal("B;X; ;D;Y", stdout_read.read.chomp)
+  end
+
+  test "create supports empty :environment string" do
+    assert_nothing_raised {
+      Process.create(
+        :creation_flags => Process::DETACHED_PROCESS,
+        :app_name       => 'cmd.exe',
+        :environment    => ''
+      )
+    }
+  end
+
+  test "create supports empty :environment array" do
+    assert_nothing_raised {
+      Process.create(
+        :creation_flags   => Process::DETACHED_PROCESS,
+        :app_name    => 'cmd.exe',
+        :environment => []
+      )
+    }
+  end
+
+  test "uid basic functionality" do
+    assert_respond_to(Process, :uid)
+    assert_kind_of(Fixnum, Process.uid)
+  end
+
+  test "uid accepts a boolean argument" do
+    assert_nothing_raised{ Process.uid(true) }
+    assert_nothing_raised{ Process.uid(true) }
+  end
+
+  test "uid returns a string if its argument is true" do
+    assert_kind_of(String, Process.uid(true))
+  end
+
+  test "uid accepts a maximum of one argument" do
+    assert_raise(ArgumentError){ Process.uid(true, true) }
+  end
+
+  test "argument to uid must be a boolean" do
+    assert_raise(TypeError){ Process.uid('test') }
+  end
+
+  test "getpriority basic functionality" do
+    assert_respond_to(Process, :getpriority)
+    assert_nothing_raised{ Process.getpriority(Process::PRIO_PROCESS, Process.pid) }
+    assert_kind_of(Fixnum, Process.getpriority(Process::PRIO_PROCESS, Process.pid))
+  end
+
+  test "getpriority treats an int argument of zero as the current process" do
+    assert_nothing_raised{ Process.getpriority(0, 0) }
+  end
+
+  test "getpriority requires both a kind and an int" do
+    assert_raise(ArgumentError){ Process.getpriority }
+    assert_raise(ArgumentError){ Process.getpriority(Process::PRIO_PROCESS) }
+  end
+
+  test "getpriority requires integer arguments" do
+    assert_raise(TypeError){ Process.getpriority('test', 0) }
+    assert_raise(TypeError){ Process.getpriority(Process::PRIO_PROCESS, 'test') }
+  end
+
+  test "setpriority basic functionality" do
+    assert_respond_to(Process, :setpriority)
+    assert_nothing_raised{ Process.setpriority(0, Process.pid, @priority) }
+    assert_equal(@priority, Process.getpriority(0, Process.pid))
+  end
+
+  test "setpriority returns zero on success" do
+    assert_equal(0, Process.setpriority(0, Process.pid, @priority))
+  end
+
+  test "setpriority treats an int argument of zero as the current process" do
+    assert_equal(0, Process.setpriority(0, 0, @priority))
+  end
+
+  test "setpriority requires at least three arguments" do
+    assert_raise(ArgumentError){ Process.setpriority }
+    assert_raise(ArgumentError){ Process.setpriority(0) }
+    assert_raise(ArgumentError){ Process.setpriority(0, 0) }
+  end
+
+  test "arguments to setpriority must be numeric" do
+    assert_raise(TypeError){ Process.setpriority('test', 0, @priority) }
+    assert_raise(TypeError){ Process.setpriority(0, 'test', @priority) }
+    assert_raise(TypeError){ Process.setpriority(0, 0, 'test') }
+  end
+
+  test "custom creation constants are defined" do
+    assert_not_nil(Process::CREATE_DEFAULT_ERROR_MODE)
+    assert_not_nil(Process::CREATE_NEW_CONSOLE)
+    assert_not_nil(Process::CREATE_NEW_PROCESS_GROUP)
+    assert_not_nil(Process::CREATE_NO_WINDOW)
+    assert_not_nil(Process::CREATE_SEPARATE_WOW_VDM)
+    assert_not_nil(Process::CREATE_SHARED_WOW_VDM)
+    assert_not_nil(Process::CREATE_SUSPENDED)
+    assert_not_nil(Process::CREATE_UNICODE_ENVIRONMENT)
+    assert_not_nil(Process::DEBUG_ONLY_THIS_PROCESS)
+    assert_not_nil(Process::DEBUG_PROCESS)
+    assert_not_nil(Process::DETACHED_PROCESS)
+  end
+
+  test "getrlimit basic functionality" do
+    assert_respond_to(Process, :getrlimit)
+    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
+    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_FSIZE) }
+  end
+
+  test "getrlimit returns an array of two numeric elements" do
+    assert_kind_of(Array, Process.getrlimit(Process::RLIMIT_CPU))
+    assert_equal(2, Process.getrlimit(Process::RLIMIT_CPU).length)
+    assert_kind_of(Integer, Process.getrlimit(Process::RLIMIT_CPU).first)
+  end
+
+  test "getrlimit can be called multiple times without issue" do
+    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
+    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
+    assert_nothing_raised{ Process.getrlimit(Process::RLIMIT_CPU) }
+  end
+
+  test "getrlimit requires a valid resource value" do
+    assert_raise(ArgumentError){ Process.getrlimit(9999) }
+  end
+
+  test "setrlimit basic functionality" do
+    assert_respond_to(Process, :setrlimit)
+    assert_nothing_raised{ Process.setrlimit(Process::RLIMIT_CPU, 1000000) }
+  end
+
+  test "setrlimit returns nil on success" do
+    assert_nil(Process.setrlimit(Process::RLIMIT_CPU, 1000000))
+  end
+
+  test "setrlimit sets the resource limit as expected" do
+    assert_nothing_raised{ Process.setrlimit(Process::RLIMIT_CPU, 1000000) }
+    assert_equal([1000000, 1000000], Process.getrlimit(Process::RLIMIT_CPU))
+  end
+
+  test "setrlimit raises an error if the resource value is invalid" do
+    assert_raise(ArgumentError){ Process.setrlimit(9999, 100) }
+  end
+
+  test "is_job basic functionality" do
+    assert_respond_to(Process, :job?)
+    assert_nothing_raised{ Process.job? }
+  end
+
+  test "is_job returns a boolean value" do
+    assert_boolean(Process.job?)
+  end
+
+  test "is_job does not accept any arguments" do
+    assert_raise(ArgumentError){ Process.job?(Process.pid) }
+  end
+
+  test "volume_type is a private method" do
+    assert_not_respond_to(Process, :volume_type)
+  end
+
+  test "ffi functions are private" do
+    assert_not_respond_to(Process, :CloseHandle)
+    assert_not_respond_to(Process, :GetCurrentProcess)
+    assert_not_respond_to(Process, :GetProcessAffinityMask)
+    assert_not_respond_to(Process, :GetPriorityClass)
+    assert_not_respond_to(Process, :IsProcessInJob)
+    assert_not_respond_to(Process, :OpenProcess)
+    assert_not_respond_to(Process, :SetHandleInformation)
+    assert_not_respond_to(Process, :SetPriorityClass)
+  end
+
+  test "get_exitcode basic functionality" do
+    assert_respond_to(Process, :get_exitcode)
+  end
+
+=begin
+  test "get_exitcode returns the process exit code" do
+    pid = Process.create(
+      :app_name         => 'cmd /c exit 7',
+      :creation_flags   => Process::DETACHED_PROCESS
+    ).process_id
+    10.times do
+      sleep(0.1)
+      break if Process.get_exitcode(pid)
+    end
+    assert_equal 7, Process.get_exitcode(pid)
+  end
+
+  test "get_exitcode returns nil while the process is running" do
+    stdin_read, stdin_write = IO.pipe
+    pid = Process.create(
+      :app_name         => 'cmd /c pause',
+      :creation_flags   => Process::DETACHED_PROCESS,
+      :startup_info     => { :stdin => stdin_read }
+    ).process_id
+    assert_equal nil, Process.get_exitcode(pid)
+    stdin_write.write("\n")
+    10.times do
+      sleep(0.1)
+      break if Process.get_exitcode(pid)
+    end
+    assert_equal 1, Process.get_exitcode(pid)
+  end
+=end
+
+  def teardown
+    @priority = nil
+  end
+
+  def self.shutdown
+    @@pids = nil
+  end
+end