beaneater 0.3.0 → 1.1.1

data/lib/beaneater/pool_command.rb

@@ -1,79 +0,0 @@
-require 'set'
-
-module Beaneater
-  # Represents collection of pool related commands.
-  class PoolCommand
-    # @!attribute pool
-    #   @return [Beaneater::Pool] returns Pool object
-    attr_reader :pool
-
-    # Initialize new connection
-    #
-    # @param [Beaneater::Pool] pool Pool object
-    def initialize(pool)
-      @pool = pool
-    end
-
-    # Delegate to Pool#transmit_to_all and if needed will merge responses from beanstalkd.
-    #
-    # @param [String] body Beanstalkd command
-    # @param [Hash{String => String, Boolean}] options socket connections options
-    # @option options [Boolean] merge Ask for merging responses or not
-    # @param [Proc] block Block passed in socket connection object
-    # @example
-    #   @pool.transmit_to_all("stats")
-    #
-    def transmit_to_all(body, options={}, &block)
-      merge = options.delete(:merge)
-      res = pool.transmit_to_all(body, options, &block)
-      first = res.find { |r| r && r[:status] }
-      if first && merge
-        res = { :status => first[:status], :body => sum_items(res.map { |r| r[:body] }) }
-      end
-      res
-    end
-
-    # Delegate missing methods to pool
-    # @api public
-    def method_missing(name, *args, &block)
-      if pool.respond_to?(name)
-        pool.send(name, *args, &block)
-      else # not a known pool command
-        super
-      end
-    end
-
-    protected
-
-    # Selects items from collection and then merges the individual values
-    # Supports array of hashes or array of arrays
-    #
-    # @param [Array<Hash, Array>] hs Collection of responses returned from beanstalkd
-    # @return [Hash{Symbol => String}] Merged responses combining values from all the hash bodies
-    # @example
-    #  self.sum_items([{ :foo => 1, :bar => 5 }, { :foo => 2, :bar => 3 }])
-    #    => { :foo => 3, :bar => 8 }
-    #  self.sum_items([['foo', 'bar'], ['foo', 'bar', 'baz']])
-    #    => ['foo', 'bar', 'baz']
-    #
-    def sum_items(items)
-      if items.first.is_a?(Hash)
-        items.select { |h| h.is_a?(Hash) }.
-          inject({}) { |a,b| a.merge(b) { |k,o,n| combine_stats(k, o, n) } }
-      elsif items.first.is_a?(Array)
-        items.flatten.uniq
-      end
-    end
-
-    # Combine two values for given key
-    #
-    # @param [String] k key name within response hash
-    # @return [Set,Integer] combined value for stat
-    # @example
-    #  self.combine_stats('total_connections', 4, 5) # => 9
-    #
-    def combine_stats(k, a, b)
-      ['name', 'version', 'pid'].include?(k) ? Set[a] + Set[b] : a + b
-    end
-  end # PoolCommand
-end # Beaneater

data/test/pool_command_test.rb

