test/test_pool.rb

   1 # -*- encoding: binary -*-
   2 require 'test/unit'
   3
   4 $TESTING = true
   5 require 'mogilefs'
   6 require 'mogilefs/pool'
   7
   8 class MogileFS::Pool
   9
  10   attr_reader :objects, :queue
  11
  12 end
  13
  14 class Resource; end
  15
  16 class ResourceWithArgs
  17
  18   def initialize(args)
  19   end
  20
  21 end
  22
  23 class TestPool < Test::Unit::TestCase
  24
  25   def setup
  26     @pool = MogileFS::Pool.new Resource
  27   end
  28
  29   def test_get
  30     o1 = @pool.get
  31     o2 = @pool.get
  32     assert_kind_of Resource, o1
  33     assert_kind_of Resource, o2
  34     assert_not_equal o1, o2
  35   end
  36
  37   def test_get_with_args
  38     @pool = MogileFS::Pool.new ResourceWithArgs, 'my arg'
  39     o = @pool.get
  40     assert_kind_of ResourceWithArgs, o
  41   end
  42
  43   def test_put
  44     o = @pool.get
  45     @pool.put o
  46
  47     assert_raises(MogileFS::Pool::BadObjectError) { @pool.put nil }
  48     assert_raises(MogileFS::Pool::BadObjectError) { @pool.put Resource.new }
  49   end
  50
  51   def test_put_destroy
  52     objs = (0...7).map { @pool.get } # pool full
  53
  54     assert_equal 7, @pool.objects.length
  55     assert_equal 0, @pool.queue.length
  56
  57     4.times { @pool.put objs.shift }
  58
  59     assert_equal 7, @pool.objects.length
  60     assert_equal 4, @pool.queue.length
  61
  62     @pool.put objs.shift # trip threshold
  63
  64     assert_equal 4, @pool.objects.length
  65     assert_equal 2, @pool.queue.length
  66
  67     @pool.put objs.shift # don't need to remove any more
  68
  69     assert_equal 4, @pool.objects.length
  70     assert_equal 3, @pool.queue.length
  71
  72     @pool.put objs.shift until objs.empty?
  73
  74     assert_equal 4, @pool.objects.length
  75     assert_equal 4, @pool.queue.length
  76   end
  77
  78   def test_use
  79     val = @pool.use { |o| assert_kind_of Resource, o }
  80     assert_equal nil, val, "Don't return object from pool"
  81   end
  82
  83   def test_use_with_exception
  84     @pool.use { |o| raise } rescue nil
  85     assert_equal 1, @pool.queue.length, "Resource not returned to pool"
  86   end
  87
  88   def test_use_reuse
  89     o1 = nil
  90     o2 = nil
  91
  92     @pool.use { |o| o1 = o }
  93     @pool.use { |o| o2 = o }
  94
  95     assert_equal o1, o2, "Objects must be reused"
  96   end
  97
  98 end
  99