beaneater 0.3.0 → 1.1.0

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