spec/zk/watch_spec.rb in zk-1.4.2 vs spec/zk/watch_spec.rb in zk-1.5.0

@@ -363,9 +363,13 @@
         @zk = ZK.new(*connection_args) do |zk|
           @cnx_reg = zk.on_connected { |event| @event = event }
         end
       end
 
+      after do
+        @zk.close! if @zk and not @zk.closed?
+      end
+
       it %[should fire the registered callback] do
         wait_while { @event.nil? }
         @event.should_not be_nil
       end
     end