costan-zerg_support 0.0.9

data/test/test_process.rb

@@ -0,0 +1,78 @@
+require 'English'
+require 'test/unit'
+
+require 'zerg_support'
+
+class TestProcess < Test::Unit::TestCase
+  def teardown
+    @pid_files.each { |f| File.delete f rescue nil } if @pid_files
+    super
+  end
+  
+  def test_process_info
+    info = Zerg::Support::Process.process_info($PID)
+    assert info, 'Failed to find current process'
+    
+    $ARGV.each do |arg|
+      assert info[:command_line].index(arg),
+             "Command line for current process does not contain argument #{arg}"
+    end
+  end
+  
+  def test_process_list_contains_current_process
+    list = Zerg::Support::Process.processes_by_id
+    
+    assert list[$PID], 'Process list does not contain current process'
+  end
+
+  # spawn 2^(exp+1)-1 children, and their command lines will contain "cookie"
+  def spawn_sleepers(exp, cookie)
+    Thread.new { system "ruby test/fork_tree.rb -- #{exp} 100 #{cookie}" }
+    on_windows = RUBY_PLATFORM =~ /win/ && RUBY_PLATFORM !~ /darwin/
+    sleep on_windows ? 0.5 : 0.1
+    pid_file = "#{cookie}.pid"
+    @pid_files ||= []
+    @pid_files << pid_file
+    pid = File.read(pid_file).to_i
+    sleep on_windows ? 0.4 : 0.5
+    return pid
+  end
+  
+  def now_cookie
+    "zerg_test_#{Time.now.to_f}s"
+  end
+  
+  def test_process_tree
+    cookie = now_cookie
+    root_pid = spawn_sleepers 3, cookie
+    
+    tree_processes = Zerg::Support::Process.process_tree root_pid    
+    assert_equal 15, tree_processes.length, 'Missed some processes'
+    assert tree_processes.all? { |info| info[:command_line].index cookie },
+           'Got the wrong processes'
+           
+    Zerg::Support::Process.kill_tree root_pid
+  end
+  
+  def test_kill
+    cookie = now_cookie
+    victim_pid = spawn_sleepers 0, cookie
+    
+    Zerg::Support::Process.kill victim_pid
+    sleep 0.5
+    processes = Zerg::Support::Process.processes
+    assert_equal nil, processes[victim_pid], 'Victim was not killed'
+  end
+  
+  def test_kill_tree
+    cookie = now_cookie
+    root_pid = spawn_sleepers 3, cookie
+     
+    Zerg::Support::Process.kill_tree root_pid
+    sleep 0.5
+    processes = Zerg::Support::Process.processes
+    escaped = processes.select { |info| info[:command_line].index cookie }
+    
+    assert_equal [], escaped, 'Some process(es) escaped'
+  end
+end

data/test/test_spawn.rb

