zk-eventmachine 0.1.13 → 0.2.0.beta.1

data/spec/z_k/z_k_event_machine/client_spec.rb

@@ -1,31 +1,23 @@
 require 'spec_helper'
 
 module ZK::ZKEventMachine
-  shared_examples_for 'Client' do
+  describe 'Client' do
     include EventedSpec::SpecHelper
     default_timeout 2.0
 
-    let(:base_path) { '/zk-em-testing' }
-
     before do
-      @zk.rm_rf(base_path)
-      @zk.mkdir_p(base_path)
+      @zk = ::ZK.new
+      @base_path = '/zk-em-testing'
+      @zk.rm_rf(@base_path)
+      @zk.mkdir_p(@base_path)
+      @zkem = ZK::ZKEventMachine::Client.new('localhost:2181')
     end
 
     after do
-      @zk.rm_rf(base_path)
+      @zk.rm_rf(@base_path)
       @zk.close!
     end
 
-    def event_mock(name=:event)
-      flexmock(name).tap do |ev|
-        ev.should_receive(:node_event?).and_return(false)
-        ev.should_receive(:state_event?).and_return(true)
-        ev.should_receive(:zk=).with_any_args
-        yield ev if block_given?
-      end
-    end
-
     describe 'connect' do
       it %[should return a deferred that fires when connected and then close] do
         em do
@@ -53,10 +45,48 @@ module ZK::ZKEventMachine
       end
     end
 
+    describe 'on_connection_loss' do
+      before do
+        @path = [@base_path, 'foo'].join('/')
+        @data = 'this is data'
+        @zk.create(@path, @data)
+      end
+
+      it %[should be called back if the connection is lost] do
+        em do
+          @zkem.on_connection_lost do |exc|
+            logger.debug { "WIN!" }
+            exc.should be_kind_of(ZK::Exceptions::ConnectionLoss)
+            @zkem.close! { done }
+          end
+
+          @zkem.connect do
+            flexmock(@zkem.cnx) do |m|
+              m.should_receive(:get).with(Hash).and_return do |hash|
+                logger.debug { "client received :get wtih #{hash.inspect}" }
+                @user_cb = hash[:callback]
+
+                EM.next_tick do
+                  logger.debug { "calling back user cb with connection loss" }
+                  @user_cb.call(:rc => ZK::Exceptions::CONNECTIONLOSS)
+                end
+
+                { :rc => Zookeeper::ZOK }
+              end
+            end
+
+            @zkem.get(@path) do |exc,data|
+              exc.should be_kind_of(ZK::Exceptions::ConnectionLoss)
+            end
+          end
+        end
+      end
+    end
+
     describe 'get' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @data = 'this is data'
           @zk.create(@path, @data)
         end
@@ -99,7 +129,7 @@ module ZK::ZKEventMachine
 
       describe 'failure' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
         end
 
@@ -137,7 +167,7 @@ module ZK::ZKEventMachine
     describe 'create' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
 
           @data = 'this is data'
@@ -203,7 +233,7 @@ module ZK::ZKEventMachine
 
       describe 'failure' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.create(@path, '')
         end
 
@@ -237,10 +267,11 @@ module ZK::ZKEventMachine
       end # failure
     end # create
 
+
     describe 'set' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @data = 'this is data'
           @new_data = 'this is better data'
           @zk.create(@path, @data)
@@ -290,7 +321,7 @@ module ZK::ZKEventMachine
 
       describe 'failure' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
         end
 
@@ -326,7 +357,7 @@ module ZK::ZKEventMachine
 
     describe 'exists?' do
       before do
-        @path = [base_path, 'foo'].join('/')
+        @path = [@base_path, 'foo'].join('/')
         @data = 'this is data'
       end
 
@@ -372,7 +403,7 @@ module ZK::ZKEventMachine
     describe 'stat' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @data = 'this is data'
           @zk.create(@path, @data)
           @orig_stat = @zk.stat(@path)
