pipemaster 0.3.1

data/test/unit/test_configurator.rb

@@ -0,0 +1,151 @@
+# -*- encoding: binary -*-
+
+require 'test/test_helper'
+require 'tempfile'
+require 'pipemaster'
+
+TestStruct = Struct.new(
+  *(Pipemaster::Configurator::DEFAULTS.keys + %w(listener_opts listeners)))
+class TestConfigurator < Test::Unit::TestCase
+
+  def test_config_init
+    assert_nothing_raised { Pipemaster::Configurator.new {} }
+  end
+
+  def test_expand_addr
+    meth = Pipemaster::Configurator.new.method(:expand_addr)
+
+    assert_equal "/var/run/pipemaster.sock", meth.call("/var/run/pipemaster.sock")
+    assert_equal "#{Dir.pwd}/foo/bar.sock", meth.call("unix:foo/bar.sock")
+
+    path = meth.call("~/foo/bar.sock")
+    assert_equal "/", path[0..0]
+    assert_match %r{/foo/bar\.sock\z}, path
+
+    path = meth.call("~root/foo/bar.sock")
+    assert_equal "/", path[0..0]
+    assert_match %r{/foo/bar\.sock\z}, path
+
+    assert_equal "1.2.3.4:2007", meth.call('1.2.3.4:2007')
+    assert_equal "0.0.0.0:2007", meth.call('0.0.0.0:2007')
+    assert_equal "0.0.0.0:2007", meth.call(':2007')
+    assert_equal "0.0.0.0:2007", meth.call('*:2007')
+    assert_equal "0.0.0.0:2007", meth.call('2007')
+    assert_equal "0.0.0.0:2007", meth.call(2007)
+
+    # the next two aren't portable, consider them unsupported for now
+    # assert_match %r{\A\d+\.\d+\.\d+\.\d+:2007\z}, meth.call('1:2007')
+    # assert_match %r{\A\d+\.\d+\.\d+\.\d+:2007\z}, meth.call('2:2007')
+  end
+
+  def test_config_invalid
+    tmp = Tempfile.new('Pipefile')
+    tmp.syswrite(%q(asdfasdf "hello-world"))
+    assert_raises(NoMethodError) do
+      Pipemaster::Configurator.new(:config_file => tmp.path)
+    end
+  end
+
+  def test_config_non_existent
+    tmp = Tempfile.new('Pipefile')
+    path = tmp.path
+    tmp.close!
+    assert_raises(Errno::ENOENT) do
+      Pipemaster::Configurator.new(:config_file => path)
+    end
+  end
+
+  def test_config_defaults
+    cfg = Pipemaster::Configurator.new(:use_defaults => true)
+    test_struct = TestStruct.new
+    assert_nothing_raised { cfg.commit!(test_struct) }
+    Pipemaster::Configurator::DEFAULTS.each do |key,value|
+      assert_equal value, test_struct.__send__(key)
+    end
+  end
+
+  def test_config_defaults_skip
+    cfg = Pipemaster::Configurator.new(:use_defaults => true)
+    skip = [ :logger ]
+    test_struct = TestStruct.new
+    assert_nothing_raised { cfg.commit!(test_struct, :skip => skip) }
+    Pipemaster::Configurator::DEFAULTS.each do |key,value|
+      next if skip.include?(key)
+      assert_equal value, test_struct.__send__(key)
+    end
+    assert_nil test_struct.logger
+  end
+
+  def test_listen_options
+    tmp = Tempfile.new('Pipefile')
+    expect = { :sndbuf => 1, :rcvbuf => 2, :backlog => 10 }.freeze
+    listener = "127.0.0.1:12345"
+    tmp.syswrite("listen '#{listener}', #{expect.inspect}\n")
+    cfg = nil
+    assert_nothing_raised do
+      cfg = Pipemaster::Configurator.new(:config_file => tmp.path)
+    end
+    test_struct = TestStruct.new
+    assert_nothing_raised { cfg.commit!(test_struct) }
+    assert(listener_opts = test_struct.listener_opts)
+    assert_equal expect, listener_opts[listener]
+  end
+
+  def test_listen_option_bad
+    tmp = Tempfile.new('Pipefile')
+    expect = { :sndbuf => "five" }
+    listener = "127.0.0.1:12345"
+    tmp.syswrite("listen '#{listener}', #{expect.inspect}\n")
+    assert_raises(ArgumentError) do
+      Pipemaster::Configurator.new(:config_file => tmp.path)
+    end
+  end
+
+  def test_listen_option_bad_delay
+    tmp = Tempfile.new('Pipefile')
+    expect = { :delay => "five" }
+    listener = "127.0.0.1:12345"
+    tmp.syswrite("listen '#{listener}', #{expect.inspect}\n")
+    assert_raises(ArgumentError) do
+      Pipemaster::Configurator.new(:config_file => tmp.path)
+    end
+  end
+
+  def test_listen_option_float_delay
+    tmp = Tempfile.new('Pipefile')
+    expect = { :delay => 0.5 }
+    listener = "127.0.0.1:12345"
+    tmp.syswrite("listen '#{listener}', #{expect.inspect}\n")
+    assert_nothing_raised do
+      Pipemaster::Configurator.new(:config_file => tmp.path)
+    end
+  end
+
+  def test_listen_option_int_delay
+    tmp = Tempfile.new('Pipefile')
+    expect = { :delay => 5 }
+    listener = "127.0.0.1:12345"
+    tmp.syswrite("listen '#{listener}', #{expect.inspect}\n")
+    assert_nothing_raised do
+      Pipemaster::Configurator.new(:config_file => tmp.path)
+    end
+  end
+
+  def test_after_fork_proc
+    test_struct = TestStruct.new
+    [ proc { |a,b| }, Proc.new { |a,b| }, lambda { |a,b| } ].each do |my_proc|
+      Pipemaster::Configurator.new(:after_fork => my_proc).commit!(test_struct)
+      assert_equal my_proc, test_struct.after_fork
+    end
+  end
+
+  def test_after_fork_wrong_arity
+    [ proc { |a| }, Proc.new { }, lambda { |a,b,c| } ].each do |my_proc|
+      assert_raises(ArgumentError) do
+        Pipemaster::Configurator.new(:after_fork => my_proc)
+      end
+    end
+  end
+
+end
+