@@ -1,90 +0,0 @@
-# test/connection_test.rb
-
-require File.expand_path('../test_helper', __FILE__)
-
-describe Beaneater::PoolCommand do
-
-  describe 'for #new' do
-    before do
-      @pool = stub
-      @command = Beaneater::PoolCommand.new(@pool)
-    end
-
-    it "should store pool" do
-      assert_equal @pool, @command.pool
-    end
-  end #new
-
-  describe 'for #transmit_to_all' do
-    describe 'for regular command' do
-      before do
-        @pool = stub(:transmit_to_all => [{ :body => "foo", :status => "OK" }])
-        @command = Beaneater::PoolCommand.new(@pool)
-      end
-
-      it "can run regular command" do
-        res = @command.transmit_to_all("foo")
-        assert_equal "OK", res[0][:status]
-        assert_equal "foo", res[0][:body]
-      end
-    end # regular command
-
-    describe 'for merge command with hashes' do
-      before do
-        @pool = stub(:transmit_to_all => [
-          { :body => { 'x' => 1, 'version' => 1.1 }, :status => "OK"},
-          { :body => { 'x' => 3,'version' => 1.2 }, :status => "OK" }
-        ])
-        @command = Beaneater::PoolCommand.new(@pool)
-      end
-
-      it "can run merge command " do
-        cmd = @command.transmit_to_all("bar", :merge => true)
-        assert_equal "OK", cmd[:status]
-        assert_equal 4, cmd[:body]['x']
-        assert_equal Set[1.1, 1.2], cmd[:body]['version']
-      end
-    end # merged command
-
-    describe 'for merge command with arrays' do
-      before do
-        @pool = stub(:transmit_to_all => [
-          { :body => ['foo', 'bar'], :status => "OK"},
-          { :body => ['foo', 'bar', 'baz'], :status => "OK" }
-        ])
-        @command = Beaneater::PoolCommand.new(@pool)
-      end
-
-      it "can run merge command " do
-        cmd = @command.transmit_to_all("bar", :merge => true)
-        assert_equal "OK", cmd[:status]
-        assert_equal ['foo', 'bar', 'baz'].sort, cmd[:body].sort
-      end
-    end # merged command
-  end # transmit_to_all
-
-  describe 'for #method_missing' do
-    describe '#transmit_to_rand' do
-      before do
-        @pool = stub
-        @pool.expects(:transmit_to_rand).with('foo').returns('OK')
-        @command = Beaneater::PoolCommand.new(@pool)
-      end
-
-      it 'delegates to connection' do
-        assert_equal 'OK', @command.transmit_to_rand('foo')
-      end
-    end # transmit_to_rand
-
-    describe 'invalid method' do
-      before do
-        @pool = stub
-        @command = Beaneater::PoolCommand.new(@pool)
-      end
-
-      it 'raises no method error' do
-        assert_raises(NoMethodError) { @command.foo('foo') }
-      end
-    end # transmit_to_rand
-  end
-end # Beaneater::PoolCommand

data/test/pool_test.rb

