zookeeper-ng 1.5.2.1-java

data/spec/chrooted_connection_spec.rb

@@ -0,0 +1,83 @@
+require 'spec_helper'
+require 'shared/connection_examples'
+
+describe 'Zookeeper chrooted' do
+  let(:path) { "/_zkchroottest_" }
+  let(:data) { "underpants" } 
+  let(:chroot_path) { '/slyphon-zookeeper-chroot' }
+
+  let(:connection_string) { "#{Zookeeper.default_cnx_str}#{chroot_path}" }
+
+  before do
+    @zk = Zookeeper.new(connection_string)
+  end
+
+  after do
+    @zk and @zk.close
+  end
+
+  def zk
+    @zk
+  end
+
+  describe 'non-existent' do
+    describe 'with existing parent' do
+      let(:chroot_path) { '/one-level' }
+
+      describe 'create' do
+        before do
+          with_open_zk { |z| rm_rf(z, chroot_path) }
+        end
+
+        after do
+          with_open_zk { |z| rm_rf(z, chroot_path) }
+        end
+
+        it %[should successfully create the path] do
+          rv = zk.create(:path => '/', :data => '')
+          rv[:rc].should be_zero
+          rv[:path].should == ''
+        end
+      end
+    end
+
+    describe 'with missing parent' do
+      let(:chroot_path) { '/deeply/nested/path' }
+
+      describe 'create' do
+        before do
+          with_open_zk do |z|
+            rm_rf(z, chroot_path)
+          end
+        end
+
+        it %[should return ZNONODE] do
+          rv = zk.create(:path => '/', :data => '')
+          rv[:rc].should_not be_zero
+          rv[:rc].should == Zookeeper::Exceptions::ZNONODE
+        end
+      end
+    end
+  end
+
+
+  describe do
+    before :all do
+      logger.warn "running before :all"
+
+      with_open_zk do |z|
+        z.create(:path => chroot_path, :data => '')
+      end
+    end
+
+    after :all do
+      with_open_zk do |z|
+        rm_rf(z, chroot_path)
+      end
+    end
+
+    it_should_behave_like "connection"
+  end
+end
+
+

data/spec/compatibilty_spec.rb

@@ -0,0 +1,8 @@
+require 'spec_helper'
+
+describe 'Compatibiliy layer' do
+  it %[should raise the correct error when a const is missing] do
+    lambda { Zookeeper::THISISANINVALIDCONST }.should raise_error(NameError)
+  end
+end
+

data/spec/default_watcher_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+
+describe Zookeeper do
+  describe :initialize, 'with watcher block' do
+    before do
+      @events = []
+      @watch_block = lambda do |hash| 
+        logger.debug "watch_block: #{hash.inspect}"
+        @events << hash
+      end
+
+      @zk = Zookeeper.new(Zookeeper.default_cnx_str, 10, @watch_block)
+
+      wait_until(2) { @zk.connected? }
+      @zk.should be_connected
+      logger.debug "connected!"
+
+      wait_until(2) { !@events.empty? }
+      logger.debug "got events!"
+    end
+
+    after do
+      @zk.close if @zk.connected?
+    end
+
+    it %[should receive initial connection state events] do
+      @events.should_not be_empty
+      @events.length.should == 1
+      @events.first[:state].should == Zookeeper::ZOO_CONNECTED_STATE
+    end
+
+    it %[should receive disconnection events] do
+      pending "the C driver doesn't appear to deliver disconnection events (?)"
+      @events.clear
+      @zk.close
+      wait_until(2) { !@events.empty? }
+      @events.should_not be_empty
+    end
+  end
+end
+

data/spec/em_spec.rb

@@ -0,0 +1,51 @@
+require File.expand_path('../spec_helper', __FILE__)
+require 'zookeeper/em_client'
+
+gem 'evented-spec', '~> 0.9.0'
+require 'evented-spec'
+
+
+describe 'ZookeeperEM' do
+  describe 'Client' do
+    include EventedSpec::SpecHelper
+    default_timeout 3.0
+
+    def setup_zk
+      @zk = ZookeeperEM::Client.new(Zookeeper.default_cnx_str)
+      em do
+        @zk.on_attached do
+          yield
+        end
+      end
+    end
+
+    def teardown_and_done
+      @zk.close do 
+        logger.debug { "TEST: about to call done" }
+        EM.next_tick do
+          done
+        end
+      end
+    end
+
+    describe 'callbacks' do
+      it %[should be called on the reactor thread] do
+        cb = lambda do |h|
+          EM.reactor_thread?.should be_true
+          logger.debug { "called back on the reactor thread? #{EM.reactor_thread?}" }
+          teardown_and_done
+        end
+
+        setup_zk do
+          @zk.on_attached do |*|
+            logger.debug { "on_attached called" }
+            rv = @zk.get(:path => '/', :callback => cb) 
+            logger.debug { "rv from @zk.get: #{rv.inspect}" }
+          end
+        end
+      end
+    end
+  end
+end
+
+

data/spec/ext/zookeeper_base_spec.rb

@@ -0,0 +1,19 @@
+require 'spec_helper'
+
+unless defined?(::JRUBY_VERSION)
+  describe Zookeeper::ZookeeperBase do
+    before do
+      @zk = described_class.new(Zookeeper.default_cnx_str)
+    end
+
+    after do
+      @zk.close unless @zk.closed?
+    end
+
+    it %[should have an original_pid assigned] do
+      @zk.original_pid.should == Process.pid
+    end
+  end
+end
+
+