data/test/unit/test_server.rb

@@ -0,0 +1,84 @@
+# -*- encoding: binary -*-
+
+# Copyright (c) 2005 Zed A. Shaw 
+# You can redistribute it and/or modify it under the same terms as Ruby.
+#
+# Additional work donated by contributors.  See http://mongrel.rubyforge.org/attributions.html
+# for more information.
+
+require 'test/test_helper'
+
+class ServerTest < Test::Unit::TestCase
+
+  def setup
+    @port = unused_port
+  end
+
+  def teardown
+    if @server
+      redirect_test_io do
+        wait_master_ready("test_stderr.#$$.log")
+        File.truncate("test_stderr.#$$.log", 0)
+        Process.kill 9, @pid
+      end
+    end
+  end
+
+  def start(options = nil)
+    redirect_test_io do
+      @server = Pipemaster::Server.new({:listeners => [ "127.0.0.1:#{@port}" ]}.merge(options || {}))
+      @pid = fork { @server.start }
+      at_exit { Process.kill @pid, 9 }
+      wait_master_ready("test_stderr.#$$.log")
+    end
+  end
+
+  def test_master_and_fork
+    start :commands => { :ppid => lambda { puts Process.ppid } }
+    results = hit("127.0.0.1:#@port", "ppid")
+    assert_equal @pid, results.last.to_i
+  end
+
+  def test_broken_pipefile
+    tmp = Tempfile.new('Pipefile')
+    ObjectSpace.undefine_finalizer(tmp)
+    tmp.write "raise RuntimeError"
+    tmp.flush
+    teardown
+    assert_raises RuntimeError do
+      start :config_file=>tmp.path
+    end
+  ensure
+    tmp.close!
+  end
+
+  def test_no_command
+    start
+    results = hit("127.0.0.1:#@port", "test")
+    assert_equal 1, results.first
+  end
+
+  def test_arguments
+    start :commands => { :args => lambda { |x,y| $stdout << "#{x} - #{y}" } }
+    results = hit("127.0.0.1:#@port", "args", "foo", "bar")
+    assert_equal "foo - bar", results.last
+  end
+
+  def test_streams
+    start :commands => { :reverse => lambda { $stdout << $stdin.read.reverse } }
+    client = Pipemaster::Client.new("127.0.0.1:#@port")
+    tmp = Tempfile.new('input')
+    ObjectSpace.undefine_finalizer(tmp)
+    tmp.write "foo bar"
+    tmp.flush
+    tmp.rewind
+    client.input = tmp
+    output = StringIO.new
+    client.output = output
+    client.request :reverse
+    assert_equal "rab oof", client.output.string
+  ensure
+    tmp.close!
+  end
+
+end

metadata

@@ -0,0 +1,84 @@
+--- !ruby/object:Gem::Specification 
+name: pipemaster
+version: !ruby/object:Gem::Version 
+  version: 0.3.1
+platform: ruby
+authors: 
+- Assaf Arkin
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-18 00:00:00 -08:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: unicorn
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        version: "0.96"
+    version: 
+description: 
+email: assaf@labnotes.org
+executables: 
+- pipemaster
+extensions: []
+
+extra_rdoc_files: 
+- README.rdoc
+- CHANGELOG
+files: 
+- bin/pipemaster
+- lib/pipemaster/client.rb
+- lib/pipemaster/configurator.rb
+- lib/pipemaster/server.rb
+- lib/pipemaster/worker.rb
+- lib/pipemaster.rb
+- test/test_helper.rb
+- test/unit/test_configurator.rb
+- test/unit/test_server.rb
+- CHANGELOG
+- LICENSE
+- README.rdoc
+- Rakefile
+- Gemfile
+- pipemaster.gemspec
+has_rdoc: true
+homepage: http://labnotes.org
+licenses: []
+
+post_install_message: To get started run pipemaster --help
+rdoc_options: 
+- --title
+- Pipemaster 0.3.1
+- --main
+- README.rdoc
+- --webcvs
+- http://github.com/assaf/pipemaster
+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: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Use the fork
+test_files: []
+