@@ -414,7 +445,7 @@ module ZK::ZKEventMachine
 
       describe 'non-existent node' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
         end
 
@@ -443,7 +474,7 @@ module ZK::ZKEventMachine
     describe 'delete' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @data = 'this is data'
           @zk.create(@path, @data)
         end
@@ -481,7 +512,7 @@ module ZK::ZKEventMachine
 
       describe 'failure' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
         end
 
@@ -518,7 +549,7 @@ module ZK::ZKEventMachine
     describe 'children' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @child_1_path = [@path, 'child_1'].join('/')
           @child_2_path = [@path, 'child_2'].join('/')
 
@@ -572,7 +603,7 @@ module ZK::ZKEventMachine
 
       describe 'failure' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
         end
 
@@ -609,7 +640,7 @@ module ZK::ZKEventMachine
     describe 'get_acl' do
       describe 'success' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @data = 'this is data'
           @zk.create(@path, @data)
         end
@@ -653,7 +684,7 @@ module ZK::ZKEventMachine
 
       describe 'failure' do
         before do
-          @path = [base_path, 'foo'].join('/')
+          @path = [@base_path, 'foo'].join('/')
           @zk.delete(@path) rescue ZK::Exceptions::NoNode
         end
 
@@ -698,157 +729,6 @@ module ZK::ZKEventMachine
         it %[should have NoNode as the first argument to the block] 
       end # failure
     end # set_acl
-
-    describe 'session_id and session_passwd' do
-      it %[should return a Fixnum for session_id once connected] do
-        em do
-          @zkem.session_id.should be_nil
-
-          @zkem.connect do
-            @zkem.session_id.should be_kind_of(Fixnum)
-            @zkem.close! { done }
-          end
-        end
-      end
-
-      it %[should return a String for session_passwd once connected] do
-        em do
-          @zkem.session_passwd.should be_nil
-
-          @zkem.connect do
-            @zkem.session_passwd.should be_kind_of(String)
-            @zkem.close! { done }
-          end
-        end
-      end
-    end
-
-    describe 'on_connection_lost' do
-      before do
-        @path = [base_path, 'foo'].join('/')
-        @data = 'this is data'
-        @zk.create(@path, @data)
-      end
-
-      it %[should be called back if the connection is lost] do
-        em do
-          @zkem.on_connection_lost do |exc|
-            logger.debug { "WIN!" }
-            exc.should be_kind_of(ZK::Exceptions::ConnectionLoss)
-            @zkem.close! { done }
-          end
-
-          @zkem.connect do
-            flexmock(@zkem.cnx) do |m|
-              m.should_receive(:get).with(Hash).and_return do |hash|
-                logger.debug { "client received :get wtih #{hash.inspect}" }
-                @user_cb = hash[:callback]
-
-                EM.next_tick do
-                  logger.debug { "calling back user cb with connection loss" }
-                  @user_cb.call(:rc => ZK::Exceptions::CONNECTIONLOSS)
-                end
-
-                { :rc => Zookeeper::ZOK }
-              end
-            end
-
-            @zkem.get(@path) do |exc,data|
-              exc.should be_kind_of(ZK::Exceptions::ConnectionLoss)
-            end
-          end
-        end
-      end
-
-      it %[should be called if we get a session expired event] do
-        @zkem.on_connection_lost do |exc|
-          logger.debug { "WIN!" }
-          exc.should be_kind_of(ZK::Exceptions::ConnectionLoss)
-          @zkem.close! { done }
-        end
-
-        em do
-          @zkem.connect do
-            event = event_mock(:connection_loss_event).tap do |ev|
-              ev.should_receive(:state).and_return(Zookeeper::ZOO_EXPIRED_SESSION_STATE)
-            end
-
-            EM.next_tick { @zkem.event_handler.process(event) }
-          end
-        end
-      end
-    end # on_connection_lost
-
-    describe 'on_connected' do
-      it %[should be called back when a ZOO_CONNECTED_STATE event is received] do
-        em do
-          @zkem.on_connected do |event|
-            logger.debug { "WIN!" }
-            @zkem.close! { done }
-          end
-
-          @zkem.connect do
-            logger.debug { "we connected" }
-          end
-        end
-      end
-    end # on_connected
-
-    describe 'on_connecting' do
-      it %[should be called back when a ZOO_CONNECTING_STATE event is received] do
-        @zkem.on_connecting do |event|
-          logger.debug { "WIN!" }
-          @zkem.close! { done }
-        end
-
-        em do
-          @zkem.connect do
-            event = event_mock(:connecting_event).tap do |ev|
-              ev.should_receive(:state).and_return(Zookeeper::ZOO_CONNECTING_STATE)
-            end
-
-            EM.next_tick { @zkem.event_handler.process(event) }
-          end
-        end
-      end
-    end # on_connecting
   end # Client