data/spec/forked_connection_spec.rb

@@ -0,0 +1,122 @@
+require 'spec_helper'
+
+unless defined?(::JRUBY_VERSION)
+  describe %[forked connection] do
+    let(:path) { "/_zktest_" }
+    let(:pids_root) { "#{path}/pids" }
+    let(:data) { "underpants" }
+    let(:connection_string) { Zookeeper.default_cnx_str }
+
+    def process_alive?(pid)
+      Process.kill(0, @pid)
+      true
+    rescue Errno::ESRCH
+      false
+    end
+
+    LBORDER = ('-' * 35) << '< '
+    RBORDER = ' >' << ('-' * 35)
+
+    def mark(thing)
+      logger << "\n#{LBORDER}#{thing}#{RBORDER}\n\n"
+    end
+
+    before do
+      mark "BEFORE: START"
+      if defined?(::Rubinius)
+        pending("this test is currently broken in rbx")
+      else
+        @zk = Zookeeper.new(connection_string)
+        rm_rf(@zk, path)
+      end
+      mark "BEFORE: END"
+    end
+
+    after do
+      mark "AFTER: START"
+
+      if @pid and process_alive?(@pid)
+        begin
+          Process.kill('KILL', @pid)
+          p Process.wait2(@pid)
+        rescue Errno::ESRCH
+        end
+      end
+
+      @zk.close if @zk and !@zk.closed?
+      with_open_zk(connection_string) { |z| rm_rf(z, path) }
+
+      mark "AFTER: END"
+    end
+
+    def wait_for_child_safely(pid, timeout=5)
+      time_to_stop = Time.now + timeout
+
+      until Time.now > time_to_stop
+        if a = Process.wait2(@pid, Process::WNOHANG)
+          return a.last
+        else
+          sleep(0.01)
+        end
+      end
+
+      nil
+    end
+
+    it %[should do the right thing and not fail] do
+      mark "TEST: START"
+
+      @zk.wait_until_connected
+
+      mkdir_p(@zk, pids_root)
+
+      # the parent's pid path
+      @zk.create(:path => "#{pids_root}/#{$$}", :data => $$.to_s)
+
+      @latch = Zookeeper::Latch.new
+      @event = nil
+
+      cb = proc do |h|
+        logger.debug { "watcher called back: #{h.inspect}" }
+        @event = h
+        @latch.release
+      end
+
+      @zk.stat(:path => "#{pids_root}/child", :watcher => cb)
+
+      @zk.pause_before_fork_in_parent
+
+      mark "FORK"
+
+      @pid = fork do
+        logger.debug { "reopening connection in child: #{$$}" }
+        @zk.reopen
+        logger.debug { "creating path" }
+        rv = @zk.create(:path => "#{pids_root}/child", :data => $$.to_s)
+        logger.debug { "created path #{rv[:path]}" }
+        @zk.close
+
+        logger.debug { "close finished" }
+        exit!(0)
+      end
+
+      @zk.resume_after_fork_in_parent
+
+      event_waiter_th = Thread.new do
+        @latch.await(5) unless @event
+        @event
+      end
+
+      logger.debug { "waiting on child #{@pid}" }
+
+      status = wait_for_child_safely(@pid)
+      raise "Child process did not exit, likely hung"  unless status
+
+      status.should_not be_signaled
+      status.should be_success
+
+      event_waiter_th.join(5).should == event_waiter_th
+      @event.should_not be_nil
+    end
+  end
+end

data/spec/latch_spec.rb

@@ -0,0 +1,24 @@
+require 'spec_helper'
+
+# this is a simple sanity check of my timeout addition
+
+describe Zookeeper::Latch do
+  subject { described_class.new }
+
+  describe %[await] do
+    describe %[with timeout] do
+      it %[should return after waiting until timeout if not released] do
+        other_latch = described_class.new
+
+        th = Thread.new do
+          subject.await(0.01)
+          other_latch.release
+        end
+
+        other_latch.await
+        th.join(1).should == th
+      end
+    end
+  end
+end
+

data/spec/log4j.properties

@@ -0,0 +1,17 @@
+log4j.rootLogger        = DEBUG,S
+
+log4j.appender.S        = org.apache.log4j.FileAppender
+log4j.appender.S.File   = /tmp/zk-test.log
+
+log4j.appender.S.layout = org.apache.log4j.PatternLayout
+log4j.appender.S.layout.ConversionPattern  = %5p [%t] (%c:%F:%L) - %m%n
+
+log4j.appender.C        = org.apache.log4j.FileAppender
+log4j.appender.C.File   = /tmp/zk-test-client.log
+
+log4j.appender.C.layout = org.apache.log4j.PatternLayout
+log4j.appender.C.layout.ConversionPattern  = %5p [%t] (%c:%F:%L) - %m%n
+
+log4j.org.apache.zookeeper.ZooKeeper = DEBUG,C
+log4j.org.apache.zookeeper.server    = DEBUG,S
+

data/spec/shared/all_success_return_values.rb

@@ -0,0 +1,10 @@
+shared_examples_for "all success return values" do
+  it %[should have a return code of Zookeeper::ZOK] do
+    @rv[:rc].should == Zookeeper::ZOK
+  end
+
+  it %[should have a req_id integer] do
+    @rv[:req_id].should be_kind_of(Integer)
+  end
+end
+