@@ -0,0 +1,163 @@
+require 'digest/sha1'
+require 'test/unit'
+
+require 'zerg_support'
+
+require 'rubygems'
+require 'flexmock/test_unit'
+
+class TestSpawn < Test::Unit::TestCase
+  def temp_file
+    now_cookie = "zerg_test_#{Time.now.to_f}s"
+    File.open(now_cookie, 'w') {}
+    return now_cookie
+  end
+  
+  def setup
+    super
+    @in_file = temp_file
+    @out_file = temp_file
+  end
+
+  def teardown
+    File.delete @in_file rescue nil
+    File.delete @out_file rescue nil
+    super
+  end
+  
+  def test_stdout_redirect
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', 'print "1234\n"'],
+        STDOUT => @out_file
+    Process.waitpid pid
+    
+    assert_equal "1234\n", File.read(@out_file)
+  end
+  
+  def test_stdin_redirect
+    File.open(@in_file, 'w') { |f| f.write "1234\n" }
+
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', "print gets"],
+        STDIN => @in_file, STDOUT => @out_file
+    Process.waitpid pid
+    assert_equal "1234\n", File.read(@out_file)
+  end
+
+  def test_stderr_redirect
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', "raise 'zerg_stderr'"],
+        STDERR => @out_file
+    Process.waitpid pid
+    assert_equal "-e:1: zerg_stderr (RuntimeError)\n", File.read(@out_file)
+  end
+  
+  def test_redirect_cascade
+    pid = Zerg::Support::Process.spawn 'ruby',
+                                       ['-e', "print 1; raise 'zerg_stderr'"],
+        STDOUT => @out_file, STDERR => STDOUT
+    Process.waitpid pid
+    assert_operator ["1-e:1: zerg_stderr (RuntimeError)\n",
+                     "-e:1: zerg_stderr (RuntimeError)\n1"],
+                    :include?, File.read(@out_file)    
+  end
+  
+  def test_redirect_share
+    pid = Zerg::Support::Process.spawn 'ruby',
+                                       ['-e', "print 1; raise 'zerg_stderr'"],
+        STDOUT => @out_file, STDERR => @out_file
+    Process.waitpid pid
+    assert_operator ["1-e:1: zerg_stderr (RuntimeError)\n",
+                     "-e:1: zerg_stderr (RuntimeError)\n1"],
+                    :include?, File.read(@out_file)    
+  end
+    
+  def test_async
+    t0 = Time.now
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', 'sleep 0.5']
+    t1 = Time.now
+    Process.waitpid pid
+    t2 = Time.now
+    
+    assert_operator t1 - t0, :<, 0.2, 'Spawning is not asynchronous'
+    assert_operator t2 - t0, :>, 0.5,
+                    'The spawned program did not sleep for 1 second'
+  end
+  
+  def test_set_environment
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', 'print ENV[\'XA\']'],
+        STDOUT => @out_file, :env => {'XA' => "1234\n"}
+    Process.waitpid pid
+    assert_equal "1234\n", File.read(@out_file)
+    
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', 'print ENV[\'PATH\']'],
+        STDOUT => @in_file, :env => {'XA' => "1234\n"}, :unsetenv_others => true
+    Process.waitpid pid
+    # clearing the environment will reset PATH, so ruby might not run at all
+    output = File.read(@in_file)
+    assert output == 'nil' || output == '', 'The environment was not cleared'
+  end
+  
+  def test_close_fd
+    in_file = File.open(@in_file, 'w')
+    in_file.sync = true
+    in_file.write "1234\n"
+    in_file.close
+    
+    in_file = File.open(@in_file, 'r')
+    
+    in_fd = in_file.fileno
+    read_fd_code = "File.open(#{in_fd}).read"
+    assert_equal "1234\n", eval(read_fd_code), 'File reading code is broken'
+    
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', "print #{read_fd_code}"],
+        STDOUT => @out_file, STDERR => STDOUT
+    Process.waitpid pid
+    in_file.close
+
+    assert_equal "-e:1:", File.read(@out_file)[0, 5],
+                 'Spawn does not close file descriptors'    
+  end
+    
+  def test_rlimit_processing
+    if RUBY_PLATFORM =~ /win/ and RUBY_PLATFORM !~ /darwin/
+      Process.const_set :RLIMIT_CPU, 'rlimit_cpu'
+      Process.const_set :RLIMIT_RSS, 'rlimit_rss'
+    end
+
+    flexmock(Process).should_receive(:setrlimit).
+                      with(Process::RLIMIT_CPU, 1, 5).once.and_return(nil)
+    flexmock(Process).should_receive(:setrlimit).
+                      with(Process::RLIMIT_RSS, 64 * 1024 * 1024).once.
+                      and_return(nil)
+    
+    Zerg::Support::Process::Helpers.set_rlimits :rlimit_cpu => [1, 5],
+                                                :rlimit_rss => 64 * 1024 * 1024
+  end
+  
+  def test_process_group_processing
+    flexmock(Process).should_receive(:setpgid).with(0, 42).once.and_return(nil)
+    flexmock(Process).should_receive(:setsid).with().once.and_return(nil)
+
+    Zerg::Support::Process::Helpers.set_process_group :pgroup => 42
+    Zerg::Support::Process::Helpers.set_process_group :pgroup => true
+  end
+  
+  def test_process_group
+    pid = Zerg::Support::Process.spawn 'ruby', ['-e', "sleep 0.5"],
+                                       :pgroup => true
+    pinfo = Zerg::Support::Process::process_info pid
+    Process.waitpid pid
+    
+    assert pinfo[:state].index('s'), 'Failed to become session leader'
+  end
+  
+  def test_rlimits
+    t0 = Time.now
+    pid = Zerg::Support::Process.spawn 'ruby',
+        ['-e', "i = 0; t0 = Time.now; i += 1 while Time.now - t0 < 3.0"],
+        :rlimit_cpu => 1
+    Process.waitpid pid
+    t1 = Time.now
+    
+    assert_operator t1 - t0, :>=, 0.95, 'Spawning failed'
+    assert_operator t1 - t0, :<=, 2.0, 'Failed to apply rlimit_cpu'
+  end
+end

data/zerg_support.gemspec