-
-  describe 'regular' do
-
-    before do
-      @zkem = ZK::ZKEventMachine::Client.new('localhost:2181')
-      @zk = ZK.new.tap { |z| wait_until { z.connected? } }
-      @zk.should be_connected
-    end
-
-    it_should_behave_like 'Client'
-  end
-
-  describe 'chrooted' do
-    let(:chroot_path) { '/_zkem_chroot_' }
-    let(:zk_connect_host) { "localhost:2181#{chroot_path}" }
-
-    before :all do
-      ZK.open('localhost:2181') do |z|
-        z.rm_rf(chroot_path)
-        z.mkdir_p(chroot_path)
-      end
-    end
-
-    after :all do
-      ZK.open('localhost:2181') do |z|
-        z.rm_rf(chroot_path)
-      end
-    end
-
-    before do
-      @zkem = ZK::ZKEventMachine::Client.new(zk_connect_host)
-      @zk = ZK.new(zk_connect_host).tap { |z| wait_until { z.connected? } }
-      @zk.should be_connected
-    end
-
-    it_should_behave_like 'Client'
-  end
 end # ZK::ZKEventMachine
 

data/spec/z_k/z_k_event_machine/synchrony_client_spec.rb

@@ -0,0 +1,339 @@
+require 'spec_helper'
+
+module ZK
+  module ZKEventMachine
+    describe 'SynchronyClient' do
+      include EventedSpec::SpecHelper
+      default_timeout 2.0
+
+      def em_synchrony(&blk)
+        em do
+          EM.next_tick { Fiber.new { blk.call }.resume }
+        end
+      end
+
+      def with_zksync
+        em_synchrony do
+          begin
+            @zksync.connect
+            yield @zksync
+          ensure
+            @zksync.close!
+            done
+          end
+        end
+      end
+
+      before do
+        @zk = ::ZK.new
+        @base_path = '/zk-em-testing'
+        @zk.rm_rf(@base_path)
+        @zk.mkdir_p(@base_path)
+        @zksync = ZK::ZKEventMachine::SynchronyClient.new('localhost:2181')
+      end
+
+      after do
+        @zk.rm_rf(@base_path)
+        @zk.close!
+      end
+
+      describe 'connect' do
+        it %[should connect to zookeeper] do
+          logger.debug { "about to call connect" }
+          em_synchrony do
+            @zksync.connect
+
+            lambda { @zksync.get(@base_path) }.should_not raise_error
+
+            done { @zksync.close }
+          end
+        end
+      end
+
+      describe 'get' do
+        before do
+          @data = "this is data"
+          @zk.set(@base_path, @data)
+        end
+
+        it %[should get the data and stat of a node that exists] do
+          with_zksync do
+            data, stat = @zksync.get(@base_path)
+            data.should == @data
+            stat.should be_kind_of(ZookeeperStat::Stat)
+          end
+        end
+
+        it %[should raise an exception if the node does not exist] do
+          with_zksync do
+            lambda { @zksync.get('/thispathdoesnotexist') }.should raise_error(ZK::Exceptions::NoNode)
+          end
+        end
+      end
+
+      describe 'create' do
+        describe 'success' do
+          before do
+            @path = [@base_path, 'foo'].join('/')
+            @zk.delete(@path) rescue ZK::Exceptions::NoNode
+
+            @data = 'this is data'
+          end
+
+          it 'should create a non-sequence node' do
+            with_zksync do
+              @zksync.create(@path, @data).should == @path
+            end
+          end
+
+          it %[should create a sequence node] do
+            with_zksync do
+              @zksync.create(@path, @data, :sequence => true).should =~ /\A#{@path}\d+\Z/
+            end
+          end
+        end
+        
+        describe 'failure' do
+          it %[should barf if the node exists] do
+            @path = [@base_path, 'foo'].join('/')
+            @zk.create(@path, '')
+
+            lambda { @zk.create(@path, '') }.should raise_error(ZK::Exceptions::NodeExists)
+          end
+        end
+      end
+
+      describe 'set' do
+        before do
+          @path = [@base_path, 'foo'].join('/')
+          @data = 'this is data'
+          @new_data = 'this is better data'
+          @zk.create(@path, @data)
+          @orig_stat = @zk.stat(@path)
+        end
+
+        it %[should set the data and return a stat] do
+          with_zksync do
+            stat = @zksync.set(@path, @new_data)
+            stat.should be_instance_of(ZookeeperStat::Stat)
+            stat.version.should > @orig_stat.version
+
+            @zksync.get(@path).first.should == @new_data
+          end
+        end
+
+        it %[should raise NoNode if the node doesn't exist] do
+          with_zksync do
+            lambda { @zksync.set('/thispathdoesnotexist', 'data') }.should raise_error(ZK::Exceptions::NoNode)
+          end
+        end
+
+        it %[should raise BadVersion if the version is wrong] do
+          with_zksync do
+            @zksync.set(@path, @new_data)
+            lambda { @zksync.set(@path, 'otherdata', :version => @orig_stat.version) }.should raise_error(ZK::Exceptions::BadVersion)
+          end
+        end
+      end
+
+      describe 'exists?' do
+        before do
+          @path = [@base_path, 'foo'].join('/')
+          @data = 'this is data'
+        end
+
+        it %[should return true if the node exists] do
+          @zk.create(@path, @data)
+
+          with_zksync do
+            @zksync.exists?(@path).should be_true
+          end
+        end
+
+        it %[should return false if the node doesn't exist] do
+          with_zksync do
+            @zksync.exists?(@path).should be_false
+          end
+        end
+      end
+
+      describe 'stat' do
+        describe 'success' do
+          before do
+            @path = [@base_path, 'foo'].join('/')
+            @data = 'this is data'
+            @zk.create(@path, @data)
+            @orig_stat = @zk.stat(@path)
+          end
+
+          it %[should get the stat] do
+            with_zksync do
+              stat = @zksync.stat(@path)
+
+              stat.should_not be_nil
+              stat.should == @orig_stat
+              stat.should be_instance_of(ZookeeperStat::Stat)
+            end
+          end
+        end
+
+        describe 'non-existent node' do
+          before do
+            @path = [@base_path, 'foo'].join('/')
+            @zk.delete(@path) rescue ZK::Exceptions::NoNode
+          end
+
+          it %[should not be an error] do
+            with_zksync do
+              stat = @zksync.stat(@path)
+              stat.should_not be_nil
+              stat.exists?.should be_false
+              stat.should be_instance_of(ZookeeperStat::Stat)
+            end
+          end
+        end
+      end
+
+      describe 'delete' do
+        it %[should delete the node] do
+          @path = [@base_path, 'foo'].join('/')
+          @data = 'this is data'
+          @zk.create(@path, @data)
+
+          with_zksync do
+            @zksync.delete(@path)
+          end
+
+          @zk.exists?(@path).should be_false
+        end
+
+        it %[should raise NoNode exception if the node does not exist] do
+          @path = [@base_path, 'foo'].join('/')
+          @zk.delete(@path) rescue ZK::Exceptions::NoNode
+
+          with_zksync do
+            lambda { @zksync.delete(@path) }.should raise_error(ZK::Exceptions::NoNode)
+          end
+        end
+      end
+
+      describe 'children' do
+        it %[should return the names of the children of the node] do
+          @path = [@base_path, 'foo'].join('/')
+          @child_1_path = [@path, 'child_1'].join('/')
+          @child_2_path = [@path, 'child_2'].join('/')
+
+          @data = 'this is data'
+          @zk.create(@path, @data)
+          @zk.create(@child_1_path, '')
+          @zk.create(@child_2_path, '')
+
+          with_zksync do
+            children, stat = @zksync.children(@path)
+            children.should be_kind_of(Array)
+            children.length.should == 2
+            children.should include('child_1')
+            children.should include('child_2')
+
+            stat.should be_instance_of(ZookeeperStat::Stat)
+          end
+        end
+
+        it %[should raise NoNode if the node doesn't exist] do
+          @path = [@base_path, 'foo'].join('/')
+          @zk.delete(@path) rescue ZK::Exceptions::NoNode
+
+          with_zksync do
+            lambda { @zksync.children(@path) }.should raise_error(ZK::Exceptions::NoNode)
+          end
+        end
+      end
+
+      describe 'mkdir_p' do
+        it %[should create the directory structure] do
+          paths = [ "#{@base_path}/bar/baz", "#{@base_path}/foo/bar/quux" ]
+
+          with_zksync do
+            @zksync.mkdir_p(paths)
+          end
+
+          @zk.exists?("#{@base_path}/bar").should be_true
+          @zk.exists?("#{@base_path}/bar/baz").should be_true
+          @zk.exists?("#{@base_path}/foo").should be_true
+          @zk.exists?("#{@base_path}/foo/bar").should be_true
+          @zk.exists?("#{@base_path}/foo/bar/quux").should be_true
+        end
+      end
+
+      describe 'rm_rf' do
+        it %[should remove all paths listed] do
+          @relpaths = ['disco/foo', 'prune/bar', 'fig/bar/one', 'apple/bar/two', 'orange/quux/c/d/e']
+
+          @roots = @relpaths.map { |p| File.join(@base_path, p.split('/').first) }.uniq
+          @paths = @relpaths.map { |n| File.join(@base_path, n) }
+
+          @paths.each { |n| @zk.mkdir_p(n) }
+
+          with_zksync do
+            @zksync.rm_rf(@roots)
+          end
+
+          @roots.each { |n| @zk.exists?(n).should be_false }
+
+          @zk.exists?(@base_path).should be_true
+        end
+      end
+
+      describe 'event delivery' do
+        default_timeout 0.5
+
+        before do
+          @path = [@base_path, 'foo'].join('/')
+          @data = 'this is data'
+          @zk.create(@path, @data)
+          @orig_stat = @zk.stat(@path)
+        end
+
+        it %[should receive an event when the node changes] do
+          em_synchrony do
+            @orig_fiber = Fiber.current
+
+            @zksync.connect
+            @zksync.should be_connected
+
+            @zksync.event_handler.register(@path) do |event|
+              Fiber.current.should_not == @orig_fiber
+
+              done { @zksync.close! }
+            end
+
+            @zksync.get(@path, :watch => true)
+
+            stat = @zksync.set(@path, 'new data')
+            logger.debug { "sksync set returned: #{stat}" }
+          end
+        end
+
+        it %[should be able to register for events on a node that doesn't exist yet] do
+          em_synchrony do
+            @new_path = "#{@path}/blah"
+
+            @zksync.connect
+            @zksync.should be_connected
+
+            @zksync.event_handler.register(@new_path) do |event|
+              logger.debug { "got event #{event}" }
+
+              done { @zksync.close! }
+            end
+
+            @zksync.stat(@new_path, :watch => true)
+
+            @zksync.create(@new_path, '')
+          end
+        end
+      end
+    end
+  end
+end
+