@@ -1,180 +0,0 @@
-# test/connection_test.rb
-
-require File.expand_path('../test_helper', __FILE__)
-
-describe Beaneater::Pool do
-
-  before do
-    @hosts = ['localhost', 'localhost']
-    @bp = Beaneater::Pool.new(@hosts)
-  end
-
-  describe 'for #new' do
-
-    describe "for multiple connection" do
-      before do
-        @host_string_port = 'localhost:11301'
-        @host_num_port = '127.0.0.1:11302'
-        @host_string = 'host.local'
-        @host_num = '1.1.1.1:11303'
-        @hosts = [@host_string_port, @host_num_port, @host_string, @host_num]
-
-        TCPSocket.expects(:new).with('localhost',11301).once
-        TCPSocket.expects(:new).with('127.0.0.1',11302).once
-        TCPSocket.expects(:new).with('host.local',11300).once
-        TCPSocket.expects(:new).with('1.1.1.1',11303).once
-
-        @bp = Beaneater::Pool.new(@hosts)
-      end
-
-      it "should init 4 connections" do
-        assert_equal 4, @bp.connections.size
-      end
-    end
-
-    describe "for invalid connection in setup" do
-      it "should raise NotConnected" do
-        assert_raises(Beaneater::NotConnected) { Beaneater::Pool.new('localhost:5679') }
-      end
-    end
-
-    describe "for clearing watch list" do
-      it "should clear connections of tube watches" do
-        @bp.tubes.watch!('foo', 'bar')
-        assert_equal ['foo', 'bar'].sort, @bp.tubes.watched.map(&:name).sort
-        @bp2 = Beaneater::Pool.new(@hosts)
-        assert_equal ['default'], @bp2.tubes.watched.map(&:name)
-      end
-    end
-
-    describe "with ENV variable set" do
-      before do
-        ENV['BEANSTALKD_URL'] = '0.0.0.0:11300,127.0.0.1:11300'
-      end
-
-      it "should create 1 connection" do
-        bp = Beaneater::Pool.new
-        bc = bp.connections.first
-        bc2 = bp.connections.last
-
-        assert_equal 2, bp.connections.size
-        assert_equal '0.0.0.0', bc.host
-        assert_equal 11300, bc.port
-
-        assert_equal '127.0.0.1', bc2.host
-        assert_equal 11300, bc2.port
-      end
-
-      after do
-        ENV['BEANSTALKD_URL'] = nil
-      end
-    end
-
-    describe "by configuring via Beaneater.configure" do
-      before do
-        Beaneater.configure.beanstalkd_url = ['0.0.0.0:11300', '127.0.0.1:11300']
-      end
-
-      it "should create 1 connection" do
-        bp = Beaneater::Pool.new
-        bc = bp.connections.first
-        bc2 = bp.connections.last
-
-        assert_equal 2, bp.connections.size
-        assert_equal '0.0.0.0', bc.host
-        assert_equal 11300, bc.port
-
-        assert_equal '127.0.0.1', bc2.host
-        assert_equal 11300, bc2.port
-      end
-
-      after do
-        Beaneater.configure.beanstalkd_url = ['0.0.0.0:11300']
-      end
-    end
-  end # new
-
-  describe 'for #transmit_to_all' do
-    it "should return yaml loaded response" do
-      res = @bp.transmit_to_all 'stats'
-      assert_equal 2, res.size
-      refute_nil res.first[:body]['current-connections']
-    end
-  end # transmit_to_all
-
-  describe 'for #transmit_to_rand' do
-    it "should return yaml loaded response" do
-      res = @bp.transmit_to_rand 'stats'
-      refute_nil res[:body]['current-connections']
-      assert_equal 'OK', res[:status]
-    end
-
-    it "should return id" do
-      Beaneater::Connection.any_instance.expects(:transmit).with("foo",{}).returns({:id => "254", :status => "INSERTED"})
-      res = @bp.transmit_to_rand 'foo'
-      assert_equal '254', res[:id]
-      assert_equal 'INSERTED', res[:status]
-    end
-  end # transmit_to_rand
-
-  describe 'for #transmit_until_res' do
-    before do
-      Beaneater::Connection.any_instance.expects(:transmit).with('foo', {}).twice.
-      returns({:status => "FAILED", :body => 'x'}).then.
-      returns({:status => "OK", :body => 'y'}).then.returns({:status => "OK", :body => 'z'})
-    end
-
-    it "should returns first matching status" do
-      assert_equal 'y', @bp.transmit_until_res('foo', :status => 'OK')[:body]
-    end
-  end # transmit_until_res
-
-  describe 'for #stats' do
-    it("should return stats object"){ assert_kind_of Beaneater::Stats, @bp.stats }
-  end # stats
-
-  describe 'for #tubes' do
-    it("should return Tubes object"){ assert_kind_of Beaneater::Tubes, @bp.tubes }
-  end # tubes
-
-  describe "for #safe_transmit" do
-    it "should retry 3 times for temporary failed connection" do
-      TCPSocket.any_instance.expects(:write).times(3)
-      TCPSocket.any_instance.expects(:gets).raises(Errno::ECONNRESET).then.
-        raises(Errno::ECONNRESET).then.returns('INSERTED 254').times(3)
-      res = @bp.transmit_to_rand "put 0 0 10 2\r\nxy"
-      assert_equal '254', res[:id]
-      assert_equal 'INSERTED', res[:status]
-    end
-
-    it 'should raise proper exception for invalid status NOT_FOUND' do
-      TCPSocket.any_instance.expects(:write).once
-      TCPSocket.any_instance.expects(:gets).returns('NOT_FOUND')
-      assert_raises(Beaneater::NotFoundError) { @bp.transmit_to_rand 'foo' }
-    end
-
-    it 'should raise proper exception for invalid status BAD_FORMAT' do
-      TCPSocket.any_instance.expects(:write).once
-      TCPSocket.any_instance.expects(:gets).returns('BAD_FORMAT')
-      assert_raises(Beaneater::BadFormatError) { @bp.transmit_to_rand 'foo' }
-    end
-
-    it 'should raise proper exception for invalid status DEADLINE_SOON' do
-      TCPSocket.any_instance.expects(:write).once
-      TCPSocket.any_instance.expects(:gets).once.returns('DEADLINE_SOON')
-      assert_raises(Beaneater::DeadlineSoonError) { @bp.transmit_to_rand 'expecting deadline' }
-    end
-  end
-
-  describe "for #close" do
-    it "should support closing the pool" do
-      connection = @bp.connections.first
-      assert_equal 2, @bp.connections.size
-      assert_kind_of Beaneater::Connection, connection
-      assert_kind_of TCPSocket, connection.connection
-      @bp.close
-      assert_equal 0, @bp.connections.size
-      assert_nil connection.connection
-    end
-  end # close
-end # Beaneater::Pool