@@ -0,0 +1,41 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{zerg_support}
+  s.version = "0.0.9"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Victor Costan"]
+  s.date = %q{2009-03-28}
+  s.description = %q{Support libraries used by Zergling.Net deployment code.}
+  s.email = %q{victor@zergling.net}
+  s.extra_rdoc_files = ["CHANGELOG", "lib/zerg_support/event_machine/connection_mocks.rb", "lib/zerg_support/event_machine/frame_protocol.rb", "lib/zerg_support/event_machine/object_protocol.rb", "lib/zerg_support/gems.rb", "lib/zerg_support/open_ssh.rb", "lib/zerg_support/process.rb", "lib/zerg_support/spawn.rb", "lib/zerg_support.rb", "LICENSE", "README"]
+  s.files = ["CHANGELOG", "lib/zerg_support/event_machine/connection_mocks.rb", "lib/zerg_support/event_machine/frame_protocol.rb", "lib/zerg_support/event_machine/object_protocol.rb", "lib/zerg_support/gems.rb", "lib/zerg_support/open_ssh.rb", "lib/zerg_support/process.rb", "lib/zerg_support/spawn.rb", "lib/zerg_support.rb", "LICENSE", "Manifest", "Rakefile", "README", "RUBYFORGE", "test/fork_tree.rb", "test/test_connection_mocks.rb", "test/test_frame_protocol.rb", "test/test_gems.rb", "test/test_object_protocol.rb", "test/test_open_ssh.rb", "test/test_process.rb", "test/test_spawn.rb", "zerg_support.gemspec"]
+  s.has_rdoc = true
+  s.homepage = %q{http://www.zergling.net}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Zerg_support", "--main", "README"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{zerglings}
+  s.rubygems_version = %q{1.3.1}
+  s.summary = %q{Support libraries used by Zergling.Net deployment code.}
+  s.test_files = ["test/test_connection_mocks.rb", "test/test_frame_protocol.rb", "test/test_gems.rb", "test/test_object_protocol.rb", "test/test_open_ssh.rb", "test/test_process.rb", "test/test_spawn.rb"]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 2
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<echoe>, [">= 3.0.2"])
+      s.add_development_dependency(%q<event_machine>, [">= 0.12.2"])
+      s.add_development_dependency(%q<flexmock>, [">= 0.8.3"])
+    else
+      s.add_dependency(%q<echoe>, [">= 3.0.2"])
+      s.add_dependency(%q<event_machine>, [">= 0.12.2"])
+      s.add_dependency(%q<flexmock>, [">= 0.8.3"])
+    end
+  else
+    s.add_dependency(%q<echoe>, [">= 3.0.2"])
+    s.add_dependency(%q<event_machine>, [">= 0.12.2"])
+    s.add_dependency(%q<flexmock>, [">= 0.8.3"])
+  end
+end

metadata

@@ -0,0 +1,125 @@
+--- !ruby/object:Gem::Specification 
+name: costan-zerg_support
+version: !ruby/object:Gem::Version 
+  version: 0.0.9
+platform: ruby
+authors: 
+- Victor Costan
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-03-28 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: echoe
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 3.0.2
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: event_machine
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.12.2
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: flexmock
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: 0.8.3
+    version: 
+description: Support libraries used by Zergling.Net deployment code.
+email: victor@zergling.net
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- CHANGELOG
+- lib/zerg_support/event_machine/connection_mocks.rb
+- lib/zerg_support/event_machine/frame_protocol.rb
+- lib/zerg_support/event_machine/object_protocol.rb
+- lib/zerg_support/gems.rb
+- lib/zerg_support/open_ssh.rb
+- lib/zerg_support/process.rb
+- lib/zerg_support/spawn.rb
+- lib/zerg_support.rb
+- LICENSE
+- README
+files: 
+- CHANGELOG
+- lib/zerg_support/event_machine/connection_mocks.rb
+- lib/zerg_support/event_machine/frame_protocol.rb
+- lib/zerg_support/event_machine/object_protocol.rb
+- lib/zerg_support/gems.rb
+- lib/zerg_support/open_ssh.rb
+- lib/zerg_support/process.rb
+- lib/zerg_support/spawn.rb
+- lib/zerg_support.rb
+- LICENSE
+- Manifest
+- Rakefile
+- README
+- RUBYFORGE
+- test/fork_tree.rb
+- test/test_connection_mocks.rb
+- test/test_frame_protocol.rb
+- test/test_gems.rb
+- test/test_object_protocol.rb
+- test/test_open_ssh.rb
+- test/test_process.rb
+- test/test_spawn.rb
+- zerg_support.gemspec
+has_rdoc: true
+homepage: http://www.zergling.net
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Zerg_support
+- --main
+- README
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "1.2"
+  version: 
+requirements: []
+
+rubyforge_project: zerglings
+rubygems_version: 1.2.0
+signing_key: 
+specification_version: 2
+summary: Support libraries used by Zergling.Net deployment code.
+test_files: 
+- test/test_connection_mocks.rb
+- test/test_frame_protocol.rb
+- test/test_gems.rb
+- test/test_object_protocol.rb
+- test/test_open_ssh.rb
+- test/test_process.rb
+- test/test_spawn.rb